@zimtsui/brainswitch 0.0.54 → 0.0.56
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/DEVELOPMENT.md +13 -14
- package/README.md +7 -14
- package/build/adaptor.d.ts +5 -5
- package/build/adaptor.js +16 -16
- package/build/adaptor.js.map +1 -1
- package/build/agentloop.d.ts +5 -5
- package/build/agentloop.js +4 -4
- package/build/agentloop.js.map +1 -1
- package/build/api-types/anthropic/billing.d.ts +12 -0
- package/build/api-types/anthropic/billing.js +15 -0
- package/build/api-types/anthropic/billing.js.map +1 -0
- package/build/api-types/anthropic/tool-codec.d.ts +17 -0
- package/build/api-types/anthropic/tool-codec.js +73 -0
- package/build/api-types/anthropic/tool-codec.js.map +1 -0
- package/build/api-types/google/billing.d.ts +12 -0
- package/build/api-types/google/billing.js +24 -0
- package/build/api-types/google/billing.js.map +1 -0
- package/build/api-types/google/restful-request.d.ts +8 -0
- package/build/api-types/google/restful-request.js +2 -0
- package/build/api-types/google/restful-request.js.map +1 -0
- package/build/api-types/google/tool-codec.d.ts +16 -0
- package/build/api-types/{google.js → google/tool-codec.js} +27 -26
- package/build/api-types/google/tool-codec.js.map +1 -0
- package/build/api-types/openai-chatcompletion/billing.d.ts +13 -0
- package/build/api-types/openai-chatcompletion/billing.js +16 -0
- package/build/api-types/openai-chatcompletion/billing.js.map +1 -0
- package/build/api-types/openai-chatcompletion/tool-codec.d.ts +18 -0
- package/build/api-types/openai-chatcompletion/tool-codec.js +86 -0
- package/build/api-types/openai-chatcompletion/tool-codec.js.map +1 -0
- package/build/api-types/openai-responses/billing.d.ts +12 -0
- package/build/api-types/openai-responses/billing.js +15 -0
- package/build/api-types/openai-responses/billing.js.map +1 -0
- package/build/api-types/openai-responses/tool-codec.d.ts +16 -0
- package/build/api-types/openai-responses/tool-codec.js +72 -0
- package/build/api-types/openai-responses/tool-codec.js.map +1 -0
- package/build/compatible/engine.d/aliyun.d.ts +23 -0
- package/build/compatible/engine.d/aliyun.js +44 -0
- package/build/compatible/engine.d/aliyun.js.map +1 -0
- package/build/compatible/engine.d/anthropic.d.ts +23 -0
- package/build/compatible/engine.d/anthropic.js +41 -0
- package/build/compatible/engine.d/anthropic.js.map +1 -0
- package/build/compatible/engine.d/google.d.ts +23 -0
- package/build/compatible/engine.d/google.js +45 -0
- package/build/compatible/engine.d/google.js.map +1 -0
- package/build/compatible/engine.d/openai-responses.d.ts +23 -0
- package/build/compatible/engine.d/openai-responses.js +41 -0
- package/build/compatible/engine.d/openai-responses.js.map +1 -0
- package/build/compatible/engine.d.ts +18 -0
- package/build/compatible/engine.js +21 -0
- package/build/compatible/engine.js.map +1 -0
- package/build/compatible/session.d.ts +78 -0
- package/build/compatible/session.js +138 -0
- package/build/compatible/session.js.map +1 -0
- package/build/compatible/tool-call-validator.d.ts +11 -0
- package/build/compatible/tool-call-validator.js +12 -0
- package/build/compatible/tool-call-validator.js.map +1 -0
- package/build/compatible.d/aliyun/transport.d.ts +11 -0
- package/build/compatible.d/aliyun/transport.js +9 -0
- package/build/compatible.d/aliyun/transport.js.map +1 -0
- package/build/compatible.d/anthropic/message-codec.d.ts +31 -0
- package/build/compatible.d/anthropic/message-codec.js +88 -0
- package/build/compatible.d/anthropic/message-codec.js.map +1 -0
- package/build/compatible.d/anthropic/transport.d.ts +31 -0
- package/build/compatible.d/anthropic/transport.js +137 -0
- package/build/compatible.d/anthropic/transport.js.map +1 -0
- package/build/compatible.d/google/message-codec.d.ts +31 -0
- package/build/compatible.d/google/message-codec.js +98 -0
- package/build/compatible.d/google/message-codec.js.map +1 -0
- package/build/compatible.d/google/transport.d.ts +28 -0
- package/build/compatible.d/google/transport.js +74 -0
- package/build/compatible.d/google/transport.js.map +1 -0
- package/build/compatible.d/openai-chatcompletions/message-codec.d.ts +19 -0
- package/build/compatible.d/openai-chatcompletions/message-codec.js +66 -0
- package/build/compatible.d/openai-chatcompletions/message-codec.js.map +1 -0
- package/build/compatible.d/openai-chatcompletions/transport.d.ts +9 -0
- package/build/compatible.d/openai-chatcompletions/transport.js +25 -0
- package/build/compatible.d/openai-chatcompletions/transport.js.map +1 -0
- package/build/compatible.d/openai-chatcompletions.d/monolith.d.ts +36 -0
- package/build/compatible.d/openai-chatcompletions.d/monolith.js +74 -0
- package/build/compatible.d/openai-chatcompletions.d/monolith.js.map +1 -0
- package/build/compatible.d/openai-chatcompletions.d/stream.d.ts +35 -0
- package/build/{compatible-engines.d → compatible.d}/openai-chatcompletions.d/stream.js +52 -42
- package/build/compatible.d/openai-chatcompletions.d/stream.js.map +1 -0
- package/build/compatible.d/openai-responses/message-codec.d.ts +32 -0
- package/build/compatible.d/openai-responses/message-codec.js +105 -0
- package/build/compatible.d/openai-responses/message-codec.js.map +1 -0
- package/build/compatible.d/openai-responses/transport.d.ts +33 -0
- package/build/compatible.d/openai-responses/transport.js +88 -0
- package/build/compatible.d/openai-responses/transport.js.map +1 -0
- package/build/config.js +1 -1
- package/build/config.js.map +1 -1
- package/build/engine.d.ts +48 -26
- package/build/engine.js +78 -29
- package/build/engine.js.map +1 -1
- package/build/exports.d.ts +14 -13
- package/build/exports.js +14 -13
- package/build/exports.js.map +1 -1
- package/build/function.d.ts +25 -13
- package/build/function.js +2 -1
- package/build/function.js.map +1 -1
- package/build/inference-context.d.ts +1 -1
- package/build/native-engines.d/google/agentloop.d.ts +6 -6
- package/build/native-engines.d/google/agentloop.js +5 -5
- package/build/native-engines.d/google/agentloop.js.map +1 -1
- package/build/native-engines.d/google/engine.d.ts +28 -103
- package/build/native-engines.d/google/engine.js +63 -304
- package/build/native-engines.d/google/engine.js.map +1 -1
- package/build/native-engines.d/google/exports.d.ts +5 -5
- package/build/native-engines.d/google/exports.js +5 -5
- package/build/native-engines.d/google/exports.js.map +1 -1
- package/build/native-engines.d/google/message-codec.d.ts +21 -0
- package/build/native-engines.d/google/message-codec.js +69 -0
- package/build/native-engines.d/google/message-codec.js.map +1 -0
- package/build/native-engines.d/google/session.d.ts +16 -15
- package/build/native-engines.d/google/session.js +2 -2
- package/build/native-engines.d/google/session.js.map +1 -1
- package/build/native-engines.d/google/transport.d.ts +31 -0
- package/build/native-engines.d/google/transport.js +81 -0
- package/build/native-engines.d/google/transport.js.map +1 -0
- package/build/native-engines.d/openai-responses/agentloop.d.ts +7 -7
- package/build/native-engines.d/openai-responses/agentloop.js +6 -6
- package/build/native-engines.d/openai-responses/agentloop.js.map +1 -1
- package/build/native-engines.d/openai-responses/engine.d.ts +26 -106
- package/build/native-engines.d/openai-responses/engine.js +53 -334
- package/build/native-engines.d/openai-responses/engine.js.map +1 -1
- package/build/native-engines.d/openai-responses/exports.d.ts +6 -7
- package/build/native-engines.d/openai-responses/exports.js +6 -7
- package/build/native-engines.d/openai-responses/exports.js.map +1 -1
- package/build/native-engines.d/openai-responses/message-codec.d.ts +21 -0
- package/build/native-engines.d/openai-responses/message-codec.js +69 -0
- package/build/native-engines.d/openai-responses/message-codec.js.map +1 -0
- package/build/native-engines.d/openai-responses/session.d.ts +27 -26
- package/build/native-engines.d/openai-responses/session.js +3 -3
- package/build/native-engines.d/openai-responses/session.js.map +1 -1
- package/build/native-engines.d/openai-responses/tool-call-validator.d.ts +12 -0
- package/build/native-engines.d/openai-responses/tool-call-validator.js +33 -0
- package/build/native-engines.d/openai-responses/tool-call-validator.js.map +1 -0
- package/build/native-engines.d/openai-responses/tool.d.ts +3 -3
- package/build/native-engines.d/openai-responses/tool.js +1 -1
- package/build/native-engines.d/openai-responses/tool.js.map +1 -1
- package/build/native-engines.d/openai-responses/transport.d.ts +36 -0
- package/build/native-engines.d/openai-responses/transport.js +104 -0
- package/build/native-engines.d/openai-responses/transport.js.map +1 -0
- package/build/session.d.ts +5 -75
- package/build/session.js +0 -136
- package/build/session.js.map +1 -1
- package/build/throttle.d.ts +1 -1
- package/build/tsconfig.tsbuildinfo +1 -1
- package/package.json +4 -4
- package/test/build/test-codecs.js +230 -0
- package/test/build/test-codecs.js.map +1 -0
- package/test/build/test-compatible-engine.d.ts +1 -0
- package/test/build/test-compatible-engine.js +27 -0
- package/test/build/test-compatible-engine.js.map +1 -0
- package/test/build/test-compatible-engines.d.ts +1 -0
- package/test/build/test-compatible-engines.js +112 -0
- package/test/build/test-compatible-engines.js.map +1 -0
- package/test/build/test-engine.d.ts +1 -0
- package/test/build/test-engine.js +97 -0
- package/test/build/test-engine.js.map +1 -0
- package/test/build/test-helpers.d.ts +46 -0
- package/test/build/test-helpers.js +45 -0
- package/test/build/test-helpers.js.map +1 -0
- package/test/build/test-native-engines.d.ts +1 -0
- package/test/build/test-native-engines.js +64 -0
- package/test/build/test-native-engines.js.map +1 -0
- package/test/build/test.d.ts +5 -0
- package/test/build/test.js +6 -0
- package/test/build/test.js.map +1 -0
- package/test/build/tsconfig.tsbuildinfo +1 -0
- package/test/src/test-codecs.ts +256 -0
- package/test/src/test-compatible-engine.ts +45 -0
- package/test/src/test-compatible-engines.ts +129 -0
- package/test/src/test-engine.ts +144 -0
- package/test/src/test-helpers.ts +53 -0
- package/test/src/test-native-engines.ts +75 -0
- package/test/src/test.ts +5 -0
- package/test/src/tsconfig.json +30 -0
- package/build/api-types/anthropic.d.ts +0 -29
- package/build/api-types/anthropic.js +0 -93
- package/build/api-types/anthropic.js.map +0 -1
- package/build/api-types/google.d.ts +0 -28
- package/build/api-types/google.js.map +0 -1
- package/build/api-types/openai-chat-completions.d.ts +0 -31
- package/build/api-types/openai-chat-completions.js +0 -113
- package/build/api-types/openai-chat-completions.js.map +0 -1
- package/build/api-types/openai-responses.d.ts +0 -23
- package/build/api-types/openai-responses.js +0 -71
- package/build/api-types/openai-responses.js.map +0 -1
- package/build/compatible-engine.d.ts +0 -51
- package/build/compatible-engine.js +0 -67
- package/build/compatible-engine.js.map +0 -1
- package/build/compatible-engines.d/aliyun.d.ts +0 -69
- package/build/compatible-engines.d/aliyun.js +0 -149
- package/build/compatible-engines.d/aliyun.js.map +0 -1
- package/build/compatible-engines.d/anthropic.d.ts +0 -78
- package/build/compatible-engines.d/anthropic.js +0 -312
- package/build/compatible-engines.d/anthropic.js.map +0 -1
- package/build/compatible-engines.d/google.d.ts +0 -83
- package/build/compatible-engines.d/google.js +0 -279
- package/build/compatible-engines.d/google.js.map +0 -1
- package/build/compatible-engines.d/openai-chatcompletions.d/monolith.d.ts +0 -15
- package/build/compatible-engines.d/openai-chatcompletions.d/monolith.js +0 -75
- package/build/compatible-engines.d/openai-chatcompletions.d/monolith.js.map +0 -1
- package/build/compatible-engines.d/openai-chatcompletions.d/stream.d.ts +0 -21
- package/build/compatible-engines.d/openai-chatcompletions.d/stream.js.map +0 -1
- package/build/compatible-engines.d/openai-chatcompletions.d.ts +0 -27
- package/build/compatible-engines.d/openai-chatcompletions.js +0 -76
- package/build/compatible-engines.d/openai-chatcompletions.js.map +0 -1
- package/build/compatible-engines.d/openai-responses.d.ts +0 -93
- package/build/compatible-engines.d/openai-responses.js +0 -300
- package/build/compatible-engines.d/openai-responses.js.map +0 -1
- package/build/test.js +0 -198
- package/build/test.js.map +0 -1
- /package/{build/test.d.ts → test/build/test-codecs.d.ts} +0 -0
|
@@ -0,0 +1,98 @@
|
|
|
1
|
+
import { ResponseInvalid } from '#@/engine.ts';
|
|
2
|
+
import { RoleMessage } from '#@/compatible/session.ts';
|
|
3
|
+
import { Function } from '#@/function.ts';
|
|
4
|
+
import * as Google from '@google/genai';
|
|
5
|
+
import {} from '#@/api-types/google/tool-codec.ts';
|
|
6
|
+
export class GoogleCompatibleMessageCodec {
|
|
7
|
+
ctx;
|
|
8
|
+
constructor(ctx) {
|
|
9
|
+
this.ctx = ctx;
|
|
10
|
+
}
|
|
11
|
+
convertFromAiMessage(aiMessage) {
|
|
12
|
+
if (aiMessage instanceof GoogleCompatibleMessageCodec.Message.Ai.Instance)
|
|
13
|
+
return aiMessage.getRaw();
|
|
14
|
+
else {
|
|
15
|
+
const parts = aiMessage.getParts().map(part => {
|
|
16
|
+
if (part instanceof RoleMessage.Part.Text.Instance)
|
|
17
|
+
return Google.createPartFromText(part.text);
|
|
18
|
+
else if (part instanceof Function.Call) {
|
|
19
|
+
if (part.args instanceof Object) { }
|
|
20
|
+
else
|
|
21
|
+
throw new Error();
|
|
22
|
+
return Google.createPartFromFunctionCall(part.name, part.args);
|
|
23
|
+
}
|
|
24
|
+
else
|
|
25
|
+
throw new Error();
|
|
26
|
+
});
|
|
27
|
+
return Google.createModelContent(parts);
|
|
28
|
+
}
|
|
29
|
+
}
|
|
30
|
+
convertFromChatMessages(chatMessages) {
|
|
31
|
+
return chatMessages.map(chatMessage => {
|
|
32
|
+
if (chatMessage instanceof RoleMessage.User.Instance)
|
|
33
|
+
return this.convertFromUserMessage(chatMessage);
|
|
34
|
+
else if (chatMessage instanceof RoleMessage.Ai.Instance)
|
|
35
|
+
return this.convertFromAiMessage(chatMessage);
|
|
36
|
+
else
|
|
37
|
+
throw new Error();
|
|
38
|
+
});
|
|
39
|
+
}
|
|
40
|
+
convertFromUserMessage(userMessage) {
|
|
41
|
+
const parts = userMessage.getParts().map(part => {
|
|
42
|
+
if (part instanceof RoleMessage.Part.Text.Instance)
|
|
43
|
+
return Google.createPartFromText(part.text);
|
|
44
|
+
else if (part instanceof Function.Response)
|
|
45
|
+
return {
|
|
46
|
+
functionResponse: { id: part.id, name: part.name, response: { returnValue: part.text } },
|
|
47
|
+
};
|
|
48
|
+
else
|
|
49
|
+
throw new Error();
|
|
50
|
+
});
|
|
51
|
+
return Google.createUserContent(parts);
|
|
52
|
+
}
|
|
53
|
+
convertFromDeveloperMessage(developerMessage) {
|
|
54
|
+
const parts = developerMessage.getParts().map(part => Google.createPartFromText(part.text));
|
|
55
|
+
return { parts };
|
|
56
|
+
}
|
|
57
|
+
convertToAiMessage(content) {
|
|
58
|
+
if (content.parts) { }
|
|
59
|
+
else
|
|
60
|
+
throw new Error();
|
|
61
|
+
return GoogleCompatibleMessageCodec.Message.Ai.create(content.parts.flatMap(part => {
|
|
62
|
+
const parts = [];
|
|
63
|
+
if (part.functionCall || part.text) { }
|
|
64
|
+
else
|
|
65
|
+
throw new ResponseInvalid('Unknown content part', { cause: content });
|
|
66
|
+
if (part.text)
|
|
67
|
+
parts.push(RoleMessage.Part.Text.create(part.text));
|
|
68
|
+
if (part.functionCall)
|
|
69
|
+
parts.push(this.ctx.toolCodec.convertToFunctionCall(part.functionCall));
|
|
70
|
+
return parts;
|
|
71
|
+
}), content);
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
(function (GoogleCompatibleMessageCodec) {
|
|
75
|
+
let Message;
|
|
76
|
+
(function (Message) {
|
|
77
|
+
let Ai;
|
|
78
|
+
(function (Ai) {
|
|
79
|
+
function create(parts, raw) {
|
|
80
|
+
return new Instance(parts, raw);
|
|
81
|
+
}
|
|
82
|
+
Ai.create = create;
|
|
83
|
+
Ai.NOMINAL = Symbol();
|
|
84
|
+
class Instance extends RoleMessage.Ai.Instance {
|
|
85
|
+
raw;
|
|
86
|
+
constructor(parts, raw) {
|
|
87
|
+
super(parts);
|
|
88
|
+
this.raw = raw;
|
|
89
|
+
}
|
|
90
|
+
getRaw() {
|
|
91
|
+
return this.raw;
|
|
92
|
+
}
|
|
93
|
+
}
|
|
94
|
+
Ai.Instance = Instance;
|
|
95
|
+
})(Ai = Message.Ai || (Message.Ai = {}));
|
|
96
|
+
})(Message = GoogleCompatibleMessageCodec.Message || (GoogleCompatibleMessageCodec.Message = {}));
|
|
97
|
+
})(GoogleCompatibleMessageCodec || (GoogleCompatibleMessageCodec = {}));
|
|
98
|
+
//# sourceMappingURL=message-codec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"message-codec.js","sourceRoot":"","sources":["../../../src/compatible.d/google/message-codec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAgB,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AACxC,OAAO,EAAwB,MAAM,mCAAmC,CAAC;AAIzE,MAAM,OAAO,4BAA4B;IACR;IAA7B,YAA6B,GAA8C;QAA9C,QAAG,GAAH,GAAG,CAA2C;IAAG,CAAC;IAGxE,oBAAoB,CACvB,SAA8B;QAE9B,IAAI,SAAS,YAAY,4BAA4B,CAAC,OAAO,CAAC,EAAE,CAAC,QAAQ;YACrE,OAAO,SAAS,CAAC,MAAM,EAAE,CAAC;aACzB,CAAC;YACF,MAAM,KAAK,GAAG,SAAS,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;gBAC1C,IAAI,IAAI,YAAY,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ;oBAC9C,OAAO,MAAM,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;qBAC3C,IAAI,IAAI,YAAY,QAAQ,CAAC,IAAI,EAAE,CAAC;oBACrC,IAAI,IAAI,CAAC,IAAI,YAAY,MAAM,EAAE,CAAC,CAAA,CAAC;;wBAAM,MAAM,IAAI,KAAK,EAAE,CAAC;oBAC3D,OAAO,MAAM,CAAC,0BAA0B,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAA+B,CAAC,CAAC;gBAC9F,CAAC;;oBAAM,MAAM,IAAI,KAAK,EAAE,CAAC;YAC7B,CAAC,CAAC,CAAC;YACH,OAAO,MAAM,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QAC5C,CAAC;IACL,CAAC;IAEM,uBAAuB,CAC1B,YAAwC;QAExC,OAAO,YAAY,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE;YAClC,IAAI,WAAW,YAAY,WAAW,CAAC,IAAI,CAAC,QAAQ;gBAAE,OAAO,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC;iBACjG,IAAI,WAAW,YAAY,WAAW,CAAC,EAAE,CAAC,QAAQ;gBAAE,OAAO,IAAI,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC;;gBAClG,MAAM,IAAI,KAAK,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;IACP,CAAC;IAEM,sBAAsB,CACzB,WAAkC;QAElC,MAAM,KAAK,GAAG,WAAW,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;YAC5C,IAAI,IAAI,YAAY,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ;gBAC9C,OAAO,MAAM,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;iBAC3C,IAAI,IAAI,YAAY,QAAQ,CAAC,QAAQ;gBACtC,OAAO;oBACH,gBAAgB,EAAE,EAAE,EAAE,EAAE,IAAI,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,QAAQ,EAAE,EAAE,WAAW,EAAE,IAAI,CAAC,IAAI,EAAE,EAAE;iBAC3F,CAAC;;gBACD,MAAM,IAAI,KAAK,EAAE,CAAC;QAC3B,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;IAC3C,CAAC;IAEM,2BAA2B,CAC9B,gBAAuC;QAEvC,MAAM,KAAK,GAAG,gBAAgB,CAAC,QAAQ,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,MAAM,CAAC,kBAAkB,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;QAC5F,OAAO,EAAE,KAAK,EAAE,CAAC;IACrB,CAAC;IAEM,kBAAkB,CACrB,OAAuB;QAEvB,IAAI,OAAO,CAAC,KAAK,EAAE,CAAC,CAAA,CAAC;;YAAM,MAAM,IAAI,KAAK,EAAE,CAAC;QAC7C,OAAO,4BAA4B,CAAC,OAAO,CAAC,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,EAAE;YAC/E,MAAM,KAAK,GAA+B,EAAE,CAAC;YAC7C,IAAI,IAAI,CAAC,YAAY,IAAI,IAAI,CAAC,IAAI,EAAE,CAAC,CAAA,CAAC;;gBAAM,MAAM,IAAI,eAAe,CAAC,sBAAsB,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;YAClH,IAAI,IAAI,CAAC,IAAI;gBAAE,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;YACnE,IAAI,IAAI,CAAC,YAAY;gBAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,qBAAqB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,CAAC;YAC/F,OAAO,KAAK,CAAC;QACjB,CAAC,CAAC,EAAE,OAAO,CAAC,CAAC;IACjB,CAAC;CACJ;AAGD,WAAiB,4BAA4B;IAKzC,IAAiB,OAAO,CAiBvB;IAjBD,WAAiB,OAAO;QAEpB,IAAiB,EAAE,CAclB;QAdD,WAAiB,EAAE;YACf,SAAgB,MAAM,CAAuC,KAAiC,EAAE,GAAmB;gBAC/G,OAAO,IAAI,QAAQ,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC;YACpC,CAAC;YAFe,SAAM,SAErB,CAAA;YACY,UAAO,GAAG,MAAM,EAAE,CAAC;YAChC,MAAa,QAAsD,SAAQ,WAAW,CAAC,EAAE,CAAC,QAAa;gBAEnC;gBAAhE,YAAmB,KAAiC,EAAY,GAAmB;oBAC/E,KAAK,CAAC,KAAK,CAAC,CAAC;oBAD+C,QAAG,GAAH,GAAG,CAAgB;gBAEnF,CAAC;gBACM,MAAM;oBACT,OAAO,IAAI,CAAC,GAAG,CAAC;gBACpB,CAAC;aACJ;YARY,WAAQ,WAQpB,CAAA;QACL,CAAC,EAdgB,EAAE,GAAF,UAAE,KAAF,UAAE,QAclB;IACL,CAAC,EAjBgB,OAAO,GAAP,oCAAO,KAAP,oCAAO,QAiBvB;AAEL,CAAC,EAxBgB,4BAA4B,KAA5B,4BAA4B,QAwB5C"}
|
|
@@ -0,0 +1,28 @@
|
|
|
1
|
+
import { type InferenceParams as InferenceParams, type ProviderSpec } from '#@/engine.ts';
|
|
2
|
+
import { RoleMessage, type Session } from '#@/compatible/session.ts';
|
|
3
|
+
import { Function } from '#@/function.ts';
|
|
4
|
+
import { type InferenceContext } from '#@/inference-context.ts';
|
|
5
|
+
import { Throttle } from '#@/throttle.ts';
|
|
6
|
+
import type { GoogleCompatibleMessageCodec } from '#@/compatible.d/google/message-codec.ts';
|
|
7
|
+
import type { GoogleToolCodec } from '#@/api-types/google/tool-codec.ts';
|
|
8
|
+
import type { GoogleBilling } from '#@/api-types/google/billing.ts';
|
|
9
|
+
import type { ToolCallValidator } from '#@/compatible/tool-call-validator.ts';
|
|
10
|
+
export declare class GoogleCompatibleTransport<in out fdm extends Function.Declaration.Map> {
|
|
11
|
+
protected ctx: GoogleCompatibleTransport.Context<fdm>;
|
|
12
|
+
protected apiURL: URL;
|
|
13
|
+
constructor(ctx: GoogleCompatibleTransport.Context<fdm>);
|
|
14
|
+
fetch(wfctx: InferenceContext, session: Session<fdm>, signal?: AbortSignal): Promise<RoleMessage.Ai<fdm>>;
|
|
15
|
+
}
|
|
16
|
+
export declare namespace GoogleCompatibleTransport {
|
|
17
|
+
interface Context<in out fdm extends Function.Declaration.Map> {
|
|
18
|
+
inferenceParams: InferenceParams;
|
|
19
|
+
providerSpec: ProviderSpec;
|
|
20
|
+
fdm: fdm;
|
|
21
|
+
throttle: Throttle;
|
|
22
|
+
toolChoice: Function.ToolChoice<fdm>;
|
|
23
|
+
messageCodec: GoogleCompatibleMessageCodec<fdm>;
|
|
24
|
+
toolCodec: GoogleToolCodec<fdm>;
|
|
25
|
+
billing: GoogleBilling;
|
|
26
|
+
toolCallValidator: ToolCallValidator<fdm>;
|
|
27
|
+
}
|
|
28
|
+
}
|
|
@@ -0,0 +1,74 @@
|
|
|
1
|
+
import { ResponseInvalid } from '#@/engine.ts';
|
|
2
|
+
import { RoleMessage } from '#@/compatible/session.ts';
|
|
3
|
+
import { Function } from '#@/function.ts';
|
|
4
|
+
import * as Google from '@google/genai';
|
|
5
|
+
import * as Undici from 'undici';
|
|
6
|
+
import {} from '#@/inference-context.ts';
|
|
7
|
+
import { Throttle } from '#@/throttle.ts';
|
|
8
|
+
import { logger } from '#@/telemetry.ts';
|
|
9
|
+
export class GoogleCompatibleTransport {
|
|
10
|
+
ctx;
|
|
11
|
+
apiURL;
|
|
12
|
+
constructor(ctx) {
|
|
13
|
+
this.ctx = ctx;
|
|
14
|
+
this.apiURL = new URL(`${this.ctx.providerSpec.baseUrl}/v1beta/models/${this.ctx.inferenceParams.model}:generateContent`);
|
|
15
|
+
}
|
|
16
|
+
async fetch(wfctx, session, signal) {
|
|
17
|
+
const systemInstruction = session.developerMessage && this.ctx.messageCodec.convertFromDeveloperMessage(session.developerMessage);
|
|
18
|
+
const contents = this.ctx.messageCodec.convertFromChatMessages(session.chatMessages);
|
|
19
|
+
await this.ctx.throttle.requests(wfctx);
|
|
20
|
+
const tools = this.ctx.toolCodec.convertFromFunctionDeclarationMap(this.ctx.fdm);
|
|
21
|
+
const reqbody = {
|
|
22
|
+
contents,
|
|
23
|
+
tools: tools.length ? [{
|
|
24
|
+
functionDeclarations: tools,
|
|
25
|
+
}] : undefined,
|
|
26
|
+
toolConfig: tools.length ? {
|
|
27
|
+
functionCallingConfig: this.ctx.toolCodec.convertFromToolChoice(this.ctx.toolChoice),
|
|
28
|
+
} : undefined,
|
|
29
|
+
systemInstruction,
|
|
30
|
+
generationConfig: this.ctx.inferenceParams.maxTokens || this.ctx.inferenceParams.additionalOptions ? {
|
|
31
|
+
maxOutputTokens: this.ctx.inferenceParams.maxTokens ?? undefined,
|
|
32
|
+
...this.ctx.inferenceParams.additionalOptions,
|
|
33
|
+
} : undefined,
|
|
34
|
+
};
|
|
35
|
+
logger.message.trace(reqbody);
|
|
36
|
+
const res = await Undici.fetch(this.apiURL, {
|
|
37
|
+
method: 'POST',
|
|
38
|
+
headers: new Headers({
|
|
39
|
+
'Content-Type': 'application/json',
|
|
40
|
+
'x-goog-api-key': this.ctx.providerSpec.apiKey,
|
|
41
|
+
}),
|
|
42
|
+
body: JSON.stringify(reqbody),
|
|
43
|
+
dispatcher: this.ctx.providerSpec.proxyAgent,
|
|
44
|
+
signal,
|
|
45
|
+
});
|
|
46
|
+
logger.message.trace(res);
|
|
47
|
+
if (res.ok) { }
|
|
48
|
+
else
|
|
49
|
+
throw new Error(undefined, { cause: res });
|
|
50
|
+
const response = await res.json();
|
|
51
|
+
if (response.candidates?.[0]?.content?.parts?.length) { }
|
|
52
|
+
else
|
|
53
|
+
throw new ResponseInvalid('Content missing', { cause: response });
|
|
54
|
+
if (response.candidates[0].finishReason === Google.FinishReason.MAX_TOKENS)
|
|
55
|
+
throw new ResponseInvalid('Token limit exceeded.', { cause: response });
|
|
56
|
+
if (response.candidates[0].finishReason === Google.FinishReason.STOP) { }
|
|
57
|
+
else
|
|
58
|
+
throw new ResponseInvalid('Abnormal finish reason', { cause: response });
|
|
59
|
+
for (const part of response.candidates[0].content.parts) {
|
|
60
|
+
if (part.text)
|
|
61
|
+
logger.inference.debug(part.text);
|
|
62
|
+
if (part.functionCall)
|
|
63
|
+
logger.message.debug(part.functionCall);
|
|
64
|
+
}
|
|
65
|
+
if (response.usageMetadata) { }
|
|
66
|
+
else
|
|
67
|
+
throw new ResponseInvalid('Usage metadata missing', { cause: response });
|
|
68
|
+
wfctx.cost?.(this.ctx.billing.charge(response.usageMetadata));
|
|
69
|
+
const aiMessage = this.ctx.messageCodec.convertToAiMessage(response.candidates[0].content);
|
|
70
|
+
this.ctx.toolCallValidator.validate(aiMessage.getFunctionCalls());
|
|
71
|
+
return aiMessage;
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
//# sourceMappingURL=transport.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transport.js","sourceRoot":"","sources":["../../../src/compatible.d/google/transport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAA4E,MAAM,cAAc,CAAC;AACzH,OAAO,EAAE,WAAW,EAAgB,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,KAAK,MAAM,MAAM,eAAe,CAAC;AACxC,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAyB,MAAM,yBAAyB,CAAC;AAEhE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AAQzC,MAAM,OAAO,yBAAyB;IAEL;IADnB,MAAM,CAAM;IACtB,YAA6B,GAA2C;QAA3C,QAAG,GAAH,GAAG,CAAwC;QACpE,IAAI,CAAC,MAAM,GAAG,IAAI,GAAG,CAAC,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,OAAO,kBAAkB,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,KAAK,kBAAkB,CAAC,CAAA;IAC7H,CAAC;IAEM,KAAK,CAAC,KAAK,CACd,KAAuB,EACvB,OAAqB,EACrB,MAAoB;QAEpB,MAAM,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,IAAI,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,2BAA2B,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC;QAClI,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,uBAAuB,CAAC,OAAO,CAAC,YAAY,CAAC,CAAC;QAErF,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAExC,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,iCAAiC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACjF,MAAM,OAAO,GAAyB;YAClC,QAAQ;YACR,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC;oBACnB,oBAAoB,EAAE,KAAK;iBAC9B,CAAC,CAAC,CAAC,CAAC,SAAS;YACd,UAAU,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;gBACvB,qBAAqB,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,qBAAqB,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC;aACvF,CAAC,CAAC,CAAC,SAAS;YACb,iBAAiB;YACjB,gBAAgB,EAAE,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,IAAI,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,iBAAiB,CAAC,CAAC,CAAC;gBACjG,eAAe,EAAE,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,SAAS,IAAI,SAAS;gBAChE,GAAG,IAAI,CAAC,GAAG,CAAC,eAAe,CAAC,iBAAiB;aAChD,CAAC,CAAC,CAAC,SAAS;SAChB,CAAC;QAEF,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAE9B,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,EAAE;YACxC,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI,OAAO,CAAC;gBACjB,cAAc,EAAE,kBAAkB;gBAClC,gBAAgB,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,MAAM;aACjD,CAAC;YACF,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC;YAC7B,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,UAAU;YAC5C,MAAM;SACT,CAAC,CAAC;QACH,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC1B,IAAI,GAAG,CAAC,EAAE,EAAE,CAAC,CAAA,CAAC;;YAAM,MAAM,IAAI,KAAK,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QAC/D,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,IAAI,EAAoC,CAAC;QAEpE,IAAI,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,CAAC,CAAA,CAAC;;YAAM,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAChI,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,YAAY,KAAK,MAAM,CAAC,YAAY,CAAC,UAAU;YACtE,MAAM,IAAI,eAAe,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC5E,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,YAAY,KAAK,MAAM,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAA,CAAC;;YACnE,MAAM,IAAI,eAAe,CAAC,wBAAwB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAG9E,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACtD,IAAI,IAAI,CAAC,IAAI;gBAAE,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjD,IAAI,IAAI,CAAC,YAAY;gBAAE,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QACnE,CAAC;QAED,IAAI,QAAQ,CAAC,aAAa,EAAE,CAAC,CAAA,CAAC;;YAAM,MAAM,IAAI,eAAe,CAAC,wBAAwB,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,CAAC,CAAC;QAC7G,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC,CAAC;QAE9D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,kBAAkB,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;QAC3F,IAAI,CAAC,GAAG,CAAC,iBAAiB,CAAC,QAAQ,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAClE,OAAO,SAAS,CAAC;IACrB,CAAC;CAGJ"}
|
|
@@ -0,0 +1,19 @@
|
|
|
1
|
+
import { RoleMessage, type Session } from '#@/compatible/session.ts';
|
|
2
|
+
import { Function } from '#@/function.ts';
|
|
3
|
+
import OpenAI from 'openai';
|
|
4
|
+
import type { OpenAIChatCompletionsToolCodec } from '#@/api-types/openai-chatcompletion/tool-codec.ts';
|
|
5
|
+
export declare class OpenAIChatCompletionsCompatibleMessageCodec<in out fdm extends Function.Declaration.Map> {
|
|
6
|
+
protected ctx: OpenAIChatCompletionsCompatibleMessageCodec.Context<fdm>;
|
|
7
|
+
constructor(ctx: OpenAIChatCompletionsCompatibleMessageCodec.Context<fdm>);
|
|
8
|
+
convertToAiMessage(message: OpenAI.ChatCompletionMessage): RoleMessage.Ai<fdm>;
|
|
9
|
+
convertFromDeveloperMessage(developerMessage: RoleMessage.Developer): OpenAI.ChatCompletionSystemMessageParam;
|
|
10
|
+
convertFromUserMessage(userMessage: RoleMessage.User<fdm>): [OpenAI.ChatCompletionUserMessageParam] | OpenAI.ChatCompletionToolMessageParam[];
|
|
11
|
+
convertFromAiMessage(aiMessage: RoleMessage.Ai<fdm>): OpenAI.ChatCompletionAssistantMessageParam;
|
|
12
|
+
convertFromRoleMessage(roleMessage: RoleMessage): OpenAI.ChatCompletionMessageParam[];
|
|
13
|
+
convertFromChatMessages(chatMessages: Session.ChatMessage<fdm>[]): OpenAI.ChatCompletionMessageParam[];
|
|
14
|
+
}
|
|
15
|
+
export declare namespace OpenAIChatCompletionsCompatibleMessageCodec {
|
|
16
|
+
interface Context<in out fdm extends Function.Declaration.Map> {
|
|
17
|
+
toolCodec: OpenAIChatCompletionsToolCodec<fdm>;
|
|
18
|
+
}
|
|
19
|
+
}
|
|
@@ -0,0 +1,66 @@
|
|
|
1
|
+
import { ResponseInvalid } from '#@/engine.ts';
|
|
2
|
+
import { RoleMessage } from '#@/compatible/session.ts';
|
|
3
|
+
import { Function } from '#@/function.ts';
|
|
4
|
+
import OpenAI from 'openai';
|
|
5
|
+
export class OpenAIChatCompletionsCompatibleMessageCodec {
|
|
6
|
+
ctx;
|
|
7
|
+
constructor(ctx) {
|
|
8
|
+
this.ctx = ctx;
|
|
9
|
+
}
|
|
10
|
+
convertToAiMessage(message) {
|
|
11
|
+
const parts = [];
|
|
12
|
+
if (message.content)
|
|
13
|
+
parts.push(RoleMessage.Part.Text.create(message.content));
|
|
14
|
+
if (message.tool_calls)
|
|
15
|
+
parts.push(...message.tool_calls.map(apifc => {
|
|
16
|
+
if (apifc.type === 'function') { }
|
|
17
|
+
else
|
|
18
|
+
throw new Error();
|
|
19
|
+
return this.ctx.toolCodec.convertToFunctionCall(apifc);
|
|
20
|
+
}));
|
|
21
|
+
if (parts.length) { }
|
|
22
|
+
else
|
|
23
|
+
throw new ResponseInvalid('Content or tool calls not found in Response', { cause: message });
|
|
24
|
+
return RoleMessage.Ai.create(parts);
|
|
25
|
+
}
|
|
26
|
+
convertFromDeveloperMessage(developerMessage) {
|
|
27
|
+
return {
|
|
28
|
+
role: 'system',
|
|
29
|
+
content: developerMessage.getOnlyText(),
|
|
30
|
+
};
|
|
31
|
+
}
|
|
32
|
+
convertFromUserMessage(userMessage) {
|
|
33
|
+
const textParts = userMessage.getParts().filter(part => part instanceof RoleMessage.Part.Text.Instance);
|
|
34
|
+
const frs = userMessage.getFunctionResponses();
|
|
35
|
+
if (textParts.length && !frs.length)
|
|
36
|
+
return [{ role: 'user', content: textParts.map(part => ({ type: 'text', text: part.text })) }];
|
|
37
|
+
else if (!textParts.length && frs.length)
|
|
38
|
+
return frs.map(fr => this.ctx.toolCodec.convertFromFunctionResponse(fr));
|
|
39
|
+
else
|
|
40
|
+
throw new Error();
|
|
41
|
+
}
|
|
42
|
+
convertFromAiMessage(aiMessage) {
|
|
43
|
+
const parts = aiMessage.getParts();
|
|
44
|
+
const textParts = parts.filter(part => part instanceof RoleMessage.Part.Text.Instance);
|
|
45
|
+
const fcParts = parts.filter(part => part instanceof Function.Call);
|
|
46
|
+
return {
|
|
47
|
+
role: 'assistant',
|
|
48
|
+
content: textParts.length ? textParts.map(part => part.text).join('') : undefined,
|
|
49
|
+
tool_calls: fcParts.length ? fcParts.map(fc => this.ctx.toolCodec.convertFromFunctionCall(fc)) : undefined,
|
|
50
|
+
};
|
|
51
|
+
}
|
|
52
|
+
convertFromRoleMessage(roleMessage) {
|
|
53
|
+
if (roleMessage instanceof RoleMessage.Developer.Instance)
|
|
54
|
+
return [this.convertFromDeveloperMessage(roleMessage)];
|
|
55
|
+
else if (roleMessage instanceof RoleMessage.User.Instance)
|
|
56
|
+
return this.convertFromUserMessage(roleMessage);
|
|
57
|
+
else if (roleMessage instanceof RoleMessage.Ai.Instance)
|
|
58
|
+
return [this.convertFromAiMessage(roleMessage)];
|
|
59
|
+
else
|
|
60
|
+
throw new Error();
|
|
61
|
+
}
|
|
62
|
+
convertFromChatMessages(chatMessages) {
|
|
63
|
+
return chatMessages.map(chatMessage => this.convertFromRoleMessage(chatMessage)).flat();
|
|
64
|
+
}
|
|
65
|
+
}
|
|
66
|
+
//# sourceMappingURL=message-codec.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"message-codec.js","sourceRoot":"","sources":["../../../src/compatible.d/openai-chatcompletions/message-codec.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAgB,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,MAAM,MAAM,QAAQ,CAAC;AAK5B,MAAM,OAAO,2CAA2C;IACvB;IAA7B,YAA6B,GAA6D;QAA7D,QAAG,GAAH,GAAG,CAA0D;IAAG,CAAC;IAEvF,kBAAkB,CACrB,OAAqC;QAErC,MAAM,KAAK,GAA+B,EAAE,CAAC;QAC7C,IAAI,OAAO,CAAC,OAAO;YACf,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC;QAC9D,IAAI,OAAO,CAAC,UAAU;YAClB,KAAK,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE;gBACzC,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC,CAAA,CAAC;;oBAAM,MAAM,IAAI,KAAK,EAAE,CAAC;gBACzD,OAAO,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,qBAAqB,CAAC,KAAK,CAAC,CAAC;YAC3D,CAAC,CAAC,CAAC,CAAC;QACR,IAAI,KAAK,CAAC,MAAM,EAAE,CAAC,CAAA,CAAC;;YAAM,MAAM,IAAI,eAAe,CAAC,6CAA6C,EAAE,EAAE,KAAK,EAAE,OAAO,EAAE,CAAC,CAAC;QACvH,OAAO,WAAW,CAAC,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IACxC,CAAC;IAEM,2BAA2B,CAAC,gBAAuC;QACtE,OAAO;YACH,IAAI,EAAE,QAAQ;YACd,OAAO,EAAE,gBAAgB,CAAC,WAAW,EAAE;SAC1C,CAAC;IACN,CAAC;IAEM,sBAAsB,CACzB,WAAkC;QAElC,MAAM,SAAS,GAAG,WAAW,CAAC,QAAQ,EAAE,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,YAAY,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACxG,MAAM,GAAG,GAAG,WAAW,CAAC,oBAAoB,EAAE,CAAC;QAC/C,IAAI,SAAS,CAAC,MAAM,IAAI,CAAC,GAAG,CAAC,MAAM;YAC/B,OAAO,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,EAAE,CAAC,CAAC;aAC9F,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,GAAG,CAAC,MAAM;YACpC,OAAO,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,2BAA2B,CAAC,EAAE,CAAC,CAAC,CAAC;;YACxE,MAAM,IAAI,KAAK,EAAE,CAAC;IAC3B,CAAC;IAEM,oBAAoB,CACvB,SAA8B;QAE9B,MAAM,KAAK,GAAG,SAAS,CAAC,QAAQ,EAAE,CAAC;QACnC,MAAM,SAAS,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,YAAY,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvF,MAAM,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,YAAY,QAAQ,CAAC,IAAI,CAAC,CAAC;QACpE,OAAO;YACH,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS;YACjF,UAAU,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,uBAAuB,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS;SAC7G,CAAC;IACN,CAAC;IAEM,sBAAsB,CACzB,WAAwB;QAExB,IAAI,WAAW,YAAY,WAAW,CAAC,SAAS,CAAC,QAAQ;YACrD,OAAO,CAAC,IAAI,CAAC,2BAA2B,CAAC,WAAW,CAAC,CAAC,CAAC;aACtD,IAAI,WAAW,YAAY,WAAW,CAAC,IAAI,CAAC,QAAQ;YACrD,OAAO,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC;aAC/C,IAAI,WAAW,YAAY,WAAW,CAAC,EAAE,CAAC,QAAQ;YACnD,OAAO,CAAC,IAAI,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,CAAC;;YAC/C,MAAM,IAAI,KAAK,EAAE,CAAC;IAC3B,CAAC;IAEM,uBAAuB,CAC1B,YAAwC;QAExC,OAAO,YAAY,CAAC,GAAG,CAAC,WAAW,CAAC,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IAC5F,CAAC;CAEJ"}
|
|
@@ -0,0 +1,9 @@
|
|
|
1
|
+
import { RoleMessage, type Session } from '#@/compatible/session.ts';
|
|
2
|
+
import { Function } from '#@/function.ts';
|
|
3
|
+
import OpenAI from 'openai';
|
|
4
|
+
import type { InferenceContext } from '#@/inference-context.ts';
|
|
5
|
+
export declare abstract class OpenAIChatCompletionsCompatibleTransport<in out fdm extends Function.Declaration.Map> {
|
|
6
|
+
fetch(wfctx: InferenceContext, session: Session<fdm>, signal?: AbortSignal): Promise<RoleMessage.Ai<fdm>>;
|
|
7
|
+
handleFinishReason(completion: OpenAI.ChatCompletion, finishReason: OpenAI.ChatCompletion.Choice['finish_reason']): void;
|
|
8
|
+
protected abstract fetchRaw(wfctx: InferenceContext, session: Session<fdm>, signal?: AbortSignal): Promise<RoleMessage.Ai<fdm>>;
|
|
9
|
+
}
|
|
@@ -0,0 +1,25 @@
|
|
|
1
|
+
import { ResponseInvalid } from '#@/engine.ts';
|
|
2
|
+
import { RoleMessage } from '#@/compatible/session.ts';
|
|
3
|
+
import { Function } from '#@/function.ts';
|
|
4
|
+
import OpenAI from 'openai';
|
|
5
|
+
export class OpenAIChatCompletionsCompatibleTransport {
|
|
6
|
+
async fetch(wfctx, session, signal) {
|
|
7
|
+
try {
|
|
8
|
+
return await this.fetchRaw(wfctx, session, signal);
|
|
9
|
+
}
|
|
10
|
+
catch (e) {
|
|
11
|
+
if (e instanceof OpenAI.APIError)
|
|
12
|
+
throw new ResponseInvalid(undefined, { cause: e });
|
|
13
|
+
else
|
|
14
|
+
throw e;
|
|
15
|
+
}
|
|
16
|
+
}
|
|
17
|
+
handleFinishReason(completion, finishReason) {
|
|
18
|
+
if (finishReason === 'length')
|
|
19
|
+
throw new ResponseInvalid('Token limit exceeded.', { cause: completion });
|
|
20
|
+
if (['stop', 'tool_calls'].includes(finishReason)) { }
|
|
21
|
+
else
|
|
22
|
+
throw new ResponseInvalid('Abnormal finish reason', { cause: finishReason });
|
|
23
|
+
}
|
|
24
|
+
}
|
|
25
|
+
//# sourceMappingURL=transport.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"transport.js","sourceRoot":"","sources":["../../../src/compatible.d/openai-chatcompletions/transport.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAgB,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC1C,OAAO,MAAM,MAAM,QAAQ,CAAC;AAK5B,MAAM,OAAgB,wCAAwC;IAEnD,KAAK,CAAC,KAAK,CACd,KAAuB,EACvB,OAAqB,EACrB,MAAoB;QAEpB,IAAI,CAAC;YACD,OAAO,MAAM,IAAI,CAAC,QAAQ,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QACvD,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,YAAY,MAAM,CAAC,QAAQ;gBAC5B,MAAM,IAAI,eAAe,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;;gBAClD,MAAM,CAAC,CAAC;QACjB,CAAC;IACL,CAAC;IAEM,kBAAkB,CAAC,UAAiC,EAAE,YAA2D;QACpH,IAAI,YAAY,KAAK,QAAQ;YACzB,MAAM,IAAI,eAAe,CAAC,uBAAuB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAC9E,IAAI,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,CAAC,CAAA,CAAC;;YAChD,MAAM,IAAI,eAAe,CAAC,wBAAwB,EAAE,EAAE,KAAK,EAAE,YAAY,EAAE,CAAC,CAAC;IACtF,CAAC;CAOJ"}
|
|
@@ -0,0 +1,36 @@
|
|
|
1
|
+
import { RoleMessage, type Session } from '#@/compatible/session.ts';
|
|
2
|
+
import { Function } from '#@/function.ts';
|
|
3
|
+
import type OpenAI from 'openai';
|
|
4
|
+
import { OpenAIChatCompletionsCompatibleTransport } from '#@/compatible.d/openai-chatcompletions/transport.ts';
|
|
5
|
+
import { type InferenceContext } from '#@/inference-context.ts';
|
|
6
|
+
import * as Undici from 'undici';
|
|
7
|
+
import type { OpenAIChatCompletionsBilling } from '#@/api-types/openai-chatcompletion/billing.ts';
|
|
8
|
+
import type { OpenAIChatCompletionsToolCodec } from '#@/api-types/openai-chatcompletion/tool-codec.ts';
|
|
9
|
+
import type { OpenAIChatCompletionsCompatibleMessageCodec } from '#@/compatible.d/openai-chatcompletions/message-codec.ts';
|
|
10
|
+
import type { Throttle } from '#@/throttle.ts';
|
|
11
|
+
import type { ToolCallValidator } from '#@/compatible/tool-call-validator.ts';
|
|
12
|
+
export declare abstract class OpenAIChatCompletionsCompatibleMonolith<in out fdm extends Function.Declaration.Map> extends OpenAIChatCompletionsCompatibleTransport<fdm> {
|
|
13
|
+
protected ctx: OpenAIChatCompletionsCompatibleMonolith.Context<fdm>;
|
|
14
|
+
constructor(ctx: OpenAIChatCompletionsCompatibleMonolith.Context<fdm>);
|
|
15
|
+
protected getApiURL(baseUrl: string): URL;
|
|
16
|
+
protected makeParams(session: Session<fdm>): OpenAI.ChatCompletionCreateParamsNonStreaming;
|
|
17
|
+
protected fetchRaw(wfctx: InferenceContext, session: Session<fdm>, signal?: AbortSignal): Promise<RoleMessage.Ai<fdm>>;
|
|
18
|
+
}
|
|
19
|
+
export declare namespace OpenAIChatCompletionsCompatibleMonolith {
|
|
20
|
+
interface Context<in out fdm extends Function.Declaration.Map> {
|
|
21
|
+
proxyAgent?: Undici.ProxyAgent;
|
|
22
|
+
apiURL: URL;
|
|
23
|
+
apiKey: string;
|
|
24
|
+
model: string;
|
|
25
|
+
fdm: fdm;
|
|
26
|
+
maxTokens?: number;
|
|
27
|
+
throttle: Throttle;
|
|
28
|
+
additionalOptions?: Record<string, unknown>;
|
|
29
|
+
toolChoice: Function.ToolChoice<fdm>;
|
|
30
|
+
parallelToolCall: boolean;
|
|
31
|
+
messageCodec: OpenAIChatCompletionsCompatibleMessageCodec<fdm>;
|
|
32
|
+
toolCodec: OpenAIChatCompletionsToolCodec<fdm>;
|
|
33
|
+
billing: OpenAIChatCompletionsBilling<fdm>;
|
|
34
|
+
toolCallValidator: ToolCallValidator<fdm>;
|
|
35
|
+
}
|
|
36
|
+
}
|
|
@@ -0,0 +1,74 @@
|
|
|
1
|
+
import { RoleMessage } from '#@/compatible/session.ts';
|
|
2
|
+
import { Function } from '#@/function.ts';
|
|
3
|
+
import { OpenAIChatCompletionsCompatibleTransport } from '#@/compatible.d/openai-chatcompletions/transport.ts';
|
|
4
|
+
import {} from '#@/inference-context.ts';
|
|
5
|
+
import * as Undici from 'undici';
|
|
6
|
+
import { ResponseInvalid } from '#@/engine.ts';
|
|
7
|
+
import { logger } from '#@/telemetry.ts';
|
|
8
|
+
export class OpenAIChatCompletionsCompatibleMonolith extends OpenAIChatCompletionsCompatibleTransport {
|
|
9
|
+
ctx;
|
|
10
|
+
constructor(ctx) {
|
|
11
|
+
super();
|
|
12
|
+
this.ctx = ctx;
|
|
13
|
+
}
|
|
14
|
+
getApiURL(baseUrl) {
|
|
15
|
+
return new URL(`${baseUrl}/chat/completions`);
|
|
16
|
+
}
|
|
17
|
+
makeParams(session) {
|
|
18
|
+
const tools = this.ctx.toolCodec.convertFromFunctionDeclarationMap(this.ctx.fdm);
|
|
19
|
+
return {
|
|
20
|
+
model: this.ctx.model,
|
|
21
|
+
stream: false,
|
|
22
|
+
messages: [
|
|
23
|
+
...(session.developerMessage ? this.ctx.messageCodec.convertFromRoleMessage(session.developerMessage) : []),
|
|
24
|
+
...this.ctx.messageCodec.convertFromChatMessages(session.chatMessages),
|
|
25
|
+
],
|
|
26
|
+
tools: tools.length ? tools : undefined,
|
|
27
|
+
tool_choice: tools.length ? this.ctx.toolCodec.convertFromToolChoice(this.ctx.toolChoice) : undefined,
|
|
28
|
+
parallel_tool_calls: tools.length ? this.ctx.parallelToolCall : undefined,
|
|
29
|
+
max_completion_tokens: this.ctx.maxTokens ?? undefined,
|
|
30
|
+
...this.ctx.additionalOptions,
|
|
31
|
+
};
|
|
32
|
+
}
|
|
33
|
+
async fetchRaw(wfctx, session, signal) {
|
|
34
|
+
const params = this.makeParams(session);
|
|
35
|
+
logger.message.trace(params);
|
|
36
|
+
await this.ctx.throttle.requests(wfctx);
|
|
37
|
+
const res = await Undici.fetch(this.ctx.apiURL, {
|
|
38
|
+
method: 'POST',
|
|
39
|
+
headers: new Headers({
|
|
40
|
+
'Content-Type': 'application/json',
|
|
41
|
+
'Authorization': `Bearer ${this.ctx.apiKey}`,
|
|
42
|
+
}),
|
|
43
|
+
body: JSON.stringify(params),
|
|
44
|
+
dispatcher: this.ctx.proxyAgent,
|
|
45
|
+
signal,
|
|
46
|
+
});
|
|
47
|
+
if (res.ok) { }
|
|
48
|
+
else
|
|
49
|
+
throw new Error(undefined, { cause: res });
|
|
50
|
+
const completion = await res.json();
|
|
51
|
+
logger.message.trace(completion);
|
|
52
|
+
const choice = completion.choices[0];
|
|
53
|
+
if (choice) { }
|
|
54
|
+
else
|
|
55
|
+
throw new ResponseInvalid('Content missing', { cause: completion });
|
|
56
|
+
this.handleFinishReason(completion, choice.finish_reason);
|
|
57
|
+
if (completion.usage) { }
|
|
58
|
+
else
|
|
59
|
+
throw new Error();
|
|
60
|
+
const cost = this.ctx.billing.charge(completion.usage);
|
|
61
|
+
const aiMessage = this.ctx.messageCodec.convertToAiMessage(choice.message);
|
|
62
|
+
const text = aiMessage.getText();
|
|
63
|
+
if (text)
|
|
64
|
+
logger.inference.debug(text);
|
|
65
|
+
const apifcs = choice.message.tool_calls;
|
|
66
|
+
if (apifcs?.length)
|
|
67
|
+
logger.message.debug(apifcs);
|
|
68
|
+
logger.message.debug(completion.usage);
|
|
69
|
+
wfctx.cost?.(cost);
|
|
70
|
+
this.ctx.toolCallValidator.validate(aiMessage.getFunctionCalls());
|
|
71
|
+
return aiMessage;
|
|
72
|
+
}
|
|
73
|
+
}
|
|
74
|
+
//# sourceMappingURL=monolith.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"monolith.js","sourceRoot":"","sources":["../../../src/compatible.d/openai-chatcompletions.d/monolith.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAgB,MAAM,0BAA0B,CAAC;AACrE,OAAO,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAE1C,OAAO,EAAE,wCAAwC,EAAE,MAAM,qDAAqD,CAAC;AAC/G,OAAO,EAAyB,MAAM,yBAAyB,CAAC;AAChE,OAAO,KAAK,MAAM,MAAM,QAAQ,CAAC;AACjC,OAAO,EAAE,eAAe,EAAE,MAAM,cAAc,CAAC;AAC/C,OAAO,EAAE,MAAM,EAAE,MAAM,iBAAiB,CAAC;AASzC,MAAM,OAAgB,uCAAqF,SACvG,wCAA6C;IAEhB;IAA7B,YAA6B,GAAyD;QAClF,KAAK,EAAE,CAAC;QADiB,QAAG,GAAH,GAAG,CAAsD;IAEtF,CAAC;IAES,SAAS,CAAC,OAAe;QAC/B,OAAO,IAAI,GAAG,CAAC,GAAG,OAAO,mBAAmB,CAAC,CAAC;IAClD,CAAC;IAES,UAAU,CAChB,OAAqB;QAErB,MAAM,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,iCAAiC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACjF,OAAO;YACH,KAAK,EAAE,IAAI,CAAC,GAAG,CAAC,KAAK;YACrB,MAAM,EAAE,KAAK;YACb,QAAQ,EAAE;gBACN,GAAG,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,sBAAsB,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;gBAC3G,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,uBAAuB,CAAC,OAAO,CAAC,YAAY,CAAC;aACzE;YACD,KAAK,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;YACvC,WAAW,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,SAAS,CAAC,qBAAqB,CAAC,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS;YACrG,mBAAmB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,gBAAgB,CAAC,CAAC,CAAC,SAAS;YACzE,qBAAqB,EAAE,IAAI,CAAC,GAAG,CAAC,SAAS,IAAI,SAAS;YACtD,GAAG,IAAI,CAAC,GAAG,CAAC,iBAAiB;SAChC,CAAC;IACN,CAAC;IAEkB,KAAK,CAAC,QAAQ,CAC7B,KAAuB,EACvB,OAAqB,EACrB,MAAoB;QAEpB,MAAM,MAAM,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QACxC,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QAE7B,MAAM,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACxC,MAAM,GAAG,GAAG,MAAM,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE;YAC5C,MAAM,EAAE,MAAM;YACd,OAAO,EAAE,IAAI,OAAO,CAAC;gBACjB,cAAc,EAAE,kBAAkB;gBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE;aAC/C,CAAC;YACF,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;YAC5B,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,UAAU;YAC/B,MAAM;SACT,CAAC,CAAC;QACH,IAAI,GAAG,CAAC,EAAE,EAAE,CAAC,CAAA,CAAC;;YAAM,MAAM,IAAI,KAAK,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QAC/D,MAAM,UAAU,GAAG,MAAM,GAAG,CAAC,IAAI,EAA2B,CAAC;QAC7D,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;QAEjC,MAAM,MAAM,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC;QACrC,IAAI,MAAM,EAAE,CAAC,CAAA,CAAC;;YAAM,MAAM,IAAI,eAAe,CAAC,iBAAiB,EAAE,EAAE,KAAK,EAAE,UAAU,EAAE,CAAC,CAAC;QAExF,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;QAE1D,IAAI,UAAU,CAAC,KAAK,EAAE,CAAC,CAAA,CAAC;;YAAM,MAAM,IAAI,KAAK,EAAE,CAAC;QAChD,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAEvD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,kBAAkB,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAE3E,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC;QACjC,IAAI,IAAI;YAAE,MAAM,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACvC,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC;QACzC,IAAI,MAAM,EAAE,MAAM;YAAE,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACjD,MAAM,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QACvC,KAAK,CAAC,IAAI,EAAE,CAAC,IAAI,CAAC,CAAC;QAEnB,IAAI,CAAC,GAAG,CAAC,iBAAiB,CAAC,QAAQ,CAAC,SAAS,CAAC,gBAAgB,EAAE,CAAC,CAAC;QAElE,OAAO,SAAS,CAAC;IACrB,CAAC;CACJ"}
|
|
@@ -0,0 +1,35 @@
|
|
|
1
|
+
import { RoleMessage, type Session } from '#@/compatible/session.ts';
|
|
2
|
+
import { Function } from '#@/function.ts';
|
|
3
|
+
import OpenAI from 'openai';
|
|
4
|
+
import { OpenAIChatCompletionsCompatibleTransport } from '#@/compatible.d/openai-chatcompletions/transport.ts';
|
|
5
|
+
import { type InferenceContext } from '#@/inference-context.ts';
|
|
6
|
+
import { type InferenceParams, type ProviderSpec } from '#@/engine.ts';
|
|
7
|
+
import type { ToolCallValidator } from '#@/compatible/tool-call-validator.ts';
|
|
8
|
+
import type { OpenAIChatCompletionsBilling } from '#@/api-types/openai-chatcompletion/billing.ts';
|
|
9
|
+
import type { OpenAIChatCompletionsToolCodec } from '#@/api-types/openai-chatcompletion/tool-codec.ts';
|
|
10
|
+
import { Throttle } from '#@/throttle.ts';
|
|
11
|
+
import { type OpenAIChatCompletionsCompatibleMessageCodec } from '#@/compatible.d/openai-chatcompletions/message-codec.ts';
|
|
12
|
+
export declare abstract class OpenAIChatCompletionsCompatibleStream<in out fdm extends Function.Declaration.Map> extends OpenAIChatCompletionsCompatibleTransport<fdm> {
|
|
13
|
+
protected ctx: OpenAIChatCompletionsCompatibleStream.Context<fdm>;
|
|
14
|
+
protected client: OpenAI;
|
|
15
|
+
constructor(ctx: OpenAIChatCompletionsCompatibleStream.Context<fdm>);
|
|
16
|
+
protected makeParams(session: Session<fdm>): OpenAI.ChatCompletionCreateParamsStreaming;
|
|
17
|
+
convertToFunctionCallFromDelta(apifc: OpenAI.ChatCompletionChunk.Choice.Delta.ToolCall): Function.Call.From<fdm>;
|
|
18
|
+
convertCompletionStockToCompletion(stock: OpenAI.ChatCompletionChunk): OpenAI.ChatCompletion;
|
|
19
|
+
fetchRaw(wfctx: InferenceContext, session: Session<fdm>, signal?: AbortSignal): Promise<RoleMessage.Ai<fdm>>;
|
|
20
|
+
protected abstract getDeltaThoughts(delta: OpenAI.ChatCompletionChunk.Choice.Delta): string;
|
|
21
|
+
}
|
|
22
|
+
export declare namespace OpenAIChatCompletionsCompatibleStream {
|
|
23
|
+
interface Context<in out fdm extends Function.Declaration.Map> {
|
|
24
|
+
inferenceParams: InferenceParams;
|
|
25
|
+
providerSpec: ProviderSpec;
|
|
26
|
+
fdm: fdm;
|
|
27
|
+
throttle: Throttle;
|
|
28
|
+
toolChoice: Function.ToolChoice<fdm>;
|
|
29
|
+
parallelToolCall: boolean;
|
|
30
|
+
messageCodec: OpenAIChatCompletionsCompatibleMessageCodec<fdm>;
|
|
31
|
+
toolCodec: OpenAIChatCompletionsToolCodec<fdm>;
|
|
32
|
+
billing: OpenAIChatCompletionsBilling<fdm>;
|
|
33
|
+
toolCallValidator: ToolCallValidator<fdm>;
|
|
34
|
+
}
|
|
35
|
+
}
|