@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.
Files changed (123) hide show
  1. package/dist/debug-conversation.d.ts +2 -0
  2. package/dist/debug-conversation.d.ts.map +1 -0
  3. package/dist/debug-conversation.js +98 -0
  4. package/dist/debug-conversation.js.map +1 -0
  5. package/dist/debug-minimal-test.d.ts +2 -0
  6. package/dist/debug-minimal-test.d.ts.map +1 -0
  7. package/dist/debug-minimal-test.js +47 -0
  8. package/dist/debug-minimal-test.js.map +1 -0
  9. package/dist/debug-sdk-tool.d.ts +2 -0
  10. package/dist/debug-sdk-tool.d.ts.map +1 -0
  11. package/dist/debug-sdk-tool.js +99 -0
  12. package/dist/debug-sdk-tool.js.map +1 -0
  13. package/dist/debug-sdk.d.ts +2 -0
  14. package/dist/debug-sdk.d.ts.map +1 -0
  15. package/dist/debug-sdk.js +59 -0
  16. package/dist/debug-sdk.js.map +1 -0
  17. package/dist/debug-session-flow.d.ts +2 -0
  18. package/dist/debug-session-flow.d.ts.map +1 -0
  19. package/dist/debug-session-flow.js +134 -0
  20. package/dist/debug-session-flow.js.map +1 -0
  21. package/dist/debug-session-hack.d.ts +2 -0
  22. package/dist/debug-session-hack.d.ts.map +1 -0
  23. package/dist/debug-session-hack.js +143 -0
  24. package/dist/debug-session-hack.js.map +1 -0
  25. package/dist/llm.d.ts.map +1 -1
  26. package/dist/llm.js +10 -1
  27. package/dist/llm.js.map +1 -1
  28. package/dist/models.generated.d.ts +19 -0
  29. package/dist/models.generated.d.ts.map +1 -1
  30. package/dist/models.generated.js +19 -0
  31. package/dist/models.generated.js.map +1 -1
  32. package/dist/providers/anthropic-agents-sdk/complete.d.ts +3 -0
  33. package/dist/providers/anthropic-agents-sdk/complete.d.ts.map +1 -0
  34. package/dist/providers/anthropic-agents-sdk/complete.js +246 -0
  35. package/dist/providers/anthropic-agents-sdk/complete.js.map +1 -0
  36. package/dist/providers/anthropic-agents-sdk/index.d.ts +4 -0
  37. package/dist/providers/anthropic-agents-sdk/index.d.ts.map +1 -0
  38. package/dist/providers/anthropic-agents-sdk/index.js +4 -0
  39. package/dist/providers/anthropic-agents-sdk/index.js.map +1 -0
  40. package/dist/providers/anthropic-agents-sdk/session.d.ts +37 -0
  41. package/dist/providers/anthropic-agents-sdk/session.d.ts.map +1 -0
  42. package/dist/providers/anthropic-agents-sdk/session.js +188 -0
  43. package/dist/providers/anthropic-agents-sdk/session.js.map +1 -0
  44. package/dist/providers/anthropic-agents-sdk/stream.d.ts +3 -0
  45. package/dist/providers/anthropic-agents-sdk/stream.d.ts.map +1 -0
  46. package/dist/providers/anthropic-agents-sdk/stream.js +402 -0
  47. package/dist/providers/anthropic-agents-sdk/stream.js.map +1 -0
  48. package/dist/providers/anthropic-agents-sdk/types.d.ts +22 -0
  49. package/dist/providers/anthropic-agents-sdk/types.d.ts.map +1 -0
  50. package/dist/providers/anthropic-agents-sdk/types.js +3 -0
  51. package/dist/providers/anthropic-agents-sdk/types.js.map +1 -0
  52. package/dist/providers/anthropic-agents-sdk/utils.d.ts +39 -0
  53. package/dist/providers/anthropic-agents-sdk/utils.d.ts.map +1 -0
  54. package/dist/providers/anthropic-agents-sdk/utils.js +251 -0
  55. package/dist/providers/anthropic-agents-sdk/utils.js.map +1 -0
  56. package/dist/providers/google/utils.d.ts.map +1 -1
  57. package/dist/providers/google/utils.js +2 -0
  58. package/dist/providers/google/utils.js.map +1 -1
  59. package/dist/providers/google-interactions/complete.d.ts +3 -0
  60. package/dist/providers/google-interactions/complete.d.ts.map +1 -0
  61. package/dist/providers/google-interactions/complete.js +63 -0
  62. package/dist/providers/google-interactions/complete.js.map +1 -0
  63. package/dist/providers/google-interactions/index.d.ts +5 -0
  64. package/dist/providers/google-interactions/index.d.ts.map +1 -0
  65. package/dist/providers/google-interactions/index.js +4 -0
  66. package/dist/providers/google-interactions/index.js.map +1 -0
  67. package/dist/providers/google-interactions/poll.d.ts +59 -0
  68. package/dist/providers/google-interactions/poll.d.ts.map +1 -0
  69. package/dist/providers/google-interactions/poll.js +161 -0
  70. package/dist/providers/google-interactions/poll.js.map +1 -0
  71. package/dist/providers/google-interactions/stream.d.ts +3 -0
  72. package/dist/providers/google-interactions/stream.d.ts.map +1 -0
  73. package/dist/providers/google-interactions/stream.js +242 -0
  74. package/dist/providers/google-interactions/stream.js.map +1 -0
  75. package/dist/providers/google-interactions/types.d.ts +10 -0
  76. package/dist/providers/google-interactions/types.d.ts.map +1 -0
  77. package/dist/providers/google-interactions/types.js +2 -0
  78. package/dist/providers/google-interactions/types.js.map +1 -0
  79. package/dist/providers/google-interactions/utils.d.ts +61 -0
  80. package/dist/providers/google-interactions/utils.d.ts.map +1 -0
  81. package/dist/providers/google-interactions/utils.js +252 -0
  82. package/dist/providers/google-interactions/utils.js.map +1 -0
  83. package/dist/providers/zai/complete.d.ts +3 -0
  84. package/dist/providers/zai/complete.d.ts.map +1 -0
  85. package/dist/providers/zai/complete.js +58 -0
  86. package/dist/providers/zai/complete.js.map +1 -0
  87. package/dist/providers/zai/index.d.ts +5 -0
  88. package/dist/providers/zai/index.d.ts.map +1 -0
  89. package/dist/providers/zai/index.js +4 -0
  90. package/dist/providers/zai/index.js.map +1 -0
  91. package/dist/providers/zai/stream.d.ts +3 -0
  92. package/dist/providers/zai/stream.d.ts.map +1 -0
  93. package/dist/providers/zai/stream.js +321 -0
  94. package/dist/providers/zai/stream.js.map +1 -0
  95. package/dist/providers/zai/types.d.ts +13 -0
  96. package/dist/providers/zai/types.d.ts.map +1 -0
  97. package/dist/providers/zai/types.js +2 -0
  98. package/dist/providers/zai/types.js.map +1 -0
  99. package/dist/providers/zai/utils.d.ts +19 -0
  100. package/dist/providers/zai/utils.d.ts.map +1 -0
  101. package/dist/providers/zai/utils.js +253 -0
  102. package/dist/providers/zai/utils.js.map +1 -0
  103. package/dist/test-sdk-direct.d.ts +2 -0
  104. package/dist/test-sdk-direct.d.ts.map +1 -0
  105. package/dist/test-sdk-direct.js +41 -0
  106. package/dist/test-sdk-direct.js.map +1 -0
  107. package/dist/test.d.ts +2 -0
  108. package/dist/test.d.ts.map +1 -0
  109. package/dist/test.js +213 -0
  110. package/dist/test.js.map +1 -0
  111. package/dist/types.d.ts +4 -1
  112. package/dist/types.d.ts.map +1 -1
  113. package/dist/types.js +1 -1
  114. package/dist/types.js.map +1 -1
  115. package/dist/utils/oauth/google-antigravity.d.ts +25 -0
  116. package/dist/utils/oauth/google-antigravity.d.ts.map +1 -0
  117. package/dist/utils/oauth/google-antigravity.js +337 -0
  118. package/dist/utils/oauth/google-antigravity.js.map +1 -0
  119. package/dist/utils/oauth/pkce.d.ts +13 -0
  120. package/dist/utils/oauth/pkce.d.ts.map +1 -0
  121. package/dist/utils/oauth/pkce.js +31 -0
  122. package/dist/utils/oauth/pkce.js.map +1 -0
  123. 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,3 @@
1
+ import { StreamFunction } from "../../types.js";
2
+ export declare const streamGoogleInteractions: StreamFunction<'google-interactions'>;
3
+ //# sourceMappingURL=stream.d.ts.map
@@ -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,2 @@
1
+ export {};
2
+ //# sourceMappingURL=types.js.map
@@ -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"}