@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.
Files changed (215) hide show
  1. package/DEVELOPMENT.md +13 -14
  2. package/README.md +7 -14
  3. package/build/adaptor.d.ts +5 -5
  4. package/build/adaptor.js +16 -16
  5. package/build/adaptor.js.map +1 -1
  6. package/build/agentloop.d.ts +5 -5
  7. package/build/agentloop.js +4 -4
  8. package/build/agentloop.js.map +1 -1
  9. package/build/api-types/anthropic/billing.d.ts +12 -0
  10. package/build/api-types/anthropic/billing.js +15 -0
  11. package/build/api-types/anthropic/billing.js.map +1 -0
  12. package/build/api-types/anthropic/tool-codec.d.ts +17 -0
  13. package/build/api-types/anthropic/tool-codec.js +73 -0
  14. package/build/api-types/anthropic/tool-codec.js.map +1 -0
  15. package/build/api-types/google/billing.d.ts +12 -0
  16. package/build/api-types/google/billing.js +24 -0
  17. package/build/api-types/google/billing.js.map +1 -0
  18. package/build/api-types/google/restful-request.d.ts +8 -0
  19. package/build/api-types/google/restful-request.js +2 -0
  20. package/build/api-types/google/restful-request.js.map +1 -0
  21. package/build/api-types/google/tool-codec.d.ts +16 -0
  22. package/build/api-types/{google.js → google/tool-codec.js} +27 -26
  23. package/build/api-types/google/tool-codec.js.map +1 -0
  24. package/build/api-types/openai-chatcompletion/billing.d.ts +13 -0
  25. package/build/api-types/openai-chatcompletion/billing.js +16 -0
  26. package/build/api-types/openai-chatcompletion/billing.js.map +1 -0
  27. package/build/api-types/openai-chatcompletion/tool-codec.d.ts +18 -0
  28. package/build/api-types/openai-chatcompletion/tool-codec.js +86 -0
  29. package/build/api-types/openai-chatcompletion/tool-codec.js.map +1 -0
  30. package/build/api-types/openai-responses/billing.d.ts +12 -0
  31. package/build/api-types/openai-responses/billing.js +15 -0
  32. package/build/api-types/openai-responses/billing.js.map +1 -0
  33. package/build/api-types/openai-responses/tool-codec.d.ts +16 -0
  34. package/build/api-types/openai-responses/tool-codec.js +72 -0
  35. package/build/api-types/openai-responses/tool-codec.js.map +1 -0
  36. package/build/compatible/engine.d/aliyun.d.ts +23 -0
  37. package/build/compatible/engine.d/aliyun.js +44 -0
  38. package/build/compatible/engine.d/aliyun.js.map +1 -0
  39. package/build/compatible/engine.d/anthropic.d.ts +23 -0
  40. package/build/compatible/engine.d/anthropic.js +41 -0
  41. package/build/compatible/engine.d/anthropic.js.map +1 -0
  42. package/build/compatible/engine.d/google.d.ts +23 -0
  43. package/build/compatible/engine.d/google.js +45 -0
  44. package/build/compatible/engine.d/google.js.map +1 -0
  45. package/build/compatible/engine.d/openai-responses.d.ts +23 -0
  46. package/build/compatible/engine.d/openai-responses.js +41 -0
  47. package/build/compatible/engine.d/openai-responses.js.map +1 -0
  48. package/build/compatible/engine.d.ts +18 -0
  49. package/build/compatible/engine.js +21 -0
  50. package/build/compatible/engine.js.map +1 -0
  51. package/build/compatible/session.d.ts +78 -0
  52. package/build/compatible/session.js +138 -0
  53. package/build/compatible/session.js.map +1 -0
  54. package/build/compatible/tool-call-validator.d.ts +11 -0
  55. package/build/compatible/tool-call-validator.js +12 -0
  56. package/build/compatible/tool-call-validator.js.map +1 -0
  57. package/build/compatible.d/aliyun/transport.d.ts +11 -0
  58. package/build/compatible.d/aliyun/transport.js +9 -0
  59. package/build/compatible.d/aliyun/transport.js.map +1 -0
  60. package/build/compatible.d/anthropic/message-codec.d.ts +31 -0
  61. package/build/compatible.d/anthropic/message-codec.js +88 -0
  62. package/build/compatible.d/anthropic/message-codec.js.map +1 -0
  63. package/build/compatible.d/anthropic/transport.d.ts +31 -0
  64. package/build/compatible.d/anthropic/transport.js +137 -0
  65. package/build/compatible.d/anthropic/transport.js.map +1 -0
  66. package/build/compatible.d/google/message-codec.d.ts +31 -0
  67. package/build/compatible.d/google/message-codec.js +98 -0
  68. package/build/compatible.d/google/message-codec.js.map +1 -0
  69. package/build/compatible.d/google/transport.d.ts +28 -0
  70. package/build/compatible.d/google/transport.js +74 -0
  71. package/build/compatible.d/google/transport.js.map +1 -0
  72. package/build/compatible.d/openai-chatcompletions/message-codec.d.ts +19 -0
  73. package/build/compatible.d/openai-chatcompletions/message-codec.js +66 -0
  74. package/build/compatible.d/openai-chatcompletions/message-codec.js.map +1 -0
  75. package/build/compatible.d/openai-chatcompletions/transport.d.ts +9 -0
  76. package/build/compatible.d/openai-chatcompletions/transport.js +25 -0
  77. package/build/compatible.d/openai-chatcompletions/transport.js.map +1 -0
  78. package/build/compatible.d/openai-chatcompletions.d/monolith.d.ts +36 -0
  79. package/build/compatible.d/openai-chatcompletions.d/monolith.js +74 -0
  80. package/build/compatible.d/openai-chatcompletions.d/monolith.js.map +1 -0
  81. package/build/compatible.d/openai-chatcompletions.d/stream.d.ts +35 -0
  82. package/build/{compatible-engines.d → compatible.d}/openai-chatcompletions.d/stream.js +52 -42
  83. package/build/compatible.d/openai-chatcompletions.d/stream.js.map +1 -0
  84. package/build/compatible.d/openai-responses/message-codec.d.ts +32 -0
  85. package/build/compatible.d/openai-responses/message-codec.js +105 -0
  86. package/build/compatible.d/openai-responses/message-codec.js.map +1 -0
  87. package/build/compatible.d/openai-responses/transport.d.ts +33 -0
  88. package/build/compatible.d/openai-responses/transport.js +88 -0
  89. package/build/compatible.d/openai-responses/transport.js.map +1 -0
  90. package/build/config.js +1 -1
  91. package/build/config.js.map +1 -1
  92. package/build/engine.d.ts +48 -26
  93. package/build/engine.js +78 -29
  94. package/build/engine.js.map +1 -1
  95. package/build/exports.d.ts +14 -13
  96. package/build/exports.js +14 -13
  97. package/build/exports.js.map +1 -1
  98. package/build/function.d.ts +25 -13
  99. package/build/function.js +2 -1
  100. package/build/function.js.map +1 -1
  101. package/build/inference-context.d.ts +1 -1
  102. package/build/native-engines.d/google/agentloop.d.ts +6 -6
  103. package/build/native-engines.d/google/agentloop.js +5 -5
  104. package/build/native-engines.d/google/agentloop.js.map +1 -1
  105. package/build/native-engines.d/google/engine.d.ts +28 -103
  106. package/build/native-engines.d/google/engine.js +63 -304
  107. package/build/native-engines.d/google/engine.js.map +1 -1
  108. package/build/native-engines.d/google/exports.d.ts +5 -5
  109. package/build/native-engines.d/google/exports.js +5 -5
  110. package/build/native-engines.d/google/exports.js.map +1 -1
  111. package/build/native-engines.d/google/message-codec.d.ts +21 -0
  112. package/build/native-engines.d/google/message-codec.js +69 -0
  113. package/build/native-engines.d/google/message-codec.js.map +1 -0
  114. package/build/native-engines.d/google/session.d.ts +16 -15
  115. package/build/native-engines.d/google/session.js +2 -2
  116. package/build/native-engines.d/google/session.js.map +1 -1
  117. package/build/native-engines.d/google/transport.d.ts +31 -0
  118. package/build/native-engines.d/google/transport.js +81 -0
  119. package/build/native-engines.d/google/transport.js.map +1 -0
  120. package/build/native-engines.d/openai-responses/agentloop.d.ts +7 -7
  121. package/build/native-engines.d/openai-responses/agentloop.js +6 -6
  122. package/build/native-engines.d/openai-responses/agentloop.js.map +1 -1
  123. package/build/native-engines.d/openai-responses/engine.d.ts +26 -106
  124. package/build/native-engines.d/openai-responses/engine.js +53 -334
  125. package/build/native-engines.d/openai-responses/engine.js.map +1 -1
  126. package/build/native-engines.d/openai-responses/exports.d.ts +6 -7
  127. package/build/native-engines.d/openai-responses/exports.js +6 -7
  128. package/build/native-engines.d/openai-responses/exports.js.map +1 -1
  129. package/build/native-engines.d/openai-responses/message-codec.d.ts +21 -0
  130. package/build/native-engines.d/openai-responses/message-codec.js +69 -0
  131. package/build/native-engines.d/openai-responses/message-codec.js.map +1 -0
  132. package/build/native-engines.d/openai-responses/session.d.ts +27 -26
  133. package/build/native-engines.d/openai-responses/session.js +3 -3
  134. package/build/native-engines.d/openai-responses/session.js.map +1 -1
  135. package/build/native-engines.d/openai-responses/tool-call-validator.d.ts +12 -0
  136. package/build/native-engines.d/openai-responses/tool-call-validator.js +33 -0
  137. package/build/native-engines.d/openai-responses/tool-call-validator.js.map +1 -0
  138. package/build/native-engines.d/openai-responses/tool.d.ts +3 -3
  139. package/build/native-engines.d/openai-responses/tool.js +1 -1
  140. package/build/native-engines.d/openai-responses/tool.js.map +1 -1
  141. package/build/native-engines.d/openai-responses/transport.d.ts +36 -0
  142. package/build/native-engines.d/openai-responses/transport.js +104 -0
  143. package/build/native-engines.d/openai-responses/transport.js.map +1 -0
  144. package/build/session.d.ts +5 -75
  145. package/build/session.js +0 -136
  146. package/build/session.js.map +1 -1
  147. package/build/throttle.d.ts +1 -1
  148. package/build/tsconfig.tsbuildinfo +1 -1
  149. package/package.json +4 -4
  150. package/test/build/test-codecs.js +230 -0
  151. package/test/build/test-codecs.js.map +1 -0
  152. package/test/build/test-compatible-engine.d.ts +1 -0
  153. package/test/build/test-compatible-engine.js +27 -0
  154. package/test/build/test-compatible-engine.js.map +1 -0
  155. package/test/build/test-compatible-engines.d.ts +1 -0
  156. package/test/build/test-compatible-engines.js +112 -0
  157. package/test/build/test-compatible-engines.js.map +1 -0
  158. package/test/build/test-engine.d.ts +1 -0
  159. package/test/build/test-engine.js +97 -0
  160. package/test/build/test-engine.js.map +1 -0
  161. package/test/build/test-helpers.d.ts +46 -0
  162. package/test/build/test-helpers.js +45 -0
  163. package/test/build/test-helpers.js.map +1 -0
  164. package/test/build/test-native-engines.d.ts +1 -0
  165. package/test/build/test-native-engines.js +64 -0
  166. package/test/build/test-native-engines.js.map +1 -0
  167. package/test/build/test.d.ts +5 -0
  168. package/test/build/test.js +6 -0
  169. package/test/build/test.js.map +1 -0
  170. package/test/build/tsconfig.tsbuildinfo +1 -0
  171. package/test/src/test-codecs.ts +256 -0
  172. package/test/src/test-compatible-engine.ts +45 -0
  173. package/test/src/test-compatible-engines.ts +129 -0
  174. package/test/src/test-engine.ts +144 -0
  175. package/test/src/test-helpers.ts +53 -0
  176. package/test/src/test-native-engines.ts +75 -0
  177. package/test/src/test.ts +5 -0
  178. package/test/src/tsconfig.json +30 -0
  179. package/build/api-types/anthropic.d.ts +0 -29
  180. package/build/api-types/anthropic.js +0 -93
  181. package/build/api-types/anthropic.js.map +0 -1
  182. package/build/api-types/google.d.ts +0 -28
  183. package/build/api-types/google.js.map +0 -1
  184. package/build/api-types/openai-chat-completions.d.ts +0 -31
  185. package/build/api-types/openai-chat-completions.js +0 -113
  186. package/build/api-types/openai-chat-completions.js.map +0 -1
  187. package/build/api-types/openai-responses.d.ts +0 -23
  188. package/build/api-types/openai-responses.js +0 -71
  189. package/build/api-types/openai-responses.js.map +0 -1
  190. package/build/compatible-engine.d.ts +0 -51
  191. package/build/compatible-engine.js +0 -67
  192. package/build/compatible-engine.js.map +0 -1
  193. package/build/compatible-engines.d/aliyun.d.ts +0 -69
  194. package/build/compatible-engines.d/aliyun.js +0 -149
  195. package/build/compatible-engines.d/aliyun.js.map +0 -1
  196. package/build/compatible-engines.d/anthropic.d.ts +0 -78
  197. package/build/compatible-engines.d/anthropic.js +0 -312
  198. package/build/compatible-engines.d/anthropic.js.map +0 -1
  199. package/build/compatible-engines.d/google.d.ts +0 -83
  200. package/build/compatible-engines.d/google.js +0 -279
  201. package/build/compatible-engines.d/google.js.map +0 -1
  202. package/build/compatible-engines.d/openai-chatcompletions.d/monolith.d.ts +0 -15
  203. package/build/compatible-engines.d/openai-chatcompletions.d/monolith.js +0 -75
  204. package/build/compatible-engines.d/openai-chatcompletions.d/monolith.js.map +0 -1
  205. package/build/compatible-engines.d/openai-chatcompletions.d/stream.d.ts +0 -21
  206. package/build/compatible-engines.d/openai-chatcompletions.d/stream.js.map +0 -1
  207. package/build/compatible-engines.d/openai-chatcompletions.d.ts +0 -27
  208. package/build/compatible-engines.d/openai-chatcompletions.js +0 -76
  209. package/build/compatible-engines.d/openai-chatcompletions.js.map +0 -1
  210. package/build/compatible-engines.d/openai-responses.d.ts +0 -93
  211. package/build/compatible-engines.d/openai-responses.js +0 -300
  212. package/build/compatible-engines.d/openai-responses.js.map +0 -1
  213. package/build/test.js +0 -198
  214. package/build/test.js.map +0 -1
  215. /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
+ }