@ank1015/providers 0.0.36 → 0.0.37
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/dist/debug-conversation.d.ts +2 -0
- package/dist/debug-conversation.d.ts.map +1 -0
- package/dist/debug-conversation.js +98 -0
- package/dist/debug-conversation.js.map +1 -0
- package/dist/debug-minimal-test.d.ts +2 -0
- package/dist/debug-minimal-test.d.ts.map +1 -0
- package/dist/debug-minimal-test.js +47 -0
- package/dist/debug-minimal-test.js.map +1 -0
- package/dist/debug-sdk-tool.d.ts +2 -0
- package/dist/debug-sdk-tool.d.ts.map +1 -0
- package/dist/debug-sdk-tool.js +99 -0
- package/dist/debug-sdk-tool.js.map +1 -0
- package/dist/debug-sdk.d.ts +2 -0
- package/dist/debug-sdk.d.ts.map +1 -0
- package/dist/debug-sdk.js +59 -0
- package/dist/debug-sdk.js.map +1 -0
- package/dist/debug-session-flow.d.ts +2 -0
- package/dist/debug-session-flow.d.ts.map +1 -0
- package/dist/debug-session-flow.js +134 -0
- package/dist/debug-session-flow.js.map +1 -0
- package/dist/debug-session-hack.d.ts +2 -0
- package/dist/debug-session-hack.d.ts.map +1 -0
- package/dist/debug-session-hack.js +143 -0
- package/dist/debug-session-hack.js.map +1 -0
- package/dist/llm.d.ts.map +1 -1
- package/dist/llm.js +10 -1
- package/dist/llm.js.map +1 -1
- package/dist/models.generated.d.ts +19 -0
- package/dist/models.generated.d.ts.map +1 -1
- package/dist/models.generated.js +19 -0
- package/dist/models.generated.js.map +1 -1
- package/dist/providers/anthropic-agents-sdk/complete.d.ts +3 -0
- package/dist/providers/anthropic-agents-sdk/complete.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/complete.js +246 -0
- package/dist/providers/anthropic-agents-sdk/complete.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/index.d.ts +4 -0
- package/dist/providers/anthropic-agents-sdk/index.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/index.js +4 -0
- package/dist/providers/anthropic-agents-sdk/index.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/session.d.ts +37 -0
- package/dist/providers/anthropic-agents-sdk/session.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/session.js +188 -0
- package/dist/providers/anthropic-agents-sdk/session.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/stream.d.ts +3 -0
- package/dist/providers/anthropic-agents-sdk/stream.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/stream.js +402 -0
- package/dist/providers/anthropic-agents-sdk/stream.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/types.d.ts +22 -0
- package/dist/providers/anthropic-agents-sdk/types.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/types.js +3 -0
- package/dist/providers/anthropic-agents-sdk/types.js.map +1 -0
- package/dist/providers/anthropic-agents-sdk/utils.d.ts +39 -0
- package/dist/providers/anthropic-agents-sdk/utils.d.ts.map +1 -0
- package/dist/providers/anthropic-agents-sdk/utils.js +251 -0
- package/dist/providers/anthropic-agents-sdk/utils.js.map +1 -0
- package/dist/providers/google/utils.d.ts.map +1 -1
- package/dist/providers/google/utils.js +2 -0
- package/dist/providers/google/utils.js.map +1 -1
- package/dist/providers/google-interactions/complete.d.ts +3 -0
- package/dist/providers/google-interactions/complete.d.ts.map +1 -0
- package/dist/providers/google-interactions/complete.js +63 -0
- package/dist/providers/google-interactions/complete.js.map +1 -0
- package/dist/providers/google-interactions/index.d.ts +5 -0
- package/dist/providers/google-interactions/index.d.ts.map +1 -0
- package/dist/providers/google-interactions/index.js +4 -0
- package/dist/providers/google-interactions/index.js.map +1 -0
- package/dist/providers/google-interactions/poll.d.ts +59 -0
- package/dist/providers/google-interactions/poll.d.ts.map +1 -0
- package/dist/providers/google-interactions/poll.js +161 -0
- package/dist/providers/google-interactions/poll.js.map +1 -0
- package/dist/providers/google-interactions/stream.d.ts +3 -0
- package/dist/providers/google-interactions/stream.d.ts.map +1 -0
- package/dist/providers/google-interactions/stream.js +242 -0
- package/dist/providers/google-interactions/stream.js.map +1 -0
- package/dist/providers/google-interactions/types.d.ts +10 -0
- package/dist/providers/google-interactions/types.d.ts.map +1 -0
- package/dist/providers/google-interactions/types.js +2 -0
- package/dist/providers/google-interactions/types.js.map +1 -0
- package/dist/providers/google-interactions/utils.d.ts +61 -0
- package/dist/providers/google-interactions/utils.d.ts.map +1 -0
- package/dist/providers/google-interactions/utils.js +252 -0
- package/dist/providers/google-interactions/utils.js.map +1 -0
- package/dist/providers/zai/complete.d.ts +3 -0
- package/dist/providers/zai/complete.d.ts.map +1 -0
- package/dist/providers/zai/complete.js +58 -0
- package/dist/providers/zai/complete.js.map +1 -0
- package/dist/providers/zai/index.d.ts +5 -0
- package/dist/providers/zai/index.d.ts.map +1 -0
- package/dist/providers/zai/index.js +4 -0
- package/dist/providers/zai/index.js.map +1 -0
- package/dist/providers/zai/stream.d.ts +3 -0
- package/dist/providers/zai/stream.d.ts.map +1 -0
- package/dist/providers/zai/stream.js +321 -0
- package/dist/providers/zai/stream.js.map +1 -0
- package/dist/providers/zai/types.d.ts +13 -0
- package/dist/providers/zai/types.d.ts.map +1 -0
- package/dist/providers/zai/types.js +2 -0
- package/dist/providers/zai/types.js.map +1 -0
- package/dist/providers/zai/utils.d.ts +19 -0
- package/dist/providers/zai/utils.d.ts.map +1 -0
- package/dist/providers/zai/utils.js +253 -0
- package/dist/providers/zai/utils.js.map +1 -0
- package/dist/test-sdk-direct.d.ts +2 -0
- package/dist/test-sdk-direct.d.ts.map +1 -0
- package/dist/test-sdk-direct.js +41 -0
- package/dist/test-sdk-direct.js.map +1 -0
- package/dist/test.d.ts +2 -0
- package/dist/test.d.ts.map +1 -0
- package/dist/test.js +213 -0
- package/dist/test.js.map +1 -0
- package/dist/types.d.ts +4 -1
- package/dist/types.d.ts.map +1 -1
- package/dist/types.js +1 -1
- package/dist/types.js.map +1 -1
- package/dist/utils/oauth/google-antigravity.d.ts +25 -0
- package/dist/utils/oauth/google-antigravity.d.ts.map +1 -0
- package/dist/utils/oauth/google-antigravity.js +337 -0
- package/dist/utils/oauth/google-antigravity.js.map +1 -0
- package/dist/utils/oauth/pkce.d.ts +13 -0
- package/dist/utils/oauth/pkce.d.ts.map +1 -0
- package/dist/utils/oauth/pkce.js +31 -0
- package/dist/utils/oauth/pkce.js.map +1 -0
- package/package.json +1 -1
|
@@ -0,0 +1,161 @@
|
|
|
1
|
+
import { createClient, getResponseAssistantResponse, getResponseUsage, mapInteractionStatus } from "./utils.js";
|
|
2
|
+
/**
|
|
3
|
+
* Retrieves the current state of an interaction by ID.
|
|
4
|
+
* Use this to check on background tasks or get the final result.
|
|
5
|
+
*/
|
|
6
|
+
export async function getInteraction(model, interactionId, options) {
|
|
7
|
+
const client = createClient(model, options?.apiKey);
|
|
8
|
+
const interaction = await client.interactions.get(interactionId);
|
|
9
|
+
const isComplete = interaction.status === 'completed' ||
|
|
10
|
+
interaction.status === 'failed' ||
|
|
11
|
+
interaction.status === 'cancelled';
|
|
12
|
+
return {
|
|
13
|
+
interaction,
|
|
14
|
+
status: interaction.status,
|
|
15
|
+
isComplete
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
/**
|
|
19
|
+
* Converts an Interaction to a BaseAssistantMessage.
|
|
20
|
+
* Useful for converting poll results to the standard SDK format.
|
|
21
|
+
*/
|
|
22
|
+
export function interactionToMessage(model, interaction, id, startTimestamp) {
|
|
23
|
+
const content = getResponseAssistantResponse(interaction);
|
|
24
|
+
const usage = getResponseUsage(interaction, model);
|
|
25
|
+
let stopReason = mapInteractionStatus(interaction.status);
|
|
26
|
+
// Check if any tool calls are present and update stopReason
|
|
27
|
+
const hasToolCall = content.some(c => c.type === 'toolCall');
|
|
28
|
+
if (hasToolCall && stopReason === 'stop') {
|
|
29
|
+
stopReason = 'toolUse';
|
|
30
|
+
}
|
|
31
|
+
const now = Date.now();
|
|
32
|
+
return {
|
|
33
|
+
role: "assistant",
|
|
34
|
+
message: interaction,
|
|
35
|
+
id,
|
|
36
|
+
model,
|
|
37
|
+
api: model.api,
|
|
38
|
+
timestamp: now,
|
|
39
|
+
duration: startTimestamp ? now - startTimestamp : 0,
|
|
40
|
+
stopReason,
|
|
41
|
+
content,
|
|
42
|
+
usage,
|
|
43
|
+
interactionId: interaction.id,
|
|
44
|
+
interactionStatus: interaction.status
|
|
45
|
+
};
|
|
46
|
+
}
|
|
47
|
+
/**
|
|
48
|
+
* Polls an interaction until it completes, fails, or times out.
|
|
49
|
+
* Returns the final interaction state as a BaseAssistantMessage.
|
|
50
|
+
*
|
|
51
|
+
* @example
|
|
52
|
+
* ```typescript
|
|
53
|
+
* // Start a background task
|
|
54
|
+
* const result = await complete(model, context, { background: true });
|
|
55
|
+
*
|
|
56
|
+
* // Poll for completion
|
|
57
|
+
* const finalResult = await waitForInteraction(
|
|
58
|
+
* model,
|
|
59
|
+
* result.interactionId!,
|
|
60
|
+
* result.id,
|
|
61
|
+
* {
|
|
62
|
+
* pollInterval: 5000,
|
|
63
|
+
* timeout: 300000,
|
|
64
|
+
* onPoll: (interaction) => console.log('Status:', interaction.status)
|
|
65
|
+
* }
|
|
66
|
+
* );
|
|
67
|
+
* ```
|
|
68
|
+
*/
|
|
69
|
+
export async function waitForInteraction(model, interactionId, messageId, options) {
|
|
70
|
+
const pollInterval = options?.pollInterval ?? 5000;
|
|
71
|
+
const timeout = options?.timeout ?? 600000;
|
|
72
|
+
const startTimestamp = Date.now();
|
|
73
|
+
const client = createClient(model, options?.apiKey);
|
|
74
|
+
while (true) {
|
|
75
|
+
// Check for abort
|
|
76
|
+
if (options?.signal?.aborted) {
|
|
77
|
+
const emptyUsage = {
|
|
78
|
+
input: 0,
|
|
79
|
+
output: 0,
|
|
80
|
+
cacheRead: 0,
|
|
81
|
+
cacheWrite: 0,
|
|
82
|
+
totalTokens: 0,
|
|
83
|
+
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
|
|
84
|
+
};
|
|
85
|
+
return {
|
|
86
|
+
role: "assistant",
|
|
87
|
+
message: { id: interactionId, status: 'cancelled', outputs: [] },
|
|
88
|
+
id: messageId,
|
|
89
|
+
model,
|
|
90
|
+
api: model.api,
|
|
91
|
+
errorMessage: "Polling was aborted",
|
|
92
|
+
timestamp: Date.now(),
|
|
93
|
+
duration: Date.now() - startTimestamp,
|
|
94
|
+
stopReason: "aborted",
|
|
95
|
+
content: [],
|
|
96
|
+
usage: emptyUsage,
|
|
97
|
+
interactionId,
|
|
98
|
+
interactionStatus: 'cancelled'
|
|
99
|
+
};
|
|
100
|
+
}
|
|
101
|
+
// Check for timeout
|
|
102
|
+
if (Date.now() - startTimestamp > timeout) {
|
|
103
|
+
const emptyUsage = {
|
|
104
|
+
input: 0,
|
|
105
|
+
output: 0,
|
|
106
|
+
cacheRead: 0,
|
|
107
|
+
cacheWrite: 0,
|
|
108
|
+
totalTokens: 0,
|
|
109
|
+
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
|
|
110
|
+
};
|
|
111
|
+
return {
|
|
112
|
+
role: "assistant",
|
|
113
|
+
message: { id: interactionId, status: 'failed', outputs: [] },
|
|
114
|
+
id: messageId,
|
|
115
|
+
model,
|
|
116
|
+
api: model.api,
|
|
117
|
+
errorMessage: `Polling timed out after ${timeout}ms`,
|
|
118
|
+
timestamp: Date.now(),
|
|
119
|
+
duration: Date.now() - startTimestamp,
|
|
120
|
+
stopReason: "error",
|
|
121
|
+
content: [],
|
|
122
|
+
usage: emptyUsage,
|
|
123
|
+
interactionId,
|
|
124
|
+
interactionStatus: 'failed'
|
|
125
|
+
};
|
|
126
|
+
}
|
|
127
|
+
// Fetch current state
|
|
128
|
+
const interaction = await client.interactions.get(interactionId);
|
|
129
|
+
// Call onPoll callback if provided
|
|
130
|
+
options?.onPoll?.(interaction);
|
|
131
|
+
// Check if complete
|
|
132
|
+
if (interaction.status === 'completed' ||
|
|
133
|
+
interaction.status === 'failed' ||
|
|
134
|
+
interaction.status === 'cancelled') {
|
|
135
|
+
return interactionToMessage(model, interaction, messageId, startTimestamp);
|
|
136
|
+
}
|
|
137
|
+
// Wait before next poll
|
|
138
|
+
await new Promise(resolve => setTimeout(resolve, pollInterval));
|
|
139
|
+
}
|
|
140
|
+
}
|
|
141
|
+
/**
|
|
142
|
+
* Cancels an in-progress interaction.
|
|
143
|
+
* Returns true if cancellation was successful.
|
|
144
|
+
*/
|
|
145
|
+
export async function cancelInteraction(model, interactionId, options) {
|
|
146
|
+
try {
|
|
147
|
+
const client = createClient(model, options?.apiKey);
|
|
148
|
+
// The SDK may provide a cancel method - try to use it
|
|
149
|
+
const interactions = client.interactions;
|
|
150
|
+
if (typeof interactions.cancel === 'function') {
|
|
151
|
+
await interactions.cancel({ id: interactionId });
|
|
152
|
+
return true;
|
|
153
|
+
}
|
|
154
|
+
// If no cancel method, return false
|
|
155
|
+
return false;
|
|
156
|
+
}
|
|
157
|
+
catch {
|
|
158
|
+
return false;
|
|
159
|
+
}
|
|
160
|
+
}
|
|
161
|
+
//# sourceMappingURL=poll.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"poll.js","sourceRoot":"","sources":["../../../src/providers/google-interactions/poll.ts"],"names":[],"mappings":"AAGA,OAAO,EACN,YAAY,EACZ,4BAA4B,EAC5B,gBAAgB,EAChB,oBAAoB,EACpB,MAAM,YAAY,CAAC;AAyBpB;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,cAAc,CACnC,KAAmC,EACnC,aAAqB,EACrB,OAAqC;IAErC,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAEpD,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAA2B,CAAC;IAE3F,MAAM,UAAU,GAAG,WAAW,CAAC,MAAM,KAAK,WAAW;QACpD,WAAW,CAAC,MAAM,KAAK,QAAQ;QAC/B,WAAW,CAAC,MAAM,KAAK,WAAW,CAAC;IAEpC,OAAO;QACN,WAAW;QACX,MAAM,EAAE,WAAW,CAAC,MAAM;QAC1B,UAAU;KACV,CAAC;AACH,CAAC;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CACnC,KAAmC,EACnC,WAAwB,EACxB,EAAU,EACV,cAAuB;IAEvB,MAAM,OAAO,GAAG,4BAA4B,CAAC,WAAW,CAAC,CAAC;IAC1D,MAAM,KAAK,GAAG,gBAAgB,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;IACnD,IAAI,UAAU,GAAG,oBAAoB,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;IAE1D,4DAA4D;IAC5D,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;IAC7D,IAAI,WAAW,IAAI,UAAU,KAAK,MAAM,EAAE,CAAC;QAC1C,UAAU,GAAG,SAAS,CAAC;IACxB,CAAC;IAED,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAEvB,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,WAAW;QACpB,EAAE;QACF,KAAK;QACL,GAAG,EAAE,KAAK,CAAC,GAAG;QACd,SAAS,EAAE,GAAG;QACd,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,GAAG,GAAG,cAAc,CAAC,CAAC,CAAC,CAAC;QACnD,UAAU;QACV,OAAO;QACP,KAAK;QACL,aAAa,EAAE,WAAW,CAAC,EAAE;QAC7B,iBAAiB,EAAE,WAAW,CAAC,MAAM;KACrC,CAAC;AACH,CAAC;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACvC,KAAmC,EACnC,aAAqB,EACrB,SAAiB,EACjB,OAAqB;IAErB,MAAM,YAAY,GAAG,OAAO,EAAE,YAAY,IAAI,IAAI,CAAC;IACnD,MAAM,OAAO,GAAG,OAAO,EAAE,OAAO,IAAI,MAAM,CAAC;IAC3C,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAElC,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;IAEpD,OAAO,IAAI,EAAE,CAAC;QACb,kBAAkB;QAClB,IAAI,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;YAC9B,MAAM,UAAU,GAAU;gBACzB,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,CAAC;gBACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;aACpE,CAAC;YAEF,OAAO;gBACN,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,EAAE,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAA4B;gBAC1F,EAAE,EAAE,SAAS;gBACb,KAAK;gBACL,GAAG,EAAE,KAAK,CAAC,GAAG;gBACd,YAAY,EAAE,qBAAqB;gBACnC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;gBACrC,UAAU,EAAE,SAAS;gBACrB,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,UAAU;gBACjB,aAAa;gBACb,iBAAiB,EAAE,WAAW;aAC9B,CAAC;QACH,CAAC;QAED,oBAAoB;QACpB,IAAI,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,GAAG,OAAO,EAAE,CAAC;YAC3C,MAAM,UAAU,GAAU;gBACzB,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,CAAC;gBACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;aACpE,CAAC;YAEF,OAAO;gBACN,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,EAAE,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,EAAE,EAA4B;gBACvF,EAAE,EAAE,SAAS;gBACb,KAAK;gBACL,GAAG,EAAE,KAAK,CAAC,GAAG;gBACd,YAAY,EAAE,2BAA2B,OAAO,IAAI;gBACpD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;gBACrC,UAAU,EAAE,OAAO;gBACnB,OAAO,EAAE,EAAE;gBACX,KAAK,EAAE,UAAU;gBACjB,aAAa;gBACb,iBAAiB,EAAE,QAAQ;aAC3B,CAAC;QACH,CAAC;QAED,sBAAsB;QACtB,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,aAAa,CAA2B,CAAC;QAE3F,mCAAmC;QACnC,OAAO,EAAE,MAAM,EAAE,CAAC,WAAW,CAAC,CAAC;QAE/B,oBAAoB;QACpB,IAAI,WAAW,CAAC,MAAM,KAAK,WAAW;YACrC,WAAW,CAAC,MAAM,KAAK,QAAQ;YAC/B,WAAW,CAAC,MAAM,KAAK,WAAW,EAAE,CAAC;YAErC,OAAO,oBAAoB,CAAC,KAAK,EAAE,WAAW,EAAE,SAAS,EAAE,cAAc,CAAC,CAAC;QAC5E,CAAC;QAED,wBAAwB;QACxB,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,YAAY,CAAC,CAAC,CAAC;IACjE,CAAC;AACF,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACtC,KAAmC,EACnC,aAAqB,EACrB,OAAqC;IAErC,IAAI,CAAC;QACJ,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAEpD,sDAAsD;QACtD,MAAM,YAAY,GAAG,MAAM,CAAC,YAAmB,CAAC;QAChD,IAAI,OAAO,YAAY,CAAC,MAAM,KAAK,UAAU,EAAE,CAAC;YAC/C,MAAM,YAAY,CAAC,MAAM,CAAC,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC,CAAC;YACjD,OAAO,IAAI,CAAC;QACb,CAAC;QAED,oCAAoC;QACpC,OAAO,KAAK,CAAC;IACd,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,KAAK,CAAC;IACd,CAAC;AACF,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.d.ts","sourceRoot":"","sources":["../../../src/providers/google-interactions/stream.ts"],"names":[],"mappings":"AAAA,OAAO,EACN,cAAc,EASd,MAAM,gBAAgB,CAAC;AAkBxB,eAAO,MAAM,wBAAwB,EAAE,cAAc,CAAC,qBAAqB,CAyQ1E,CAAC"}
|
|
@@ -0,0 +1,242 @@
|
|
|
1
|
+
import { createClient, buildParams, mapInteractionStatus } from "./utils.js";
|
|
2
|
+
import { AssistantMessageEventStream } from "../../utils/event-stream.js";
|
|
3
|
+
import { calculateCost } from "../../models.js";
|
|
4
|
+
export const streamGoogleInteractions = (model, context, options, id) => {
|
|
5
|
+
const stream = new AssistantMessageEventStream();
|
|
6
|
+
(async () => {
|
|
7
|
+
const startTimestamp = Date.now();
|
|
8
|
+
const output = {
|
|
9
|
+
role: "assistant",
|
|
10
|
+
api: model.api,
|
|
11
|
+
model: model,
|
|
12
|
+
id,
|
|
13
|
+
content: [],
|
|
14
|
+
usage: {
|
|
15
|
+
input: 0,
|
|
16
|
+
output: 0,
|
|
17
|
+
cacheRead: 0,
|
|
18
|
+
cacheWrite: 0,
|
|
19
|
+
totalTokens: 0,
|
|
20
|
+
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 },
|
|
21
|
+
},
|
|
22
|
+
stopReason: "stop",
|
|
23
|
+
timestamp: startTimestamp,
|
|
24
|
+
duration: 0
|
|
25
|
+
};
|
|
26
|
+
let finalInteraction = {
|
|
27
|
+
id: '',
|
|
28
|
+
status: 'in_progress',
|
|
29
|
+
outputs: []
|
|
30
|
+
};
|
|
31
|
+
try {
|
|
32
|
+
const client = createClient(model, options?.apiKey);
|
|
33
|
+
const params = buildParams(model, context, options);
|
|
34
|
+
// Enable streaming
|
|
35
|
+
params.stream = true;
|
|
36
|
+
const interactionStream = await client.interactions.create(params);
|
|
37
|
+
stream.push({ type: "start", message: { ...output, timestamp: Date.now() } });
|
|
38
|
+
let currentBlock = null;
|
|
39
|
+
const blocks = output.content;
|
|
40
|
+
const blockIndex = () => blocks.length - 1;
|
|
41
|
+
// Track content indices for proper block management
|
|
42
|
+
const contentIndices = new Map(); // Maps stream index to our content index
|
|
43
|
+
for await (const event of interactionStream) {
|
|
44
|
+
const eventType = event.event_type;
|
|
45
|
+
// Handle interaction.start event
|
|
46
|
+
if (eventType === 'interaction.start') {
|
|
47
|
+
const startEvent = event;
|
|
48
|
+
if (startEvent.interaction) {
|
|
49
|
+
finalInteraction = { ...finalInteraction, ...startEvent.interaction };
|
|
50
|
+
output.interactionId = startEvent.interaction.id;
|
|
51
|
+
output.interactionStatus = startEvent.interaction.status;
|
|
52
|
+
}
|
|
53
|
+
}
|
|
54
|
+
// Handle content.start event
|
|
55
|
+
else if (eventType === 'content.start') {
|
|
56
|
+
const startEvent = event;
|
|
57
|
+
const contentType = startEvent.content?.type;
|
|
58
|
+
const streamIndex = startEvent.index ?? 0;
|
|
59
|
+
if (contentType === 'text') {
|
|
60
|
+
currentBlock = { type: "response", content: [{ type: 'text', content: "" }] };
|
|
61
|
+
output.content.push(currentBlock);
|
|
62
|
+
contentIndices.set(streamIndex, blockIndex());
|
|
63
|
+
stream.push({ type: "text_start", contentIndex: blockIndex(), message: { ...output, timestamp: Date.now() } });
|
|
64
|
+
}
|
|
65
|
+
else if (contentType === 'thought') {
|
|
66
|
+
currentBlock = { type: "thinking", thinkingText: "" };
|
|
67
|
+
output.content.push(currentBlock);
|
|
68
|
+
contentIndices.set(streamIndex, blockIndex());
|
|
69
|
+
stream.push({ type: "thinking_start", contentIndex: blockIndex(), message: { ...output, timestamp: Date.now() } });
|
|
70
|
+
}
|
|
71
|
+
else if (contentType === 'function_call') {
|
|
72
|
+
const fnContent = startEvent.content;
|
|
73
|
+
currentBlock = {
|
|
74
|
+
type: "toolCall",
|
|
75
|
+
toolCallId: fnContent.id || '',
|
|
76
|
+
name: fnContent.name || '',
|
|
77
|
+
arguments: {},
|
|
78
|
+
partialJson: ''
|
|
79
|
+
};
|
|
80
|
+
output.content.push(currentBlock);
|
|
81
|
+
contentIndices.set(streamIndex, blockIndex());
|
|
82
|
+
stream.push({ type: "toolcall_start", contentIndex: blockIndex(), message: { ...output, timestamp: Date.now() } });
|
|
83
|
+
}
|
|
84
|
+
}
|
|
85
|
+
// Handle content.delta event
|
|
86
|
+
else if (eventType === 'content.delta') {
|
|
87
|
+
const deltaEvent = event;
|
|
88
|
+
const delta = deltaEvent.delta;
|
|
89
|
+
const streamIndex = deltaEvent.index ?? 0;
|
|
90
|
+
const contentIdx = contentIndices.get(streamIndex) ?? blockIndex();
|
|
91
|
+
if (delta?.type === 'text' && delta.text) {
|
|
92
|
+
const block = output.content[contentIdx];
|
|
93
|
+
if (block && block.type === 'response') {
|
|
94
|
+
const textIdx = block.content.findIndex(c => c.type === 'text');
|
|
95
|
+
if (textIdx !== -1) {
|
|
96
|
+
block.content[textIdx].content += delta.text;
|
|
97
|
+
}
|
|
98
|
+
}
|
|
99
|
+
stream.push({
|
|
100
|
+
type: "text_delta",
|
|
101
|
+
contentIndex: contentIdx,
|
|
102
|
+
delta: delta.text,
|
|
103
|
+
message: { ...output, timestamp: Date.now() }
|
|
104
|
+
});
|
|
105
|
+
}
|
|
106
|
+
else if (delta?.type === 'thought_summary' && delta.content?.text) {
|
|
107
|
+
const block = output.content[contentIdx];
|
|
108
|
+
if (block && block.type === 'thinking') {
|
|
109
|
+
block.thinkingText += delta.content.text;
|
|
110
|
+
}
|
|
111
|
+
stream.push({
|
|
112
|
+
type: "thinking_delta",
|
|
113
|
+
contentIndex: contentIdx,
|
|
114
|
+
delta: delta.content.text,
|
|
115
|
+
message: { ...output, timestamp: Date.now() }
|
|
116
|
+
});
|
|
117
|
+
}
|
|
118
|
+
else if (delta?.type === 'function_call') {
|
|
119
|
+
const block = output.content[contentIdx];
|
|
120
|
+
if (block && block.type === 'toolCall') {
|
|
121
|
+
// Accumulate arguments delta
|
|
122
|
+
if (delta.arguments) {
|
|
123
|
+
const argStr = JSON.stringify(delta.arguments);
|
|
124
|
+
block.partialJson = (block.partialJson || '') + argStr;
|
|
125
|
+
try {
|
|
126
|
+
block.arguments = { ...block.arguments, ...delta.arguments };
|
|
127
|
+
}
|
|
128
|
+
catch { /* ignore parse errors during streaming */ }
|
|
129
|
+
}
|
|
130
|
+
stream.push({
|
|
131
|
+
type: "toolcall_delta",
|
|
132
|
+
contentIndex: contentIdx,
|
|
133
|
+
delta: JSON.stringify(delta.arguments || {}),
|
|
134
|
+
message: { ...output, timestamp: Date.now() }
|
|
135
|
+
});
|
|
136
|
+
}
|
|
137
|
+
}
|
|
138
|
+
}
|
|
139
|
+
// Handle content.stop event
|
|
140
|
+
else if (eventType === 'content.stop') {
|
|
141
|
+
const stopEvent = event;
|
|
142
|
+
const streamIndex = stopEvent.index ?? 0;
|
|
143
|
+
const contentIdx = contentIndices.get(streamIndex) ?? blockIndex();
|
|
144
|
+
const block = output.content[contentIdx];
|
|
145
|
+
if (block?.type === 'response') {
|
|
146
|
+
stream.push({
|
|
147
|
+
type: "text_end",
|
|
148
|
+
contentIndex: contentIdx,
|
|
149
|
+
content: block.content,
|
|
150
|
+
message: { ...output, timestamp: Date.now() }
|
|
151
|
+
});
|
|
152
|
+
}
|
|
153
|
+
else if (block?.type === 'thinking') {
|
|
154
|
+
stream.push({
|
|
155
|
+
type: "thinking_end",
|
|
156
|
+
contentIndex: contentIdx,
|
|
157
|
+
content: block.thinkingText,
|
|
158
|
+
message: { ...output, timestamp: Date.now() }
|
|
159
|
+
});
|
|
160
|
+
}
|
|
161
|
+
else if (block?.type === 'toolCall') {
|
|
162
|
+
// Clean up partialJson before emitting end
|
|
163
|
+
const toolCall = {
|
|
164
|
+
type: 'toolCall',
|
|
165
|
+
toolCallId: block.toolCallId,
|
|
166
|
+
name: block.name,
|
|
167
|
+
arguments: block.arguments
|
|
168
|
+
};
|
|
169
|
+
// Update the block in content array
|
|
170
|
+
output.content[contentIdx] = toolCall;
|
|
171
|
+
stream.push({
|
|
172
|
+
type: "toolcall_end",
|
|
173
|
+
contentIndex: contentIdx,
|
|
174
|
+
toolCall,
|
|
175
|
+
message: { ...output, timestamp: Date.now() }
|
|
176
|
+
});
|
|
177
|
+
}
|
|
178
|
+
currentBlock = null;
|
|
179
|
+
}
|
|
180
|
+
// Handle interaction.complete event
|
|
181
|
+
else if (eventType === 'interaction.complete') {
|
|
182
|
+
const completeEvent = event;
|
|
183
|
+
if (completeEvent.interaction) {
|
|
184
|
+
finalInteraction = completeEvent.interaction;
|
|
185
|
+
output.interactionId = completeEvent.interaction.id;
|
|
186
|
+
output.interactionStatus = completeEvent.interaction.status;
|
|
187
|
+
// Update usage from final interaction
|
|
188
|
+
const interactionUsage = completeEvent.interaction.usage;
|
|
189
|
+
if (interactionUsage) {
|
|
190
|
+
output.usage = {
|
|
191
|
+
input: (interactionUsage.total_input_tokens || 0) - (interactionUsage.total_cached_tokens || 0),
|
|
192
|
+
output: (interactionUsage.total_output_tokens || 0) + (interactionUsage.total_reasoning_tokens || 0),
|
|
193
|
+
cacheRead: interactionUsage.total_cached_tokens || 0,
|
|
194
|
+
cacheWrite: 0,
|
|
195
|
+
totalTokens: interactionUsage.total_tokens || 0,
|
|
196
|
+
cost: { input: 0, output: 0, cacheRead: 0, cacheWrite: 0, total: 0 }
|
|
197
|
+
};
|
|
198
|
+
calculateCost(model, output.usage);
|
|
199
|
+
}
|
|
200
|
+
output.stopReason = mapInteractionStatus(completeEvent.interaction.status);
|
|
201
|
+
if (output.content.some(b => b.type === 'toolCall') && output.stopReason === 'stop') {
|
|
202
|
+
output.stopReason = 'toolUse';
|
|
203
|
+
}
|
|
204
|
+
}
|
|
205
|
+
}
|
|
206
|
+
// Handle error event
|
|
207
|
+
else if (eventType === 'error') {
|
|
208
|
+
const errorEvent = event;
|
|
209
|
+
throw new Error(errorEvent.error?.message || 'Unknown streaming error');
|
|
210
|
+
}
|
|
211
|
+
}
|
|
212
|
+
if (options?.signal?.aborted) {
|
|
213
|
+
throw new Error("Request was aborted");
|
|
214
|
+
}
|
|
215
|
+
if (output.stopReason === "aborted" || output.stopReason === "error") {
|
|
216
|
+
throw new Error(`Stream ended with status: ${output.stopReason}${output.errorMessage ? ` - ${output.errorMessage}` : ""}`);
|
|
217
|
+
}
|
|
218
|
+
stream.push({ type: "done", reason: output.stopReason, message: { ...output, timestamp: Date.now() } });
|
|
219
|
+
const baseAssistantMessage = {
|
|
220
|
+
...output,
|
|
221
|
+
message: finalInteraction,
|
|
222
|
+
timestamp: Date.now(),
|
|
223
|
+
duration: Date.now() - startTimestamp
|
|
224
|
+
};
|
|
225
|
+
stream.end(baseAssistantMessage);
|
|
226
|
+
}
|
|
227
|
+
catch (error) {
|
|
228
|
+
output.stopReason = options?.signal?.aborted ? "aborted" : "error";
|
|
229
|
+
output.errorMessage = error instanceof Error ? error.message : JSON.stringify(error);
|
|
230
|
+
stream.push({ type: "error", reason: output.stopReason, message: { ...output, timestamp: Date.now() } });
|
|
231
|
+
const baseAssistantMessage = {
|
|
232
|
+
...output,
|
|
233
|
+
message: finalInteraction,
|
|
234
|
+
timestamp: Date.now(),
|
|
235
|
+
duration: Date.now() - startTimestamp
|
|
236
|
+
};
|
|
237
|
+
stream.end(baseAssistantMessage);
|
|
238
|
+
}
|
|
239
|
+
})();
|
|
240
|
+
return stream;
|
|
241
|
+
};
|
|
242
|
+
//# sourceMappingURL=stream.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"stream.js","sourceRoot":"","sources":["../../../src/providers/google-interactions/stream.ts"],"names":[],"mappings":"AAaA,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAC7E,OAAO,EAAE,2BAA2B,EAAE,MAAM,6BAA6B,CAAC;AAC1E,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAahD,MAAM,CAAC,MAAM,wBAAwB,GAA0C,CAC9E,KAAmC,EACnC,OAAgB,EAChB,OAA0C,EAC1C,EAAU,EACT,EAAE;IACH,MAAM,MAAM,GAAG,IAAI,2BAA2B,EAAyB,CAAC;IAExE,CAAC,KAAK,IAAI,EAAE;QACX,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAClC,MAAM,MAAM,GAAqD;YAChE,IAAI,EAAE,WAAW;YACjB,GAAG,EAAE,KAAK,CAAC,GAAG;YACd,KAAK,EAAE,KAAK;YACZ,EAAE;YACF,OAAO,EAAE,EAAE;YACX,KAAK,EAAE;gBACN,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,UAAU,EAAE,CAAC;gBACb,WAAW,EAAE,CAAC;gBACd,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;aACpE;YACD,UAAU,EAAE,MAAM;YAClB,SAAS,EAAE,cAAc;YACzB,QAAQ,EAAE,CAAC;SACX,CAAC;QAEF,IAAI,gBAAgB,GAAgB;YACnC,EAAE,EAAE,EAAE;YACN,MAAM,EAAE,aAAa;YACrB,OAAO,EAAE,EAAE;SACI,CAAC;QAEjB,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,YAAY,CAAC,KAAK,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;YACpD,MAAM,MAAM,GAAG,WAAW,CAAC,KAAK,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAEpD,mBAAmB;YAClB,MAAc,CAAC,MAAM,GAAG,IAAI,CAAC;YAE9B,MAAM,iBAAiB,GAAG,MAAM,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAiC,CAAC;YAEnG,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;YAE9E,IAAI,YAAY,GAAgH,IAAI,CAAC;YACrI,MAAM,MAAM,GAAG,MAAM,CAAC,OAAO,CAAC;YAC9B,MAAM,UAAU,GAAG,GAAG,EAAE,CAAC,MAAM,CAAC,MAAM,GAAG,CAAC,CAAC;YAE3C,oDAAoD;YACpD,MAAM,cAAc,GAAG,IAAI,GAAG,EAAkB,CAAC,CAAC,yCAAyC;YAE3F,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,iBAAiB,EAAE,CAAC;gBAC7C,MAAM,SAAS,GAAI,KAAa,CAAC,UAAU,CAAC;gBAE5C,iCAAiC;gBACjC,IAAI,SAAS,KAAK,mBAAmB,EAAE,CAAC;oBACvC,MAAM,UAAU,GAAG,KAAsC,CAAC;oBAC1D,IAAI,UAAU,CAAC,WAAW,EAAE,CAAC;wBAC5B,gBAAgB,GAAG,EAAE,GAAG,gBAAgB,EAAE,GAAG,UAAU,CAAC,WAAW,EAAE,CAAC;wBACtE,MAAM,CAAC,aAAa,GAAG,UAAU,CAAC,WAAW,CAAC,EAAE,CAAC;wBACjD,MAAM,CAAC,iBAAiB,GAAG,UAAU,CAAC,WAAW,CAAC,MAAM,CAAC;oBAC1D,CAAC;gBACF,CAAC;gBACD,6BAA6B;qBACxB,IAAI,SAAS,KAAK,eAAe,EAAE,CAAC;oBACxC,MAAM,UAAU,GAAG,KAA0C,CAAC;oBAC9D,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,EAAE,IAAI,CAAC;oBAC7C,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;oBAE1C,IAAI,WAAW,KAAK,MAAM,EAAE,CAAC;wBAC5B,YAAY,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC;wBAC9E,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;wBAClC,cAAc,CAAC,GAAG,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC;wBAC9C,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;oBAChH,CAAC;yBAAM,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;wBACtC,YAAY,GAAG,EAAE,IAAI,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,CAAC;wBACtD,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;wBAClC,cAAc,CAAC,GAAG,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC;wBAC9C,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;oBACpH,CAAC;yBAAM,IAAI,WAAW,KAAK,eAAe,EAAE,CAAC;wBAC5C,MAAM,SAAS,GAAG,UAAU,CAAC,OAAyC,CAAC;wBACvE,YAAY,GAAG;4BACd,IAAI,EAAE,UAAU;4BAChB,UAAU,EAAE,SAAS,CAAC,EAAE,IAAI,EAAE;4BAC9B,IAAI,EAAE,SAAS,CAAC,IAAI,IAAI,EAAE;4BAC1B,SAAS,EAAE,EAAE;4BACb,WAAW,EAAE,EAAE;yBACf,CAAC;wBACF,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;wBAClC,cAAc,CAAC,GAAG,CAAC,WAAW,EAAE,UAAU,EAAE,CAAC,CAAC;wBAC9C,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,YAAY,EAAE,UAAU,EAAE,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;oBACpH,CAAC;gBACF,CAAC;gBACD,6BAA6B;qBACxB,IAAI,SAAS,KAAK,eAAe,EAAE,CAAC;oBACxC,MAAM,UAAU,GAAG,KAAwC,CAAC;oBAC5D,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC;oBAC/B,MAAM,WAAW,GAAG,UAAU,CAAC,KAAK,IAAI,CAAC,CAAC;oBAC1C,MAAM,UAAU,GAAG,cAAc,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,UAAU,EAAE,CAAC;oBAEnE,IAAI,KAAK,EAAE,IAAI,KAAK,MAAM,IAAI,KAAK,CAAC,IAAI,EAAE,CAAC;wBAC1C,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;wBACzC,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;4BACxC,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC;4BAChE,IAAI,OAAO,KAAK,CAAC,CAAC,EAAE,CAAC;gCACnB,KAAK,CAAC,OAAO,CAAC,OAAO,CAAiB,CAAC,OAAO,IAAI,KAAK,CAAC,IAAI,CAAC;4BAC/D,CAAC;wBACF,CAAC;wBACD,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,YAAY;4BAClB,YAAY,EAAE,UAAU;4BACxB,KAAK,EAAE,KAAK,CAAC,IAAI;4BACjB,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;yBAC7C,CAAC,CAAC;oBACJ,CAAC;yBAAM,IAAI,KAAK,EAAE,IAAI,KAAK,iBAAiB,IAAI,KAAK,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC;wBACrE,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;wBACzC,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;4BACxC,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;wBAC1C,CAAC;wBACD,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,gBAAgB;4BACtB,YAAY,EAAE,UAAU;4BACxB,KAAK,EAAE,KAAK,CAAC,OAAO,CAAC,IAAI;4BACzB,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;yBAC7C,CAAC,CAAC;oBACJ,CAAC;yBAAM,IAAI,KAAK,EAAE,IAAI,KAAK,eAAe,EAAE,CAAC;wBAC5C,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAA+D,CAAC;wBACvG,IAAI,KAAK,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;4BACxC,6BAA6B;4BAC7B,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;gCACrB,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;gCAC/C,KAAK,CAAC,WAAW,GAAG,CAAC,KAAK,CAAC,WAAW,IAAI,EAAE,CAAC,GAAG,MAAM,CAAC;gCACvD,IAAI,CAAC;oCACJ,KAAK,CAAC,SAAS,GAAG,EAAE,GAAG,KAAK,CAAC,SAAS,EAAE,GAAG,KAAK,CAAC,SAAS,EAAE,CAAC;gCAC9D,CAAC;gCAAC,MAAM,CAAC,CAAC,0CAA0C,CAAC,CAAC;4BACvD,CAAC;4BACD,MAAM,CAAC,IAAI,CAAC;gCACX,IAAI,EAAE,gBAAgB;gCACtB,YAAY,EAAE,UAAU;gCACxB,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,SAAS,IAAI,EAAE,CAAC;gCAC5C,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;6BAC7C,CAAC,CAAC;wBACJ,CAAC;oBACF,CAAC;gBACF,CAAC;gBACD,4BAA4B;qBACvB,IAAI,SAAS,KAAK,cAAc,EAAE,CAAC;oBACvC,MAAM,SAAS,GAAG,KAA0C,CAAC;oBAC7D,MAAM,WAAW,GAAG,SAAS,CAAC,KAAK,IAAI,CAAC,CAAC;oBACzC,MAAM,UAAU,GAAG,cAAc,CAAC,GAAG,CAAC,WAAW,CAAC,IAAI,UAAU,EAAE,CAAC;oBACnE,MAAM,KAAK,GAAG,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;oBAEzC,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;wBAChC,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,UAAU;4BAChB,YAAY,EAAE,UAAU;4BACxB,OAAO,EAAE,KAAK,CAAC,OAAO;4BACtB,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;yBAC7C,CAAC,CAAC;oBACJ,CAAC;yBAAM,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;wBACvC,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,cAAc;4BACpB,YAAY,EAAE,UAAU;4BACxB,OAAO,EAAE,KAAK,CAAC,YAAY;4BAC3B,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;yBAC7C,CAAC,CAAC;oBACJ,CAAC;yBAAM,IAAI,KAAK,EAAE,IAAI,KAAK,UAAU,EAAE,CAAC;wBACvC,2CAA2C;wBAC3C,MAAM,QAAQ,GAAsB;4BACnC,IAAI,EAAE,UAAU;4BAChB,UAAU,EAAE,KAAK,CAAC,UAAU;4BAC5B,IAAI,EAAE,KAAK,CAAC,IAAI;4BAChB,SAAS,EAAE,KAAK,CAAC,SAAS;yBAC1B,CAAC;wBACF,oCAAoC;wBACpC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,GAAG,QAAQ,CAAC;wBACtC,MAAM,CAAC,IAAI,CAAC;4BACX,IAAI,EAAE,cAAc;4BACpB,YAAY,EAAE,UAAU;4BACxB,QAAQ;4BACR,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE;yBAC7C,CAAC,CAAC;oBACJ,CAAC;oBACD,YAAY,GAAG,IAAI,CAAC;gBACrB,CAAC;gBACD,oCAAoC;qBAC/B,IAAI,SAAS,KAAK,sBAAsB,EAAE,CAAC;oBAC/C,MAAM,aAAa,GAAG,KAAsC,CAAC;oBAC7D,IAAI,aAAa,CAAC,WAAW,EAAE,CAAC;wBAC/B,gBAAgB,GAAG,aAAa,CAAC,WAAW,CAAC;wBAC7C,MAAM,CAAC,aAAa,GAAG,aAAa,CAAC,WAAW,CAAC,EAAE,CAAC;wBACpD,MAAM,CAAC,iBAAiB,GAAG,aAAa,CAAC,WAAW,CAAC,MAAM,CAAC;wBAE5D,sCAAsC;wBACtC,MAAM,gBAAgB,GAAG,aAAa,CAAC,WAAW,CAAC,KAMtC,CAAC;wBAEd,IAAI,gBAAgB,EAAE,CAAC;4BACtB,MAAM,CAAC,KAAK,GAAG;gCACd,KAAK,EAAE,CAAC,gBAAgB,CAAC,kBAAkB,IAAI,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,mBAAmB,IAAI,CAAC,CAAC;gCAC/F,MAAM,EAAE,CAAC,gBAAgB,CAAC,mBAAmB,IAAI,CAAC,CAAC,GAAG,CAAC,gBAAgB,CAAC,sBAAsB,IAAI,CAAC,CAAC;gCACpG,SAAS,EAAE,gBAAgB,CAAC,mBAAmB,IAAI,CAAC;gCACpD,UAAU,EAAE,CAAC;gCACb,WAAW,EAAE,gBAAgB,CAAC,YAAY,IAAI,CAAC;gCAC/C,IAAI,EAAE,EAAE,KAAK,EAAE,CAAC,EAAE,MAAM,EAAE,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE;6BACpE,CAAC;4BACF,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC;wBACpC,CAAC;wBAED,MAAM,CAAC,UAAU,GAAG,oBAAoB,CAAC,aAAa,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;wBAC3E,IAAI,MAAM,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,IAAI,MAAM,CAAC,UAAU,KAAK,MAAM,EAAE,CAAC;4BACrF,MAAM,CAAC,UAAU,GAAG,SAAS,CAAC;wBAC/B,CAAC;oBACF,CAAC;gBACF,CAAC;gBACD,qBAAqB;qBAChB,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;oBAChC,MAAM,UAAU,GAAG,KAAyC,CAAC;oBAC7D,MAAM,IAAI,KAAK,CAAC,UAAU,CAAC,KAAK,EAAE,OAAO,IAAI,yBAAyB,CAAC,CAAC;gBACzE,CAAC;YACF,CAAC;YAED,IAAI,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC;gBAC9B,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACxC,CAAC;YAED,IAAI,MAAM,CAAC,UAAU,KAAK,SAAS,IAAI,MAAM,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;gBACtE,MAAM,IAAI,KAAK,CACd,6BAA6B,MAAM,CAAC,UAAU,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,MAAM,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CACzG,CAAC;YACH,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;YAExG,MAAM,oBAAoB,GAAgD;gBACzE,GAAG,MAAM;gBACT,OAAO,EAAE,gBAAgB;gBACzB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;aACrC,CAAC;YACF,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QAElC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YAChB,MAAM,CAAC,UAAU,GAAG,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO,CAAC;YACnE,MAAM,CAAC,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;YACrF,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,MAAM,CAAC,UAAU,EAAE,OAAO,EAAE,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,CAAC;YAEzG,MAAM,oBAAoB,GAAgD;gBACzE,GAAG,MAAM;gBACT,OAAO,EAAE,gBAAgB;gBACzB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;gBACrB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc;aACrC,CAAC;YACF,MAAM,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC;QAClC,CAAC;IACF,CAAC,CAAC,EAAE,CAAC;IAEL,OAAO,MAAM,CAAC;AACf,CAAC,CAAC"}
|
|
@@ -0,0 +1,10 @@
|
|
|
1
|
+
import type { Interactions } from "@google/genai";
|
|
2
|
+
type CreateModelParams = Parameters<Interactions['create']>[0];
|
|
3
|
+
type Props = {
|
|
4
|
+
apiKey?: string;
|
|
5
|
+
signal?: AbortSignal;
|
|
6
|
+
};
|
|
7
|
+
export type GoogleInteractionsProviderOptions = Omit<CreateModelParams, 'model' | 'agent' | 'input' | 'system_instruction' | 'stream'> & Props;
|
|
8
|
+
export type InteractionStatus = 'in_progress' | 'requires_action' | 'completed' | 'failed' | 'cancelled';
|
|
9
|
+
export {};
|
|
10
|
+
//# sourceMappingURL=types.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/providers/google-interactions/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAGlD,KAAK,iBAAiB,GAAG,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;AAE/D,KAAK,KAAK,GAAG;IACZ,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,MAAM,CAAC,EAAE,WAAW,CAAC;CACrB,CAAA;AAOD,MAAM,MAAM,iCAAiC,GAAG,IAAI,CACnD,iBAAiB,EACjB,OAAO,GAAG,OAAO,GAAG,OAAO,GAAG,oBAAoB,GAAG,QAAQ,CAC7D,GAAG,KAAK,CAAC;AAGV,MAAM,MAAM,iBAAiB,GAAG,aAAa,GAAG,iBAAiB,GAAG,WAAW,GAAG,QAAQ,GAAG,WAAW,CAAC"}
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/providers/google-interactions/types.ts"],"names":[],"mappings":""}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
import { GoogleGenAI, Interactions } from "@google/genai";
|
|
2
|
+
import { Model, AssistantResponse, StopReason, Usage, Context, Tool } from "../../types.js";
|
|
3
|
+
import { GoogleInteractionsProviderOptions } from "./types.js";
|
|
4
|
+
type Interaction = Awaited<ReturnType<Interactions['create']>> extends infer T ? T extends AsyncIterable<any> ? never : T : never;
|
|
5
|
+
type Turn = {
|
|
6
|
+
role?: string;
|
|
7
|
+
content?: string | InteractionContentItem[];
|
|
8
|
+
};
|
|
9
|
+
type InteractionContentItem = {
|
|
10
|
+
type: 'text';
|
|
11
|
+
text?: string;
|
|
12
|
+
} | {
|
|
13
|
+
type: 'image';
|
|
14
|
+
data?: string;
|
|
15
|
+
mime_type?: string;
|
|
16
|
+
uri?: string;
|
|
17
|
+
} | {
|
|
18
|
+
type: 'audio';
|
|
19
|
+
data?: string;
|
|
20
|
+
mime_type?: string;
|
|
21
|
+
} | {
|
|
22
|
+
type: 'document';
|
|
23
|
+
data?: string;
|
|
24
|
+
mime_type?: string;
|
|
25
|
+
} | {
|
|
26
|
+
type: 'video';
|
|
27
|
+
data?: string;
|
|
28
|
+
mime_type?: string;
|
|
29
|
+
} | {
|
|
30
|
+
type: 'function_call';
|
|
31
|
+
id: string;
|
|
32
|
+
name: string;
|
|
33
|
+
arguments: Record<string, unknown>;
|
|
34
|
+
} | {
|
|
35
|
+
type: 'function_result';
|
|
36
|
+
call_id: string;
|
|
37
|
+
name?: string;
|
|
38
|
+
result: unknown;
|
|
39
|
+
is_error?: boolean;
|
|
40
|
+
} | {
|
|
41
|
+
type: 'thought';
|
|
42
|
+
summary?: Array<{
|
|
43
|
+
type: 'text';
|
|
44
|
+
text?: string;
|
|
45
|
+
}>;
|
|
46
|
+
};
|
|
47
|
+
export declare function createClient(model: Model<"google-interactions">, apiKey?: string): GoogleGenAI;
|
|
48
|
+
export declare function getResponseAssistantResponse(interaction: Interaction): AssistantResponse;
|
|
49
|
+
export declare function getResponseUsage(interaction: Interaction, model: Model<'google-interactions'>): Usage;
|
|
50
|
+
export declare function mapInteractionStatus(status: Interaction['status']): StopReason;
|
|
51
|
+
export declare function buildParams(model: Model<"google-interactions">, context: Context, options: GoogleInteractionsProviderOptions): Parameters<Interactions['create']>[0];
|
|
52
|
+
export declare function buildInteractionInput(model: Model<'google-interactions'>, context: Context): Turn[] | string;
|
|
53
|
+
export declare function convertTools(tools: readonly Tool[]): Array<{
|
|
54
|
+
type: 'function';
|
|
55
|
+
name: string;
|
|
56
|
+
description: string;
|
|
57
|
+
parameters: unknown;
|
|
58
|
+
}>;
|
|
59
|
+
export declare function getMockInteraction(): Interaction;
|
|
60
|
+
export {};
|
|
61
|
+
//# sourceMappingURL=utils.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/providers/google-interactions/utils.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,YAAY,EAAE,MAAM,eAAe,CAAC;AAC1D,OAAO,EACN,KAAK,EACL,iBAAiB,EACjB,UAAU,EACV,KAAK,EACL,OAAO,EAEP,IAAI,EAEJ,MAAM,gBAAgB,CAAC;AAExB,OAAO,EAAE,iCAAiC,EAAE,MAAM,YAAY,CAAC;AAI/D,KAAK,WAAW,GAAG,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,MAAM,CAAC,GAC3E,CAAC,SAAS,aAAa,CAAC,GAAG,CAAC,GAAG,KAAK,GAAG,CAAC,GACxC,KAAK,CAAC;AAGT,KAAK,IAAI,GAAG;IACX,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,OAAO,CAAC,EAAE,MAAM,GAAG,sBAAsB,EAAE,CAAC;CAC5C,CAAC;AAEF,KAAK,sBAAsB,GACxB;IAAE,IAAI,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAA;CAAE,GAC/B;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAC;IAAC,GAAG,CAAC,EAAE,MAAM,CAAA;CAAE,GAClE;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GACpD;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GACvD;IAAE,IAAI,EAAE,OAAO,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,SAAS,CAAC,EAAE,MAAM,CAAA;CAAE,GACpD;IAAE,IAAI,EAAE,eAAe,CAAC;IAAC,EAAE,EAAE,MAAM,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,SAAS,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAA;CAAE,GACvF;IAAE,IAAI,EAAE,iBAAiB,CAAC;IAAC,OAAO,EAAE,MAAM,CAAC;IAAC,IAAI,CAAC,EAAE,MAAM,CAAC;IAAC,MAAM,EAAE,OAAO,CAAC;IAAC,QAAQ,CAAC,EAAE,OAAO,CAAA;CAAE,GAChG;IAAE,IAAI,EAAE,SAAS,CAAC;IAAC,OAAO,CAAC,EAAE,KAAK,CAAC;QAAE,IAAI,EAAE,MAAM,CAAC;QAAC,IAAI,CAAC,EAAE,MAAM,CAAA;KAAE,CAAC,CAAA;CAAE,CAAC;AAEzE,wBAAgB,YAAY,CAAC,KAAK,EAAE,KAAK,CAAC,qBAAqB,CAAC,EAAE,MAAM,CAAC,EAAE,MAAM,GAAG,WAAW,CAa9F;AAED,wBAAgB,4BAA4B,CAAC,WAAW,EAAE,WAAW,GAAG,iBAAiB,CA8DxF;AAED,wBAAgB,gBAAgB,CAAC,WAAW,EAAE,WAAW,EAAE,KAAK,EAAE,KAAK,CAAC,qBAAqB,CAAC,GAAG,KAAK,CAoBrG;AAED,wBAAgB,oBAAoB,CAAC,MAAM,EAAE,WAAW,CAAC,QAAQ,CAAC,GAAG,UAAU,CAe9E;AAED,wBAAgB,WAAW,CAC1B,KAAK,EAAE,KAAK,CAAC,qBAAqB,CAAC,EACnC,OAAO,EAAE,OAAO,EAChB,OAAO,EAAE,iCAAiC,GACxC,UAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC,CA6BvC;AAED,wBAAgB,qBAAqB,CACpC,KAAK,EAAE,KAAK,CAAC,qBAAqB,CAAC,EACnC,OAAO,EAAE,OAAO,GACd,IAAI,EAAE,GAAG,MAAM,CA6GjB;AAED,wBAAgB,YAAY,CAAC,KAAK,EAAE,SAAS,IAAI,EAAE,GAAG,KAAK,CAAC;IAAE,IAAI,EAAE,UAAU,CAAC;IAAC,IAAI,EAAE,MAAM,CAAC;IAAC,WAAW,EAAE,MAAM,CAAC;IAAC,UAAU,EAAE,OAAO,CAAA;CAAE,CAAC,CAOxI;AAED,wBAAgB,kBAAkB,IAAI,WAAW,CAWhD"}
|