@matthesketh/utopia-ai 0.0.1

package/dist/adapters/google.cjs

@@ -0,0 +1,220 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/adapters/google.ts
+var google_exports = {};
+__export(google_exports, {
+  googleAdapter: () => googleAdapter
+});
+module.exports = __toCommonJS(google_exports);
+function googleAdapter(config) {
+  let genAI = null;
+  async function getGenAI() {
+    if (genAI) return genAI;
+    let GoogleGenerativeAI;
+    try {
+      const mod = await import("@google/generative-ai");
+      GoogleGenerativeAI = mod.GoogleGenerativeAI;
+    } catch {
+      throw new Error(
+        '@matthesketh/utopia-ai: "@google/generative-ai" package is required for the Google adapter. Install it with: npm install @google/generative-ai'
+      );
+    }
+    genAI = new GoogleGenerativeAI(config.apiKey);
+    return genAI;
+  }
+  return {
+    async chat(request) {
+      const ai = await getGenAI();
+      const modelName = request.model ?? config.defaultModel ?? "gemini-2.0-flash";
+      const modelConfig = {};
+      if (request.temperature !== void 0) modelConfig.temperature = request.temperature;
+      if (request.maxTokens !== void 0) modelConfig.maxOutputTokens = request.maxTokens;
+      if (request.topP !== void 0) modelConfig.topP = request.topP;
+      if (request.stop) modelConfig.stopSequences = request.stop;
+      const model = ai.getGenerativeModel({
+        model: modelName,
+        generationConfig: modelConfig,
+        ...request.tools?.length ? { tools: [{ functionDeclarations: request.tools.map(toGeminiTool) }] } : {},
+        ...request.extra
+      });
+      const { system, contents } = toGeminiContents(request.messages);
+      const result = await model.generateContent({
+        contents,
+        ...system ? { systemInstruction: { parts: [{ text: system }] } } : {}
+      });
+      const response = result.response;
+      const candidate = response.candidates?.[0];
+      const parts = candidate?.content?.parts ?? [];
+      const textParts = parts.filter((p) => p.text).map((p) => p.text);
+      const fnCalls = parts.filter((p) => p.functionCall);
+      const toolCalls = fnCalls.map((p) => ({
+        id: `call_${crypto.randomUUID()}`,
+        name: p.functionCall.name,
+        arguments: p.functionCall.args ?? {}
+      }));
+      return {
+        content: textParts.join(""),
+        toolCalls: toolCalls.length > 0 ? toolCalls : void 0,
+        finishReason: mapFinishReason(candidate?.finishReason),
+        usage: response.usageMetadata ? {
+          promptTokens: response.usageMetadata.promptTokenCount ?? 0,
+          completionTokens: response.usageMetadata.candidatesTokenCount ?? 0,
+          totalTokens: response.usageMetadata.totalTokenCount ?? 0
+        } : void 0,
+        raw: response
+      };
+    },
+    async *stream(request) {
+      const ai = await getGenAI();
+      const modelName = request.model ?? config.defaultModel ?? "gemini-2.0-flash";
+      const modelConfig = {};
+      if (request.temperature !== void 0) modelConfig.temperature = request.temperature;
+      if (request.maxTokens !== void 0) modelConfig.maxOutputTokens = request.maxTokens;
+      if (request.topP !== void 0) modelConfig.topP = request.topP;
+      if (request.stop) modelConfig.stopSequences = request.stop;
+      const model = ai.getGenerativeModel({
+        model: modelName,
+        generationConfig: modelConfig,
+        ...request.tools?.length ? { tools: [{ functionDeclarations: request.tools.map(toGeminiTool) }] } : {},
+        ...request.extra
+      });
+      const { system, contents } = toGeminiContents(request.messages);
+      const result = await model.generateContentStream({
+        contents,
+        ...system ? { systemInstruction: { parts: [{ text: system }] } } : {}
+      });
+      for await (const chunk of result.stream) {
+        const parts = chunk.candidates?.[0]?.content?.parts ?? [];
+        const text = parts.filter((p) => p.text).map((p) => p.text).join("");
+        const finishReason = chunk.candidates?.[0]?.finishReason;
+        yield {
+          delta: text,
+          finishReason: finishReason ? mapFinishReason(finishReason) : void 0,
+          usage: chunk.usageMetadata ? {
+            promptTokens: chunk.usageMetadata.promptTokenCount ?? 0,
+            completionTokens: chunk.usageMetadata.candidatesTokenCount ?? 0,
+            totalTokens: chunk.usageMetadata.totalTokenCount ?? 0
+          } : void 0
+        };
+      }
+    },
+    async embeddings(request) {
+      const ai = await getGenAI();
+      const model = ai.getGenerativeModel({
+        model: request.model ?? "text-embedding-004"
+      });
+      const inputs = Array.isArray(request.input) ? request.input : [request.input];
+      const result = await model.batchEmbedContents({
+        requests: inputs.map((text) => ({
+          content: { parts: [{ text }] }
+        }))
+      });
+      return {
+        embeddings: result.embeddings.map((e) => e.values),
+        raw: result
+      };
+    }
+  };
+}
+function toGeminiContents(messages) {
+  let system;
+  const contents = [];
+  for (const msg of messages) {
+    if (msg.role === "system") {
+      system = typeof msg.content === "string" ? msg.content : "";
+      continue;
+    }
+    const role = msg.role === "assistant" ? "model" : "user";
+    if (typeof msg.content === "string") {
+      contents.push({ role, parts: [{ text: msg.content }] });
+      continue;
+    }
+    if (Array.isArray(msg.content)) {
+      const parts = [];
+      for (const part2 of msg.content) {
+        if (typeof part2 === "string") {
+          parts.push({ text: part2 });
+        } else if (part2.type === "text") {
+          parts.push({ text: part2.text });
+        } else if (part2.type === "image") {
+          parts.push({
+            inlineData: {
+              mimeType: part2.mediaType ?? "image/png",
+              data: part2.source
+            }
+          });
+        } else if (part2.type === "tool_call") {
+          parts.push({
+            functionCall: { name: part2.name, args: part2.arguments }
+          });
+        } else if (part2.type === "tool_result") {
+          parts.push({
+            functionResponse: {
+              name: part2.id,
+              response: { content: part2.content }
+            }
+          });
+        }
+      }
+      contents.push({ role, parts });
+      continue;
+    }
+    const part = msg.content;
+    if (part.type === "text") {
+      contents.push({ role, parts: [{ text: part.text }] });
+    }
+  }
+  return { system, contents };
+}
+function toGeminiTool(tool) {
+  return {
+    name: tool.name,
+    description: tool.description,
+    parameters: tool.parameters
+  };
+}
+function mapFinishReason(reason) {
+  switch (reason) {
+    case "STOP":
+      return "stop";
+    case "MAX_TOKENS":
+      return "length";
+    case "SAFETY":
+    case "RECITATION":
+    case "OTHER":
+      return "stop";
+    default:
+      return "stop";
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  googleAdapter
+});

package/dist/adapters/google.d.cts

@@ -0,0 +1,17 @@
+import { G as GoogleConfig, A as AIAdapter } from '../types-FSnS43LM.cjs';
+
+/**
+ * Create a Google Gemini adapter.
+ *
+ * Requires `@google/generative-ai` as a peer dependency.
+ *
+ * ```ts
+ * import { createAI } from '@matthesketh/utopia-ai';
+ * import { googleAdapter } from '@matthesketh/utopia-ai/google';
+ *
+ * const ai = createAI(googleAdapter({ apiKey: process.env.GOOGLE_API_KEY }));
+ * ```
+ */
+declare function googleAdapter(config: GoogleConfig): AIAdapter;
+
+export { googleAdapter };

package/dist/adapters/google.d.ts

@@ -0,0 +1,17 @@
+import { G as GoogleConfig, A as AIAdapter } from '../types-FSnS43LM.js';
+
+/**
+ * Create a Google Gemini adapter.
+ *
+ * Requires `@google/generative-ai` as a peer dependency.
+ *
+ * ```ts
+ * import { createAI } from '@matthesketh/utopia-ai';
+ * import { googleAdapter } from '@matthesketh/utopia-ai/google';
+ *
+ * const ai = createAI(googleAdapter({ apiKey: process.env.GOOGLE_API_KEY }));
+ * ```
+ */
+declare function googleAdapter(config: GoogleConfig): AIAdapter;
+
+export { googleAdapter };

package/dist/adapters/google.js

@@ -0,0 +1,185 @@
+// src/adapters/google.ts
+function googleAdapter(config) {
+  let genAI = null;
+  async function getGenAI() {
+    if (genAI) return genAI;
+    let GoogleGenerativeAI;
+    try {
+      const mod = await import("@google/generative-ai");
+      GoogleGenerativeAI = mod.GoogleGenerativeAI;
+    } catch {
+      throw new Error(
+        '@matthesketh/utopia-ai: "@google/generative-ai" package is required for the Google adapter. Install it with: npm install @google/generative-ai'
+      );
+    }
+    genAI = new GoogleGenerativeAI(config.apiKey);
+    return genAI;
+  }
+  return {
+    async chat(request) {
+      const ai = await getGenAI();
+      const modelName = request.model ?? config.defaultModel ?? "gemini-2.0-flash";
+      const modelConfig = {};
+      if (request.temperature !== void 0) modelConfig.temperature = request.temperature;
+      if (request.maxTokens !== void 0) modelConfig.maxOutputTokens = request.maxTokens;
+      if (request.topP !== void 0) modelConfig.topP = request.topP;
+      if (request.stop) modelConfig.stopSequences = request.stop;
+      const model = ai.getGenerativeModel({
+        model: modelName,
+        generationConfig: modelConfig,
+        ...request.tools?.length ? { tools: [{ functionDeclarations: request.tools.map(toGeminiTool) }] } : {},
+        ...request.extra
+      });
+      const { system, contents } = toGeminiContents(request.messages);
+      const result = await model.generateContent({
+        contents,
+        ...system ? { systemInstruction: { parts: [{ text: system }] } } : {}
+      });
+      const response = result.response;
+      const candidate = response.candidates?.[0];
+      const parts = candidate?.content?.parts ?? [];
+      const textParts = parts.filter((p) => p.text).map((p) => p.text);
+      const fnCalls = parts.filter((p) => p.functionCall);
+      const toolCalls = fnCalls.map((p) => ({
+        id: `call_${crypto.randomUUID()}`,
+        name: p.functionCall.name,
+        arguments: p.functionCall.args ?? {}
+      }));
+      return {
+        content: textParts.join(""),
+        toolCalls: toolCalls.length > 0 ? toolCalls : void 0,
+        finishReason: mapFinishReason(candidate?.finishReason),
+        usage: response.usageMetadata ? {
+          promptTokens: response.usageMetadata.promptTokenCount ?? 0,
+          completionTokens: response.usageMetadata.candidatesTokenCount ?? 0,
+          totalTokens: response.usageMetadata.totalTokenCount ?? 0
+        } : void 0,
+        raw: response
+      };
+    },
+    async *stream(request) {
+      const ai = await getGenAI();
+      const modelName = request.model ?? config.defaultModel ?? "gemini-2.0-flash";
+      const modelConfig = {};
+      if (request.temperature !== void 0) modelConfig.temperature = request.temperature;
+      if (request.maxTokens !== void 0) modelConfig.maxOutputTokens = request.maxTokens;
+      if (request.topP !== void 0) modelConfig.topP = request.topP;
+      if (request.stop) modelConfig.stopSequences = request.stop;
+      const model = ai.getGenerativeModel({
+        model: modelName,
+        generationConfig: modelConfig,
+        ...request.tools?.length ? { tools: [{ functionDeclarations: request.tools.map(toGeminiTool) }] } : {},
+        ...request.extra
+      });
+      const { system, contents } = toGeminiContents(request.messages);
+      const result = await model.generateContentStream({
+        contents,
+        ...system ? { systemInstruction: { parts: [{ text: system }] } } : {}
+      });
+      for await (const chunk of result.stream) {
+        const parts = chunk.candidates?.[0]?.content?.parts ?? [];
+        const text = parts.filter((p) => p.text).map((p) => p.text).join("");
+        const finishReason = chunk.candidates?.[0]?.finishReason;
+        yield {
+          delta: text,
+          finishReason: finishReason ? mapFinishReason(finishReason) : void 0,
+          usage: chunk.usageMetadata ? {
+            promptTokens: chunk.usageMetadata.promptTokenCount ?? 0,
+            completionTokens: chunk.usageMetadata.candidatesTokenCount ?? 0,
+            totalTokens: chunk.usageMetadata.totalTokenCount ?? 0
+          } : void 0
+        };
+      }
+    },
+    async embeddings(request) {
+      const ai = await getGenAI();
+      const model = ai.getGenerativeModel({
+        model: request.model ?? "text-embedding-004"
+      });
+      const inputs = Array.isArray(request.input) ? request.input : [request.input];
+      const result = await model.batchEmbedContents({
+        requests: inputs.map((text) => ({
+          content: { parts: [{ text }] }
+        }))
+      });
+      return {
+        embeddings: result.embeddings.map((e) => e.values),
+        raw: result
+      };
+    }
+  };
+}
+function toGeminiContents(messages) {
+  let system;
+  const contents = [];
+  for (const msg of messages) {
+    if (msg.role === "system") {
+      system = typeof msg.content === "string" ? msg.content : "";
+      continue;
+    }
+    const role = msg.role === "assistant" ? "model" : "user";
+    if (typeof msg.content === "string") {
+      contents.push({ role, parts: [{ text: msg.content }] });
+      continue;
+    }
+    if (Array.isArray(msg.content)) {
+      const parts = [];
+      for (const part2 of msg.content) {
+        if (typeof part2 === "string") {
+          parts.push({ text: part2 });
+        } else if (part2.type === "text") {
+          parts.push({ text: part2.text });
+        } else if (part2.type === "image") {
+          parts.push({
+            inlineData: {
+              mimeType: part2.mediaType ?? "image/png",
+              data: part2.source
+            }
+          });
+        } else if (part2.type === "tool_call") {
+          parts.push({
+            functionCall: { name: part2.name, args: part2.arguments }
+          });
+        } else if (part2.type === "tool_result") {
+          parts.push({
+            functionResponse: {
+              name: part2.id,
+              response: { content: part2.content }
+            }
+          });
+        }
+      }
+      contents.push({ role, parts });
+      continue;
+    }
+    const part = msg.content;
+    if (part.type === "text") {
+      contents.push({ role, parts: [{ text: part.text }] });
+    }
+  }
+  return { system, contents };
+}
+function toGeminiTool(tool) {
+  return {
+    name: tool.name,
+    description: tool.description,
+    parameters: tool.parameters
+  };
+}
+function mapFinishReason(reason) {
+  switch (reason) {
+    case "STOP":
+      return "stop";
+    case "MAX_TOKENS":
+      return "length";
+    case "SAFETY":
+    case "RECITATION":
+    case "OTHER":
+      return "stop";
+    default:
+      return "stop";
+  }
+}
+export {
+  googleAdapter
+};

package/dist/adapters/ollama.cjs

@@ -0,0 +1,187 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/adapters/ollama.ts
+var ollama_exports = {};
+__export(ollama_exports, {
+  ollamaAdapter: () => ollamaAdapter
+});
+module.exports = __toCommonJS(ollama_exports);
+function ollamaAdapter(config = {}) {
+  const baseURL = (config.baseURL ?? "http://localhost:11434").replace(/\/$/, "");
+  return {
+    async chat(request) {
+      const model = request.model ?? config.defaultModel ?? "llama3.2";
+      const body = {
+        model,
+        messages: toOllamaMessages(request.messages),
+        stream: false,
+        options: {}
+      };
+      if (request.temperature !== void 0) body.options.temperature = request.temperature;
+      if (request.topP !== void 0) body.options.top_p = request.topP;
+      if (request.maxTokens !== void 0) body.options.num_predict = request.maxTokens;
+      if (request.stop) body.options.stop = request.stop;
+      if (request.tools?.length) {
+        body.tools = request.tools.map(toOllamaTool);
+      }
+      const response = await fetch(`${baseURL}/api/chat`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify(body)
+      });
+      if (!response.ok) {
+        const text = await response.text();
+        throw new Error(`Ollama error ${response.status}: ${text}`);
+      }
+      const data = await response.json();
+      const toolCalls = (data.message?.tool_calls ?? []).map(
+        (tc, i) => ({
+          id: `call_${i}`,
+          name: tc.function.name,
+          arguments: tc.function.arguments ?? {}
+        })
+      );
+      return {
+        content: data.message?.content ?? "",
+        toolCalls: toolCalls.length > 0 ? toolCalls : void 0,
+        finishReason: data.done_reason === "length" ? "length" : "stop",
+        usage: {
+          promptTokens: data.prompt_eval_count ?? 0,
+          completionTokens: data.eval_count ?? 0,
+          totalTokens: (data.prompt_eval_count ?? 0) + (data.eval_count ?? 0)
+        },
+        raw: data
+      };
+    },
+    async *stream(request) {
+      const model = request.model ?? config.defaultModel ?? "llama3.2";
+      const body = {
+        model,
+        messages: toOllamaMessages(request.messages),
+        stream: true,
+        options: {}
+      };
+      if (request.temperature !== void 0) body.options.temperature = request.temperature;
+      if (request.topP !== void 0) body.options.top_p = request.topP;
+      if (request.maxTokens !== void 0) body.options.num_predict = request.maxTokens;
+      if (request.stop) body.options.stop = request.stop;
+      if (request.tools?.length) {
+        body.tools = request.tools.map(toOllamaTool);
+      }
+      const response = await fetch(`${baseURL}/api/chat`, {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify(body)
+      });
+      if (!response.ok) {
+        const text = await response.text();
+        throw new Error(`Ollama error ${response.status}: ${text}`);
+      }
+      const reader = response.body.getReader();
+      const decoder = new TextDecoder();
+      let buffer = "";
+      while (true) {
+        const { done, value } = await reader.read();
+        if (done) break;
+        buffer += decoder.decode(value, { stream: true });
+        const lines = buffer.split("\n");
+        buffer = lines.pop() ?? "";
+        for (const line of lines) {
+          if (!line.trim()) continue;
+          const data = JSON.parse(line);
+          yield {
+            delta: data.message?.content ?? "",
+            finishReason: data.done ? "stop" : void 0,
+            usage: data.done ? {
+              promptTokens: data.prompt_eval_count ?? 0,
+              completionTokens: data.eval_count ?? 0,
+              totalTokens: (data.prompt_eval_count ?? 0) + (data.eval_count ?? 0)
+            } : void 0
+          };
+        }
+      }
+    },
+    async embeddings(request) {
+      const model = request.model ?? "nomic-embed-text";
+      const inputs = Array.isArray(request.input) ? request.input : [request.input];
+      const results = [];
+      for (const input of inputs) {
+        const response = await fetch(`${baseURL}/api/embed`, {
+          method: "POST",
+          headers: { "Content-Type": "application/json" },
+          body: JSON.stringify({ model, input })
+        });
+        if (!response.ok) {
+          const text = await response.text();
+          throw new Error(`Ollama error ${response.status}: ${text}`);
+        }
+        const data = await response.json();
+        results.push(...data.embeddings ?? [data.embedding]);
+      }
+      return { embeddings: results };
+    }
+  };
+}
+function toOllamaMessages(messages) {
+  return messages.map((msg) => {
+    if (typeof msg.content === "string") {
+      return { role: msg.role, content: msg.content };
+    }
+    if (Array.isArray(msg.content)) {
+      const texts = [];
+      const images = [];
+      for (const part2 of msg.content) {
+        if (typeof part2 === "string") {
+          texts.push(part2);
+        } else if (part2.type === "text") {
+          texts.push(part2.text);
+        } else if (part2.type === "image") {
+          images.push(part2.source);
+        } else if (part2.type === "tool_result") {
+          texts.push(part2.content);
+        }
+      }
+      return {
+        role: msg.role === "tool" ? "user" : msg.role,
+        content: texts.join("\n"),
+        ...images.length > 0 ? { images } : {}
+      };
+    }
+    const part = msg.content;
+    if (part.type === "text") {
+      return { role: msg.role, content: part.text };
+    }
+    return { role: msg.role, content: "" };
+  });
+}
+function toOllamaTool(tool) {
+  return {
+    type: "function",
+    function: {
+      name: tool.name,
+      description: tool.description,
+      parameters: tool.parameters
+    }
+  };
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ollamaAdapter
+});

package/dist/adapters/ollama.d.cts

@@ -0,0 +1,17 @@
+import { O as OllamaConfig, A as AIAdapter } from '../types-FSnS43LM.cjs';
+
+/**
+ * Create an Ollama adapter for local models.
+ *
+ * No external dependencies required — uses native fetch.
+ *
+ * ```ts
+ * import { createAI } from '@matthesketh/utopia-ai';
+ * import { ollamaAdapter } from '@matthesketh/utopia-ai/ollama';
+ *
+ * const ai = createAI(ollamaAdapter({ defaultModel: 'llama3.2' }));
+ * ```
+ */
+declare function ollamaAdapter(config?: OllamaConfig): AIAdapter;
+
+export { ollamaAdapter };

package/dist/adapters/ollama.d.ts

@@ -0,0 +1,17 @@
+import { O as OllamaConfig, A as AIAdapter } from '../types-FSnS43LM.js';
+
+/**
+ * Create an Ollama adapter for local models.
+ *
+ * No external dependencies required — uses native fetch.
+ *
+ * ```ts
+ * import { createAI } from '@matthesketh/utopia-ai';
+ * import { ollamaAdapter } from '@matthesketh/utopia-ai/ollama';
+ *
+ * const ai = createAI(ollamaAdapter({ defaultModel: 'llama3.2' }));
+ * ```
+ */
+declare function ollamaAdapter(config?: OllamaConfig): AIAdapter;
+
+export { ollamaAdapter };