@effect-uai/google 0.9.0 → 0.11.0

package/src/geminiTools.test.ts

@@ -0,0 +1,88 @@
+import { Effect, Option, Result, Schema } from "effect"
+import * as Tool from "@effect-uai/core/Tool"
+import { descriptorsOf, providerToolsOf } from "@effect-uai/core/Tool"
+import { describe, expect, it } from "vitest"
+import { WireChunk, buildRequestBody } from "./codec.js"
+import { codeExecutionTool, googleSearchTool, renderProviderTools } from "./GeminiTools.js"
+
+const localTool = Tool.make({
+  name: "get_weather",
+  description: "Look up the weather.",
+  inputSchema: Tool.fromEffectSchema(Schema.Struct({ city: Schema.String })),
+  run: ({ city }) => Effect.succeed(`sunny in ${city}`),
+})
+
+const build = (toolkit: Record<string, Tool.AnyTool>) => {
+  const entries = Result.getOrThrow(renderProviderTools(providerToolsOf(toolkit)))
+  return buildRequestBody([], Option.none(), descriptorsOf(toolkit), entries)
+}
+
+describe("Gemini provider tools", () => {
+  it("renders google_search to the native googleSearch entry", () => {
+    const body = build({ google_search: googleSearchTool })
+    expect(body.tools).toEqual([{ googleSearch: {} }])
+  })
+
+  it("renders a mixed toolkit to both a functionDeclarations entry and the native entry", () => {
+    const body = build({ get_weather: localTool, code_execution: codeExecutionTool })
+    expect(body.tools).toEqual([
+      {
+        functionDeclarations: [
+          expect.objectContaining({ name: "get_weather", description: "Look up the weather." }),
+        ],
+      },
+      { codeExecution: {} },
+    ])
+  })
+
+  it("omits toolConfig for a grounding-only request (no forced function-calling mode)", () => {
+    const body = build({ google_search: googleSearchTool })
+    expect(body.toolConfig).toBeUndefined()
+  })
+
+  it("sets toolConfig when a function declaration is present", () => {
+    const body = build({ get_weather: localTool, google_search: googleSearchTool })
+    expect(body.toolConfig).toEqual({ functionCallingConfig: { mode: "AUTO" } })
+  })
+
+  it("fails on a foreign provider tool", () => {
+    const foreign = Tool.provider({
+      name: "web_search",
+      description: "Anthropic web search.",
+      inputSchema: Tool.noInput,
+      provider: "anthropic",
+      config: { type: "web_search_20250305" },
+    })
+    const result = renderProviderTools(providerToolsOf({ web_search: foreign }))
+    expect(Result.isFailure(result)).toBe(true)
+  })
+
+  it("fails on an unrecognized config kind", () => {
+    const bogus = Tool.provider({
+      name: "mystery",
+      description: "Unknown Gemini tool.",
+      inputSchema: Tool.noInput,
+      provider: "gemini",
+      config: { kind: "telepathy" },
+    })
+    const result = renderProviderTools(providerToolsOf({ mystery: bogus }))
+    expect(Result.isFailure(result)).toBe(true)
+  })
+
+  it("decodes a grounded chunk without throwing (groundingMetadata is dropped)", () => {
+    const grounded = {
+      candidates: [
+        {
+          content: { role: "model", parts: [{ text: "Grounded answer." }] },
+          groundingMetadata: {
+            webSearchQueries: ["effect ts"],
+            groundingChunks: [{ web: { uri: "https://example.com", title: "Example" } }],
+          },
+          finishReason: "STOP",
+        },
+      ],
+    }
+    const chunk = Schema.decodeUnknownSync(WireChunk)(grounded)
+    expect(chunk.candidates?.[0]?.content?.parts?.[0]).toEqual({ text: "Grounded answer." })
+  })
+})

package/src/index.ts

@@ -1,6 +1,8 @@
 export * as Gemini from "./Gemini.js"
+export * as GeminiTools from "./GeminiTools.js"
 export * as GeminiEmbedding from "./GeminiEmbedding.js"
 export * as GeminiSynthesizer from "./GeminiSynthesizer.js"
+export * as GoogleDeepResearch from "./GoogleDeepResearch.js"
 export * as LyriaGenerator from "./LyriaGenerator.js"
 export * as codec from "./codec.js"
 export * from "./models.js"

package/src/models.ts

@@ -17,6 +17,21 @@ export type GoogleModel =
   // eslint-disable-next-line @typescript-eslint/ban-types
   | (string & {})
 
+/**
+ * Deep-research agent identifiers for the Interactions API (`POST
+ * /v1beta/interactions`). `-preview` is the fast tier, `-max-` the
+ * comprehensive one. Preview surface: verify against live docs before relying on
+ * exact ids. The `(string & {})` tail accepts any string so a renamed agent
+ * works without an SDK update.
+ *
+ * Reference: https://ai.google.dev/gemini-api/docs/interactions/deep-research
+ */
+export type GoogleDeepResearchModel =
+  | "deep-research-preview-04-2026"
+  | "deep-research-max-preview-04-2026"
+  // eslint-disable-next-line @typescript-eslint/ban-types
+  | (string & {})
+
 /**
  * Known Gemini embedding model identifiers.
  *