@workglow/ai-provider 0.0.125 → 0.1.0

package/dist/provider-gemini/runtime.js

@@ -48,6 +48,24 @@ function getModelName(model) {
   }
   return name;
 }
+// src/common/modelSearchQuery.ts
+function normalizedModelSearchQuery(query) {
+  const t = query?.trim();
+  return t ? t.toLowerCase() : undefined;
+}
+function filterLabeledModelsByQuery(models, query) {
+  const q = normalizedModelSearchQuery(query);
+  if (!q)
+    return [...models];
+  return models.filter((m) => m.value.toLowerCase().includes(q) || m.label.toLowerCase().includes(q));
+}
+function filterModelSearchResultsByQuery(results, query) {
+  const q = normalizedModelSearchQuery(query);
+  if (!q)
+    return results;
+  return results.filter((m) => m.id.toLowerCase().includes(q) || m.label.toLowerCase().includes(q) || m.description.toLowerCase().includes(q));
+}
+
 // src/provider-gemini/common/Gemini_Constants.ts
 var GOOGLE_GEMINI = "GOOGLE_GEMINI";
 
@@ -61,8 +79,9 @@ var GEMINI_MODELS = [
   { label: "gemini-1.5-pro", value: "gemini-1.5-pro" },
   { label: "gemini-1.5-flash", value: "gemini-1.5-flash" }
 ];
-var Gemini_ModelSearch = async () => {
-  const results = GEMINI_MODELS.map((m) => ({
+var Gemini_ModelSearch = async (input) => {
+  const models = filterLabeledModelsByQuery(GEMINI_MODELS, input.query);
+  const results = models.map((m) => ({
     id: m.value,
     label: m.label,
     description: "",
@@ -96,18 +115,7 @@ function sanitizeSchemaForGemini(schema) {
 }
 
 // src/provider-gemini/common/Gemini_CountTokens.ts
-import { getLogger } from "@workglow/util/worker";
 var Gemini_CountTokens = async (input, model, onProgress, signal) => {
-  if (Array.isArray(input.text)) {
-    getLogger().warn("Gemini_CountTokens: array input received; processing sequentially (no native batch support)");
-    const texts = input.text;
-    const counts = [];
-    for (const item of texts) {
-      const r = await Gemini_CountTokens({ ...input, text: item }, model, onProgress, signal);
-      counts.push(r.count);
-    }
-    return { count: counts };
-  }
   const GoogleGenerativeAI = await loadGeminiSDK();
   const genAI = new GoogleGenerativeAI(getApiKey(model));
   const genModel = genAI.getGenerativeModel({ model: getModelName(model) });
@@ -115,9 +123,6 @@ var Gemini_CountTokens = async (input, model, onProgress, signal) => {
   return { count: result.totalTokens };
 };
 var Gemini_CountTokens_Reactive = async (input, _output, _model) => {
-  if (Array.isArray(input.text)) {
-    return { count: input.text.map((t) => Math.ceil(t.length / 4)) };
-  }
   return { count: Math.ceil(input.text.length / 4) };
 };
 
@@ -195,9 +200,9 @@ var Gemini_StructuredGeneration_Stream = async function* (input, model, signal,
 };
 
 // src/provider-gemini/common/Gemini_TextEmbedding.ts
-import { getLogger as getLogger2 } from "@workglow/util/worker";
+import { getLogger } from "@workglow/util/worker";
 var Gemini_TextEmbedding = async (input, model, update_progress, signal) => {
-  const logger = getLogger2();
+  const logger = getLogger();
   const timerLabel = `gemini:TextEmbedding:${model?.provider_config?.model_name}`;
   logger.time(timerLabel, { model: model?.provider_config?.model_name });
   update_progress(0, "Starting Gemini text embedding");
@@ -230,19 +235,9 @@ var Gemini_TextEmbedding = async (input, model, update_progress, signal) => {
 };
 
 // src/provider-gemini/common/Gemini_TextGeneration.ts
-import { getLogger as getLogger3 } from "@workglow/util/worker";
+import { getLogger as getLogger2 } from "@workglow/util/worker";
 var Gemini_TextGeneration = async (input, model, update_progress, signal) => {
-  if (Array.isArray(input.prompt)) {
-    getLogger3().warn("Gemini_TextGeneration: array input received; processing sequentially (no native batch support)");
-    const prompts = input.prompt;
-    const results = [];
-    for (const item of prompts) {
-      const r = await Gemini_TextGeneration({ ...input, prompt: item }, model, update_progress, signal);
-      results.push(r.text);
-    }
-    return { text: results };
-  }
-  const logger = getLogger3();
+  const logger = getLogger2();
   const timerLabel = `gemini:TextGeneration:${model?.provider_config?.model_name}`;
   logger.time(timerLabel, { model: model?.provider_config?.model_name });
   update_progress(0, "Starting Gemini text generation");
@@ -286,18 +281,7 @@ var Gemini_TextGeneration_Stream = async function* (input, model, signal) {
 };
 
 // src/provider-gemini/common/Gemini_TextRewriter.ts
-import { getLogger as getLogger4 } from "@workglow/util/worker";
 var Gemini_TextRewriter = async (input, model, update_progress, signal) => {
-  if (Array.isArray(input.text)) {
-    getLogger4().warn("Gemini_TextRewriter: array input received; processing sequentially (no native batch support)");
-    const texts = input.text;
-    const results = [];
-    for (const item of texts) {
-      const r = await Gemini_TextRewriter({ ...input, text: item }, model, update_progress, signal);
-      results.push(r.text);
-    }
-    return { text: results };
-  }
   update_progress(0, "Starting Gemini text rewriting");
   const GoogleGenerativeAI = await loadGeminiSDK();
   const genAI = new GoogleGenerativeAI(getApiKey(model));
@@ -330,18 +314,7 @@ var Gemini_TextRewriter_Stream = async function* (input, model, signal) {
 };
 
 // src/provider-gemini/common/Gemini_TextSummary.ts
-import { getLogger as getLogger5 } from "@workglow/util/worker";
 var Gemini_TextSummary = async (input, model, update_progress, signal) => {
-  if (Array.isArray(input.text)) {
-    getLogger5().warn("Gemini_TextSummary: array input received; processing sequentially (no native batch support)");
-    const texts = input.text;
-    const results = [];
-    for (const item of texts) {
-      const r = await Gemini_TextSummary({ ...input, text: item }, model, update_progress, signal);
-      results.push(r.text);
-    }
-    return { text: results };
-  }
   update_progress(0, "Starting Gemini text summarization");
   const GoogleGenerativeAI = await loadGeminiSDK();
   const genAI = new GoogleGenerativeAI(getApiKey(model));
@@ -373,201 +346,6 @@ var Gemini_TextSummary_Stream = async function* (input, model, signal) {
   yield { type: "finish", data: {} };
 };
 
-// src/provider-gemini/common/Gemini_ToolCalling.ts
-import { buildToolDescription, filterValidToolCalls } from "@workglow/ai/worker";
-import { getLogger as getLogger6 } from "@workglow/util/worker";
-function buildGeminiContents(input) {
-  const inputMessages = input.messages;
-  if (!inputMessages || inputMessages.length === 0) {
-    return [{ role: "user", parts: [{ text: input.prompt }] }];
-  }
-  const toolUseNames = new Map;
-  for (const msg of inputMessages) {
-    if (msg.role === "assistant" && Array.isArray(msg.content)) {
-      for (const block of msg.content) {
-        if (block.type === "tool_use") {
-          toolUseNames.set(block.id, block.name);
-        }
-      }
-    }
-  }
-  const contents = [];
-  for (const msg of inputMessages) {
-    if (msg.role === "user") {
-      if (typeof msg.content === "string") {
-        contents.push({ role: "user", parts: [{ text: msg.content }] });
-      } else if (Array.isArray(msg.content)) {
-        const parts = [];
-        for (const block of msg.content) {
-          if (block.type === "text") {
-            parts.push({ text: block.text });
-          } else if (block.type === "image" || block.type === "audio") {
-            parts.push({ inlineData: { mimeType: block.mimeType, data: block.data } });
-          }
-        }
-        contents.push({ role: "user", parts });
-      } else {
-        contents.push({ role: "user", parts: [{ text: msg.content }] });
-      }
-    } else if (msg.role === "assistant" && Array.isArray(msg.content)) {
-      const parts = [];
-      for (const block of msg.content) {
-        if (block.type === "text" && block.text) {
-          parts.push({ text: block.text });
-        } else if (block.type === "tool_use") {
-          parts.push({ functionCall: { name: block.name, args: block.input } });
-        }
-      }
-      if (parts.length > 0) {
-        contents.push({ role: "model", parts });
-      }
-    } else if (msg.role === "tool" && Array.isArray(msg.content)) {
-      const parts = msg.content.map((block) => {
-        const name = toolUseNames.get(block.tool_use_id) ?? "unknown";
-        let response;
-        if (typeof block.content === "string") {
-          try {
-            response = JSON.parse(block.content);
-          } catch {
-            response = { result: block.content };
-          }
-        } else if (Array.isArray(block.content)) {
-          const textParts = block.content.filter((b) => b.type === "text").map((b) => b.text);
-          try {
-            response = JSON.parse(textParts.join(""));
-          } catch {
-            response = { result: textParts.join("") };
-          }
-        } else {
-          response = {};
-        }
-        return { functionResponse: { name, response } };
-      });
-      contents.push({ role: "user", parts });
-    }
-  }
-  return contents;
-}
-function mapGeminiToolConfig(toolChoice) {
-  if (!toolChoice || toolChoice === "auto") {
-    return { functionCallingConfig: { mode: "AUTO" } };
-  }
-  if (toolChoice === "none") {
-    return { functionCallingConfig: { mode: "NONE" } };
-  }
-  if (toolChoice === "required") {
-    return { functionCallingConfig: { mode: "ANY" } };
-  }
-  return {
-    functionCallingConfig: {
-      mode: "ANY",
-      allowedFunctionNames: [toolChoice]
-    }
-  };
-}
-var Gemini_ToolCalling = async (input, model, update_progress, signal) => {
-  if (Array.isArray(input.prompt)) {
-    getLogger6().warn("Gemini_ToolCalling: array input received; processing sequentially (no native batch support)");
-    const prompts = input.prompt;
-    const texts = [];
-    const toolCallsList = [];
-    for (const item of prompts) {
-      const r = await Gemini_ToolCalling({ ...input, prompt: item }, model, update_progress, signal);
-      texts.push(r.text);
-      toolCallsList.push(r.toolCalls);
-    }
-    return { text: texts, toolCalls: toolCallsList };
-  }
-  update_progress(0, "Starting Gemini tool calling");
-  const GoogleGenerativeAI = await loadGeminiSDK();
-  const genAI = new GoogleGenerativeAI(getApiKey(model));
-  const functionDeclarations = input.tools.map((t) => ({
-    name: t.name,
-    description: buildToolDescription(t),
-    parameters: sanitizeSchemaForGemini(t.inputSchema)
-  }));
-  const toolConfig = mapGeminiToolConfig(input.toolChoice);
-  const genModel = genAI.getGenerativeModel({
-    model: getModelName(model),
-    tools: [{ functionDeclarations }],
-    toolConfig,
-    systemInstruction: input.systemPrompt || undefined,
-    generationConfig: {
-      maxOutputTokens: input.maxTokens,
-      temperature: input.temperature
-    }
-  });
-  const contents = buildGeminiContents(input);
-  const result = await genModel.generateContent({ contents });
-  const parts = result.response.candidates?.[0]?.content?.parts ?? [];
-  const textParts = [];
-  const toolCalls = [];
-  let callIndex = 0;
-  for (const part of parts) {
-    if ("text" in part && part.text) {
-      textParts.push(part.text);
-    }
-    if ("functionCall" in part && part.functionCall) {
-      const id = `call_${callIndex++}`;
-      toolCalls.push({
-        id,
-        name: part.functionCall.name,
-        input: part.functionCall.args ?? {}
-      });
-    }
-  }
-  update_progress(100, "Completed Gemini tool calling");
-  return { text: textParts.join(""), toolCalls: filterValidToolCalls(toolCalls, input.tools) };
-};
-var Gemini_ToolCalling_Stream = async function* (input, model, signal) {
-  const GoogleGenerativeAI = await loadGeminiSDK();
-  const genAI = new GoogleGenerativeAI(getApiKey(model));
-  const functionDeclarations = input.tools.map((t) => ({
-    name: t.name,
-    description: buildToolDescription(t),
-    parameters: sanitizeSchemaForGemini(t.inputSchema)
-  }));
-  const toolConfig = mapGeminiToolConfig(input.toolChoice);
-  const genModel = genAI.getGenerativeModel({
-    model: getModelName(model),
-    tools: [{ functionDeclarations }],
-    toolConfig,
-    systemInstruction: input.systemPrompt || undefined,
-    generationConfig: {
-      maxOutputTokens: input.maxTokens,
-      temperature: input.temperature
-    }
-  });
-  const contents = buildGeminiContents(input);
-  const result = await genModel.generateContentStream({ contents }, { signal });
-  let accumulatedText = "";
-  const toolCalls = [];
-  let callIndex = 0;
-  for await (const chunk of result.stream) {
-    const parts = chunk.candidates?.[0]?.content?.parts ?? [];
-    for (const part of parts) {
-      if ("text" in part && part.text) {
-        accumulatedText += part.text;
-        yield { type: "text-delta", port: "text", textDelta: part.text };
-      }
-      if ("functionCall" in part && part.functionCall) {
-        const id = `call_${callIndex++}`;
-        toolCalls.push({
-          id,
-          name: part.functionCall.name,
-          input: part.functionCall.args ?? {}
-        });
-        yield { type: "object-delta", port: "toolCalls", objectDelta: [...toolCalls] };
-      }
-    }
-  }
-  const validToolCalls = filterValidToolCalls(toolCalls, input.tools);
-  yield {
-    type: "finish",
-    data: { text: accumulatedText, toolCalls: validToolCalls }
-  };
-};
-
 // src/provider-gemini/common/Gemini_JobRunFns.ts
 var GEMINI_TASKS = {
   CountTokensTask: Gemini_CountTokens,
@@ -577,15 +355,13 @@ var GEMINI_TASKS = {
   TextRewriterTask: Gemini_TextRewriter,
   TextSummaryTask: Gemini_TextSummary,
   StructuredGenerationTask: Gemini_StructuredGeneration,
-  ToolCallingTask: Gemini_ToolCalling,
   ModelSearchTask: Gemini_ModelSearch
 };
 var GEMINI_STREAM_TASKS = {
   TextGenerationTask: Gemini_TextGeneration_Stream,
   TextRewriterTask: Gemini_TextRewriter_Stream,
   TextSummaryTask: Gemini_TextSummary_Stream,
-  StructuredGenerationTask: Gemini_StructuredGeneration_Stream,
-  ToolCallingTask: Gemini_ToolCalling_Stream
+  StructuredGenerationTask: Gemini_StructuredGeneration_Stream
 };
 var GEMINI_REACTIVE_TASKS = {
   CountTokensTask: Gemini_CountTokens_Reactive
@@ -593,9 +369,9 @@ var GEMINI_REACTIVE_TASKS = {
 
 // src/provider-gemini/GoogleGeminiQueuedProvider.ts
 import {
-  QueuedAiProvider
+  AiProvider
 } from "@workglow/ai";
-class GoogleGeminiQueuedProvider extends QueuedAiProvider {
+class GoogleGeminiQueuedProvider extends AiProvider {
   name = GOOGLE_GEMINI;
   displayName = "Google Gemini";
   isLocal = false;
@@ -608,7 +384,6 @@ class GoogleGeminiQueuedProvider extends QueuedAiProvider {
     "TextRewriterTask",
     "TextSummaryTask",
     "StructuredGenerationTask",
-    "ToolCallingTask",
     "ModelSearchTask"
   ];
   constructor(tasks, streamTasks, reactiveTasks) {
@@ -621,13 +396,13 @@ async function registerGeminiInline(options) {
   await new GoogleGeminiQueuedProvider(GEMINI_TASKS, GEMINI_STREAM_TASKS, GEMINI_REACTIVE_TASKS).register(options ?? {});
 }
 // src/provider-gemini/registerGeminiWorker.ts
-import { getLogger as getLogger7, globalServiceRegistry, WORKER_SERVER } from "@workglow/util/worker";
+import { getLogger as getLogger3, globalServiceRegistry, WORKER_SERVER } from "@workglow/util/worker";
 
 // src/provider-gemini/GoogleGeminiProvider.ts
 import {
-  AiProvider
+  AiProvider as AiProvider2
 } from "@workglow/ai/worker";
-class GoogleGeminiProvider extends AiProvider {
+class GoogleGeminiProvider extends AiProvider2 {
   name = GOOGLE_GEMINI;
   displayName = "Google Gemini";
   isLocal = false;
@@ -640,7 +415,6 @@ class GoogleGeminiProvider extends AiProvider {
     "TextRewriterTask",
     "TextSummaryTask",
     "StructuredGenerationTask",
-    "ToolCallingTask",
     "ModelSearchTask"
   ];
   constructor(tasks, streamTasks, reactiveTasks) {
@@ -653,7 +427,7 @@ async function registerGeminiWorker() {
   const workerServer = globalServiceRegistry.get(WORKER_SERVER);
   new GoogleGeminiProvider(GEMINI_TASKS, GEMINI_STREAM_TASKS, GEMINI_REACTIVE_TASKS).registerOnWorkerServer(workerServer);
   workerServer.sendReady();
-  getLogger7().info("Google Gemini worker job run functions registered");
+  getLogger3().info("Google Gemini worker job run functions registered");
 }
 export {
   registerGeminiWorker,
@@ -663,4 +437,4 @@ export {
   getApiKey
 };
 
-//# debugId=5CC2029E1D06664464756E2164756E21
+//# debugId=265C7048DA37E5B964756E2164756E21