langchain 0.0.203 → 0.0.204-rc.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (494) hide show
  1. package/dist/cache/cloudflare_kv.cjs +15 -69
  2. package/dist/cache/cloudflare_kv.d.ts +1 -42
  3. package/dist/cache/cloudflare_kv.js +1 -67
  4. package/dist/cache/momento.cjs +15 -135
  5. package/dist/cache/momento.d.ts +1 -91
  6. package/dist/cache/momento.js +1 -133
  7. package/dist/cache/upstash_redis.cjs +15 -72
  8. package/dist/cache/upstash_redis.d.ts +1 -45
  9. package/dist/cache/upstash_redis.js +1 -70
  10. package/dist/callbacks/handlers/llmonitor.cjs +14 -238
  11. package/dist/callbacks/handlers/llmonitor.d.ts +1 -35
  12. package/dist/callbacks/handlers/llmonitor.js +1 -233
  13. package/dist/chains/openai_moderation.cjs +2 -2
  14. package/dist/chains/openai_moderation.d.ts +1 -1
  15. package/dist/chains/openai_moderation.js +1 -1
  16. package/dist/chat_models/anthropic.d.ts +4 -4
  17. package/dist/chat_models/baiduwenxin.cjs +15 -411
  18. package/dist/chat_models/baiduwenxin.d.ts +1 -150
  19. package/dist/chat_models/baiduwenxin.js +1 -409
  20. package/dist/chat_models/bedrock/index.cjs +15 -33
  21. package/dist/chat_models/bedrock/index.d.ts +1 -23
  22. package/dist/chat_models/bedrock/index.js +1 -29
  23. package/dist/chat_models/bedrock/web.cjs +15 -360
  24. package/dist/chat_models/bedrock/web.d.ts +1 -82
  25. package/dist/chat_models/bedrock/web.js +1 -356
  26. package/dist/chat_models/cloudflare_workersai.cjs +15 -216
  27. package/dist/chat_models/cloudflare_workersai.d.ts +1 -68
  28. package/dist/chat_models/cloudflare_workersai.js +1 -214
  29. package/dist/chat_models/fireworks.cjs +15 -89
  30. package/dist/chat_models/fireworks.d.ts +1 -43
  31. package/dist/chat_models/fireworks.js +1 -87
  32. package/dist/chat_models/googlepalm.cjs +15 -235
  33. package/dist/chat_models/googlepalm.d.ts +1 -108
  34. package/dist/chat_models/googlepalm.js +1 -233
  35. package/dist/chat_models/googlevertexai/index.cjs +15 -40
  36. package/dist/chat_models/googlevertexai/index.d.ts +1 -28
  37. package/dist/chat_models/googlevertexai/index.js +1 -38
  38. package/dist/chat_models/googlevertexai/web.cjs +15 -39
  39. package/dist/chat_models/googlevertexai/web.d.ts +1 -28
  40. package/dist/chat_models/googlevertexai/web.js +1 -37
  41. package/dist/chat_models/iflytek_xinghuo/index.cjs +14 -40
  42. package/dist/chat_models/iflytek_xinghuo/index.d.ts +1 -13
  43. package/dist/chat_models/iflytek_xinghuo/index.js +1 -36
  44. package/dist/chat_models/iflytek_xinghuo/web.cjs +15 -36
  45. package/dist/chat_models/iflytek_xinghuo/web.d.ts +1 -13
  46. package/dist/chat_models/iflytek_xinghuo/web.js +1 -34
  47. package/dist/chat_models/llama_cpp.cjs +15 -275
  48. package/dist/chat_models/llama_cpp.d.ts +1 -71
  49. package/dist/chat_models/llama_cpp.js +1 -273
  50. package/dist/chat_models/minimax.cjs +15 -578
  51. package/dist/chat_models/minimax.d.ts +1 -369
  52. package/dist/chat_models/minimax.js +1 -576
  53. package/dist/chat_models/ollama.cjs +15 -394
  54. package/dist/chat_models/ollama.d.ts +1 -127
  55. package/dist/chat_models/ollama.js +1 -392
  56. package/dist/chat_models/openai.cjs +4 -698
  57. package/dist/chat_models/openai.d.ts +4 -137
  58. package/dist/chat_models/openai.js +2 -695
  59. package/dist/chat_models/portkey.cjs +15 -167
  60. package/dist/chat_models/portkey.d.ts +1 -17
  61. package/dist/chat_models/portkey.js +1 -165
  62. package/dist/chat_models/yandex.cjs +15 -132
  63. package/dist/chat_models/yandex.d.ts +1 -30
  64. package/dist/chat_models/yandex.js +1 -130
  65. package/dist/document_loaders/fs/openai_whisper_audio.cjs +2 -2
  66. package/dist/document_loaders/fs/openai_whisper_audio.d.ts +1 -1
  67. package/dist/document_loaders/fs/openai_whisper_audio.js +1 -1
  68. package/dist/document_transformers/html_to_text.cjs +15 -47
  69. package/dist/document_transformers/html_to_text.d.ts +1 -30
  70. package/dist/document_transformers/html_to_text.js +1 -45
  71. package/dist/document_transformers/mozilla_readability.cjs +15 -53
  72. package/dist/document_transformers/mozilla_readability.d.ts +1 -31
  73. package/dist/document_transformers/mozilla_readability.js +1 -51
  74. package/dist/embeddings/bedrock.cjs +15 -109
  75. package/dist/embeddings/bedrock.d.ts +1 -71
  76. package/dist/embeddings/bedrock.js +1 -107
  77. package/dist/embeddings/cloudflare_workersai.cjs +15 -67
  78. package/dist/embeddings/cloudflare_workersai.d.ts +1 -28
  79. package/dist/embeddings/cloudflare_workersai.js +1 -65
  80. package/dist/embeddings/cohere.cjs +15 -124
  81. package/dist/embeddings/cohere.d.ts +1 -66
  82. package/dist/embeddings/cohere.js +1 -122
  83. package/dist/embeddings/googlepalm.cjs +15 -88
  84. package/dist/embeddings/googlepalm.d.ts +1 -61
  85. package/dist/embeddings/googlepalm.js +1 -86
  86. package/dist/embeddings/googlevertexai.cjs +15 -81
  87. package/dist/embeddings/googlevertexai.d.ts +1 -54
  88. package/dist/embeddings/googlevertexai.js +1 -79
  89. package/dist/embeddings/gradient_ai.cjs +15 -101
  90. package/dist/embeddings/gradient_ai.d.ts +1 -48
  91. package/dist/embeddings/gradient_ai.js +1 -99
  92. package/dist/embeddings/hf.cjs +15 -75
  93. package/dist/embeddings/hf.d.ts +1 -41
  94. package/dist/embeddings/hf.js +1 -73
  95. package/dist/embeddings/hf_transformers.cjs +15 -87
  96. package/dist/embeddings/hf_transformers.d.ts +1 -47
  97. package/dist/embeddings/hf_transformers.js +1 -85
  98. package/dist/embeddings/llama_cpp.cjs +15 -81
  99. package/dist/embeddings/llama_cpp.d.ts +1 -42
  100. package/dist/embeddings/llama_cpp.js +1 -79
  101. package/dist/embeddings/minimax.cjs +15 -165
  102. package/dist/embeddings/minimax.d.ts +1 -119
  103. package/dist/embeddings/minimax.js +1 -163
  104. package/dist/embeddings/ollama.cjs +15 -118
  105. package/dist/embeddings/ollama.d.ts +1 -34
  106. package/dist/embeddings/ollama.js +1 -116
  107. package/dist/embeddings/openai.cjs +2 -240
  108. package/dist/embeddings/openai.d.ts +1 -82
  109. package/dist/embeddings/openai.js +1 -239
  110. package/dist/embeddings/tensorflow.cjs +3 -96
  111. package/dist/embeddings/tensorflow.d.ts +1 -59
  112. package/dist/embeddings/tensorflow.js +1 -83
  113. package/dist/embeddings/voyage.cjs +15 -118
  114. package/dist/embeddings/voyage.d.ts +1 -66
  115. package/dist/embeddings/voyage.js +1 -116
  116. package/dist/experimental/chat_models/ollama_functions.d.ts +1 -1
  117. package/dist/experimental/openai_assistant/index.cjs +3 -3
  118. package/dist/experimental/openai_assistant/index.d.ts +1 -1
  119. package/dist/experimental/openai_assistant/index.js +1 -1
  120. package/dist/experimental/openai_assistant/schema.d.ts +1 -1
  121. package/dist/experimental/openai_files/index.cjs +2 -2
  122. package/dist/experimental/openai_files/index.d.ts +1 -1
  123. package/dist/experimental/openai_files/index.js +1 -1
  124. package/dist/graphs/neo4j_graph.cjs +3 -254
  125. package/dist/graphs/neo4j_graph.d.ts +1 -63
  126. package/dist/graphs/neo4j_graph.js +1 -241
  127. package/dist/llms/ai21.cjs +15 -200
  128. package/dist/llms/ai21.d.ts +1 -100
  129. package/dist/llms/ai21.js +1 -198
  130. package/dist/llms/aleph_alpha.cjs +15 -374
  131. package/dist/llms/aleph_alpha.d.ts +1 -161
  132. package/dist/llms/aleph_alpha.js +1 -372
  133. package/dist/llms/bedrock/web.cjs +15 -310
  134. package/dist/llms/bedrock/web.d.ts +1 -66
  135. package/dist/llms/bedrock/web.js +1 -308
  136. package/dist/llms/cloudflare_workersai.cjs +15 -168
  137. package/dist/llms/cloudflare_workersai.d.ts +1 -55
  138. package/dist/llms/cloudflare_workersai.js +1 -166
  139. package/dist/llms/cohere.cjs +15 -110
  140. package/dist/llms/cohere.d.ts +1 -54
  141. package/dist/llms/cohere.js +1 -108
  142. package/dist/llms/fireworks.cjs +15 -90
  143. package/dist/llms/fireworks.d.ts +1 -33
  144. package/dist/llms/fireworks.js +1 -88
  145. package/dist/llms/googlepalm.cjs +15 -134
  146. package/dist/llms/googlepalm.d.ts +1 -92
  147. package/dist/llms/googlepalm.js +1 -132
  148. package/dist/llms/googlevertexai/index.cjs +15 -45
  149. package/dist/llms/googlevertexai/index.d.ts +1 -38
  150. package/dist/llms/googlevertexai/index.js +1 -43
  151. package/dist/llms/googlevertexai/web.cjs +15 -42
  152. package/dist/llms/googlevertexai/web.d.ts +1 -36
  153. package/dist/llms/googlevertexai/web.js +1 -40
  154. package/dist/llms/gradient_ai.cjs +15 -110
  155. package/dist/llms/gradient_ai.d.ts +1 -55
  156. package/dist/llms/gradient_ai.js +1 -108
  157. package/dist/llms/hf.cjs +15 -136
  158. package/dist/llms/hf.d.ts +1 -68
  159. package/dist/llms/hf.js +1 -134
  160. package/dist/llms/llama_cpp.cjs +15 -111
  161. package/dist/llms/llama_cpp.d.ts +1 -41
  162. package/dist/llms/llama_cpp.js +1 -109
  163. package/dist/llms/ollama.cjs +15 -353
  164. package/dist/llms/ollama.d.ts +1 -106
  165. package/dist/llms/ollama.js +1 -351
  166. package/dist/llms/openai-chat.cjs +3 -445
  167. package/dist/llms/openai-chat.d.ts +4 -123
  168. package/dist/llms/openai-chat.js +2 -443
  169. package/dist/llms/openai.cjs +6 -530
  170. package/dist/llms/openai.d.ts +4 -123
  171. package/dist/llms/openai.js +2 -525
  172. package/dist/llms/portkey.cjs +14 -168
  173. package/dist/llms/portkey.d.ts +1 -57
  174. package/dist/llms/portkey.js +1 -162
  175. package/dist/llms/raycast.cjs +15 -94
  176. package/dist/llms/raycast.d.ts +1 -50
  177. package/dist/llms/raycast.js +1 -92
  178. package/dist/llms/replicate.cjs +15 -132
  179. package/dist/llms/replicate.d.ts +1 -51
  180. package/dist/llms/replicate.js +1 -130
  181. package/dist/llms/sagemaker_endpoint.cjs +15 -203
  182. package/dist/llms/sagemaker_endpoint.d.ts +1 -121
  183. package/dist/llms/sagemaker_endpoint.js +1 -200
  184. package/dist/llms/watsonx_ai.cjs +15 -152
  185. package/dist/llms/watsonx_ai.d.ts +1 -72
  186. package/dist/llms/watsonx_ai.js +1 -150
  187. package/dist/llms/writer.cjs +15 -165
  188. package/dist/llms/writer.d.ts +1 -60
  189. package/dist/llms/writer.js +1 -163
  190. package/dist/llms/yandex.cjs +15 -98
  191. package/dist/llms/yandex.d.ts +1 -40
  192. package/dist/llms/yandex.js +1 -96
  193. package/dist/load/index.cjs +3 -1
  194. package/dist/load/index.d.ts +1 -1
  195. package/dist/load/index.js +3 -1
  196. package/dist/retrievers/amazon_kendra.cjs +15 -273
  197. package/dist/retrievers/amazon_kendra.d.ts +1 -142
  198. package/dist/retrievers/amazon_kendra.js +1 -271
  199. package/dist/retrievers/chaindesk.cjs +15 -88
  200. package/dist/retrievers/chaindesk.d.ts +1 -31
  201. package/dist/retrievers/chaindesk.js +1 -86
  202. package/dist/retrievers/databerry.cjs +15 -82
  203. package/dist/retrievers/databerry.d.ts +1 -34
  204. package/dist/retrievers/databerry.js +1 -80
  205. package/dist/retrievers/metal.cjs +15 -51
  206. package/dist/retrievers/metal.d.ts +1 -36
  207. package/dist/retrievers/metal.js +1 -49
  208. package/dist/retrievers/supabase.cjs +15 -167
  209. package/dist/retrievers/supabase.d.ts +1 -82
  210. package/dist/retrievers/supabase.js +1 -165
  211. package/dist/retrievers/tavily_search_api.cjs +15 -142
  212. package/dist/retrievers/tavily_search_api.d.ts +1 -38
  213. package/dist/retrievers/tavily_search_api.js +1 -140
  214. package/dist/retrievers/zep.cjs +15 -150
  215. package/dist/retrievers/zep.d.ts +1 -76
  216. package/dist/retrievers/zep.js +1 -148
  217. package/dist/schema/document.cjs +2 -16
  218. package/dist/schema/document.d.ts +1 -11
  219. package/dist/schema/document.js +1 -16
  220. package/dist/schema/index.cjs +2 -7
  221. package/dist/schema/index.d.ts +2 -10
  222. package/dist/schema/index.js +1 -6
  223. package/dist/storage/convex.cjs +15 -143
  224. package/dist/storage/convex.d.ts +1 -106
  225. package/dist/storage/convex.js +1 -141
  226. package/dist/storage/ioredis.cjs +15 -158
  227. package/dist/storage/ioredis.d.ts +1 -71
  228. package/dist/storage/ioredis.js +1 -156
  229. package/dist/storage/upstash_redis.cjs +15 -172
  230. package/dist/storage/upstash_redis.d.ts +1 -78
  231. package/dist/storage/upstash_redis.js +1 -170
  232. package/dist/storage/vercel_kv.cjs +15 -161
  233. package/dist/storage/vercel_kv.d.ts +1 -62
  234. package/dist/storage/vercel_kv.js +1 -159
  235. package/dist/stores/doc/in_memory.cjs +15 -105
  236. package/dist/stores/doc/in_memory.d.ts +1 -46
  237. package/dist/stores/doc/in_memory.js +1 -102
  238. package/dist/stores/message/cassandra.cjs +15 -158
  239. package/dist/stores/message/cassandra.d.ts +1 -69
  240. package/dist/stores/message/cassandra.js +1 -156
  241. package/dist/stores/message/cloudflare_d1.cjs +15 -150
  242. package/dist/stores/message/cloudflare_d1.d.ts +1 -67
  243. package/dist/stores/message/cloudflare_d1.js +1 -148
  244. package/dist/stores/message/convex.cjs +15 -118
  245. package/dist/stores/message/convex.d.ts +1 -81
  246. package/dist/stores/message/convex.js +1 -116
  247. package/dist/stores/message/dynamodb.cjs +15 -159
  248. package/dist/stores/message/dynamodb.d.ts +1 -51
  249. package/dist/stores/message/dynamodb.js +1 -157
  250. package/dist/stores/message/firestore.cjs +15 -185
  251. package/dist/stores/message/firestore.d.ts +1 -68
  252. package/dist/stores/message/firestore.js +1 -183
  253. package/dist/stores/message/ioredis.cjs +15 -98
  254. package/dist/stores/message/ioredis.d.ts +1 -64
  255. package/dist/stores/message/ioredis.js +1 -96
  256. package/dist/stores/message/momento.cjs +15 -160
  257. package/dist/stores/message/momento.d.ts +1 -97
  258. package/dist/stores/message/momento.js +1 -158
  259. package/dist/stores/message/mongodb.cjs +15 -57
  260. package/dist/stores/message/mongodb.d.ts +1 -26
  261. package/dist/stores/message/mongodb.js +1 -55
  262. package/dist/stores/message/planetscale.cjs +15 -170
  263. package/dist/stores/message/planetscale.d.ts +1 -75
  264. package/dist/stores/message/planetscale.js +1 -168
  265. package/dist/stores/message/redis.cjs +15 -114
  266. package/dist/stores/message/redis.d.ts +1 -69
  267. package/dist/stores/message/redis.js +1 -112
  268. package/dist/stores/message/upstash_redis.cjs +15 -85
  269. package/dist/stores/message/upstash_redis.d.ts +1 -43
  270. package/dist/stores/message/upstash_redis.js +1 -83
  271. package/dist/stores/message/utils.cjs +3 -23
  272. package/dist/stores/message/utils.d.ts +1 -17
  273. package/dist/stores/message/utils.js +1 -21
  274. package/dist/stores/message/xata.cjs +15 -216
  275. package/dist/stores/message/xata.d.ts +1 -76
  276. package/dist/stores/message/xata.js +1 -214
  277. package/dist/tools/IFTTTWebhook.cjs +15 -87
  278. package/dist/tools/IFTTTWebhook.d.ts +1 -50
  279. package/dist/tools/IFTTTWebhook.js +1 -85
  280. package/dist/tools/aiplugin.cjs +15 -73
  281. package/dist/tools/aiplugin.d.ts +1 -34
  282. package/dist/tools/aiplugin.js +1 -71
  283. package/dist/tools/aws_sfn.cjs +15 -201
  284. package/dist/tools/aws_sfn.d.ts +1 -57
  285. package/dist/tools/aws_sfn.js +1 -197
  286. package/dist/tools/bingserpapi.cjs +15 -76
  287. package/dist/tools/bingserpapi.d.ts +1 -23
  288. package/dist/tools/bingserpapi.js +1 -75
  289. package/dist/tools/brave_search.cjs +15 -64
  290. package/dist/tools/brave_search.d.ts +1 -23
  291. package/dist/tools/brave_search.js +1 -62
  292. package/dist/tools/connery.cjs +15 -277
  293. package/dist/tools/connery.d.ts +1 -145
  294. package/dist/tools/connery.js +1 -274
  295. package/dist/tools/convert_to_openai.d.ts +1 -1
  296. package/dist/tools/dadjokeapi.cjs +15 -46
  297. package/dist/tools/dadjokeapi.d.ts +1 -15
  298. package/dist/tools/dadjokeapi.js +1 -45
  299. package/dist/tools/dataforseo_api_search.cjs +15 -280
  300. package/dist/tools/dataforseo_api_search.d.ts +1 -170
  301. package/dist/tools/dataforseo_api_search.js +1 -278
  302. package/dist/tools/gmail/index.cjs +15 -11
  303. package/dist/tools/gmail/index.d.ts +1 -11
  304. package/dist/tools/gmail/index.js +1 -5
  305. package/dist/tools/google_custom_search.cjs +15 -70
  306. package/dist/tools/google_custom_search.d.ts +1 -25
  307. package/dist/tools/google_custom_search.js +1 -68
  308. package/dist/tools/google_places.cjs +15 -79
  309. package/dist/tools/google_places.d.ts +1 -21
  310. package/dist/tools/google_places.js +1 -77
  311. package/dist/tools/searchapi.cjs +15 -164
  312. package/dist/tools/searchapi.d.ts +1 -91
  313. package/dist/tools/searchapi.js +1 -162
  314. package/dist/tools/searxng_search.cjs +15 -156
  315. package/dist/tools/searxng_search.d.ts +1 -114
  316. package/dist/tools/searxng_search.js +1 -154
  317. package/dist/tools/serpapi.cjs +15 -199
  318. package/dist/tools/serpapi.d.ts +1 -302
  319. package/dist/tools/serpapi.js +1 -197
  320. package/dist/tools/serper.cjs +15 -94
  321. package/dist/tools/serper.d.ts +1 -32
  322. package/dist/tools/serper.js +1 -92
  323. package/dist/tools/wikipedia_query_run.cjs +15 -133
  324. package/dist/tools/wikipedia_query_run.d.ts +1 -54
  325. package/dist/tools/wikipedia_query_run.js +1 -131
  326. package/dist/tools/wolframalpha.cjs +15 -47
  327. package/dist/tools/wolframalpha.d.ts +1 -21
  328. package/dist/tools/wolframalpha.js +1 -45
  329. package/dist/types/openai-types.d.ts +1 -1
  330. package/dist/util/convex.d.ts +1 -1
  331. package/dist/util/event-source-parse.cjs +15 -224
  332. package/dist/util/event-source-parse.d.ts +1 -39
  333. package/dist/util/event-source-parse.js +1 -219
  334. package/dist/util/math.cjs +15 -123
  335. package/dist/util/math.d.ts +1 -40
  336. package/dist/util/math.js +1 -116
  337. package/dist/util/openapi.d.ts +2 -2
  338. package/dist/vectorstores/analyticdb.cjs +3 -333
  339. package/dist/vectorstores/analyticdb.d.ts +1 -110
  340. package/dist/vectorstores/analyticdb.js +1 -317
  341. package/dist/vectorstores/base.cjs +15 -165
  342. package/dist/vectorstores/base.d.ts +1 -104
  343. package/dist/vectorstores/base.js +1 -161
  344. package/dist/vectorstores/cassandra.cjs +15 -478
  345. package/dist/vectorstores/cassandra.d.ts +1 -162
  346. package/dist/vectorstores/cassandra.js +1 -476
  347. package/dist/vectorstores/chroma.cjs +3 -312
  348. package/dist/vectorstores/chroma.d.ts +1 -133
  349. package/dist/vectorstores/chroma.js +1 -299
  350. package/dist/vectorstores/clickhouse.cjs +3 -272
  351. package/dist/vectorstores/clickhouse.d.ts +1 -126
  352. package/dist/vectorstores/clickhouse.js +1 -259
  353. package/dist/vectorstores/closevector/node.cjs +15 -107
  354. package/dist/vectorstores/closevector/node.d.ts +1 -83
  355. package/dist/vectorstores/closevector/node.js +1 -105
  356. package/dist/vectorstores/closevector/web.cjs +15 -107
  357. package/dist/vectorstores/closevector/web.d.ts +1 -80
  358. package/dist/vectorstores/closevector/web.js +1 -105
  359. package/dist/vectorstores/cloudflare_vectorize.cjs +3 -186
  360. package/dist/vectorstores/cloudflare_vectorize.d.ts +1 -90
  361. package/dist/vectorstores/cloudflare_vectorize.js +1 -173
  362. package/dist/vectorstores/convex.cjs +15 -175
  363. package/dist/vectorstores/convex.d.ts +1 -134
  364. package/dist/vectorstores/convex.js +1 -173
  365. package/dist/vectorstores/elasticsearch.cjs +3 -287
  366. package/dist/vectorstores/elasticsearch.d.ts +1 -143
  367. package/dist/vectorstores/elasticsearch.js +1 -274
  368. package/dist/vectorstores/faiss.cjs +3 -378
  369. package/dist/vectorstores/faiss.d.ts +1 -127
  370. package/dist/vectorstores/faiss.js +1 -365
  371. package/dist/vectorstores/googlevertexai.cjs +3 -538
  372. package/dist/vectorstores/googlevertexai.d.ts +1 -181
  373. package/dist/vectorstores/googlevertexai.js +1 -521
  374. package/dist/vectorstores/hnswlib.cjs +15 -273
  375. package/dist/vectorstores/hnswlib.d.ts +1 -122
  376. package/dist/vectorstores/hnswlib.js +1 -271
  377. package/dist/vectorstores/lancedb.cjs +15 -126
  378. package/dist/vectorstores/lancedb.d.ts +1 -61
  379. package/dist/vectorstores/lancedb.js +1 -124
  380. package/dist/vectorstores/memory.cjs +15 -143
  381. package/dist/vectorstores/memory.d.ts +1 -92
  382. package/dist/vectorstores/memory.js +1 -141
  383. package/dist/vectorstores/milvus.cjs +3 -592
  384. package/dist/vectorstores/milvus.d.ts +1 -139
  385. package/dist/vectorstores/milvus.js +1 -579
  386. package/dist/vectorstores/momento_vector_index.cjs +3 -305
  387. package/dist/vectorstores/momento_vector_index.d.ts +1 -143
  388. package/dist/vectorstores/momento_vector_index.js +1 -292
  389. package/dist/vectorstores/mongodb_atlas.cjs +15 -218
  390. package/dist/vectorstores/mongodb_atlas.d.ts +1 -118
  391. package/dist/vectorstores/mongodb_atlas.js +1 -216
  392. package/dist/vectorstores/myscale.cjs +3 -263
  393. package/dist/vectorstores/myscale.d.ts +1 -129
  394. package/dist/vectorstores/myscale.js +1 -250
  395. package/dist/vectorstores/neo4j_vector.cjs +3 -578
  396. package/dist/vectorstores/neo4j_vector.d.ts +1 -75
  397. package/dist/vectorstores/neo4j_vector.js +1 -562
  398. package/dist/vectorstores/opensearch.cjs +3 -274
  399. package/dist/vectorstores/opensearch.d.ts +1 -119
  400. package/dist/vectorstores/opensearch.js +1 -261
  401. package/dist/vectorstores/pgvector.cjs +14 -404
  402. package/dist/vectorstores/pgvector.d.ts +1 -160
  403. package/dist/vectorstores/pgvector.js +1 -400
  404. package/dist/vectorstores/pinecone.cjs +3 -280
  405. package/dist/vectorstores/pinecone.d.ts +1 -120
  406. package/dist/vectorstores/pinecone.js +1 -264
  407. package/dist/vectorstores/prisma.cjs +15 -305
  408. package/dist/vectorstores/prisma.d.ts +1 -180
  409. package/dist/vectorstores/prisma.js +1 -303
  410. package/dist/vectorstores/qdrant.cjs +15 -195
  411. package/dist/vectorstores/qdrant.d.ts +1 -95
  412. package/dist/vectorstores/qdrant.js +1 -193
  413. package/dist/vectorstores/redis.cjs +15 -333
  414. package/dist/vectorstores/redis.d.ts +1 -186
  415. package/dist/vectorstores/redis.js +1 -331
  416. package/dist/vectorstores/rockset.cjs +15 -351
  417. package/dist/vectorstores/rockset.d.ts +1 -202
  418. package/dist/vectorstores/rockset.js +1 -347
  419. package/dist/vectorstores/singlestore.cjs +15 -220
  420. package/dist/vectorstores/singlestore.d.ts +1 -88
  421. package/dist/vectorstores/singlestore.js +1 -218
  422. package/dist/vectorstores/supabase.cjs +15 -217
  423. package/dist/vectorstores/supabase.d.ts +1 -118
  424. package/dist/vectorstores/supabase.js +1 -215
  425. package/dist/vectorstores/tigris.cjs +3 -143
  426. package/dist/vectorstores/tigris.d.ts +1 -74
  427. package/dist/vectorstores/tigris.js +1 -130
  428. package/dist/vectorstores/typeorm.cjs +15 -259
  429. package/dist/vectorstores/typeorm.d.ts +1 -109
  430. package/dist/vectorstores/typeorm.js +1 -256
  431. package/dist/vectorstores/typesense.cjs +15 -223
  432. package/dist/vectorstores/typesense.d.ts +1 -131
  433. package/dist/vectorstores/typesense.js +1 -221
  434. package/dist/vectorstores/usearch.cjs +3 -203
  435. package/dist/vectorstores/usearch.d.ts +1 -85
  436. package/dist/vectorstores/usearch.js +1 -187
  437. package/dist/vectorstores/vectara.cjs +3 -409
  438. package/dist/vectorstores/vectara.d.ts +1 -148
  439. package/dist/vectorstores/vectara.js +1 -396
  440. package/dist/vectorstores/vercel_postgres.cjs +15 -320
  441. package/dist/vectorstores/vercel_postgres.d.ts +1 -145
  442. package/dist/vectorstores/vercel_postgres.js +1 -318
  443. package/dist/vectorstores/voy.cjs +15 -156
  444. package/dist/vectorstores/voy.d.ts +1 -73
  445. package/dist/vectorstores/voy.js +1 -154
  446. package/dist/vectorstores/weaviate.cjs +3 -347
  447. package/dist/vectorstores/weaviate.d.ts +1 -140
  448. package/dist/vectorstores/weaviate.js +1 -333
  449. package/dist/vectorstores/xata.cjs +15 -108
  450. package/dist/vectorstores/xata.d.ts +1 -70
  451. package/dist/vectorstores/xata.js +1 -106
  452. package/dist/vectorstores/zep.cjs +15 -313
  453. package/dist/vectorstores/zep.d.ts +1 -146
  454. package/dist/vectorstores/zep.js +1 -311
  455. package/package.json +4 -26
  456. package/dist/chat_models/googlevertexai/common.cjs +0 -274
  457. package/dist/chat_models/googlevertexai/common.d.ts +0 -123
  458. package/dist/chat_models/googlevertexai/common.js +0 -269
  459. package/dist/chat_models/iflytek_xinghuo/common.cjs +0 -340
  460. package/dist/chat_models/iflytek_xinghuo/common.d.ts +0 -165
  461. package/dist/chat_models/iflytek_xinghuo/common.js +0 -336
  462. package/dist/llms/googlevertexai/common.cjs +0 -168
  463. package/dist/llms/googlevertexai/common.d.ts +0 -82
  464. package/dist/llms/googlevertexai/common.js +0 -164
  465. package/dist/tools/gmail/base.cjs +0 -69
  466. package/dist/tools/gmail/base.d.ts +0 -19
  467. package/dist/tools/gmail/base.js +0 -65
  468. package/dist/tools/gmail/create_draft.cjs +0 -62
  469. package/dist/tools/gmail/create_draft.d.ts +0 -35
  470. package/dist/tools/gmail/create_draft.js +0 -58
  471. package/dist/tools/gmail/descriptions.cjs +0 -118
  472. package/dist/tools/gmail/descriptions.d.ts +0 -5
  473. package/dist/tools/gmail/descriptions.js +0 -115
  474. package/dist/tools/gmail/get_message.cjs +0 -83
  475. package/dist/tools/gmail/get_message.d.ts +0 -18
  476. package/dist/tools/gmail/get_message.js +0 -79
  477. package/dist/tools/gmail/get_thread.cjs +0 -89
  478. package/dist/tools/gmail/get_thread.d.ts +0 -18
  479. package/dist/tools/gmail/get_thread.js +0 -85
  480. package/dist/tools/gmail/search.cjs +0 -118
  481. package/dist/tools/gmail/search.d.ts +0 -29
  482. package/dist/tools/gmail/search.js +0 -114
  483. package/dist/tools/gmail/send_message.cjs +0 -74
  484. package/dist/tools/gmail/send_message.d.ts +0 -35
  485. package/dist/tools/gmail/send_message.js +0 -70
  486. package/dist/util/openai-format-fndef.cjs +0 -81
  487. package/dist/util/openai-format-fndef.d.ts +0 -44
  488. package/dist/util/openai-format-fndef.js +0 -77
  489. package/dist/util/openai.cjs +0 -21
  490. package/dist/util/openai.d.ts +0 -1
  491. package/dist/util/openai.js +0 -17
  492. package/dist/vectorstores/closevector/common.cjs +0 -128
  493. package/dist/vectorstores/closevector/common.d.ts +0 -82
  494. package/dist/vectorstores/closevector/common.js +0 -124
@@ -1,116 +1 @@
1
- import { Embeddings } from "./base.js";
2
- export class OllamaEmbeddings extends Embeddings {
3
- constructor(params) {
4
- super(params || {});
5
- Object.defineProperty(this, "model", {
6
- enumerable: true,
7
- configurable: true,
8
- writable: true,
9
- value: "llama2"
10
- });
11
- Object.defineProperty(this, "baseUrl", {
12
- enumerable: true,
13
- configurable: true,
14
- writable: true,
15
- value: "http://localhost:11434"
16
- });
17
- Object.defineProperty(this, "requestOptions", {
18
- enumerable: true,
19
- configurable: true,
20
- writable: true,
21
- value: void 0
22
- });
23
- if (params?.model) {
24
- this.model = params.model;
25
- }
26
- if (params?.baseUrl) {
27
- this.baseUrl = params.baseUrl;
28
- }
29
- if (params?.requestOptions) {
30
- this.requestOptions = this._convertOptions(params.requestOptions);
31
- }
32
- }
33
- /** convert camelCased Ollama request options like "useMMap" to
34
- * the snake_cased equivalent which the ollama API actually uses.
35
- * Used only for consistency with the llms/Ollama and chatModels/Ollama classes
36
- */
37
- _convertOptions(requestOptions) {
38
- const snakeCasedOptions = {};
39
- const mapping = {
40
- embeddingOnly: "embedding_only",
41
- f16KV: "f16_kv",
42
- frequencyPenalty: "frequency_penalty",
43
- logitsAll: "logits_all",
44
- lowVram: "low_vram",
45
- mainGpu: "main_gpu",
46
- mirostat: "mirostat",
47
- mirostatEta: "mirostat_eta",
48
- mirostatTau: "mirostat_tau",
49
- numBatch: "num_batch",
50
- numCtx: "num_ctx",
51
- numGpu: "num_gpu",
52
- numGqa: "num_gqa",
53
- numKeep: "num_keep",
54
- numThread: "num_thread",
55
- penalizeNewline: "penalize_newline",
56
- presencePenalty: "presence_penalty",
57
- repeatLastN: "repeat_last_n",
58
- repeatPenalty: "repeat_penalty",
59
- ropeFrequencyBase: "rope_frequency_base",
60
- ropeFrequencyScale: "rope_frequency_scale",
61
- temperature: "temperature",
62
- stop: "stop",
63
- tfsZ: "tfs_z",
64
- topK: "top_k",
65
- topP: "top_p",
66
- typicalP: "typical_p",
67
- useMLock: "use_mlock",
68
- useMMap: "use_mmap",
69
- vocabOnly: "vocab_only",
70
- };
71
- for (const [key, value] of Object.entries(requestOptions)) {
72
- const snakeCasedOption = mapping[key];
73
- if (snakeCasedOption) {
74
- snakeCasedOptions[snakeCasedOption] = value;
75
- }
76
- }
77
- return snakeCasedOptions;
78
- }
79
- async _request(prompt) {
80
- const { model, baseUrl, requestOptions } = this;
81
- let formattedBaseUrl = baseUrl;
82
- if (formattedBaseUrl.startsWith("http://localhost:")) {
83
- // Node 18 has issues with resolving "localhost"
84
- // See https://github.com/node-fetch/node-fetch/issues/1624
85
- formattedBaseUrl = formattedBaseUrl.replace("http://localhost:", "http://127.0.0.1:");
86
- }
87
- const response = await fetch(`${formattedBaseUrl}/api/embeddings`, {
88
- method: "POST",
89
- headers: { "Content-Type": "application/json" },
90
- body: JSON.stringify({
91
- prompt,
92
- model,
93
- options: requestOptions,
94
- }),
95
- });
96
- if (!response.ok) {
97
- throw new Error(`Request to Ollama server failed: ${response.status} ${response.statusText}`);
98
- }
99
- const json = await response.json();
100
- return json.embedding;
101
- }
102
- async _embed(strings) {
103
- const embeddings = [];
104
- for await (const prompt of strings) {
105
- const embedding = await this.caller.call(() => this._request(prompt));
106
- embeddings.push(embedding);
107
- }
108
- return embeddings;
109
- }
110
- async embedDocuments(documents) {
111
- return this._embed(documents);
112
- }
113
- async embedQuery(document) {
114
- return (await this.embedDocuments([document]))[0];
115
- }
116
- }
1
+ export * from "@langchain/community/embeddings/ollama";
@@ -1,243 +1,5 @@
1
1
  "use strict";
2
2
  Object.defineProperty(exports, "__esModule", { value: true });
3
3
  exports.OpenAIEmbeddings = void 0;
4
- const openai_1 = require("openai");
5
- const env_js_1 = require("../util/env.cjs");
6
- const chunk_js_1 = require("../util/chunk.cjs");
7
- const base_js_1 = require("./base.cjs");
8
- const azure_js_1 = require("../util/azure.cjs");
9
- const openai_js_1 = require("../util/openai.cjs");
10
- /**
11
- * Class for generating embeddings using the OpenAI API. Extends the
12
- * Embeddings class and implements OpenAIEmbeddingsParams and
13
- * AzureOpenAIInput.
14
- * @example
15
- * ```typescript
16
- * // Embed a query using OpenAIEmbeddings to generate embeddings for a given text
17
- * const model = new OpenAIEmbeddings();
18
- * const res = await model.embedQuery(
19
- * "What would be a good company name for a company that makes colorful socks?",
20
- * );
21
- * console.log({ res });
22
- *
23
- * ```
24
- */
25
- class OpenAIEmbeddings extends base_js_1.Embeddings {
26
- constructor(fields, configuration) {
27
- const fieldsWithDefaults = { maxConcurrency: 2, ...fields };
28
- super(fieldsWithDefaults);
29
- Object.defineProperty(this, "modelName", {
30
- enumerable: true,
31
- configurable: true,
32
- writable: true,
33
- value: "text-embedding-ada-002"
34
- });
35
- Object.defineProperty(this, "batchSize", {
36
- enumerable: true,
37
- configurable: true,
38
- writable: true,
39
- value: 512
40
- });
41
- Object.defineProperty(this, "stripNewLines", {
42
- enumerable: true,
43
- configurable: true,
44
- writable: true,
45
- value: true
46
- });
47
- Object.defineProperty(this, "timeout", {
48
- enumerable: true,
49
- configurable: true,
50
- writable: true,
51
- value: void 0
52
- });
53
- Object.defineProperty(this, "azureOpenAIApiVersion", {
54
- enumerable: true,
55
- configurable: true,
56
- writable: true,
57
- value: void 0
58
- });
59
- Object.defineProperty(this, "azureOpenAIApiKey", {
60
- enumerable: true,
61
- configurable: true,
62
- writable: true,
63
- value: void 0
64
- });
65
- Object.defineProperty(this, "azureOpenAIApiInstanceName", {
66
- enumerable: true,
67
- configurable: true,
68
- writable: true,
69
- value: void 0
70
- });
71
- Object.defineProperty(this, "azureOpenAIApiDeploymentName", {
72
- enumerable: true,
73
- configurable: true,
74
- writable: true,
75
- value: void 0
76
- });
77
- Object.defineProperty(this, "azureOpenAIBasePath", {
78
- enumerable: true,
79
- configurable: true,
80
- writable: true,
81
- value: void 0
82
- });
83
- Object.defineProperty(this, "organization", {
84
- enumerable: true,
85
- configurable: true,
86
- writable: true,
87
- value: void 0
88
- });
89
- Object.defineProperty(this, "client", {
90
- enumerable: true,
91
- configurable: true,
92
- writable: true,
93
- value: void 0
94
- });
95
- Object.defineProperty(this, "clientConfig", {
96
- enumerable: true,
97
- configurable: true,
98
- writable: true,
99
- value: void 0
100
- });
101
- let apiKey = fieldsWithDefaults?.openAIApiKey ??
102
- (0, env_js_1.getEnvironmentVariable)("OPENAI_API_KEY");
103
- const azureApiKey = fieldsWithDefaults?.azureOpenAIApiKey ??
104
- (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_KEY");
105
- if (!azureApiKey && !apiKey) {
106
- throw new Error("OpenAI or Azure OpenAI API key not found");
107
- }
108
- const azureApiInstanceName = fieldsWithDefaults?.azureOpenAIApiInstanceName ??
109
- (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_INSTANCE_NAME");
110
- const azureApiDeploymentName = (fieldsWithDefaults?.azureOpenAIApiEmbeddingsDeploymentName ||
111
- fieldsWithDefaults?.azureOpenAIApiDeploymentName) ??
112
- ((0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_EMBEDDINGS_DEPLOYMENT_NAME") ||
113
- (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_DEPLOYMENT_NAME"));
114
- const azureApiVersion = fieldsWithDefaults?.azureOpenAIApiVersion ??
115
- (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_API_VERSION");
116
- this.azureOpenAIBasePath =
117
- fieldsWithDefaults?.azureOpenAIBasePath ??
118
- (0, env_js_1.getEnvironmentVariable)("AZURE_OPENAI_BASE_PATH");
119
- this.organization =
120
- fieldsWithDefaults?.configuration?.organization ??
121
- (0, env_js_1.getEnvironmentVariable)("OPENAI_ORGANIZATION");
122
- this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
123
- this.batchSize =
124
- fieldsWithDefaults?.batchSize ?? (azureApiKey ? 1 : this.batchSize);
125
- this.stripNewLines =
126
- fieldsWithDefaults?.stripNewLines ?? this.stripNewLines;
127
- this.timeout = fieldsWithDefaults?.timeout;
128
- this.azureOpenAIApiVersion = azureApiVersion;
129
- this.azureOpenAIApiKey = azureApiKey;
130
- this.azureOpenAIApiInstanceName = azureApiInstanceName;
131
- this.azureOpenAIApiDeploymentName = azureApiDeploymentName;
132
- if (this.azureOpenAIApiKey) {
133
- if (!this.azureOpenAIApiInstanceName && !this.azureOpenAIBasePath) {
134
- throw new Error("Azure OpenAI API instance name not found");
135
- }
136
- if (!this.azureOpenAIApiDeploymentName) {
137
- throw new Error("Azure OpenAI API deployment name not found");
138
- }
139
- if (!this.azureOpenAIApiVersion) {
140
- throw new Error("Azure OpenAI API version not found");
141
- }
142
- apiKey = apiKey ?? "";
143
- }
144
- this.clientConfig = {
145
- apiKey,
146
- organization: this.organization,
147
- baseURL: configuration?.basePath,
148
- dangerouslyAllowBrowser: true,
149
- defaultHeaders: configuration?.baseOptions?.headers,
150
- defaultQuery: configuration?.baseOptions?.params,
151
- ...configuration,
152
- ...fields?.configuration,
153
- };
154
- }
155
- /**
156
- * Method to generate embeddings for an array of documents. Splits the
157
- * documents into batches and makes requests to the OpenAI API to generate
158
- * embeddings.
159
- * @param texts Array of documents to generate embeddings for.
160
- * @returns Promise that resolves to a 2D array of embeddings for each document.
161
- */
162
- async embedDocuments(texts) {
163
- const batches = (0, chunk_js_1.chunkArray)(this.stripNewLines ? texts.map((t) => t.replace(/\n/g, " ")) : texts, this.batchSize);
164
- const batchRequests = batches.map((batch) => this.embeddingWithRetry({
165
- model: this.modelName,
166
- input: batch,
167
- }));
168
- const batchResponses = await Promise.all(batchRequests);
169
- const embeddings = [];
170
- for (let i = 0; i < batchResponses.length; i += 1) {
171
- const batch = batches[i];
172
- const { data: batchResponse } = batchResponses[i];
173
- for (let j = 0; j < batch.length; j += 1) {
174
- embeddings.push(batchResponse[j].embedding);
175
- }
176
- }
177
- return embeddings;
178
- }
179
- /**
180
- * Method to generate an embedding for a single document. Calls the
181
- * embeddingWithRetry method with the document as the input.
182
- * @param text Document to generate an embedding for.
183
- * @returns Promise that resolves to an embedding for the document.
184
- */
185
- async embedQuery(text) {
186
- const { data } = await this.embeddingWithRetry({
187
- model: this.modelName,
188
- input: this.stripNewLines ? text.replace(/\n/g, " ") : text,
189
- });
190
- return data[0].embedding;
191
- }
192
- /**
193
- * Private method to make a request to the OpenAI API to generate
194
- * embeddings. Handles the retry logic and returns the response from the
195
- * API.
196
- * @param request Request to send to the OpenAI API.
197
- * @returns Promise that resolves to the response from the API.
198
- */
199
- async embeddingWithRetry(request) {
200
- if (!this.client) {
201
- const openAIEndpointConfig = {
202
- azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
203
- azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
204
- azureOpenAIApiKey: this.azureOpenAIApiKey,
205
- azureOpenAIBasePath: this.azureOpenAIBasePath,
206
- baseURL: this.clientConfig.baseURL,
207
- };
208
- const endpoint = (0, azure_js_1.getEndpoint)(openAIEndpointConfig);
209
- const params = {
210
- ...this.clientConfig,
211
- baseURL: endpoint,
212
- timeout: this.timeout,
213
- maxRetries: 0,
214
- };
215
- if (!params.baseURL) {
216
- delete params.baseURL;
217
- }
218
- this.client = new openai_1.OpenAI(params);
219
- }
220
- const requestOptions = {};
221
- if (this.azureOpenAIApiKey) {
222
- requestOptions.headers = {
223
- "api-key": this.azureOpenAIApiKey,
224
- ...requestOptions.headers,
225
- };
226
- requestOptions.query = {
227
- "api-version": this.azureOpenAIApiVersion,
228
- ...requestOptions.query,
229
- };
230
- }
231
- return this.caller.call(async () => {
232
- try {
233
- const res = await this.client.embeddings.create(request, requestOptions);
234
- return res;
235
- }
236
- catch (e) {
237
- const error = (0, openai_js_1.wrapOpenAIClientError)(e);
238
- throw error;
239
- }
240
- });
241
- }
242
- }
243
- exports.OpenAIEmbeddings = OpenAIEmbeddings;
4
+ var openai_1 = require("@langchain/openai");
5
+ Object.defineProperty(exports, "OpenAIEmbeddings", { enumerable: true, get: function () { return openai_1.OpenAIEmbeddings; } });
@@ -1,82 +1 @@
1
- import { type ClientOptions } from "openai";
2
- import { AzureOpenAIInput, LegacyOpenAIInput } from "../types/openai-types.js";
3
- import { Embeddings, EmbeddingsParams } from "./base.js";
4
- /**
5
- * Interface for OpenAIEmbeddings parameters. Extends EmbeddingsParams and
6
- * defines additional parameters specific to the OpenAIEmbeddings class.
7
- */
8
- export interface OpenAIEmbeddingsParams extends EmbeddingsParams {
9
- /** Model name to use */
10
- modelName: string;
11
- /**
12
- * Timeout to use when making requests to OpenAI.
13
- */
14
- timeout?: number;
15
- /**
16
- * The maximum number of documents to embed in a single request. This is
17
- * limited by the OpenAI API to a maximum of 2048.
18
- */
19
- batchSize?: number;
20
- /**
21
- * Whether to strip new lines from the input text. This is recommended by
22
- * OpenAI, but may not be suitable for all use cases.
23
- */
24
- stripNewLines?: boolean;
25
- }
26
- /**
27
- * Class for generating embeddings using the OpenAI API. Extends the
28
- * Embeddings class and implements OpenAIEmbeddingsParams and
29
- * AzureOpenAIInput.
30
- * @example
31
- * ```typescript
32
- * // Embed a query using OpenAIEmbeddings to generate embeddings for a given text
33
- * const model = new OpenAIEmbeddings();
34
- * const res = await model.embedQuery(
35
- * "What would be a good company name for a company that makes colorful socks?",
36
- * );
37
- * console.log({ res });
38
- *
39
- * ```
40
- */
41
- export declare class OpenAIEmbeddings extends Embeddings implements OpenAIEmbeddingsParams, AzureOpenAIInput {
42
- modelName: string;
43
- batchSize: number;
44
- stripNewLines: boolean;
45
- timeout?: number;
46
- azureOpenAIApiVersion?: string;
47
- azureOpenAIApiKey?: string;
48
- azureOpenAIApiInstanceName?: string;
49
- azureOpenAIApiDeploymentName?: string;
50
- azureOpenAIBasePath?: string;
51
- organization?: string;
52
- private client;
53
- private clientConfig;
54
- constructor(fields?: Partial<OpenAIEmbeddingsParams> & Partial<AzureOpenAIInput> & {
55
- verbose?: boolean;
56
- openAIApiKey?: string;
57
- configuration?: ClientOptions;
58
- }, configuration?: ClientOptions & LegacyOpenAIInput);
59
- /**
60
- * Method to generate embeddings for an array of documents. Splits the
61
- * documents into batches and makes requests to the OpenAI API to generate
62
- * embeddings.
63
- * @param texts Array of documents to generate embeddings for.
64
- * @returns Promise that resolves to a 2D array of embeddings for each document.
65
- */
66
- embedDocuments(texts: string[]): Promise<number[][]>;
67
- /**
68
- * Method to generate an embedding for a single document. Calls the
69
- * embeddingWithRetry method with the document as the input.
70
- * @param text Document to generate an embedding for.
71
- * @returns Promise that resolves to an embedding for the document.
72
- */
73
- embedQuery(text: string): Promise<number[]>;
74
- /**
75
- * Private method to make a request to the OpenAI API to generate
76
- * embeddings. Handles the retry logic and returns the response from the
77
- * API.
78
- * @param request Request to send to the OpenAI API.
79
- * @returns Promise that resolves to the response from the API.
80
- */
81
- private embeddingWithRetry;
82
- }
1
+ export { type OpenAIEmbeddingsParams, OpenAIEmbeddings, } from "@langchain/openai";
@@ -1,239 +1 @@
1
- import { OpenAI as OpenAIClient } from "openai";
2
- import { getEnvironmentVariable } from "../util/env.js";
3
- import { chunkArray } from "../util/chunk.js";
4
- import { Embeddings } from "./base.js";
5
- import { getEndpoint } from "../util/azure.js";
6
- import { wrapOpenAIClientError } from "../util/openai.js";
7
- /**
8
- * Class for generating embeddings using the OpenAI API. Extends the
9
- * Embeddings class and implements OpenAIEmbeddingsParams and
10
- * AzureOpenAIInput.
11
- * @example
12
- * ```typescript
13
- * // Embed a query using OpenAIEmbeddings to generate embeddings for a given text
14
- * const model = new OpenAIEmbeddings();
15
- * const res = await model.embedQuery(
16
- * "What would be a good company name for a company that makes colorful socks?",
17
- * );
18
- * console.log({ res });
19
- *
20
- * ```
21
- */
22
- export class OpenAIEmbeddings extends Embeddings {
23
- constructor(fields, configuration) {
24
- const fieldsWithDefaults = { maxConcurrency: 2, ...fields };
25
- super(fieldsWithDefaults);
26
- Object.defineProperty(this, "modelName", {
27
- enumerable: true,
28
- configurable: true,
29
- writable: true,
30
- value: "text-embedding-ada-002"
31
- });
32
- Object.defineProperty(this, "batchSize", {
33
- enumerable: true,
34
- configurable: true,
35
- writable: true,
36
- value: 512
37
- });
38
- Object.defineProperty(this, "stripNewLines", {
39
- enumerable: true,
40
- configurable: true,
41
- writable: true,
42
- value: true
43
- });
44
- Object.defineProperty(this, "timeout", {
45
- enumerable: true,
46
- configurable: true,
47
- writable: true,
48
- value: void 0
49
- });
50
- Object.defineProperty(this, "azureOpenAIApiVersion", {
51
- enumerable: true,
52
- configurable: true,
53
- writable: true,
54
- value: void 0
55
- });
56
- Object.defineProperty(this, "azureOpenAIApiKey", {
57
- enumerable: true,
58
- configurable: true,
59
- writable: true,
60
- value: void 0
61
- });
62
- Object.defineProperty(this, "azureOpenAIApiInstanceName", {
63
- enumerable: true,
64
- configurable: true,
65
- writable: true,
66
- value: void 0
67
- });
68
- Object.defineProperty(this, "azureOpenAIApiDeploymentName", {
69
- enumerable: true,
70
- configurable: true,
71
- writable: true,
72
- value: void 0
73
- });
74
- Object.defineProperty(this, "azureOpenAIBasePath", {
75
- enumerable: true,
76
- configurable: true,
77
- writable: true,
78
- value: void 0
79
- });
80
- Object.defineProperty(this, "organization", {
81
- enumerable: true,
82
- configurable: true,
83
- writable: true,
84
- value: void 0
85
- });
86
- Object.defineProperty(this, "client", {
87
- enumerable: true,
88
- configurable: true,
89
- writable: true,
90
- value: void 0
91
- });
92
- Object.defineProperty(this, "clientConfig", {
93
- enumerable: true,
94
- configurable: true,
95
- writable: true,
96
- value: void 0
97
- });
98
- let apiKey = fieldsWithDefaults?.openAIApiKey ??
99
- getEnvironmentVariable("OPENAI_API_KEY");
100
- const azureApiKey = fieldsWithDefaults?.azureOpenAIApiKey ??
101
- getEnvironmentVariable("AZURE_OPENAI_API_KEY");
102
- if (!azureApiKey && !apiKey) {
103
- throw new Error("OpenAI or Azure OpenAI API key not found");
104
- }
105
- const azureApiInstanceName = fieldsWithDefaults?.azureOpenAIApiInstanceName ??
106
- getEnvironmentVariable("AZURE_OPENAI_API_INSTANCE_NAME");
107
- const azureApiDeploymentName = (fieldsWithDefaults?.azureOpenAIApiEmbeddingsDeploymentName ||
108
- fieldsWithDefaults?.azureOpenAIApiDeploymentName) ??
109
- (getEnvironmentVariable("AZURE_OPENAI_API_EMBEDDINGS_DEPLOYMENT_NAME") ||
110
- getEnvironmentVariable("AZURE_OPENAI_API_DEPLOYMENT_NAME"));
111
- const azureApiVersion = fieldsWithDefaults?.azureOpenAIApiVersion ??
112
- getEnvironmentVariable("AZURE_OPENAI_API_VERSION");
113
- this.azureOpenAIBasePath =
114
- fieldsWithDefaults?.azureOpenAIBasePath ??
115
- getEnvironmentVariable("AZURE_OPENAI_BASE_PATH");
116
- this.organization =
117
- fieldsWithDefaults?.configuration?.organization ??
118
- getEnvironmentVariable("OPENAI_ORGANIZATION");
119
- this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
120
- this.batchSize =
121
- fieldsWithDefaults?.batchSize ?? (azureApiKey ? 1 : this.batchSize);
122
- this.stripNewLines =
123
- fieldsWithDefaults?.stripNewLines ?? this.stripNewLines;
124
- this.timeout = fieldsWithDefaults?.timeout;
125
- this.azureOpenAIApiVersion = azureApiVersion;
126
- this.azureOpenAIApiKey = azureApiKey;
127
- this.azureOpenAIApiInstanceName = azureApiInstanceName;
128
- this.azureOpenAIApiDeploymentName = azureApiDeploymentName;
129
- if (this.azureOpenAIApiKey) {
130
- if (!this.azureOpenAIApiInstanceName && !this.azureOpenAIBasePath) {
131
- throw new Error("Azure OpenAI API instance name not found");
132
- }
133
- if (!this.azureOpenAIApiDeploymentName) {
134
- throw new Error("Azure OpenAI API deployment name not found");
135
- }
136
- if (!this.azureOpenAIApiVersion) {
137
- throw new Error("Azure OpenAI API version not found");
138
- }
139
- apiKey = apiKey ?? "";
140
- }
141
- this.clientConfig = {
142
- apiKey,
143
- organization: this.organization,
144
- baseURL: configuration?.basePath,
145
- dangerouslyAllowBrowser: true,
146
- defaultHeaders: configuration?.baseOptions?.headers,
147
- defaultQuery: configuration?.baseOptions?.params,
148
- ...configuration,
149
- ...fields?.configuration,
150
- };
151
- }
152
- /**
153
- * Method to generate embeddings for an array of documents. Splits the
154
- * documents into batches and makes requests to the OpenAI API to generate
155
- * embeddings.
156
- * @param texts Array of documents to generate embeddings for.
157
- * @returns Promise that resolves to a 2D array of embeddings for each document.
158
- */
159
- async embedDocuments(texts) {
160
- const batches = chunkArray(this.stripNewLines ? texts.map((t) => t.replace(/\n/g, " ")) : texts, this.batchSize);
161
- const batchRequests = batches.map((batch) => this.embeddingWithRetry({
162
- model: this.modelName,
163
- input: batch,
164
- }));
165
- const batchResponses = await Promise.all(batchRequests);
166
- const embeddings = [];
167
- for (let i = 0; i < batchResponses.length; i += 1) {
168
- const batch = batches[i];
169
- const { data: batchResponse } = batchResponses[i];
170
- for (let j = 0; j < batch.length; j += 1) {
171
- embeddings.push(batchResponse[j].embedding);
172
- }
173
- }
174
- return embeddings;
175
- }
176
- /**
177
- * Method to generate an embedding for a single document. Calls the
178
- * embeddingWithRetry method with the document as the input.
179
- * @param text Document to generate an embedding for.
180
- * @returns Promise that resolves to an embedding for the document.
181
- */
182
- async embedQuery(text) {
183
- const { data } = await this.embeddingWithRetry({
184
- model: this.modelName,
185
- input: this.stripNewLines ? text.replace(/\n/g, " ") : text,
186
- });
187
- return data[0].embedding;
188
- }
189
- /**
190
- * Private method to make a request to the OpenAI API to generate
191
- * embeddings. Handles the retry logic and returns the response from the
192
- * API.
193
- * @param request Request to send to the OpenAI API.
194
- * @returns Promise that resolves to the response from the API.
195
- */
196
- async embeddingWithRetry(request) {
197
- if (!this.client) {
198
- const openAIEndpointConfig = {
199
- azureOpenAIApiDeploymentName: this.azureOpenAIApiDeploymentName,
200
- azureOpenAIApiInstanceName: this.azureOpenAIApiInstanceName,
201
- azureOpenAIApiKey: this.azureOpenAIApiKey,
202
- azureOpenAIBasePath: this.azureOpenAIBasePath,
203
- baseURL: this.clientConfig.baseURL,
204
- };
205
- const endpoint = getEndpoint(openAIEndpointConfig);
206
- const params = {
207
- ...this.clientConfig,
208
- baseURL: endpoint,
209
- timeout: this.timeout,
210
- maxRetries: 0,
211
- };
212
- if (!params.baseURL) {
213
- delete params.baseURL;
214
- }
215
- this.client = new OpenAIClient(params);
216
- }
217
- const requestOptions = {};
218
- if (this.azureOpenAIApiKey) {
219
- requestOptions.headers = {
220
- "api-key": this.azureOpenAIApiKey,
221
- ...requestOptions.headers,
222
- };
223
- requestOptions.query = {
224
- "api-version": this.azureOpenAIApiVersion,
225
- ...requestOptions.query,
226
- };
227
- }
228
- return this.caller.call(async () => {
229
- try {
230
- const res = await this.client.embeddings.create(request, requestOptions);
231
- return res;
232
- }
233
- catch (e) {
234
- const error = wrapOpenAIClientError(e);
235
- throw error;
236
- }
237
- });
238
- }
239
- }
1
+ export { OpenAIEmbeddings, } from "@langchain/openai";