@lobehub/chat 1.57.1 → 1.59.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 (92) hide show
  1. package/CHANGELOG.md +50 -0
  2. package/changelog/v1.json +18 -0
  3. package/docker-compose/local/docker-compose.yml +1 -0
  4. package/locales/ar/modelProvider.json +24 -0
  5. package/locales/ar/models.json +60 -0
  6. package/locales/ar/providers.json +15 -0
  7. package/locales/bg-BG/modelProvider.json +24 -0
  8. package/locales/bg-BG/models.json +60 -0
  9. package/locales/bg-BG/providers.json +15 -0
  10. package/locales/de-DE/modelProvider.json +24 -0
  11. package/locales/de-DE/models.json +60 -0
  12. package/locales/de-DE/providers.json +15 -0
  13. package/locales/en-US/modelProvider.json +24 -0
  14. package/locales/en-US/models.json +60 -0
  15. package/locales/en-US/providers.json +15 -0
  16. package/locales/es-ES/modelProvider.json +24 -0
  17. package/locales/es-ES/models.json +60 -0
  18. package/locales/es-ES/providers.json +15 -0
  19. package/locales/fa-IR/modelProvider.json +24 -0
  20. package/locales/fa-IR/models.json +60 -0
  21. package/locales/fa-IR/providers.json +15 -0
  22. package/locales/fr-FR/modelProvider.json +24 -0
  23. package/locales/fr-FR/models.json +60 -0
  24. package/locales/fr-FR/providers.json +15 -0
  25. package/locales/it-IT/modelProvider.json +24 -0
  26. package/locales/it-IT/models.json +60 -0
  27. package/locales/it-IT/providers.json +15 -0
  28. package/locales/ja-JP/modelProvider.json +24 -0
  29. package/locales/ja-JP/models.json +60 -0
  30. package/locales/ja-JP/providers.json +15 -0
  31. package/locales/ko-KR/modelProvider.json +24 -0
  32. package/locales/ko-KR/models.json +60 -0
  33. package/locales/ko-KR/providers.json +15 -0
  34. package/locales/nl-NL/modelProvider.json +24 -0
  35. package/locales/nl-NL/models.json +60 -0
  36. package/locales/nl-NL/providers.json +15 -0
  37. package/locales/pl-PL/modelProvider.json +24 -0
  38. package/locales/pl-PL/models.json +60 -0
  39. package/locales/pl-PL/providers.json +15 -0
  40. package/locales/pt-BR/modelProvider.json +24 -0
  41. package/locales/pt-BR/models.json +60 -0
  42. package/locales/pt-BR/providers.json +15 -0
  43. package/locales/ru-RU/modelProvider.json +24 -0
  44. package/locales/ru-RU/models.json +60 -0
  45. package/locales/ru-RU/providers.json +15 -0
  46. package/locales/tr-TR/modelProvider.json +24 -0
  47. package/locales/tr-TR/models.json +60 -0
  48. package/locales/tr-TR/providers.json +15 -0
  49. package/locales/vi-VN/modelProvider.json +24 -0
  50. package/locales/vi-VN/models.json +60 -0
  51. package/locales/vi-VN/providers.json +15 -0
  52. package/locales/zh-CN/modelProvider.json +24 -0
  53. package/locales/zh-CN/models.json +1109 -1049
  54. package/locales/zh-CN/providers.json +82 -67
  55. package/locales/zh-TW/modelProvider.json +24 -0
  56. package/locales/zh-TW/models.json +60 -0
  57. package/locales/zh-TW/providers.json +15 -0
  58. package/next.config.ts +5 -0
  59. package/package.json +4 -2
  60. package/src/app/[variants]/(main)/settings/provider/(detail)/azureai/page.tsx +58 -0
  61. package/src/app/[variants]/(main)/settings/provider/(list)/ProviderGrid/index.tsx +6 -3
  62. package/src/app/[variants]/(main)/settings/provider/features/CreateNewProvider/index.tsx +13 -2
  63. package/src/app/[variants]/(main)/settings/provider/features/ModelList/CreateNewModelModal/Form.tsx +6 -8
  64. package/src/app/[variants]/(main)/settings/provider/features/ModelList/CreateNewModelModal/index.tsx +5 -6
  65. package/src/app/[variants]/(main)/settings/provider/features/ModelList/ModelConfigModal/index.tsx +4 -3
  66. package/src/app/[variants]/(main)/settings/provider/features/ModelList/ProviderSettingsContext.ts +2 -0
  67. package/src/app/[variants]/(main)/settings/provider/features/ModelList/index.tsx +6 -7
  68. package/src/app/[variants]/(main)/settings/provider/features/ProviderConfig/index.tsx +1 -1
  69. package/src/config/aiModels/azureai.ts +18 -0
  70. package/src/config/aiModels/index.ts +6 -0
  71. package/src/config/aiModels/volcengine.ts +83 -0
  72. package/src/config/llm.ts +6 -0
  73. package/src/config/modelProviders/azure.ts +2 -1
  74. package/src/config/modelProviders/azureai.ts +19 -0
  75. package/src/config/modelProviders/index.ts +6 -0
  76. package/src/config/modelProviders/volcengine.ts +23 -0
  77. package/src/database/server/models/aiProvider.ts +2 -0
  78. package/src/libs/agent-runtime/AgentRuntime.ts +17 -8
  79. package/src/libs/agent-runtime/azureai/index.ts +109 -0
  80. package/src/libs/agent-runtime/baichuan/index.test.ts +8 -250
  81. package/src/libs/agent-runtime/cloudflare/index.ts +22 -18
  82. package/src/libs/agent-runtime/index.ts +2 -1
  83. package/src/libs/agent-runtime/types/type.ts +5 -0
  84. package/src/libs/agent-runtime/utils/streams/__snapshots__/protocol.test.ts.snap +331 -0
  85. package/src/libs/agent-runtime/utils/streams/protocol.test.ts +137 -0
  86. package/src/libs/agent-runtime/utils/streams/protocol.ts +34 -0
  87. package/src/libs/agent-runtime/{doubao → volcengine}/index.ts +3 -3
  88. package/src/locales/default/modelProvider.ts +25 -0
  89. package/src/server/modules/AgentRuntime/index.ts +8 -1
  90. package/src/services/chat.ts +16 -4
  91. package/src/types/aiProvider.ts +5 -0
  92. package/src/types/user/settings/keyVaults.ts +2 -0
@@ -1,255 +1,13 @@
1
1
  // @vitest-environment node
2
- import OpenAI from 'openai';
3
- import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
2
+ import { ModelProvider } from '@/libs/agent-runtime';
4
3
 
5
- import {
6
- ChatStreamCallbacks,
7
- LobeOpenAICompatibleRuntime,
8
- ModelProvider,
9
- } from '@/libs/agent-runtime';
10
-
11
- import * as debugStreamModule from '../utils/debugStream';
4
+ import { testProvider } from '../providerTestUtils';
12
5
  import { LobeBaichuanAI } from './index';
13
6
 
14
- const provider = ModelProvider.Baichuan;
15
- const defaultBaseURL = 'https://api.baichuan-ai.com/v1';
16
-
17
- const bizErrorType = 'ProviderBizError';
18
- const invalidErrorType = 'InvalidProviderAPIKey';
19
-
20
- // Mock the console.error to avoid polluting test output
21
- vi.spyOn(console, 'error').mockImplementation(() => {});
22
-
23
- let instance: LobeOpenAICompatibleRuntime;
24
-
25
- beforeEach(() => {
26
- instance = new LobeBaichuanAI({ apiKey: 'test' });
27
-
28
- // 使用 vi.spyOn 来模拟 chat.completions.create 方法
29
- vi.spyOn(instance['client'].chat.completions, 'create').mockResolvedValue(
30
- new ReadableStream() as any,
31
- );
32
- });
33
-
34
- afterEach(() => {
35
- vi.clearAllMocks();
36
- });
37
-
38
- describe('LobeBaichuanAI', () => {
39
- describe('init', () => {
40
- it('should correctly initialize with an API key', async () => {
41
- const instance = new LobeBaichuanAI({ apiKey: 'test_api_key' });
42
- expect(instance).toBeInstanceOf(LobeBaichuanAI);
43
- expect(instance.baseURL).toEqual(defaultBaseURL);
44
- });
45
- });
46
-
47
- describe('chat', () => {
48
- describe('Error', () => {
49
- it('should return OpenAIBizError with an openai error response when OpenAI.APIError is thrown', async () => {
50
- // Arrange
51
- const apiError = new OpenAI.APIError(
52
- 400,
53
- {
54
- status: 400,
55
- error: {
56
- message: 'Bad Request',
57
- },
58
- },
59
- 'Error message',
60
- {},
61
- );
62
-
63
- vi.spyOn(instance['client'].chat.completions, 'create').mockRejectedValue(apiError);
64
-
65
- // Act
66
- try {
67
- await instance.chat({
68
- messages: [{ content: 'Hello', role: 'user' }],
69
- model: 'Baichuan4',
70
- temperature: 0,
71
- });
72
- } catch (e) {
73
- expect(e).toEqual({
74
- endpoint: defaultBaseURL,
75
- error: {
76
- error: { message: 'Bad Request' },
77
- status: 400,
78
- },
79
- errorType: bizErrorType,
80
- provider,
81
- });
82
- }
83
- });
84
-
85
- it('should throw AgentRuntimeError with NoOpenAIAPIKey if no apiKey is provided', async () => {
86
- try {
87
- new LobeBaichuanAI({});
88
- } catch (e) {
89
- expect(e).toEqual({ errorType: invalidErrorType });
90
- }
91
- });
92
-
93
- it('should return OpenAIBizError with the cause when OpenAI.APIError is thrown with cause', async () => {
94
- // Arrange
95
- const errorInfo = {
96
- stack: 'abc',
97
- cause: {
98
- message: 'api is undefined',
99
- },
100
- };
101
- const apiError = new OpenAI.APIError(400, errorInfo, 'module error', {});
102
-
103
- vi.spyOn(instance['client'].chat.completions, 'create').mockRejectedValue(apiError);
104
-
105
- // Act
106
- try {
107
- await instance.chat({
108
- messages: [{ content: 'Hello', role: 'user' }],
109
- model: 'Baichuan4',
110
- temperature: 0,
111
- });
112
- } catch (e) {
113
- expect(e).toEqual({
114
- endpoint: defaultBaseURL,
115
- error: {
116
- cause: { message: 'api is undefined' },
117
- stack: 'abc',
118
- },
119
- errorType: bizErrorType,
120
- provider,
121
- });
122
- }
123
- });
124
-
125
- it('should return OpenAIBizError with an cause response with desensitize Url', async () => {
126
- // Arrange
127
- const errorInfo = {
128
- stack: 'abc',
129
- cause: { message: 'api is undefined' },
130
- };
131
- const apiError = new OpenAI.APIError(400, errorInfo, 'module error', {});
132
-
133
- instance = new LobeBaichuanAI({
134
- apiKey: 'test',
135
-
136
- baseURL: 'https://api.abc.com/v1',
137
- });
138
-
139
- vi.spyOn(instance['client'].chat.completions, 'create').mockRejectedValue(apiError);
140
-
141
- // Act
142
- try {
143
- await instance.chat({
144
- messages: [{ content: 'Hello', role: 'user' }],
145
- model: 'Baichuan4',
146
- temperature: 0,
147
- });
148
- } catch (e) {
149
- expect(e).toEqual({
150
- endpoint: 'https://api.***.com/v1',
151
- error: {
152
- cause: { message: 'api is undefined' },
153
- stack: 'abc',
154
- },
155
- errorType: bizErrorType,
156
- provider,
157
- });
158
- }
159
- });
160
-
161
- it('should throw an InvalidBaichuanAPIKey error type on 401 status code', async () => {
162
- // Mock the API call to simulate a 401 error
163
- const error = new Error('Unauthorized') as any;
164
- error.status = 401;
165
- vi.mocked(instance['client'].chat.completions.create).mockRejectedValue(error);
166
-
167
- try {
168
- await instance.chat({
169
- messages: [{ content: 'Hello', role: 'user' }],
170
- model: 'Baichuan4',
171
- temperature: 0,
172
- });
173
- } catch (e) {
174
- // Expect the chat method to throw an error with InvalidBaichuanAPIKey
175
- expect(e).toEqual({
176
- endpoint: defaultBaseURL,
177
- error: new Error('Unauthorized'),
178
- errorType: invalidErrorType,
179
- provider,
180
- });
181
- }
182
- });
183
-
184
- it('should return AgentRuntimeError for non-OpenAI errors', async () => {
185
- // Arrange
186
- const genericError = new Error('Generic Error');
187
-
188
- vi.spyOn(instance['client'].chat.completions, 'create').mockRejectedValue(genericError);
189
-
190
- // Act
191
- try {
192
- await instance.chat({
193
- messages: [{ content: 'Hello', role: 'user' }],
194
- model: 'Baichuan4',
195
- temperature: 0,
196
- });
197
- } catch (e) {
198
- expect(e).toEqual({
199
- endpoint: defaultBaseURL,
200
- errorType: 'AgentRuntimeError',
201
- provider,
202
- error: {
203
- name: genericError.name,
204
- cause: genericError.cause,
205
- message: genericError.message,
206
- stack: genericError.stack,
207
- },
208
- });
209
- }
210
- });
211
- });
212
-
213
- describe('DEBUG', () => {
214
- it('should call debugStream and return StreamingTextResponse when DEBUG_BAICHUAN_CHAT_COMPLETION is 1', async () => {
215
- // Arrange
216
- const mockProdStream = new ReadableStream() as any; // 模拟的 prod 流
217
- const mockDebugStream = new ReadableStream({
218
- start(controller) {
219
- controller.enqueue('Debug stream content');
220
- controller.close();
221
- },
222
- }) as any;
223
- mockDebugStream.toReadableStream = () => mockDebugStream; // 添加 toReadableStream 方法
224
-
225
- // 模拟 chat.completions.create 返回值,包括模拟的 tee 方法
226
- (instance['client'].chat.completions.create as Mock).mockResolvedValue({
227
- tee: () => [mockProdStream, { toReadableStream: () => mockDebugStream }],
228
- });
229
-
230
- // 保存原始环境变量值
231
- const originalDebugValue = process.env.DEBUG_BAICHUAN_CHAT_COMPLETION;
232
-
233
- // 模拟环境变量
234
- process.env.DEBUG_BAICHUAN_CHAT_COMPLETION = '1';
235
- vi.spyOn(debugStreamModule, 'debugStream').mockImplementation(() => Promise.resolve());
236
-
237
- // 执行测试
238
- // 运行你的测试函数,确保它会在条件满足时调用 debugStream
239
- // 假设的测试函数调用,你可能需要根据实际情况调整
240
- await instance.chat({
241
- messages: [{ content: 'Hello', role: 'user' }],
242
- model: 'Baichuan4',
243
- stream: true,
244
- temperature: 0,
245
- });
246
-
247
- // 验证 debugStream 被调用
248
- expect(debugStreamModule.debugStream).toHaveBeenCalled();
249
-
250
- // 恢复原始环境变量值
251
- process.env.DEBUG_BAICHUAN_CHAT_COMPLETION = originalDebugValue;
252
- });
253
- });
254
- });
7
+ testProvider({
8
+ Runtime: LobeBaichuanAI,
9
+ provider: ModelProvider.Baichuan,
10
+ defaultBaseURL: 'https://api.baichuan-ai.com/v1',
11
+ chatDebugEnv: 'DEBUG_BAICHUAN_CHAT_COMPLETION',
12
+ chatModel: 'hunyuan-lite',
255
13
  });
@@ -1,3 +1,5 @@
1
+ import { ChatModelCard } from '@/types/llm';
2
+
1
3
  import { LobeRuntimeAI } from '../BaseAI';
2
4
  import { AgentRuntimeErrorType } from '../error';
3
5
  import { ChatCompetitionOptions, ChatStreamPayload, ModelProvider } from '../types';
@@ -12,8 +14,6 @@ import { debugStream } from '../utils/debugStream';
12
14
  import { StreamingResponse } from '../utils/response';
13
15
  import { createCallbacksTransformer } from '../utils/streams';
14
16
 
15
- import { ChatModelCard } from '@/types/llm';
16
-
17
17
  export interface CloudflareModelCard {
18
18
  description: string;
19
19
  name: string;
@@ -34,7 +34,7 @@ export class LobeCloudflareAI implements LobeRuntimeAI {
34
34
  accountID: string;
35
35
  apiKey?: string;
36
36
 
37
- constructor({ apiKey, baseURLOrAccountID }: LobeCloudflareParams) {
37
+ constructor({ apiKey, baseURLOrAccountID }: LobeCloudflareParams = {}) {
38
38
  if (!baseURLOrAccountID) {
39
39
  throw AgentRuntimeError.createError(AgentRuntimeErrorType.InvalidProviderAPIKey);
40
40
  }
@@ -128,30 +128,34 @@ export class LobeCloudflareAI implements LobeRuntimeAI {
128
128
 
129
129
  return modelList
130
130
  .map((model) => {
131
- const knownModel = LOBE_DEFAULT_MODEL_LIST.find((m) => model.name.toLowerCase() === m.id.toLowerCase());
131
+ const knownModel = LOBE_DEFAULT_MODEL_LIST.find(
132
+ (m) => model.name.toLowerCase() === m.id.toLowerCase(),
133
+ );
132
134
 
133
135
  return {
134
136
  contextWindowTokens: model.properties?.max_total_tokens
135
137
  ? Number(model.properties.max_total_tokens)
136
- : knownModel?.contextWindowTokens ?? undefined,
137
- displayName: knownModel?.displayName ?? (model.properties?.["beta"] === "true" ? `${model.name} (Beta)` : undefined),
138
+ : (knownModel?.contextWindowTokens ?? undefined),
139
+ displayName:
140
+ knownModel?.displayName ??
141
+ (model.properties?.['beta'] === 'true' ? `${model.name} (Beta)` : undefined),
138
142
  enabled: knownModel?.enabled || false,
139
143
  functionCall:
140
- model.description.toLowerCase().includes('function call')
141
- || model.properties?.["function_calling"] === "true"
142
- || knownModel?.abilities?.functionCall
143
- || false,
144
+ model.description.toLowerCase().includes('function call') ||
145
+ model.properties?.['function_calling'] === 'true' ||
146
+ knownModel?.abilities?.functionCall ||
147
+ false,
144
148
  id: model.name,
145
149
  reasoning:
146
- model.name.toLowerCase().includes('deepseek-r1')
147
- || knownModel?.abilities?.reasoning
148
- || false,
150
+ model.name.toLowerCase().includes('deepseek-r1') ||
151
+ knownModel?.abilities?.reasoning ||
152
+ false,
149
153
  vision:
150
- model.name.toLowerCase().includes('vision')
151
- || model.task?.name.toLowerCase().includes('image-to-text')
152
- || model.description.toLowerCase().includes('vision')
153
- || knownModel?.abilities?.vision
154
- || false,
154
+ model.name.toLowerCase().includes('vision') ||
155
+ model.task?.name.toLowerCase().includes('image-to-text') ||
156
+ model.description.toLowerCase().includes('vision') ||
157
+ knownModel?.abilities?.vision ||
158
+ false,
155
159
  };
156
160
  })
157
161
  .filter(Boolean) as ChatModelCard[];
@@ -1,10 +1,10 @@
1
1
  export { default as AgentRuntime } from './AgentRuntime';
2
2
  export { LobeAnthropicAI } from './anthropic';
3
+ export { LobeAzureAI } from './azureai';
3
4
  export { LobeAzureOpenAI } from './azureOpenai';
4
5
  export * from './BaseAI';
5
6
  export { LobeBedrockAI } from './bedrock';
6
7
  export { LobeDeepSeekAI } from './deepseek';
7
- export { LobeDoubaoAI } from './doubao';
8
8
  export * from './error';
9
9
  export { LobeGoogleAI } from './google';
10
10
  export { LobeGroq } from './groq';
@@ -19,5 +19,6 @@ export { LobeQwenAI } from './qwen';
19
19
  export { LobeTogetherAI } from './togetherai';
20
20
  export * from './types';
21
21
  export { AgentRuntimeError } from './utils/createError';
22
+ export { LobeVolcengineAI } from './volcengine';
22
23
  export { LobeZeroOneAI } from './zeroone';
23
24
  export { LobeZhipuAI } from './zhipu';
@@ -26,10 +26,14 @@ export enum ModelProvider {
26
26
  Ai360 = 'ai360',
27
27
  Anthropic = 'anthropic',
28
28
  Azure = 'azure',
29
+ AzureAI = 'azureai',
29
30
  Baichuan = 'baichuan',
30
31
  Bedrock = 'bedrock',
31
32
  Cloudflare = 'cloudflare',
32
33
  DeepSeek = 'deepseek',
34
+ /**
35
+ * @deprecated
36
+ */
33
37
  Doubao = 'doubao',
34
38
  FireworksAI = 'fireworksai',
35
39
  GiteeAI = 'giteeai',
@@ -61,6 +65,7 @@ export enum ModelProvider {
61
65
  TogetherAI = 'togetherai',
62
66
  Upstage = 'upstage',
63
67
  VLLM = 'vllm',
68
+ Volcengine = 'volcengine',
64
69
  Wenxin = 'wenxin',
65
70
  XAI = 'xai',
66
71
  ZeroOne = 'zeroone',