@eko-ai/eko 1.0.9 → 1.0.11

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 (286) hide show
  1. package/LICENSE +21 -21
  2. package/README.md +139 -139
  3. package/dist/core/eko.d.ts +3 -2
  4. package/dist/extension/script/bing.js +0 -0
  5. package/dist/extension/script/build_dom_tree.js +661 -661
  6. package/dist/extension/script/common.js +12 -4
  7. package/dist/extension/script/duckduckgo.js +0 -0
  8. package/dist/extension/script/google.js +0 -0
  9. package/dist/extension/tools/cancel_workflow.d.ts +9 -0
  10. package/dist/extension/utils.d.ts +1 -1
  11. package/dist/extension.cjs.js +117 -111
  12. package/dist/extension.esm.js +117 -111
  13. package/dist/extension_content_script.js +79 -61
  14. package/dist/fellou.cjs.js +18 -18
  15. package/dist/fellou.esm.js +18 -18
  16. package/dist/index.cjs.js +262 -172
  17. package/dist/index.esm.js +262 -172
  18. package/dist/jest.config.js +10 -0
  19. package/dist/jest.config.js.map +1 -0
  20. package/dist/models/workflow.d.ts +3 -1
  21. package/dist/nodejs.cjs.js +1651 -1633
  22. package/dist/nodejs.esm.js +1651 -1633
  23. package/dist/rollup.config.js +171 -0
  24. package/dist/rollup.config.js.map +1 -0
  25. package/dist/script.js +10 -0
  26. package/dist/script.js.map +1 -0
  27. package/dist/services/llm/deepseek-provider.d.ts +13 -0
  28. package/dist/services/llm/provider-factory.d.ts +4 -0
  29. package/dist/services/llm/siliconflow-provider.d.ts +13 -0
  30. package/dist/services/parser/workflow-parser.d.ts +2 -1
  31. package/dist/services/workflow/generator.d.ts +3 -2
  32. package/dist/src/core/eko.js +99 -0
  33. package/dist/src/core/eko.js.map +1 -0
  34. package/dist/src/core/tool-registry.js +51 -0
  35. package/dist/src/core/tool-registry.js.map +1 -0
  36. package/dist/src/extension/content/index.js +409 -0
  37. package/dist/src/extension/content/index.js.map +1 -0
  38. package/dist/src/extension/core.js +29 -0
  39. package/dist/src/extension/core.js.map +1 -0
  40. package/dist/src/extension/index.js +12 -0
  41. package/dist/src/extension/index.js.map +1 -0
  42. package/dist/src/extension/script/bing.js +25 -0
  43. package/dist/src/extension/script/bing.js.map +1 -0
  44. package/dist/src/extension/script/build_dom_tree.js +585 -0
  45. package/dist/src/extension/script/build_dom_tree.js.map +1 -0
  46. package/dist/src/extension/script/common.js +203 -0
  47. package/dist/src/extension/script/common.js.map +1 -0
  48. package/dist/src/extension/script/duckduckgo.js +25 -0
  49. package/dist/src/extension/script/duckduckgo.js.map +1 -0
  50. package/dist/src/extension/script/google.js +26 -0
  51. package/dist/src/extension/script/google.js.map +1 -0
  52. package/dist/src/extension/tools/browser.js +174 -0
  53. package/dist/src/extension/tools/browser.js.map +1 -0
  54. package/dist/src/extension/tools/browser_use.js +186 -0
  55. package/dist/src/extension/tools/browser_use.js.map +1 -0
  56. package/dist/src/extension/tools/element_click.js +123 -0
  57. package/dist/src/extension/tools/element_click.js.map +1 -0
  58. package/dist/src/extension/tools/export_file.js +93 -0
  59. package/dist/src/extension/tools/export_file.js.map +1 -0
  60. package/dist/src/extension/tools/extract_content.js +38 -0
  61. package/dist/src/extension/tools/extract_content.js.map +1 -0
  62. package/dist/src/extension/tools/find_element_position.js +125 -0
  63. package/dist/src/extension/tools/find_element_position.js.map +1 -0
  64. package/dist/src/extension/tools/html_script.js +219 -0
  65. package/dist/src/extension/tools/html_script.js.map +1 -0
  66. package/dist/src/extension/tools/index.js +12 -0
  67. package/dist/src/extension/tools/index.js.map +1 -0
  68. package/dist/src/extension/tools/open_url.js +68 -0
  69. package/dist/src/extension/tools/open_url.js.map +1 -0
  70. package/dist/src/extension/tools/request_login.js +87 -0
  71. package/dist/src/extension/tools/request_login.js.map +1 -0
  72. package/dist/src/extension/tools/screenshot.js +26 -0
  73. package/dist/src/extension/tools/screenshot.js.map +1 -0
  74. package/dist/src/extension/tools/tab_management.js +160 -0
  75. package/dist/src/extension/tools/tab_management.js.map +1 -0
  76. package/dist/src/extension/tools/web_search.js +281 -0
  77. package/dist/src/extension/tools/web_search.js.map +1 -0
  78. package/dist/src/extension/utils.js +244 -0
  79. package/dist/src/extension/utils.js.map +1 -0
  80. package/dist/src/fellou/computer.js +104 -0
  81. package/dist/src/fellou/computer.js.map +1 -0
  82. package/dist/src/fellou/index.js +7 -0
  83. package/dist/src/fellou/index.js.map +1 -0
  84. package/dist/src/fellou/tools/computer_use.js +111 -0
  85. package/dist/src/fellou/tools/computer_use.js.map +1 -0
  86. package/dist/src/index.js +9 -0
  87. package/dist/src/index.js.map +1 -0
  88. package/dist/src/models/action.js +364 -0
  89. package/dist/src/models/action.js.map +1 -0
  90. package/dist/src/models/workflow.js +120 -0
  91. package/dist/src/models/workflow.js.map +1 -0
  92. package/dist/src/nodejs/core.js +18 -0
  93. package/dist/src/nodejs/core.js.map +1 -0
  94. package/dist/src/nodejs/index.js +6 -0
  95. package/dist/src/nodejs/index.js.map +1 -0
  96. package/dist/src/nodejs/script/build_dom_tree.js +586 -0
  97. package/dist/src/nodejs/script/build_dom_tree.js.map +1 -0
  98. package/dist/src/nodejs/tools/browser_use.js +458 -0
  99. package/dist/src/nodejs/tools/browser_use.js.map +1 -0
  100. package/dist/src/nodejs/tools/command_execute.js +65 -0
  101. package/dist/src/nodejs/tools/command_execute.js.map +1 -0
  102. package/dist/src/nodejs/tools/file_read.js +45 -0
  103. package/dist/src/nodejs/tools/file_read.js.map +1 -0
  104. package/dist/src/nodejs/tools/file_write.js +95 -0
  105. package/dist/src/nodejs/tools/file_write.js.map +1 -0
  106. package/dist/src/nodejs/tools/index.js +5 -0
  107. package/dist/src/nodejs/tools/index.js.map +1 -0
  108. package/dist/src/schemas/workflow.schema.js +64 -0
  109. package/dist/src/schemas/workflow.schema.js.map +1 -0
  110. package/dist/src/services/llm/claude-provider.js +140 -0
  111. package/dist/src/services/llm/claude-provider.js.map +1 -0
  112. package/dist/src/services/llm/deepseek-provider.js +432 -0
  113. package/dist/src/services/llm/deepseek-provider.js.map +1 -0
  114. package/dist/src/services/llm/glm-provider.js +317 -0
  115. package/dist/src/services/llm/glm-provider.js.map +1 -0
  116. package/dist/src/services/llm/openai-provider copy.js +317 -0
  117. package/dist/src/services/llm/openai-provider copy.js.map +1 -0
  118. package/dist/src/services/llm/openai-provider.js +317 -0
  119. package/dist/src/services/llm/openai-provider.js.map +1 -0
  120. package/dist/src/services/parser/workflow-parser.js +208 -0
  121. package/dist/src/services/parser/workflow-parser.js.map +1 -0
  122. package/dist/src/services/workflow/generator.js +105 -0
  123. package/dist/src/services/workflow/generator.js.map +1 -0
  124. package/dist/src/services/workflow/templates.js +42 -0
  125. package/dist/src/services/workflow/templates.js.map +1 -0
  126. package/dist/src/types/action.types.js +2 -0
  127. package/dist/src/types/action.types.js.map +1 -0
  128. package/dist/src/types/eko.types.js +2 -0
  129. package/dist/src/types/eko.types.js.map +1 -0
  130. package/dist/src/types/index.js +6 -0
  131. package/dist/src/types/index.js.map +1 -0
  132. package/dist/src/types/llm.types.js +2 -0
  133. package/dist/src/types/llm.types.js.map +1 -0
  134. package/dist/src/types/parser.types.js +2 -0
  135. package/dist/src/types/parser.types.js.map +1 -0
  136. package/dist/src/types/tools.types.js +2 -0
  137. package/dist/src/types/tools.types.js.map +1 -0
  138. package/dist/src/types/workflow.types.js +3 -0
  139. package/dist/src/types/workflow.types.js.map +1 -0
  140. package/dist/src/web/core.js +18 -0
  141. package/dist/src/web/core.js.map +1 -0
  142. package/dist/src/web/index.js +9 -0
  143. package/dist/src/web/index.js.map +1 -0
  144. package/dist/src/web/script/build_dom_tree.js +584 -0
  145. package/dist/src/web/script/build_dom_tree.js.map +1 -0
  146. package/dist/src/web/tools/browser.js +249 -0
  147. package/dist/src/web/tools/browser.js.map +1 -0
  148. package/dist/src/web/tools/browser_use.js +176 -0
  149. package/dist/src/web/tools/browser_use.js.map +1 -0
  150. package/dist/src/web/tools/element_click.js +121 -0
  151. package/dist/src/web/tools/element_click.js.map +1 -0
  152. package/dist/src/web/tools/export_file.js +74 -0
  153. package/dist/src/web/tools/export_file.js.map +1 -0
  154. package/dist/src/web/tools/extract_content.js +24 -0
  155. package/dist/src/web/tools/extract_content.js.map +1 -0
  156. package/dist/src/web/tools/find_element_position.js +121 -0
  157. package/dist/src/web/tools/find_element_position.js.map +1 -0
  158. package/dist/src/web/tools/html_script.js +219 -0
  159. package/dist/src/web/tools/html_script.js.map +1 -0
  160. package/dist/src/web/tools/index.js +8 -0
  161. package/dist/src/web/tools/index.js.map +1 -0
  162. package/dist/src/web/tools/screenshot.js +24 -0
  163. package/dist/src/web/tools/screenshot.js.map +1 -0
  164. package/dist/test/integration/claude-provider.test.js +170 -0
  165. package/dist/test/integration/claude-provider.test.js.map +1 -0
  166. package/dist/test/integration/deepseek-provider.test.js +171 -0
  167. package/dist/test/integration/deepseek-provider.test.js.map +1 -0
  168. package/dist/test/integration/glm-provider.test.js +173 -0
  169. package/dist/test/integration/glm-provider.test.js.map +1 -0
  170. package/dist/test/integration/openai-provider.test copy.js +170 -0
  171. package/dist/test/integration/openai-provider.test copy.js.map +1 -0
  172. package/dist/test/integration/openai-provider.test.js +170 -0
  173. package/dist/test/integration/openai-provider.test.js.map +1 -0
  174. package/dist/test/integration/qwen-provider.js +170 -0
  175. package/dist/test/integration/qwen-provider.js.map +1 -0
  176. package/dist/test/integration/qwen-provider.test copy.js +170 -0
  177. package/dist/test/integration/qwen-provider.test copy.js.map +1 -0
  178. package/dist/test/integration/qwen-provider.test.js +170 -0
  179. package/dist/test/integration/qwen-provider.test.js.map +1 -0
  180. package/dist/test/integration/workflow.execution.test.js +152 -0
  181. package/dist/test/integration/workflow.execution.test.js.map +1 -0
  182. package/dist/test/integration/workflow.generation-and-execution.test.js +131 -0
  183. package/dist/test/integration/workflow.generation-and-execution.test.js.map +1 -0
  184. package/dist/test/integration/workflow.generator.test.js +207 -0
  185. package/dist/test/integration/workflow.generator.test.js.map +1 -0
  186. package/dist/test/unit/action.test.js +186 -0
  187. package/dist/test/unit/action.test.js.map +1 -0
  188. package/dist/test/unit/tool-registry.test.js +99 -0
  189. package/dist/test/unit/tool-registry.test.js.map +1 -0
  190. package/dist/test/unit/workflow-parser.test.js +189 -0
  191. package/dist/test/unit/workflow-parser.test.js.map +1 -0
  192. package/dist/test/unit/workflow.test.js +102 -0
  193. package/dist/test/unit/workflow.test.js.map +1 -0
  194. package/dist/types/action.types.d.ts +2 -0
  195. package/dist/types/eko.types.d.ts +4 -1
  196. package/dist/types/jest.config.d.ts +10 -0
  197. package/dist/types/llm.types.d.ts +0 -2
  198. package/dist/types/rollup.config.d.ts +16 -0
  199. package/dist/types/script.d.ts +1 -0
  200. package/dist/types/src/core/eko.d.ts +20 -0
  201. package/dist/types/src/core/tool-registry.d.ts +13 -0
  202. package/dist/types/src/extension/content/index.d.ts +16 -0
  203. package/dist/types/src/extension/core.d.ts +11 -0
  204. package/dist/types/src/extension/index.d.ts +7 -0
  205. package/dist/types/src/extension/script/bing.d.ts +0 -0
  206. package/dist/types/src/extension/script/build_dom_tree.d.ts +38 -0
  207. package/dist/types/src/extension/script/common.d.ts +0 -0
  208. package/dist/types/src/extension/script/duckduckgo.d.ts +0 -0
  209. package/dist/types/src/extension/script/google.d.ts +0 -0
  210. package/dist/types/src/extension/tools/browser.d.ts +22 -0
  211. package/dist/types/src/extension/tools/browser_use.d.ts +19 -0
  212. package/dist/types/src/extension/tools/element_click.d.ts +12 -0
  213. package/dist/types/src/extension/tools/export_file.d.ts +18 -0
  214. package/dist/types/src/extension/tools/extract_content.d.ts +18 -0
  215. package/dist/types/src/extension/tools/find_element_position.d.ts +12 -0
  216. package/dist/types/src/extension/tools/html_script.d.ts +10 -0
  217. package/dist/types/src/extension/tools/index.d.ts +11 -0
  218. package/dist/types/src/extension/tools/open_url.d.ts +18 -0
  219. package/dist/{extension/tools/form_autofill.d.ts → types/src/extension/tools/request_login.d.ts} +3 -4
  220. package/dist/types/src/extension/tools/screenshot.d.ts +18 -0
  221. package/dist/types/src/extension/tools/tab_management.d.ts +19 -0
  222. package/dist/types/src/extension/tools/web_search.d.ts +18 -0
  223. package/dist/types/src/extension/utils.d.ts +31 -0
  224. package/dist/types/src/fellou/computer.d.ts +20 -0
  225. package/dist/types/src/fellou/index.d.ts +6 -0
  226. package/dist/types/src/fellou/tools/computer_use.d.ts +18 -0
  227. package/dist/types/src/index.d.ts +8 -0
  228. package/dist/types/src/models/action.d.ts +22 -0
  229. package/dist/types/src/models/workflow.d.ts +16 -0
  230. package/dist/types/src/nodejs/core.d.ts +2 -0
  231. package/dist/types/src/nodejs/index.d.ts +3 -0
  232. package/dist/types/src/nodejs/script/build_dom_tree.d.ts +1 -0
  233. package/dist/types/src/nodejs/tools/browser_use.d.ts +28 -0
  234. package/dist/types/src/nodejs/tools/command_execute.d.ts +12 -0
  235. package/dist/types/src/nodejs/tools/file_read.d.ts +11 -0
  236. package/dist/types/src/nodejs/tools/file_write.d.ts +15 -0
  237. package/dist/types/src/nodejs/tools/index.d.ts +4 -0
  238. package/dist/types/src/schemas/workflow.schema.d.ts +88 -0
  239. package/dist/types/src/services/llm/claude-provider.d.ts +11 -0
  240. package/dist/types/src/services/llm/deepseek-provider.d.ts +15 -0
  241. package/dist/types/src/services/llm/glm-provider.d.ts +11 -0
  242. package/dist/types/src/services/llm/openai-provider copy.d.ts +11 -0
  243. package/dist/types/src/services/llm/openai-provider.d.ts +11 -0
  244. package/dist/types/src/services/parser/workflow-parser.d.ts +29 -0
  245. package/dist/types/src/services/workflow/generator.d.ts +13 -0
  246. package/dist/types/src/services/workflow/templates.d.ts +8 -0
  247. package/dist/types/src/types/action.types.d.ts +38 -0
  248. package/dist/types/src/types/eko.types.d.ts +21 -0
  249. package/dist/types/src/types/index.d.ts +5 -0
  250. package/dist/types/src/types/llm.types.d.ts +54 -0
  251. package/dist/types/src/types/parser.types.d.ts +9 -0
  252. package/dist/types/src/types/tools.types.d.ts +88 -0
  253. package/dist/types/src/types/workflow.types.d.ts +39 -0
  254. package/dist/types/src/web/core.d.ts +2 -0
  255. package/dist/types/src/web/index.d.ts +5 -0
  256. package/dist/types/src/web/script/build_dom_tree.d.ts +10 -0
  257. package/dist/types/src/web/tools/browser.d.ts +21 -0
  258. package/dist/types/src/web/tools/browser_use.d.ts +19 -0
  259. package/dist/types/src/web/tools/element_click.d.ts +12 -0
  260. package/dist/types/src/web/tools/export_file.d.ts +18 -0
  261. package/dist/types/src/web/tools/extract_content.d.ts +17 -0
  262. package/dist/types/src/web/tools/find_element_position.d.ts +12 -0
  263. package/dist/types/src/web/tools/html_script.d.ts +10 -0
  264. package/dist/types/src/web/tools/index.d.ts +7 -0
  265. package/dist/types/src/web/tools/screenshot.d.ts +18 -0
  266. package/dist/types/test/integration/claude-provider.test.d.ts +1 -0
  267. package/dist/types/test/integration/deepseek-provider.test.d.ts +1 -0
  268. package/dist/types/test/integration/glm-provider.test.d.ts +1 -0
  269. package/dist/types/test/integration/openai-provider.test copy.d.ts +1 -0
  270. package/dist/types/test/integration/openai-provider.test.d.ts +1 -0
  271. package/dist/types/test/integration/qwen-provider.d.ts +1 -0
  272. package/dist/types/test/integration/qwen-provider.test copy.d.ts +1 -0
  273. package/dist/types/test/integration/qwen-provider.test.d.ts +1 -0
  274. package/dist/types/test/integration/workflow.execution.test.d.ts +1 -0
  275. package/dist/types/test/integration/workflow.generation-and-execution.test.d.ts +1 -0
  276. package/dist/types/test/integration/workflow.generator.test.d.ts +1 -0
  277. package/dist/types/test/unit/action.test.d.ts +1 -0
  278. package/dist/types/test/unit/tool-registry.test.d.ts +1 -0
  279. package/dist/types/test/unit/workflow-parser.test.d.ts +1 -0
  280. package/dist/types/test/unit/workflow.test.d.ts +1 -0
  281. package/dist/universal_tools/human/text.d.ts +9 -0
  282. package/dist/web.cjs.js +128 -121
  283. package/dist/web.esm.js +128 -121
  284. package/package.json +106 -106
  285. package/dist/fellou/tools/index.d.ts +0 -2
  286. package/dist/types/framework.types.d.ts +0 -11
@@ -0,0 +1,171 @@
1
+ import { DeepseekProvider } from '../../src/services/llm/deepseek-provider';
2
+ import dotenv from 'dotenv';
3
+ dotenv.config();
4
+ const OPENAI_API_KEY = process.env.OPENAI_API_KEY;
5
+ const OPENAI_BASE_URL = "https://api.deepseek.com";
6
+ if (!OPENAI_API_KEY) {
7
+ throw new Error('OPENAI_API_KEY environment variable is required for integration tests');
8
+ }
9
+ // Only run these tests if explicitly enabled
10
+ const ENABLE_INTEGRATION_TESTS = process.env.ENABLE_INTEGRATION_TESTS === 'true';
11
+ const describeIntegration = ENABLE_INTEGRATION_TESTS ? describe : describe.skip;
12
+ // Default model for all tests
13
+ const DEFAULT_MODEL = 'deepseek-chat';
14
+ describeIntegration('DeepseekProvider Integration', () => {
15
+ let provider;
16
+ beforeAll(() => {
17
+ provider = new DeepseekProvider(OPENAI_API_KEY, DEFAULT_MODEL, {
18
+ baseURL: OPENAI_BASE_URL,
19
+ });
20
+ });
21
+ describe('generateText', () => {
22
+ const params = {
23
+ temperature: 0.7,
24
+ maxTokens: 1000,
25
+ };
26
+ const toolParams = {
27
+ ...params,
28
+ tools: [
29
+ {
30
+ name: 'calculate',
31
+ description: 'Performs a calculation and returns the result',
32
+ input_schema: {
33
+ type: 'object',
34
+ properties: {
35
+ expression: {
36
+ type: 'string',
37
+ description: 'The mathematical expression to calculate',
38
+ },
39
+ },
40
+ required: ['expression'],
41
+ },
42
+ },
43
+ ],
44
+ toolChoice: { type: 'auto' },
45
+ };
46
+ test('should generate simple text response', async () => {
47
+ const messages = [
48
+ {
49
+ role: 'user',
50
+ content: 'What is 2+2? Please respond with just the number.',
51
+ },
52
+ ];
53
+ const result = await provider.generateText(messages, params);
54
+ expect(result.textContent).toBe('4');
55
+ expect(result.toolCalls).toHaveLength(0);
56
+ expect(result.stop_reason).toBe('stop');
57
+ }, 30000);
58
+ test('should use tools when provided', async () => {
59
+ const messages = [
60
+ {
61
+ role: 'user',
62
+ content: 'What is 234 * 456?',
63
+ },
64
+ ];
65
+ const result = await provider.generateText(messages, toolParams);
66
+ expect(result.toolCalls).toHaveLength(1);
67
+ expect(result.toolCalls[0].name).toBe('calculate');
68
+ expect(result.toolCalls[0].input).toHaveProperty('expression');
69
+ expect(result.stop_reason).toBe('tool_calls');
70
+ }, 30000);
71
+ it('should handle multi-turn conversation', async () => {
72
+ const user_message = {
73
+ role: 'user',
74
+ content: 'What is 234 * 456?',
75
+ };
76
+ const messages_1 = [user_message];
77
+ const result_1 = await provider.generateText(messages_1, toolParams);
78
+ const tool_use_id = result_1.toolCalls[0].id;
79
+ const messages_2 = [
80
+ user_message,
81
+ {
82
+ role: 'assistant',
83
+ content: result_1.content,
84
+ },
85
+ {
86
+ role: 'user',
87
+ content: [
88
+ {
89
+ type: 'tool_result',
90
+ tool_use_id: tool_use_id,
91
+ content: [{ type: 'text', text: '106704' }],
92
+ },
93
+ ],
94
+ },
95
+ ];
96
+ const result = await provider.generateText(messages_2, toolParams);
97
+ console.log(result);
98
+ expect(result.textContent).toMatch(/106,?704/);
99
+ expect(result.stop_reason).toBe('stop');
100
+ }, 30000);
101
+ });
102
+ describe('generateStream', () => {
103
+ it('should stream text content', async () => {
104
+ const accumulated = [];
105
+ let isStarted = false;
106
+ let isCompleted = false;
107
+ const handler = {
108
+ onStart: () => {
109
+ isStarted = true;
110
+ },
111
+ onContent: (content) => {
112
+ accumulated.push(content);
113
+ },
114
+ onComplete: () => {
115
+ isCompleted = true;
116
+ },
117
+ onError: (error) => {
118
+ throw error;
119
+ },
120
+ };
121
+ const messages = [
122
+ {
123
+ role: 'user',
124
+ content: 'Count from 1 to 3, with each number on a new line (\n).',
125
+ },
126
+ ];
127
+ await provider.generateStream(messages, {
128
+ temperature: 0,
129
+ maxTokens: 100,
130
+ }, handler);
131
+ expect(isStarted).toBe(true);
132
+ expect(isCompleted).toBe(true);
133
+ expect(accumulated.join('')).toMatch(/1\s+2\s+3\s?/);
134
+ }, 30000);
135
+ it('should stream tool use', async () => {
136
+ const toolCalls = [];
137
+ const handler = {
138
+ onToolUse: (toolCall) => {
139
+ toolCalls.push(toolCall);
140
+ },
141
+ };
142
+ const messages = [
143
+ {
144
+ role: 'user',
145
+ content: 'What is 123 + 456?',
146
+ },
147
+ ];
148
+ await provider.generateStream(messages, {
149
+ temperature: 0,
150
+ tools: [
151
+ {
152
+ name: 'calculate',
153
+ description: 'Performs a calculation',
154
+ input_schema: {
155
+ type: 'object',
156
+ properties: {
157
+ expression: { type: 'string' },
158
+ },
159
+ required: ['expression'],
160
+ },
161
+ },
162
+ ],
163
+ toolChoice: { type: 'auto' },
164
+ }, handler);
165
+ expect(toolCalls).toHaveLength(1);
166
+ expect(toolCalls[0].name).toBe('calculate');
167
+ expect(toolCalls[0].input).toHaveProperty('expression');
168
+ }, 30000);
169
+ });
170
+ });
171
+ //# sourceMappingURL=deepseek-provider.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"deepseek-provider.test.js","sourceRoot":"","sources":["../../../test/integration/deepseek-provider.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAC;AAE5E,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,MAAM,CAAC,MAAM,EAAE,CAAC;AAEhB,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;AAClD,MAAM,eAAe,GAAG,0BAA0B,CAAC;AACnD,IAAI,CAAC,cAAc,EAAE,CAAC;IACpB,MAAM,IAAI,KAAK,CAAC,uEAAuE,CAAC,CAAC;AAC3F,CAAC;AAED,6CAA6C;AAC7C,MAAM,wBAAwB,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,KAAK,MAAM,CAAC;AACjF,MAAM,mBAAmB,GAAG,wBAAwB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;AAEhF,8BAA8B;AAC9B,MAAM,aAAa,GAAG,eAAe,CAAC;AAEtC,mBAAmB,CAAC,8BAA8B,EAAE,GAAG,EAAE;IACvD,IAAI,QAA0B,CAAC;IAE/B,SAAS,CAAC,GAAG,EAAE;QACb,QAAQ,GAAG,IAAI,gBAAgB,CAAC,cAAc,EAAE,aAAa,EAAE;YAC7D,OAAO,EAAE,eAAe;SACzB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,MAAM,MAAM,GAAkB;YAC5B,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,IAAI;SAChB,CAAC;QACF,MAAM,UAAU,GAAkB;YAChC,GAAG,MAAM;YACT,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,WAAW,EAAE,+CAA+C;oBAC5D,YAAY,EAAE;wBACZ,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,UAAU,EAAE;gCACV,IAAI,EAAE,QAAQ;gCACd,WAAW,EAAE,0CAA0C;6BACxD;yBACF;wBACD,QAAQ,EAAE,CAAC,YAAY,CAAC;qBACzB;iBACF;aACF;YACD,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;SAC7B,CAAC;QAEF,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,mDAAmD;iBAC7D;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAI,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,oBAAoB;iBAC9B;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAChD,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,YAAY,GAAY;gBAC5B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,oBAAoB;aAC9B,CAAC;YACF,MAAM,UAAU,GAAc,CAAC,YAAY,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YACrE,MAAM,WAAW,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7C,MAAM,UAAU,GAAc;gBAC5B,YAAY;gBACZ;oBACE,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO;iBAC1B;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,aAAa;4BACnB,WAAW,EAAE,WAAW;4BACxB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;yBAC5C;qBACF;iBACF;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YACnE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACpB,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,WAAW,GAAa,EAAE,CAAC;YACjC,IAAI,SAAS,GAAG,KAAK,CAAC;YACtB,IAAI,WAAW,GAAG,KAAK,CAAC;YAExB,MAAM,OAAO,GAAqB;gBAChC,OAAO,EAAE,GAAG,EAAE;oBACZ,SAAS,GAAG,IAAI,CAAC;gBACnB,CAAC;gBACD,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;oBACrB,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC5B,CAAC;gBACD,UAAU,EAAE,GAAG,EAAE;oBACf,WAAW,GAAG,IAAI,CAAC;gBACrB,CAAC;gBACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,MAAM,KAAK,CAAC;gBACd,CAAC;aACF,CAAC;YAEF,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,yDAAyD;iBACnE;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,cAAc,CAC3B,QAAQ,EACR;gBACE,WAAW,EAAE,CAAC;gBACd,SAAS,EAAE,GAAG;aACf,EACD,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,SAAS,GAAU,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAqB;gBAChC,SAAS,EAAE,CAAC,QAAQ,EAAE,EAAE;oBACtB,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC3B,CAAC;aACF,CAAC;YAEF,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,oBAAoB;iBAC9B;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,cAAc,CAC3B,QAAQ,EACR;gBACE,WAAW,EAAE,CAAC;gBACd,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,WAAW;wBACjB,WAAW,EAAE,wBAAwB;wBACrC,YAAY,EAAE;4BACZ,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE;gCACV,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;6BAC/B;4BACD,QAAQ,EAAE,CAAC,YAAY,CAAC;yBACzB;qBACF;iBACF;gBACD,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;aAC7B,EACD,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC1D,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,173 @@
1
+ import { OpenaiProvider } from '../../src/services/llm/openai-provider';
2
+ import dotenv from 'dotenv';
3
+ dotenv.config();
4
+ const OPENAI_API_KEY = process.env.OPENAI_API_KEY;
5
+ const OPENAI_BASE_URL = "https://open.bigmodel.cn/api/paas/v4/";
6
+ if (!OPENAI_API_KEY) {
7
+ throw new Error('OPENAI_API_KEY environment variable is required for integration tests');
8
+ }
9
+ // Only run these tests if explicitly enabled
10
+ const ENABLE_INTEGRATION_TESTS = process.env.ENABLE_INTEGRATION_TESTS === 'true';
11
+ const describeIntegration = ENABLE_INTEGRATION_TESTS ? describe : describe.skip;
12
+ // Default model for all tests
13
+ const DEFAULT_MODEL = 'glm-4v-flash';
14
+ describeIntegration('OpenaiProvider Integration', () => {
15
+ let provider;
16
+ beforeAll(() => {
17
+ provider = new OpenaiProvider(OPENAI_API_KEY, DEFAULT_MODEL, {
18
+ baseURL: OPENAI_BASE_URL,
19
+ });
20
+ });
21
+ describe('generateText', () => {
22
+ const params = {
23
+ temperature: 0.7,
24
+ maxTokens: 1000,
25
+ };
26
+ const toolParams = {
27
+ ...params,
28
+ tools: [
29
+ {
30
+ name: 'calculate',
31
+ description: 'Performs a calculation and returns the result',
32
+ input_schema: {
33
+ type: 'object',
34
+ properties: {
35
+ expression: {
36
+ type: 'string',
37
+ description: 'The mathematical expression to calculate',
38
+ },
39
+ },
40
+ required: ['expression'],
41
+ },
42
+ },
43
+ ],
44
+ toolChoice: { type: 'auto' },
45
+ };
46
+ test('should generate simple text response', async () => {
47
+ const messages = [
48
+ {
49
+ role: 'user',
50
+ content: 'What is 2+2? Please respond with just the number.',
51
+ },
52
+ ];
53
+ const result = await provider.generateText(messages, params);
54
+ expect(result.textContent).toBe('4');
55
+ expect(result.toolCalls).toHaveLength(0);
56
+ expect(result.stop_reason).toBe('stop');
57
+ }, 30000);
58
+ test('should use tools when provided', async () => {
59
+ const messages = [
60
+ {
61
+ role: 'user',
62
+ content: 'What is 114514 * 1919810?',
63
+ },
64
+ ];
65
+ console.log(messages);
66
+ console.log(toolParams);
67
+ const result = await provider.generateText(messages, toolParams);
68
+ console.log(result);
69
+ expect(result.toolCalls).toHaveLength(1);
70
+ expect(result.toolCalls[0].name).toBe('calculate');
71
+ expect(result.toolCalls[0].input).toHaveProperty('expression');
72
+ expect(result.stop_reason).toBe('tool_calls');
73
+ }, 30000);
74
+ it('should handle multi-turn conversation', async () => {
75
+ const user_message = {
76
+ role: 'user',
77
+ content: 'What is 234 * 456?',
78
+ };
79
+ const messages_1 = [user_message];
80
+ const result_1 = await provider.generateText(messages_1, toolParams);
81
+ const tool_use_id = result_1.toolCalls[0].id;
82
+ const messages_2 = [
83
+ user_message,
84
+ {
85
+ role: 'assistant',
86
+ content: result_1.content,
87
+ },
88
+ {
89
+ role: 'user',
90
+ content: [
91
+ {
92
+ type: 'tool_result',
93
+ tool_use_id: tool_use_id,
94
+ content: [{ type: 'text', text: '106704' }],
95
+ },
96
+ ],
97
+ },
98
+ ];
99
+ const result = await provider.generateText(messages_2, toolParams);
100
+ expect(result.textContent).toMatch(/106,?704/);
101
+ expect(result.stop_reason).toBe('stop');
102
+ }, 30000);
103
+ });
104
+ describe('generateStream', () => {
105
+ it('should stream text content', async () => {
106
+ const accumulated = [];
107
+ let isStarted = false;
108
+ let isCompleted = false;
109
+ const handler = {
110
+ onStart: () => {
111
+ isStarted = true;
112
+ },
113
+ onContent: (content) => {
114
+ accumulated.push(content);
115
+ },
116
+ onComplete: () => {
117
+ isCompleted = true;
118
+ },
119
+ onError: (error) => {
120
+ throw error;
121
+ },
122
+ };
123
+ const messages = [
124
+ {
125
+ role: 'user',
126
+ content: 'Count from 1 to 3, with each number on a new line (\n).',
127
+ },
128
+ ];
129
+ await provider.generateStream(messages, {
130
+ temperature: 0,
131
+ maxTokens: 100,
132
+ }, handler);
133
+ expect(isStarted).toBe(true);
134
+ expect(isCompleted).toBe(true);
135
+ expect(accumulated.join('')).toMatch(/1\s+2\s+3\s?/);
136
+ }, 30000);
137
+ it('should stream tool use', async () => {
138
+ const toolCalls = [];
139
+ const handler = {
140
+ onToolUse: (toolCall) => {
141
+ toolCalls.push(toolCall);
142
+ },
143
+ };
144
+ const messages = [
145
+ {
146
+ role: 'user',
147
+ content: 'What is 123 + 456?',
148
+ },
149
+ ];
150
+ await provider.generateStream(messages, {
151
+ temperature: 0,
152
+ tools: [
153
+ {
154
+ name: 'calculate',
155
+ description: 'Performs a calculation',
156
+ input_schema: {
157
+ type: 'object',
158
+ properties: {
159
+ expression: { type: 'string' },
160
+ },
161
+ required: ['expression'],
162
+ },
163
+ },
164
+ ],
165
+ toolChoice: { type: 'auto' },
166
+ }, handler);
167
+ expect(toolCalls).toHaveLength(1);
168
+ expect(toolCalls[0].name).toBe('calculate');
169
+ expect(toolCalls[0].input).toHaveProperty('expression');
170
+ }, 30000);
171
+ });
172
+ });
173
+ //# sourceMappingURL=glm-provider.test.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"glm-provider.test.js","sourceRoot":"","sources":["../../../test/integration/glm-provider.test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAExE,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,MAAM,CAAC,MAAM,EAAE,CAAC;AAEhB,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;AAClD,MAAM,eAAe,GAAG,uCAAuC,CAAC;AAChE,IAAI,CAAC,cAAc,EAAE,CAAC;IACpB,MAAM,IAAI,KAAK,CAAC,uEAAuE,CAAC,CAAC;AAC3F,CAAC;AAED,6CAA6C;AAC7C,MAAM,wBAAwB,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,KAAK,MAAM,CAAC;AACjF,MAAM,mBAAmB,GAAG,wBAAwB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;AAEhF,8BAA8B;AAC9B,MAAM,aAAa,GAAG,cAAc,CAAC;AAErC,mBAAmB,CAAC,4BAA4B,EAAE,GAAG,EAAE;IACrD,IAAI,QAAwB,CAAC;IAE7B,SAAS,CAAC,GAAG,EAAE;QACb,QAAQ,GAAG,IAAI,cAAc,CAAC,cAAc,EAAE,aAAa,EAAE;YAC3D,OAAO,EAAE,eAAe;SACzB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,MAAM,MAAM,GAAkB;YAC5B,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,IAAI;SAChB,CAAC;QACF,MAAM,UAAU,GAAkB;YAChC,GAAG,MAAM;YACT,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,WAAW,EAAE,+CAA+C;oBAC5D,YAAY,EAAE;wBACZ,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,UAAU,EAAE;gCACV,IAAI,EAAE,QAAQ;gCACd,WAAW,EAAE,0CAA0C;6BACxD;yBACF;wBACD,QAAQ,EAAE,CAAC,YAAY,CAAC;qBACzB;iBACF;aACF;YACD,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;SAC7B,CAAC;QAEF,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,mDAAmD;iBAC7D;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAI,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,2BAA2B;iBACrC;aACF,CAAC;YAEF,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;YACtB,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;YACxB,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACjE,OAAO,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;YACpB,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAChD,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,YAAY,GAAY;gBAC5B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,oBAAoB;aAC9B,CAAC;YACF,MAAM,UAAU,GAAc,CAAC,YAAY,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YACrE,MAAM,WAAW,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7C,MAAM,UAAU,GAAc;gBAC5B,YAAY;gBACZ;oBACE,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO;iBAC1B;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,aAAa;4BACnB,WAAW,EAAE,WAAW;4BACxB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;yBAC5C;qBACF;iBACF;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,WAAW,GAAa,EAAE,CAAC;YACjC,IAAI,SAAS,GAAG,KAAK,CAAC;YACtB,IAAI,WAAW,GAAG,KAAK,CAAC;YAExB,MAAM,OAAO,GAAqB;gBAChC,OAAO,EAAE,GAAG,EAAE;oBACZ,SAAS,GAAG,IAAI,CAAC;gBACnB,CAAC;gBACD,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;oBACrB,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC5B,CAAC;gBACD,UAAU,EAAE,GAAG,EAAE;oBACf,WAAW,GAAG,IAAI,CAAC;gBACrB,CAAC;gBACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,MAAM,KAAK,CAAC;gBACd,CAAC;aACF,CAAC;YAEF,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,yDAAyD;iBACnE;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,cAAc,CAC3B,QAAQ,EACR;gBACE,WAAW,EAAE,CAAC;gBACd,SAAS,EAAE,GAAG;aACf,EACD,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,SAAS,GAAU,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAqB;gBAChC,SAAS,EAAE,CAAC,QAAQ,EAAE,EAAE;oBACtB,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC3B,CAAC;aACF,CAAC;YAEF,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,oBAAoB;iBAC9B;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,cAAc,CAC3B,QAAQ,EACR;gBACE,WAAW,EAAE,CAAC;gBACd,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,WAAW;wBACjB,WAAW,EAAE,wBAAwB;wBACrC,YAAY,EAAE;4BACZ,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE;gCACV,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;6BAC/B;4BACD,QAAQ,EAAE,CAAC,YAAY,CAAC;yBACzB;qBACF;iBACF;gBACD,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;aAC7B,EACD,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC1D,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}
@@ -0,0 +1,170 @@
1
+ import { OpenaiProvider } from '../../src/services/llm/openai-provider';
2
+ import dotenv from 'dotenv';
3
+ dotenv.config();
4
+ const OPENAI_API_KEY = process.env.OPENAI_API_KEY;
5
+ const OPENAI_BASE_URL = process.env.OPENAI_BASE_URL;
6
+ if (!OPENAI_API_KEY) {
7
+ throw new Error('OPENAI_API_KEY environment variable is required for integration tests');
8
+ }
9
+ // Only run these tests if explicitly enabled
10
+ const ENABLE_INTEGRATION_TESTS = process.env.ENABLE_INTEGRATION_TESTS === 'true';
11
+ const describeIntegration = ENABLE_INTEGRATION_TESTS ? describe : describe.skip;
12
+ // Default model for all tests
13
+ const DEFAULT_MODEL = 'gpt-4o-mini';
14
+ describeIntegration('OpenaiProvider Integration', () => {
15
+ let provider;
16
+ beforeAll(() => {
17
+ provider = new OpenaiProvider(OPENAI_API_KEY, DEFAULT_MODEL, {
18
+ baseURL: OPENAI_BASE_URL,
19
+ });
20
+ });
21
+ describe('generateText', () => {
22
+ const params = {
23
+ temperature: 0.7,
24
+ maxTokens: 1000,
25
+ };
26
+ const toolParams = {
27
+ ...params,
28
+ tools: [
29
+ {
30
+ name: 'calculate',
31
+ description: 'Performs a calculation and returns the result',
32
+ input_schema: {
33
+ type: 'object',
34
+ properties: {
35
+ expression: {
36
+ type: 'string',
37
+ description: 'The mathematical expression to calculate',
38
+ },
39
+ },
40
+ required: ['expression'],
41
+ },
42
+ },
43
+ ],
44
+ toolChoice: { type: 'auto' },
45
+ };
46
+ test('should generate simple text response', async () => {
47
+ const messages = [
48
+ {
49
+ role: 'user',
50
+ content: 'What is 2+2? Please respond with just the number.',
51
+ },
52
+ ];
53
+ const result = await provider.generateText(messages, params);
54
+ expect(result.textContent).toBe('4');
55
+ expect(result.toolCalls).toHaveLength(0);
56
+ expect(result.stop_reason).toBe('stop');
57
+ }, 30000);
58
+ test('should use tools when provided', async () => {
59
+ const messages = [
60
+ {
61
+ role: 'user',
62
+ content: 'What is 234 * 456?',
63
+ },
64
+ ];
65
+ const result = await provider.generateText(messages, toolParams);
66
+ expect(result.toolCalls).toHaveLength(1);
67
+ expect(result.toolCalls[0].name).toBe('calculate');
68
+ expect(result.toolCalls[0].input).toHaveProperty('expression');
69
+ expect(result.stop_reason).toBe('tool_calls');
70
+ }, 30000);
71
+ it('should handle multi-turn conversation', async () => {
72
+ const user_message = {
73
+ role: 'user',
74
+ content: 'What is 234 * 456?',
75
+ };
76
+ const messages_1 = [user_message];
77
+ const result_1 = await provider.generateText(messages_1, toolParams);
78
+ const tool_use_id = result_1.toolCalls[0].id;
79
+ const messages_2 = [
80
+ user_message,
81
+ {
82
+ role: 'assistant',
83
+ content: result_1.content,
84
+ },
85
+ {
86
+ role: 'user',
87
+ content: [
88
+ {
89
+ type: 'tool_result',
90
+ tool_use_id: tool_use_id,
91
+ content: [{ type: 'text', text: '106704' }],
92
+ },
93
+ ],
94
+ },
95
+ ];
96
+ const result = await provider.generateText(messages_2, toolParams);
97
+ expect(result.textContent).toMatch(/106,?704/);
98
+ expect(result.stop_reason).toBe('stop');
99
+ }, 30000);
100
+ });
101
+ describe('generateStream', () => {
102
+ it('should stream text content', async () => {
103
+ const accumulated = [];
104
+ let isStarted = false;
105
+ let isCompleted = false;
106
+ const handler = {
107
+ onStart: () => {
108
+ isStarted = true;
109
+ },
110
+ onContent: (content) => {
111
+ accumulated.push(content);
112
+ },
113
+ onComplete: () => {
114
+ isCompleted = true;
115
+ },
116
+ onError: (error) => {
117
+ throw error;
118
+ },
119
+ };
120
+ const messages = [
121
+ {
122
+ role: 'user',
123
+ content: 'Count from 1 to 3, with each number on a new line (\n).',
124
+ },
125
+ ];
126
+ await provider.generateStream(messages, {
127
+ temperature: 0,
128
+ maxTokens: 100,
129
+ }, handler);
130
+ expect(isStarted).toBe(true);
131
+ expect(isCompleted).toBe(true);
132
+ expect(accumulated.join('')).toMatch(/1\s+2\s+3\s?/);
133
+ }, 30000);
134
+ it('should stream tool use', async () => {
135
+ const toolCalls = [];
136
+ const handler = {
137
+ onToolUse: (toolCall) => {
138
+ toolCalls.push(toolCall);
139
+ },
140
+ };
141
+ const messages = [
142
+ {
143
+ role: 'user',
144
+ content: 'What is 123 + 456?',
145
+ },
146
+ ];
147
+ await provider.generateStream(messages, {
148
+ temperature: 0,
149
+ tools: [
150
+ {
151
+ name: 'calculate',
152
+ description: 'Performs a calculation',
153
+ input_schema: {
154
+ type: 'object',
155
+ properties: {
156
+ expression: { type: 'string' },
157
+ },
158
+ required: ['expression'],
159
+ },
160
+ },
161
+ ],
162
+ toolChoice: { type: 'auto' },
163
+ }, handler);
164
+ expect(toolCalls).toHaveLength(1);
165
+ expect(toolCalls[0].name).toBe('calculate');
166
+ expect(toolCalls[0].input).toHaveProperty('expression');
167
+ }, 30000);
168
+ });
169
+ });
170
+ //# sourceMappingURL=openai-provider.test%20copy.js.map
@@ -0,0 +1 @@
1
+ {"version":3,"file":"openai-provider.test copy.js","sourceRoot":"","sources":["../../../test/integration/openai-provider.test copy.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,cAAc,EAAE,MAAM,wCAAwC,CAAC;AAExE,OAAO,MAAM,MAAM,QAAQ,CAAC;AAE5B,MAAM,CAAC,MAAM,EAAE,CAAC;AAEhB,MAAM,cAAc,GAAG,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;AAClD,MAAM,eAAe,GAAG,OAAO,CAAC,GAAG,CAAC,eAAe,CAAC;AACpD,IAAI,CAAC,cAAc,EAAE,CAAC;IACpB,MAAM,IAAI,KAAK,CAAC,uEAAuE,CAAC,CAAC;AAC3F,CAAC;AAED,6CAA6C;AAC7C,MAAM,wBAAwB,GAAG,OAAO,CAAC,GAAG,CAAC,wBAAwB,KAAK,MAAM,CAAC;AACjF,MAAM,mBAAmB,GAAG,wBAAwB,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC;AAEhF,8BAA8B;AAC9B,MAAM,aAAa,GAAG,aAAa,CAAC;AAEpC,mBAAmB,CAAC,4BAA4B,EAAE,GAAG,EAAE;IACrD,IAAI,QAAwB,CAAC;IAE7B,SAAS,CAAC,GAAG,EAAE;QACb,QAAQ,GAAG,IAAI,cAAc,CAAC,cAAc,EAAE,aAAa,EAAE;YAC3D,OAAO,EAAE,eAAe;SACzB,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,MAAM,MAAM,GAAkB;YAC5B,WAAW,EAAE,GAAG;YAChB,SAAS,EAAE,IAAI;SAChB,CAAC;QACF,MAAM,UAAU,GAAkB;YAChC,GAAG,MAAM;YACT,KAAK,EAAE;gBACL;oBACE,IAAI,EAAE,WAAW;oBACjB,WAAW,EAAE,+CAA+C;oBAC5D,YAAY,EAAE;wBACZ,IAAI,EAAE,QAAQ;wBACd,UAAU,EAAE;4BACV,UAAU,EAAE;gCACV,IAAI,EAAE,QAAQ;gCACd,WAAW,EAAE,0CAA0C;6BACxD;yBACF;wBACD,QAAQ,EAAE,CAAC,YAAY,CAAC;qBACzB;iBACF;aACF;YACD,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;SAC7B,CAAC;QAEF,IAAI,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;YACtD,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,mDAAmD;iBAC7D;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;YAC7D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACrC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,IAAI,CAAC,gCAAgC,EAAE,KAAK,IAAI,EAAE;YAChD,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,oBAAoB;iBAC9B;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YACnD,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;YAC/D,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAChD,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,EAAE,CAAC,uCAAuC,EAAE,KAAK,IAAI,EAAE;YACrD,MAAM,YAAY,GAAY;gBAC5B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,oBAAoB;aAC9B,CAAC;YACF,MAAM,UAAU,GAAc,CAAC,YAAY,CAAC,CAAC;YAC7C,MAAM,QAAQ,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YACrE,MAAM,WAAW,GAAG,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7C,MAAM,UAAU,GAAc;gBAC5B,YAAY;gBACZ;oBACE,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,QAAQ,CAAC,OAAO;iBAC1B;gBACD;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE;wBACP;4BACE,IAAI,EAAE,aAAa;4BACnB,WAAW,EAAE,WAAW;4BACxB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;yBAC5C;qBACF;iBACF;aACF,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,QAAQ,CAAC,YAAY,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YACnE,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAC/C,MAAM,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC1C,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;QAC9B,EAAE,CAAC,4BAA4B,EAAE,KAAK,IAAI,EAAE;YAC1C,MAAM,WAAW,GAAa,EAAE,CAAC;YACjC,IAAI,SAAS,GAAG,KAAK,CAAC;YACtB,IAAI,WAAW,GAAG,KAAK,CAAC;YAExB,MAAM,OAAO,GAAqB;gBAChC,OAAO,EAAE,GAAG,EAAE;oBACZ,SAAS,GAAG,IAAI,CAAC;gBACnB,CAAC;gBACD,SAAS,EAAE,CAAC,OAAO,EAAE,EAAE;oBACrB,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBAC5B,CAAC;gBACD,UAAU,EAAE,GAAG,EAAE;oBACf,WAAW,GAAG,IAAI,CAAC;gBACrB,CAAC;gBACD,OAAO,EAAE,CAAC,KAAK,EAAE,EAAE;oBACjB,MAAM,KAAK,CAAC;gBACd,CAAC;aACF,CAAC;YAEF,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,yDAAyD;iBACnE;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,cAAc,CAC3B,QAAQ,EACR;gBACE,WAAW,EAAE,CAAC;gBACd,SAAS,EAAE,GAAG;aACf,EACD,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC7B,MAAM,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAC/B,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QACvD,CAAC,EAAE,KAAK,CAAC,CAAC;QAEV,EAAE,CAAC,wBAAwB,EAAE,KAAK,IAAI,EAAE;YACtC,MAAM,SAAS,GAAU,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAqB;gBAChC,SAAS,EAAE,CAAC,QAAQ,EAAE,EAAE;oBACtB,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBAC3B,CAAC;aACF,CAAC;YAEF,MAAM,QAAQ,GAAc;gBAC1B;oBACE,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,oBAAoB;iBAC9B;aACF,CAAC;YAEF,MAAM,QAAQ,CAAC,cAAc,CAC3B,QAAQ,EACR;gBACE,WAAW,EAAE,CAAC;gBACd,KAAK,EAAE;oBACL;wBACE,IAAI,EAAE,WAAW;wBACjB,WAAW,EAAE,wBAAwB;wBACrC,YAAY,EAAE;4BACZ,IAAI,EAAE,QAAQ;4BACd,UAAU,EAAE;gCACV,UAAU,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE;6BAC/B;4BACD,QAAQ,EAAE,CAAC,YAAY,CAAC;yBACzB;qBACF;iBACF;gBACD,UAAU,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE;aAC7B,EACD,OAAO,CACR,CAAC;YAEF,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YAClC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,cAAc,CAAC,YAAY,CAAC,CAAC;QAC1D,CAAC,EAAE,KAAK,CAAC,CAAC;IACZ,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}