@ftarganski/omni-ai 0.1.0

package/dist/provider-google.js

@@ -0,0 +1,141 @@
+import {
+  registerProvider
+} from "./chunk-6APAA6WD.js";
+
+// ../provider-google/src/provider.ts
+import { GoogleGenerativeAI } from "@google/generative-ai";
+
+// ../provider-google/src/mappers.ts
+function toGeminiPart(part) {
+  if (part.type === "text") return { text: part.text };
+  return { inlineData: { mimeType: part.mimeType, data: part.data } };
+}
+function contentToParts(content) {
+  if (typeof content === "string") return [{ text: content }];
+  return content.map(toGeminiPart);
+}
+function extractSystemInstruction(request) {
+  const systemMsg = request.messages.find((m) => m.role === "system");
+  const raw = request.systemPrompt ?? systemMsg?.content;
+  if (raw === void 0) return void 0;
+  return typeof raw === "string" ? raw : raw.map((p) => p.type === "text" ? p.text : "").join("");
+}
+function toGeminiContents(messages) {
+  return messages.filter((m) => m.role !== "system").map((m) => ({
+    role: m.role === "assistant" ? "model" : "user",
+    parts: contentToParts(m.content)
+  }));
+}
+function toGeminiTools(tools) {
+  const functionDeclarations = tools.map((t) => ({
+    name: t.name,
+    description: t.description,
+    parameters: t.parameters
+  }));
+  return [{ functionDeclarations }];
+}
+function fromGeminiResponse(response, modelName, providerName) {
+  const candidate = response.candidates?.[0];
+  const parts = candidate?.content?.parts ?? [];
+  let content = "";
+  const toolCalls = [];
+  for (const part of parts) {
+    if ("text" in part && typeof part.text === "string") {
+      content += part.text;
+    } else if ("functionCall" in part && part.functionCall) {
+      toolCalls.push({
+        id: part.functionCall.name,
+        name: part.functionCall.name,
+        arguments: part.functionCall.args ?? {}
+      });
+    }
+  }
+  const usage = response.usageMetadata ? {
+    inputTokens: response.usageMetadata.promptTokenCount ?? 0,
+    outputTokens: response.usageMetadata.candidatesTokenCount ?? 0
+  } : void 0;
+  return {
+    content,
+    toolCalls: toolCalls.length > 0 ? toolCalls : void 0,
+    usage,
+    model: modelName,
+    provider: providerName
+  };
+}
+
+// ../provider-google/src/provider.ts
+var GoogleProvider = class {
+  name;
+  capabilities = {
+    chat: true,
+    embedding: true,
+    streaming: true,
+    toolUse: true,
+    vision: true
+  };
+  client;
+  defaultModel;
+  constructor(options) {
+    this.name = options.name ?? "google";
+    this.defaultModel = options.defaultModel ?? "gemini-2.0-flash";
+    this.client = new GoogleGenerativeAI(options.apiKey);
+  }
+  async complete(request) {
+    const modelName = request.model ?? this.defaultModel;
+    const systemInstruction = extractSystemInstruction(request);
+    const contents = toGeminiContents(request.messages);
+    const tools = request.tools && request.tools.length > 0 ? toGeminiTools(request.tools) : void 0;
+    const model = this.client.getGenerativeModel({
+      model: modelName,
+      systemInstruction,
+      tools,
+      generationConfig: {
+        temperature: request.temperature,
+        maxOutputTokens: request.maxTokens
+      }
+    });
+    if (request.onToken) {
+      const result2 = await model.generateContentStream({ contents });
+      let fullContent = "";
+      for await (const chunk of result2.stream) {
+        const text = chunk.text();
+        if (text) {
+          fullContent += text;
+          request.onToken(text);
+        }
+      }
+      const finalResponse = await result2.response;
+      const response = fromGeminiResponse(finalResponse, modelName, this.name);
+      return { ...response, content: fullContent };
+    }
+    const result = await model.generateContent({ contents });
+    return fromGeminiResponse(result.response, modelName, this.name);
+  }
+  async embed(request) {
+    const modelName = "text-embedding-004";
+    const model = this.client.getGenerativeModel({ model: modelName });
+    const inputs = Array.isArray(request.input) ? request.input : [request.input];
+    const embeddings = await Promise.all(
+      inputs.map(async (text) => {
+        const result = await model.embedContent(text);
+        return result.embedding.values;
+      })
+    );
+    return { embeddings, model: modelName, provider: this.name };
+  }
+};
+
+// ../provider-google/src/index.ts
+registerProvider("google", (config) => {
+  if (!config.apiKey) {
+    throw new Error(`Provider "${config.name}" (google) requires GOOGLE_API_KEY to be set in the environment.`);
+  }
+  return new GoogleProvider({
+    apiKey: config.apiKey,
+    defaultModel: config.defaultModel,
+    name: config.name
+  });
+});
+export {
+  GoogleProvider
+};

package/dist/provider-openai.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/provider-openai';

package/dist/provider-openai.js

@@ -0,0 +1,214 @@
+import {
+  registerProvider
+} from "./chunk-6APAA6WD.js";
+
+// ../provider-openai/src/provider.ts
+import OpenAI from "openai";
+
+// ../provider-openai/src/mappers.ts
+function toOpenAIUserContentPart(part) {
+  if (part.type === "text") return { type: "text", text: part.text };
+  return {
+    type: "image_url",
+    image_url: { url: `data:${part.mimeType};base64,${part.data}` }
+  };
+}
+function contentAsString(content) {
+  if (typeof content === "string") return content;
+  return content.map((p) => p.type === "text" ? p.text : "").join("");
+}
+function toOpenAIMessages(messages) {
+  return messages.map((m) => {
+    if (m.role === "system") {
+      return { role: "system", content: contentAsString(m.content) };
+    }
+    if (m.role === "assistant") {
+      return { role: "assistant", content: contentAsString(m.content) };
+    }
+    return {
+      role: "user",
+      content: typeof m.content === "string" ? m.content : m.content.map(toOpenAIUserContentPart)
+    };
+  });
+}
+function toOpenAITools(tools) {
+  return tools.map((t) => ({
+    type: "function",
+    function: {
+      name: t.name,
+      description: t.description,
+      parameters: t.parameters
+    }
+  }));
+}
+function fromOpenAIResponse(response, providerName) {
+  const choice = response.choices[0];
+  const message = choice.message;
+  const toolCalls = [];
+  for (const tc of message.tool_calls ?? []) {
+    toolCalls.push({
+      id: tc.id,
+      name: tc.function.name,
+      arguments: JSON.parse(tc.function.arguments || "{}")
+    });
+  }
+  return {
+    content: message.content ?? "",
+    toolCalls: toolCalls.length > 0 ? toolCalls : void 0,
+    usage: response.usage ? {
+      inputTokens: response.usage.prompt_tokens,
+      outputTokens: response.usage.completion_tokens
+    } : void 0,
+    model: response.model,
+    provider: providerName
+  };
+}
+
+// ../provider-openai/src/provider.ts
+var OpenAIProvider = class {
+  name;
+  capabilities = {
+    chat: true,
+    embedding: true,
+    streaming: true,
+    toolUse: true,
+    vision: true
+  };
+  client;
+  defaultModel;
+  constructor(options) {
+    this.name = options.name ?? "openai";
+    this.defaultModel = options.defaultModel ?? "gpt-4o";
+    this.client = new OpenAI({
+      apiKey: options.apiKey,
+      baseURL: options.baseUrl
+    });
+  }
+  async complete(request) {
+    const model = request.model ?? this.defaultModel;
+    const messages = toOpenAIMessages(request.messages);
+    const tools = request.tools && request.tools.length > 0 ? toOpenAITools(request.tools) : void 0;
+    if (request.onToken) {
+      const stream = await this.client.chat.completions.create({
+        model,
+        temperature: request.temperature,
+        max_tokens: request.maxTokens,
+        messages,
+        tools,
+        tool_choice: tools ? "auto" : void 0,
+        stream: true,
+        stream_options: { include_usage: true }
+      });
+      let content = "";
+      const toolCallAccum = {};
+      let inputTokens = 0;
+      let outputTokens = 0;
+      for await (const chunk of stream) {
+        const delta = chunk.choices[0]?.delta;
+        if (delta?.content) {
+          content += delta.content;
+          request.onToken(delta.content);
+        }
+        if (delta?.tool_calls) {
+          for (const tc of delta.tool_calls) {
+            const idx = tc.index;
+            if (!toolCallAccum[idx]) {
+              toolCallAccum[idx] = { id: "", name: "", args: "" };
+            }
+            if (tc.id) toolCallAccum[idx].id = tc.id;
+            if (tc.function?.name) toolCallAccum[idx].name = tc.function.name;
+            if (tc.function?.arguments) toolCallAccum[idx].args += tc.function.arguments;
+          }
+        }
+        if (chunk.usage) {
+          inputTokens = chunk.usage.prompt_tokens;
+          outputTokens = chunk.usage.completion_tokens;
+        }
+      }
+      const toolCalls = Object.values(toolCallAccum);
+      return {
+        content,
+        toolCalls: toolCalls.length > 0 ? toolCalls.map((tc) => ({
+          id: tc.id,
+          name: tc.name,
+          arguments: (() => {
+            try {
+              return JSON.parse(tc.args);
+            } catch {
+              return {};
+            }
+          })()
+        })) : void 0,
+        usage: inputTokens > 0 ? { inputTokens, outputTokens } : void 0,
+        model,
+        provider: this.name
+      };
+    }
+    const response = await this.client.chat.completions.create({
+      model,
+      temperature: request.temperature,
+      max_tokens: request.maxTokens,
+      messages,
+      tools,
+      tool_choice: tools ? "auto" : void 0
+    });
+    return fromOpenAIResponse(response, this.name);
+  }
+  async embed(request) {
+    const model = request.model ?? "text-embedding-3-small";
+    const input = Array.isArray(request.input) ? request.input : [request.input];
+    const response = await this.client.embeddings.create({ model, input });
+    return {
+      embeddings: response.data.map((d) => d.embedding),
+      model: response.model,
+      provider: this.name
+    };
+  }
+};
+
+// ../provider-openai/src/index.ts
+registerProvider("openai", (config) => {
+  if (!config.apiKey) {
+    throw new Error(`Provider "${config.name}" (openai) requires OPENAI_API_KEY to be set in the environment.`);
+  }
+  return new OpenAIProvider({
+    apiKey: config.apiKey,
+    defaultModel: config.defaultModel,
+    baseUrl: config.baseUrl,
+    name: config.name
+  });
+});
+registerProvider("copilot", (config) => {
+  if (!config.apiKey) {
+    throw new Error(`Provider "${config.name}" (copilot) requires GITHUB_TOKEN to be set in the environment.`);
+  }
+  return new OpenAIProvider({
+    apiKey: config.apiKey,
+    defaultModel: config.defaultModel ?? "gpt-4o",
+    baseUrl: config.baseUrl ?? "https://api.githubcopilot.com",
+    name: config.name
+  });
+});
+registerProvider("groq", (config) => {
+  if (!config.apiKey) {
+    throw new Error(`Provider "${config.name}" (groq) requires GROQ_API_KEY to be set in the environment.`);
+  }
+  return new OpenAIProvider({
+    apiKey: config.apiKey,
+    defaultModel: config.defaultModel ?? "llama-3.3-70b-versatile",
+    baseUrl: config.baseUrl ?? "https://api.groq.com/openai/v1",
+    name: config.name
+  });
+});
+registerProvider("ollama", (config) => {
+  return new OpenAIProvider({
+    // Ollama does not require an API key; use a placeholder so the SDK does not complain
+    apiKey: config.apiKey ?? "ollama",
+    defaultModel: config.defaultModel ?? "llama3.2",
+    baseUrl: config.baseUrl ?? "http://localhost:11434/v1",
+    name: config.name
+  });
+});
+export {
+  OpenAIProvider
+};

package/dist/skills/backend.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills/backend';

package/dist/skills/backend.js

@@ -0,0 +1,12 @@
+import {
+  analyzeDynamoSchemaSkill,
+  analyzeGraphqlSchemaSkill,
+  analyzeNestjsModuleSkill,
+  findCodePatternSkill
+} from "../chunk-3RZELMF2.js";
+export {
+  analyzeDynamoSchemaSkill,
+  analyzeGraphqlSchemaSkill,
+  analyzeNestjsModuleSkill,
+  findCodePatternSkill
+};

package/dist/skills/code.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills/code';

package/dist/skills/code.js

@@ -0,0 +1,6 @@
+import {
+  searchCodeSkill
+} from "../chunk-M4QJF7CV.js";
+export {
+  searchCodeSkill
+};

package/dist/skills/frontend.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills/frontend';

package/dist/skills/frontend.js

@@ -0,0 +1,10 @@
+import {
+  analyzeComponentSkill,
+  analyzeModuleStructureSkill,
+  findComponentPatternSkill
+} from "../chunk-S5MK6LBH.js";
+export {
+  analyzeComponentSkill,
+  analyzeModuleStructureSkill,
+  findComponentPatternSkill
+};

package/dist/skills/fs.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills/fs';

package/dist/skills/fs.js

@@ -0,0 +1,10 @@
+import {
+  listDirectorySkill,
+  readFileSkill,
+  writeFileSkill
+} from "../chunk-TFU437SW.js";
+export {
+  listDirectorySkill,
+  readFileSkill,
+  writeFileSkill
+};

package/dist/skills/git.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills/git';

package/dist/skills/git.js

@@ -0,0 +1,12 @@
+import {
+  gitCommitMessageSkill,
+  gitDiffSkill,
+  gitLogSkill,
+  gitStatusSkill
+} from "../chunk-6OPRALDC.js";
+export {
+  gitCommitMessageSkill,
+  gitDiffSkill,
+  gitLogSkill,
+  gitStatusSkill
+};

package/dist/skills/http.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills/http';

package/dist/skills/http.js

@@ -0,0 +1,6 @@
+import {
+  httpRequestSkill
+} from "../chunk-6YFFZMXY.js";
+export {
+  httpRequestSkill
+};

package/dist/skills/index.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills';

package/dist/skills/index.js

@@ -0,0 +1,60 @@
+import {
+  analyzeDynamoSchemaSkill,
+  analyzeGraphqlSchemaSkill,
+  analyzeNestjsModuleSkill,
+  findCodePatternSkill
+} from "../chunk-3RZELMF2.js";
+import {
+  analyzeComponentSkill,
+  analyzeModuleStructureSkill,
+  findComponentPatternSkill
+} from "../chunk-S5MK6LBH.js";
+import {
+  analyzeTestCoverageSkill,
+  findTestPatternSkill
+} from "../chunk-PPTEJ2FH.js";
+import {
+  listDirectorySkill,
+  readFileSkill,
+  writeFileSkill
+} from "../chunk-TFU437SW.js";
+import {
+  searchCodeSkill
+} from "../chunk-M4QJF7CV.js";
+import {
+  auditAccessibilitySkill
+} from "../chunk-JTXDF5KG.js";
+import {
+  gitCommitMessageSkill,
+  gitDiffSkill,
+  gitLogSkill,
+  gitStatusSkill
+} from "../chunk-6OPRALDC.js";
+import {
+  httpRequestSkill
+} from "../chunk-6YFFZMXY.js";
+import {
+  analyzeImageSkill
+} from "../chunk-5WELBZWN.js";
+export {
+  analyzeComponentSkill,
+  analyzeDynamoSchemaSkill,
+  analyzeGraphqlSchemaSkill,
+  analyzeImageSkill,
+  analyzeModuleStructureSkill,
+  analyzeNestjsModuleSkill,
+  analyzeTestCoverageSkill,
+  auditAccessibilitySkill,
+  findCodePatternSkill,
+  findComponentPatternSkill,
+  findTestPatternSkill,
+  gitCommitMessageSkill,
+  gitDiffSkill,
+  gitLogSkill,
+  gitStatusSkill,
+  httpRequestSkill,
+  listDirectorySkill,
+  readFileSkill,
+  searchCodeSkill,
+  writeFileSkill
+};

package/dist/skills/multimodal.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills/multimodal';

package/dist/skills/multimodal.js

@@ -0,0 +1,6 @@
+import {
+  analyzeImageSkill
+} from "../chunk-5WELBZWN.js";
+export {
+  analyzeImageSkill
+};

package/dist/skills/qa.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills/qa';

package/dist/skills/qa.js

@@ -0,0 +1,8 @@
+import {
+  analyzeTestCoverageSkill,
+  findTestPatternSkill
+} from "../chunk-PPTEJ2FH.js";
+export {
+  analyzeTestCoverageSkill,
+  findTestPatternSkill
+};

package/dist/skills/ux.d.ts

@@ -0,0 +1 @@
+export * from '@omni-ai/skills/ux';

package/dist/skills/ux.js

@@ -0,0 +1,6 @@
+import {
+  auditAccessibilitySkill
+} from "../chunk-JTXDF5KG.js";
+export {
+  auditAccessibilitySkill
+};

package/dist/src-6MUVU5ML.js

@@ -0,0 +1,8 @@
+#!/usr/bin/env node
+import {
+  OpenAIProvider
+} from "./chunk-Y4EYGADJ.js";
+import "./chunk-AWMSN7OB.js";
+export {
+  OpenAIProvider
+};

package/dist/src-ZHTGR7T6.js

@@ -0,0 +1,8 @@
+#!/usr/bin/env node
+import {
+  AnthropicProvider
+} from "./chunk-AG6NZIJ3.js";
+import "./chunk-AWMSN7OB.js";
+export {
+  AnthropicProvider
+};