@robota-sdk/agent-provider 3.0.0-beta.64
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.
- package/LICENSE +21 -0
- package/dist/browser/index.d.ts +1104 -0
- package/dist/browser/index.d.ts.map +1 -0
- package/dist/browser/index.js +7 -0
- package/dist/browser/index.js.map +1 -0
- package/dist/loggers/index.cjs +1 -0
- package/dist/loggers/index.d.ts +151 -0
- package/dist/loggers/index.d.ts.map +1 -0
- package/dist/loggers/index.js +2 -0
- package/dist/loggers/index.js.map +1 -0
- package/dist/node/anthropic/index.cjs +1 -0
- package/dist/node/anthropic/index.d.ts +158 -0
- package/dist/node/anthropic/index.d.ts.map +1 -0
- package/dist/node/anthropic/index.js +1 -0
- package/dist/node/anthropic--1vgLC-e.js +5 -0
- package/dist/node/anthropic--1vgLC-e.js.map +1 -0
- package/dist/node/anthropic-BFQ6DSCP.cjs +4 -0
- package/dist/node/bytedance/index.cjs +1 -0
- package/dist/node/bytedance/index.d.ts +74 -0
- package/dist/node/bytedance/index.d.ts.map +1 -0
- package/dist/node/bytedance/index.js +1 -0
- package/dist/node/bytedance-C_0sF_pJ.js +2 -0
- package/dist/node/bytedance-C_0sF_pJ.js.map +1 -0
- package/dist/node/bytedance-DVPxqEiC.cjs +1 -0
- package/dist/node/chunk-Bmb41Sf3.cjs +1 -0
- package/dist/node/deepseek/index.cjs +1 -0
- package/dist/node/deepseek/index.d.ts +2 -0
- package/dist/node/deepseek/index.js +1 -0
- package/dist/node/deepseek-_8Ixx7rA.js +2 -0
- package/dist/node/deepseek-_8Ixx7rA.js.map +1 -0
- package/dist/node/deepseek-oA2Y6bD0.cjs +1 -0
- package/dist/node/gemini/index.cjs +1 -0
- package/dist/node/gemini/index.d.ts +173 -0
- package/dist/node/gemini/index.d.ts.map +1 -0
- package/dist/node/gemini/index.js +1 -0
- package/dist/node/gemini-Bh2U87MY.js +4 -0
- package/dist/node/gemini-Bh2U87MY.js.map +1 -0
- package/dist/node/gemini-DSaNCxZj.cjs +3 -0
- package/dist/node/gemma/index.cjs +1 -0
- package/dist/node/gemma/index.d.ts +2 -0
- package/dist/node/gemma/index.js +1 -0
- package/dist/node/gemma-Dp_AfCUR.js +2 -0
- package/dist/node/gemma-Dp_AfCUR.js.map +1 -0
- package/dist/node/gemma-G-Pf_PnX.cjs +1 -0
- package/dist/node/google/index.cjs +1 -0
- package/dist/node/google/index.d.ts +14 -0
- package/dist/node/google/index.d.ts.map +1 -0
- package/dist/node/google/index.js +2 -0
- package/dist/node/google/index.js.map +1 -0
- package/dist/node/index-B6PnlDMd.d.ts +82 -0
- package/dist/node/index-B6PnlDMd.d.ts.map +1 -0
- package/dist/node/index-B7UvPJcI.d.ts +315 -0
- package/dist/node/index-B7UvPJcI.d.ts.map +1 -0
- package/dist/node/index-BLPOTNb5.d.ts +98 -0
- package/dist/node/index-BLPOTNb5.d.ts.map +1 -0
- package/dist/node/index-BqixM_XD.d.ts +231 -0
- package/dist/node/index-BqixM_XD.d.ts.map +1 -0
- package/dist/node/index-C3beaqKO.d.ts +231 -0
- package/dist/node/index-C3beaqKO.d.ts.map +1 -0
- package/dist/node/index-Cp2XRh9G.d.ts +82 -0
- package/dist/node/index-Cp2XRh9G.d.ts.map +1 -0
- package/dist/node/index-DSv5xruI.d.ts +98 -0
- package/dist/node/index-DSv5xruI.d.ts.map +1 -0
- package/dist/node/index-w0bV1uaP.d.ts +315 -0
- package/dist/node/index-w0bV1uaP.d.ts.map +1 -0
- package/dist/node/index.cjs +1 -0
- package/dist/node/index.d.ts +8 -0
- package/dist/node/index.js +1 -0
- package/dist/node/openai/index.cjs +1 -0
- package/dist/node/openai/index.d.ts +2 -0
- package/dist/node/openai/index.js +1 -0
- package/dist/node/openai-CRQjg4xF.js +2 -0
- package/dist/node/openai-CRQjg4xF.js.map +1 -0
- package/dist/node/openai-compatible-BYfyY5lb.cjs +1 -0
- package/dist/node/openai-compatible-Dm4Sof9e.js +2 -0
- package/dist/node/openai-compatible-Dm4Sof9e.js.map +1 -0
- package/dist/node/openai-xWC6pY7r.cjs +1 -0
- package/dist/node/qwen/index.cjs +1 -0
- package/dist/node/qwen/index.d.ts +2 -0
- package/dist/node/qwen/index.js +1 -0
- package/dist/node/qwen-ChUZobTL.js +2 -0
- package/dist/node/qwen-ChUZobTL.js.map +1 -0
- package/dist/node/qwen-CjT71vSM.cjs +1 -0
- package/package.json +157 -0
- package/src/anthropic/__tests__/abort-streaming.test.ts +199 -0
- package/src/anthropic/__tests__/model-catalog-refresh.test.ts +92 -0
- package/src/anthropic/__tests__/provider-definition.test.ts +55 -0
- package/src/anthropic/__tests__/provider.test.ts +1357 -0
- package/src/anthropic/__tests__/response-parser.test.ts +326 -0
- package/src/anthropic/index.ts +22 -0
- package/src/anthropic/message-converter.ts +181 -0
- package/src/anthropic/model-catalog-refresh.ts +128 -0
- package/src/anthropic/parsers/response-parser.ts +184 -0
- package/src/anthropic/provider-definition.ts +93 -0
- package/src/anthropic/provider.ts +290 -0
- package/src/anthropic/streaming-handler.ts +204 -0
- package/src/anthropic/types/api-types.ts +158 -0
- package/src/anthropic/types.ts +79 -0
- package/src/bytedance/http-client.test.ts +288 -0
- package/src/bytedance/http-client.ts +163 -0
- package/src/bytedance/index.ts +2 -0
- package/src/bytedance/provider.spec.ts +320 -0
- package/src/bytedance/provider.ts +171 -0
- package/src/bytedance/status-mapper.test.ts +299 -0
- package/src/bytedance/status-mapper.ts +141 -0
- package/src/bytedance/types.ts +68 -0
- package/src/deepseek/defaults.ts +4 -0
- package/src/deepseek/index.ts +22 -0
- package/src/deepseek/model-catalog-refresh.test.ts +57 -0
- package/src/deepseek/model-catalog-refresh.ts +105 -0
- package/src/deepseek/model-catalog.ts +55 -0
- package/src/deepseek/provider-definition.test.ts +109 -0
- package/src/deepseek/provider-definition.ts +132 -0
- package/src/deepseek/provider.test.ts +324 -0
- package/src/deepseek/provider.ts +298 -0
- package/src/deepseek/types.ts +37 -0
- package/src/gemini/execution-helpers.ts +233 -0
- package/src/gemini/genai-transport.test.ts +208 -0
- package/src/gemini/image-operations.test.ts +448 -0
- package/src/gemini/image-operations.ts +261 -0
- package/src/gemini/index.ts +11 -0
- package/src/gemini/message-converter.test.ts +616 -0
- package/src/gemini/message-converter.ts +140 -0
- package/src/gemini/model-catalog-refresh.test.ts +107 -0
- package/src/gemini/model-catalog-refresh.ts +92 -0
- package/src/gemini/provider-definition.test.ts +70 -0
- package/src/gemini/provider-definition.ts +78 -0
- package/src/gemini/provider-extended.test.ts +898 -0
- package/src/gemini/provider.spec.ts +216 -0
- package/src/gemini/provider.ts +279 -0
- package/src/gemini/request-converter.ts +226 -0
- package/src/gemini/tool-schema-converter.ts +78 -0
- package/src/gemini/types/api-types.ts +235 -0
- package/src/gemini/types.ts +121 -0
- package/src/gemma/index.ts +5 -0
- package/src/gemma/message-factory.ts +38 -0
- package/src/gemma/provider-definition.test.ts +43 -0
- package/src/gemma/provider-definition.ts +84 -0
- package/src/gemma/provider-projection.ts +49 -0
- package/src/gemma/provider.test.ts +628 -0
- package/src/gemma/provider.ts +308 -0
- package/src/gemma/pseudo-command-envelope.ts +58 -0
- package/src/gemma/pseudo-tool-call-projector.ts +243 -0
- package/src/gemma/pseudo-tool-call-tag-parser.ts +153 -0
- package/src/gemma/pseudo-tool-call-types.ts +31 -0
- package/src/gemma/reasoning-projector.test.ts +52 -0
- package/src/gemma/reasoning-projector.ts +144 -0
- package/src/gemma/streaming-projection.ts +79 -0
- package/src/gemma/tool-call-argument-parser.ts +126 -0
- package/src/gemma/tool-call-projector.test.ts +227 -0
- package/src/gemma/tool-call-projector.ts +264 -0
- package/src/gemma/types.ts +27 -0
- package/src/google/index.ts +11 -0
- package/src/google/provider-compat.test.ts +19 -0
- package/src/google/provider-definition.ts +6 -0
- package/src/google/provider.ts +10 -0
- package/src/google/types.ts +5 -0
- package/src/index.ts +9 -0
- package/src/openai/adapter.test.ts +494 -0
- package/src/openai/adapter.ts +145 -0
- package/src/openai/chat-completions-chat.ts +189 -0
- package/src/openai/executor-integration.test.ts +206 -0
- package/src/openai/index.ts +21 -0
- package/src/openai/interfaces/payload-logger.ts +48 -0
- package/src/openai/loggers/console-payload-logger.test.ts +173 -0
- package/src/openai/loggers/console-payload-logger.ts +94 -0
- package/src/openai/loggers/console.ts +9 -0
- package/src/openai/loggers/file-payload-logger.test.ts +238 -0
- package/src/openai/loggers/file-payload-logger.ts +112 -0
- package/src/openai/loggers/file.ts +9 -0
- package/src/openai/loggers/index.ts +12 -0
- package/src/openai/loggers/sanitize-openai-log-data.test.ts +89 -0
- package/src/openai/loggers/sanitize-openai-log-data.ts +14 -0
- package/src/openai/message-converter.ts +22 -0
- package/src/openai/model-catalog-refresh.test.ts +92 -0
- package/src/openai/model-catalog-refresh.ts +115 -0
- package/src/openai/openai-request-format.ts +92 -0
- package/src/openai/parsers/response-parser.test.ts +407 -0
- package/src/openai/parsers/response-parser.ts +47 -0
- package/src/openai/provider-definition.test.ts +75 -0
- package/src/openai/provider-definition.ts +132 -0
- package/src/openai/provider.test.ts +1402 -0
- package/src/openai/provider.ts +237 -0
- package/src/openai/responses-chat.ts +258 -0
- package/src/openai/responses-converter.ts +112 -0
- package/src/openai/responses-parser.ts +285 -0
- package/src/openai/responses-stream-utils.ts +45 -0
- package/src/openai/responses-types.ts +195 -0
- package/src/openai/streaming/stream-assembler.ts +3 -0
- package/src/openai/streaming/stream-handler.test.ts +367 -0
- package/src/openai/streaming/stream-handler.ts +119 -0
- package/src/openai/types/api-types.ts +112 -0
- package/src/openai/types.ts +194 -0
- package/src/qwen/defaults.ts +26 -0
- package/src/qwen/index.ts +5 -0
- package/src/qwen/model-catalog-refresh.test.ts +91 -0
- package/src/qwen/model-catalog-refresh.ts +97 -0
- package/src/qwen/provider-capabilities.ts +34 -0
- package/src/qwen/provider-definition.test.ts +139 -0
- package/src/qwen/provider-definition.ts +173 -0
- package/src/qwen/provider-streaming-assembly.ts +40 -0
- package/src/qwen/provider.test.ts +640 -0
- package/src/qwen/provider.ts +293 -0
- package/src/qwen/responses-chat.ts +194 -0
- package/src/qwen/responses-converter.ts +104 -0
- package/src/qwen/responses-parser.ts +299 -0
- package/src/qwen/responses-stream-utils.ts +38 -0
- package/src/qwen/types.ts +228 -0
- package/src/shared/openai-compatible/endpoint-probe.test.ts +52 -0
- package/src/shared/openai-compatible/endpoint-probe.ts +43 -0
- package/src/shared/openai-compatible/index.ts +6 -0
- package/src/shared/openai-compatible/message-converter.test.ts +111 -0
- package/src/shared/openai-compatible/message-converter.ts +84 -0
- package/src/shared/openai-compatible/native-payload-observer.test.ts +43 -0
- package/src/shared/openai-compatible/native-payload-observer.ts +26 -0
- package/src/shared/openai-compatible/response-parser.test.ts +172 -0
- package/src/shared/openai-compatible/response-parser.ts +180 -0
- package/src/shared/openai-compatible/stream-assembler.test.ts +266 -0
- package/src/shared/openai-compatible/stream-assembler.ts +248 -0
- package/src/shared/openai-compatible/types.ts +59 -0
|
@@ -0,0 +1,173 @@
|
|
|
1
|
+
import type { IProviderDefinition } from '@robota-sdk/agent-core';
|
|
2
|
+
import type { TUniversalValue } from '@robota-sdk/agent-core';
|
|
3
|
+
import { probeOpenAICompatibleProfile } from '../shared/openai-compatible/index.js';
|
|
4
|
+
import { QwenProvider } from './provider';
|
|
5
|
+
import {
|
|
6
|
+
DEFAULT_QWEN_PROVIDER_API_KEY_REFERENCE,
|
|
7
|
+
DEFAULT_QWEN_PROVIDER_BASE_URL,
|
|
8
|
+
DEFAULT_QWEN_PROVIDER_MODEL,
|
|
9
|
+
QWEN_MODEL_LAST_VERIFIED_AT,
|
|
10
|
+
QWEN_MODEL_SOURCE_URL,
|
|
11
|
+
} from './defaults';
|
|
12
|
+
import type { IQwenBuiltInWebToolsOptions } from './types';
|
|
13
|
+
import { refreshQwenModelCatalog } from './model-catalog-refresh';
|
|
14
|
+
|
|
15
|
+
const QWEN_API_KEY_URL = 'https://modelstudio.console.alibabacloud.com/?tab=api#/api-key';
|
|
16
|
+
const QWEN_SETUP_SOURCE_URL = 'https://www.alibabacloud.com/help/en/model-studio/get-api-key';
|
|
17
|
+
const QWEN_SETUP_LAST_VERIFIED_AT = '2026-05-08';
|
|
18
|
+
const QWEN_SETUP_HELP_LINKS: NonNullable<IProviderDefinition['setupHelpLinks']> = [
|
|
19
|
+
{
|
|
20
|
+
kind: 'api-key',
|
|
21
|
+
label: 'Alibaba Cloud Model Studio API keys',
|
|
22
|
+
url: QWEN_API_KEY_URL,
|
|
23
|
+
sourceUrl: QWEN_SETUP_SOURCE_URL,
|
|
24
|
+
lastVerifiedAt: QWEN_SETUP_LAST_VERIFIED_AT,
|
|
25
|
+
},
|
|
26
|
+
];
|
|
27
|
+
const QWEN_MODEL_CATALOG: NonNullable<IProviderDefinition['modelCatalog']> = {
|
|
28
|
+
status: 'fallback',
|
|
29
|
+
sourceUrl: QWEN_MODEL_SOURCE_URL,
|
|
30
|
+
lastVerifiedAt: QWEN_MODEL_LAST_VERIFIED_AT,
|
|
31
|
+
entries: [
|
|
32
|
+
{
|
|
33
|
+
id: DEFAULT_QWEN_PROVIDER_MODEL,
|
|
34
|
+
displayName: 'Qwen Plus',
|
|
35
|
+
capabilities: ['tools', 'reasoning', 'native_web', 'streaming'],
|
|
36
|
+
lifecycle: 'active',
|
|
37
|
+
sourceUrl: QWEN_MODEL_SOURCE_URL,
|
|
38
|
+
lastVerifiedAt: QWEN_MODEL_LAST_VERIFIED_AT,
|
|
39
|
+
},
|
|
40
|
+
{
|
|
41
|
+
id: 'qwen-max',
|
|
42
|
+
displayName: 'Qwen Max',
|
|
43
|
+
capabilities: ['tools', 'reasoning', 'native_web', 'streaming'],
|
|
44
|
+
lifecycle: 'active',
|
|
45
|
+
sourceUrl: QWEN_MODEL_SOURCE_URL,
|
|
46
|
+
lastVerifiedAt: QWEN_MODEL_LAST_VERIFIED_AT,
|
|
47
|
+
},
|
|
48
|
+
{
|
|
49
|
+
id: 'qwen-flash',
|
|
50
|
+
displayName: 'Qwen Flash',
|
|
51
|
+
capabilities: ['tools', 'native_web', 'streaming'],
|
|
52
|
+
lifecycle: 'active',
|
|
53
|
+
sourceUrl: QWEN_MODEL_SOURCE_URL,
|
|
54
|
+
lastVerifiedAt: QWEN_MODEL_LAST_VERIFIED_AT,
|
|
55
|
+
},
|
|
56
|
+
],
|
|
57
|
+
};
|
|
58
|
+
|
|
59
|
+
export {
|
|
60
|
+
DEFAULT_QWEN_PROVIDER_API_KEY_ENV,
|
|
61
|
+
DEFAULT_QWEN_PROVIDER_API_KEY_REFERENCE,
|
|
62
|
+
DEFAULT_QWEN_PROVIDER_BASE_URL,
|
|
63
|
+
DEFAULT_QWEN_PROVIDER_MODEL,
|
|
64
|
+
DEFAULT_QWEN_PROVIDER_RESPONSES_BASE_URL,
|
|
65
|
+
QWEN_PROVIDER_BASE_URLS,
|
|
66
|
+
QWEN_PROVIDER_RESPONSES_BASE_URLS,
|
|
67
|
+
} from './defaults';
|
|
68
|
+
export type { TQwenProviderRegion, TQwenProviderResponsesRegion } from './defaults';
|
|
69
|
+
|
|
70
|
+
export function createQwenProviderDefinition(): IProviderDefinition {
|
|
71
|
+
return {
|
|
72
|
+
type: 'qwen',
|
|
73
|
+
displayName: 'Qwen',
|
|
74
|
+
description: 'Alibaba Cloud Model Studio OpenAI-compatible endpoint',
|
|
75
|
+
defaults: {
|
|
76
|
+
model: DEFAULT_QWEN_PROVIDER_MODEL,
|
|
77
|
+
apiKey: DEFAULT_QWEN_PROVIDER_API_KEY_REFERENCE,
|
|
78
|
+
baseURL: DEFAULT_QWEN_PROVIDER_BASE_URL,
|
|
79
|
+
},
|
|
80
|
+
modelCatalog: QWEN_MODEL_CATALOG,
|
|
81
|
+
refreshModelCatalog: ({ profile }) => refreshQwenModelCatalog(profile),
|
|
82
|
+
modelCatalogCacheTtlSeconds: 86400,
|
|
83
|
+
setupHelpLinks: QWEN_SETUP_HELP_LINKS,
|
|
84
|
+
setupSteps: [
|
|
85
|
+
{
|
|
86
|
+
key: 'baseURL',
|
|
87
|
+
title: 'Qwen OpenAI-compatible base URL',
|
|
88
|
+
defaultValue: DEFAULT_QWEN_PROVIDER_BASE_URL,
|
|
89
|
+
},
|
|
90
|
+
{
|
|
91
|
+
key: 'model',
|
|
92
|
+
title: 'Qwen model',
|
|
93
|
+
defaultValue: DEFAULT_QWEN_PROVIDER_MODEL,
|
|
94
|
+
},
|
|
95
|
+
{
|
|
96
|
+
key: 'apiKey',
|
|
97
|
+
title: 'Qwen Model Studio API key',
|
|
98
|
+
defaultValue: DEFAULT_QWEN_PROVIDER_API_KEY_REFERENCE,
|
|
99
|
+
masked: true,
|
|
100
|
+
},
|
|
101
|
+
],
|
|
102
|
+
requiresApiKey: true,
|
|
103
|
+
probeProfile: probeOpenAICompatibleProfile,
|
|
104
|
+
createProvider: (config) => {
|
|
105
|
+
const qwenOptions = parseQwenProviderOptions(config.options);
|
|
106
|
+
return new QwenProvider({
|
|
107
|
+
apiKey: requireApiKey(config.apiKey),
|
|
108
|
+
...(config.baseURL !== undefined && { baseURL: config.baseURL }),
|
|
109
|
+
...(qwenOptions.responsesBaseURL !== undefined && {
|
|
110
|
+
responsesBaseURL: qwenOptions.responsesBaseURL,
|
|
111
|
+
}),
|
|
112
|
+
...(config.timeout !== undefined && { timeout: config.timeout }),
|
|
113
|
+
...(qwenOptions.builtInWebTools !== undefined && {
|
|
114
|
+
builtInWebTools: qwenOptions.builtInWebTools,
|
|
115
|
+
}),
|
|
116
|
+
defaultModel: config.model,
|
|
117
|
+
});
|
|
118
|
+
},
|
|
119
|
+
};
|
|
120
|
+
}
|
|
121
|
+
|
|
122
|
+
function requireApiKey(apiKey: string | undefined): string {
|
|
123
|
+
if (!apiKey) {
|
|
124
|
+
throw new Error('Provider qwen requires apiKey');
|
|
125
|
+
}
|
|
126
|
+
return apiKey;
|
|
127
|
+
}
|
|
128
|
+
|
|
129
|
+
function parseQwenProviderOptions(options: Record<string, TUniversalValue> | undefined): {
|
|
130
|
+
responsesBaseURL?: string;
|
|
131
|
+
builtInWebTools?: IQwenBuiltInWebToolsOptions;
|
|
132
|
+
} {
|
|
133
|
+
const builtInWebTools = parseBuiltInWebTools(asRecord(options?.['builtInWebTools']));
|
|
134
|
+
const responsesBaseURL = asString(options?.['responsesBaseURL']);
|
|
135
|
+
return {
|
|
136
|
+
...(responsesBaseURL !== undefined && { responsesBaseURL }),
|
|
137
|
+
...(builtInWebTools !== undefined && { builtInWebTools }),
|
|
138
|
+
};
|
|
139
|
+
}
|
|
140
|
+
|
|
141
|
+
function parseBuiltInWebTools(
|
|
142
|
+
options: Record<string, TUniversalValue> | undefined,
|
|
143
|
+
): IQwenBuiltInWebToolsOptions | undefined {
|
|
144
|
+
if (options === undefined) {
|
|
145
|
+
return undefined;
|
|
146
|
+
}
|
|
147
|
+
return {
|
|
148
|
+
...(asBoolean(options['webSearch']) !== undefined && {
|
|
149
|
+
webSearch: asBoolean(options['webSearch']),
|
|
150
|
+
}),
|
|
151
|
+
...(asBoolean(options['webFetch']) !== undefined && {
|
|
152
|
+
webFetch: asBoolean(options['webFetch']),
|
|
153
|
+
}),
|
|
154
|
+
...(asBoolean(options['enableThinking']) !== undefined && {
|
|
155
|
+
enableThinking: asBoolean(options['enableThinking']),
|
|
156
|
+
}),
|
|
157
|
+
};
|
|
158
|
+
}
|
|
159
|
+
|
|
160
|
+
function asRecord(value: TUniversalValue | undefined): Record<string, TUniversalValue> | undefined {
|
|
161
|
+
if (value === null || value === undefined || value instanceof Date || Array.isArray(value)) {
|
|
162
|
+
return undefined;
|
|
163
|
+
}
|
|
164
|
+
return typeof value === 'object' ? value : undefined;
|
|
165
|
+
}
|
|
166
|
+
|
|
167
|
+
function asBoolean(value: TUniversalValue | undefined): boolean | undefined {
|
|
168
|
+
return typeof value === 'boolean' ? value : undefined;
|
|
169
|
+
}
|
|
170
|
+
|
|
171
|
+
function asString(value: TUniversalValue | undefined): string | undefined {
|
|
172
|
+
return typeof value === 'string' && value.length > 0 ? value : undefined;
|
|
173
|
+
}
|
|
@@ -0,0 +1,40 @@
|
|
|
1
|
+
import OpenAI from 'openai';
|
|
2
|
+
import type { IChatOptions, TUniversalMessage } from '@robota-sdk/agent-core';
|
|
3
|
+
import {
|
|
4
|
+
assembleOpenAICompatibleStream,
|
|
5
|
+
observeProviderNativeRawPayloadStream,
|
|
6
|
+
} from '../shared/openai-compatible/index.js';
|
|
7
|
+
import type { IOpenAICompatibleError } from '../shared/openai-compatible/index.js';
|
|
8
|
+
|
|
9
|
+
export async function qwenChatWithStreamingAssembly(
|
|
10
|
+
client: OpenAI,
|
|
11
|
+
requestParams: OpenAI.Chat.ChatCompletionCreateParamsStreaming,
|
|
12
|
+
options: IChatOptions,
|
|
13
|
+
): Promise<TUniversalMessage> {
|
|
14
|
+
try {
|
|
15
|
+
options.onProviderNativeRawPayload?.({
|
|
16
|
+
provider: 'qwen',
|
|
17
|
+
apiSurface: 'chat-completions',
|
|
18
|
+
payloadKind: 'request',
|
|
19
|
+
payload: requestParams,
|
|
20
|
+
});
|
|
21
|
+
const stream = await client.chat.completions.create(
|
|
22
|
+
requestParams,
|
|
23
|
+
options.signal ? { signal: options.signal } : undefined,
|
|
24
|
+
);
|
|
25
|
+
|
|
26
|
+
return assembleOpenAICompatibleStream({
|
|
27
|
+
stream: observeProviderNativeRawPayloadStream(stream, {
|
|
28
|
+
provider: 'qwen',
|
|
29
|
+
apiSurface: 'chat-completions',
|
|
30
|
+
onProviderNativeRawPayload: options.onProviderNativeRawPayload,
|
|
31
|
+
}),
|
|
32
|
+
onTextDelta: options.onTextDelta,
|
|
33
|
+
signal: options.signal,
|
|
34
|
+
});
|
|
35
|
+
} catch (error) {
|
|
36
|
+
const qwenError = error as IOpenAICompatibleError;
|
|
37
|
+
const errorMessage = qwenError.message || 'Qwen streaming request failed';
|
|
38
|
+
throw new Error(`Qwen stream failed: ${errorMessage}`);
|
|
39
|
+
}
|
|
40
|
+
}
|