drizzle-cube 0.4.20 → 0.4.22

package/dist/adapters/google-DficVAsJ.js

@@ -0,0 +1,146 @@
+function l(u) {
+  const t = {};
+  for (const [e, s] of Object.entries(u))
+    if (e === "type" && typeof s == "string")
+      t[e] = s.toUpperCase();
+    else if (e === "properties" && typeof s == "object" && s !== null) {
+      const n = {};
+      for (const [a, o] of Object.entries(s))
+        typeof o == "object" && o !== null ? n[a] = l(o) : n[a] = o;
+      t[e] = n;
+    } else e === "items" && typeof s == "object" && s !== null ? t[e] = l(s) : t[e] = s;
+  return t;
+}
+class p {
+  apiKey;
+  sdk;
+  initialized = !1;
+  constructor(t) {
+    this.apiKey = t;
+  }
+  async ensureSDK() {
+    if (!this.initialized) {
+      try {
+        this.sdk = await import(
+          /* webpackIgnore: true */
+          "./index-C45_meK_.js"
+        );
+      } catch {
+        throw new Error("@google/generative-ai is required for the Google provider. Install it with: npm install @google/generative-ai");
+      }
+      this.initialized = !0;
+    }
+  }
+  async createStream(t) {
+    await this.ensureSDK();
+    const { GoogleGenerativeAI: e } = this.sdk, n = new e(this.apiKey).getGenerativeModel({
+      model: t.model,
+      systemInstruction: t.system,
+      generationConfig: {
+        maxOutputTokens: t.maxTokens
+      }
+    }), { messages: a } = this.formatMessages(t.messages, t.system), o = this.formatTools(t.tools);
+    return (await n.generateContentStream({
+      contents: a,
+      tools: o.length > 0 ? [{ functionDeclarations: o }] : void 0
+    })).stream;
+  }
+  async *parseStreamEvents(t) {
+    let e = "stop", s = !1;
+    for await (const n of t) {
+      const a = n;
+      a.usageMetadata && (yield {
+        type: "message_meta",
+        inputTokens: a.usageMetadata.promptTokenCount,
+        outputTokens: a.usageMetadata.candidatesTokenCount,
+        stopReason: ""
+      });
+      const o = a.candidates?.[0];
+      if (!o) continue;
+      o.finishReason && (o.finishReason === "STOP" ? e = "stop" : o.finishReason === "MAX_TOKENS" ? e = "max_tokens" : e = o.finishReason);
+      const i = o.content?.parts;
+      if (i) {
+        for (const r of i)
+          if (r.text && !r.thought && (yield { type: "text_delta", text: r.text }), r.functionCall) {
+            s = !0;
+            const f = `gemini-tc-${Date.now()}-${Math.random().toString(36).slice(2, 7)}`, c = r.thoughtSignature ? { thoughtSignature: r.thoughtSignature } : void 0;
+            yield { type: "tool_use_start", id: f, name: r.functionCall.name, ...c ? { metadata: c } : {} }, yield { type: "tool_input_delta", json: JSON.stringify(r.functionCall.args || {}) }, yield { type: "tool_use_end" };
+          }
+      }
+    }
+    yield {
+      type: "message_meta",
+      stopReason: s ? "tool_use" : e
+    };
+  }
+  formatTools(t) {
+    return t.map((e) => ({
+      name: e.name,
+      description: e.description,
+      parameters: l(e.parameters)
+    }));
+  }
+  formatMessages(t, e) {
+    const s = [];
+    for (const n of t)
+      if (n.role === "tool_result") {
+        const a = n._toolResults;
+        if (a && a.length > 0)
+          s.push({
+            role: "user",
+            parts: a.map((o) => ({
+              functionResponse: {
+                name: o.toolName || o.toolUseId,
+                response: { content: o.content, isError: o.isError || !1 }
+              }
+            }))
+          });
+        else {
+          const o = typeof n.content == "string" ? n.content : JSON.stringify(n.content);
+          o && s.push({ role: "user", parts: [{ text: o }] });
+        }
+      } else if (n.role === "user") {
+        const a = typeof n.content == "string" ? n.content : JSON.stringify(n.content);
+        s.push({ role: "user", parts: [{ text: a }] });
+      } else if (n.role === "assistant")
+        if (typeof n.content == "string")
+          s.push({ role: "model", parts: [{ text: n.content }] });
+        else {
+          const a = n.content, o = [];
+          for (const i of a)
+            if (i.type === "text" && i.text)
+              o.push({ text: i.text });
+            else if (i.type === "tool_use") {
+              const r = {
+                functionCall: {
+                  name: i.name,
+                  args: i.input || {}
+                }
+              };
+              i.metadata?.thoughtSignature && (r.thoughtSignature = i.metadata.thoughtSignature), o.push(r);
+            }
+          o.length > 0 && s.push({ role: "model", parts: o });
+        }
+    return { messages: s };
+  }
+  formatToolResults(t) {
+    return {
+      role: "tool_result",
+      content: t.map((e) => `${e.toolName || e.toolUseId}: ${e.content}`).join(`
+`),
+      _toolResults: t
+    };
+  }
+  shouldContinue(t) {
+    return t === "tool_use";
+  }
+  formatError(t) {
+    if (!t || !(t instanceof Error))
+      return "Something went wrong. Please try again.";
+    const e = t.message || "", s = t;
+    return s.status === 429 ? "Too many requests. Please wait a moment and try again." : s.status === 403 || s.status === 401 ? "Authentication failed. Please check your API key configuration." : s.status === 503 || s.status === 500 ? "The AI service is temporarily unavailable. Please try again in a moment." : e.includes("SAFETY") ? "The request was blocked by safety filters. Please rephrase your request." : e.includes("RECITATION") ? "The response was blocked due to recitation concerns. Please try a different query." : e.startsWith("{") || e.startsWith("[") ? "The AI service encountered an error. Please try again." : e;
+  }
+}
+export {
+  p as GoogleProvider
+};