@radaros/core 0.3.5 → 0.3.6

package/src/models/providers/vertex.ts

@@ -1,427 +0,0 @@
-import { createRequire } from "node:module";
-import type { ModelProvider } from "../provider.js";
-import {
-  getTextContent,
-  isMultiModal,
-  type ChatMessage,
-  type ContentPart,
-  type ModelConfig,
-  type ModelResponse,
-  type StreamChunk,
-  type ToolDefinition,
-  type TokenUsage,
-  type ToolCall,
-} from "../types.js";
-
-const _require = createRequire(import.meta.url);
-
-export interface VertexAIConfig {
-  project?: string;
-  location?: string;
-  /** Service account key JSON string or path (optional — uses ADC by default). */
-  credentials?: string;
-}
-
-/**
- * Vertex AI provider using Google's @google/genai SDK in Vertex mode.
- *
- * Authentication (in order of precedence):
- *   1. Explicit `project` + `location` in config
- *   2. GOOGLE_CLOUD_PROJECT / GOOGLE_CLOUD_LOCATION env vars
- *   3. Application Default Credentials (gcloud auth)
- */
-export class VertexAIProvider implements ModelProvider {
-  readonly providerId = "vertex";
-  readonly modelId: string;
-  private ai: any = null;
-  private GoogleGenAICtor: any;
-  private project: string;
-  private location: string;
-
-  constructor(modelId: string, config?: VertexAIConfig) {
-    this.modelId = modelId;
-    this.project =
-      config?.project ??
-      process.env.GOOGLE_CLOUD_PROJECT ??
-      process.env.GCLOUD_PROJECT ??
-      "";
-    this.location =
-      config?.location ??
-      process.env.GOOGLE_CLOUD_LOCATION ??
-      process.env.GOOGLE_CLOUD_REGION ??
-      "us-central1";
-
-    if (!this.project) {
-      throw new Error(
-        "VertexAIProvider: 'project' is required. Pass it in config or set GOOGLE_CLOUD_PROJECT env var."
-      );
-    }
-
-    try {
-      const { GoogleGenAI } = _require("@google/genai");
-      this.GoogleGenAICtor = GoogleGenAI;
-    } catch {
-      throw new Error(
-        "@google/genai is required for VertexAIProvider. Install it: npm install @google/genai"
-      );
-    }
-  }
-
-  private getClient(): any {
-    if (this.ai) return this.ai;
-    this.ai = new this.GoogleGenAICtor({
-      vertexai: true,
-      project: this.project,
-      location: this.location,
-    });
-    return this.ai;
-  }
-
-  async generate(
-    messages: ChatMessage[],
-    options?: ModelConfig & { tools?: ToolDefinition[] }
-  ): Promise<ModelResponse> {
-    const { systemInstruction, contents } = this.toGoogleMessages(messages);
-
-    const config: Record<string, unknown> = {};
-    if (options?.temperature !== undefined)
-      config.temperature = options.temperature;
-    if (options?.maxTokens !== undefined)
-      config.maxOutputTokens = options.maxTokens;
-    if (options?.topP !== undefined) config.topP = options.topP;
-    if (options?.stop) config.stopSequences = options.stop;
-
-    if (options?.responseFormat) {
-      config.responseMimeType = "application/json";
-      const rf = options.responseFormat;
-      if (
-        typeof rf === "object" &&
-        rf !== null &&
-        "type" in rf &&
-        rf.type === "json_schema" &&
-        "schema" in rf &&
-        rf.schema
-      ) {
-        config.responseSchema = this.cleanJsonSchema(
-          rf.schema as Record<string, unknown>
-        );
-      }
-    }
-
-    if (options?.reasoning?.enabled) {
-      config.thinkingConfig = {
-        thinkingBudget: options.reasoning.budgetTokens ?? 10000,
-      };
-    }
-
-    const params: Record<string, unknown> = {
-      model: this.modelId,
-      contents,
-      config,
-    };
-
-    if (systemInstruction) params.systemInstruction = systemInstruction;
-    if (options?.tools?.length) {
-      params.tools = [
-        { functionDeclarations: this.toGoogleTools(options.tools) },
-      ];
-    }
-
-    const client = this.getClient();
-    const response = await client.models.generateContent(params);
-    return this.normalizeResponse(response);
-  }
-
-  async *stream(
-    messages: ChatMessage[],
-    options?: ModelConfig & { tools?: ToolDefinition[] }
-  ): AsyncGenerator<StreamChunk> {
-    const { systemInstruction, contents } = this.toGoogleMessages(messages);
-
-    const config: Record<string, unknown> = {};
-    if (options?.temperature !== undefined)
-      config.temperature = options.temperature;
-    if (options?.maxTokens !== undefined)
-      config.maxOutputTokens = options.maxTokens;
-    if (options?.topP !== undefined) config.topP = options.topP;
-    if (options?.stop) config.stopSequences = options.stop;
-
-    if (options?.reasoning?.enabled) {
-      config.thinkingConfig = {
-        thinkingBudget: options.reasoning.budgetTokens ?? 10000,
-      };
-    }
-
-    const params: Record<string, unknown> = {
-      model: this.modelId,
-      contents,
-      config,
-    };
-
-    if (systemInstruction) params.systemInstruction = systemInstruction;
-    if (options?.tools?.length) {
-      params.tools = [
-        { functionDeclarations: this.toGoogleTools(options.tools) },
-      ];
-    }
-
-    const client = this.getClient();
-    const streamResult = await client.models.generateContentStream(params);
-
-    let toolCallCounter = 0;
-
-    for await (const chunk of streamResult) {
-      const candidate = chunk.candidates?.[0];
-      if (!candidate?.content?.parts) continue;
-
-      for (const part of candidate.content.parts) {
-        if (part.thought) {
-          yield { type: "thinking", text: part.text ?? "" };
-        } else if (part.text) {
-          yield { type: "text", text: part.text };
-        }
-
-        if (part.functionCall) {
-          const id = `vertex_tc_${toolCallCounter++}`;
-          yield {
-            type: "tool_call_start",
-            toolCall: { id, name: part.functionCall.name },
-          };
-          yield {
-            type: "tool_call_delta",
-            toolCallId: id,
-            argumentsDelta: JSON.stringify(part.functionCall.args ?? {}),
-          };
-          yield { type: "tool_call_end", toolCallId: id };
-        }
-      }
-
-      if (candidate.finishReason) {
-        let finishReason = "stop";
-        if (
-          candidate.finishReason === "STOP" ||
-          candidate.finishReason === "END_TURN"
-        )
-          finishReason = "stop";
-        else if (candidate.finishReason === "MAX_TOKENS")
-          finishReason = "length";
-        else if (candidate.finishReason === "SAFETY")
-          finishReason = "content_filter";
-
-        const hasToolCalls = candidate.content?.parts?.some(
-          (p: any) => p.functionCall
-        );
-        if (hasToolCalls) finishReason = "tool_calls";
-
-        yield {
-          type: "finish",
-          finishReason,
-          usage: chunk.usageMetadata
-            ? {
-                promptTokens: chunk.usageMetadata.promptTokenCount ?? 0,
-                completionTokens:
-                  chunk.usageMetadata.candidatesTokenCount ?? 0,
-                totalTokens: chunk.usageMetadata.totalTokenCount ?? 0,
-              }
-            : undefined,
-        };
-      }
-    }
-  }
-
-  // ── Message conversion (identical to GoogleProvider) ─────────────────────
-
-  private toGoogleMessages(messages: ChatMessage[]): {
-    systemInstruction: string | undefined;
-    contents: unknown[];
-  } {
-    let systemInstruction: string | undefined;
-    const contents: unknown[] = [];
-
-    for (const msg of messages) {
-      if (msg.role === "system") {
-        systemInstruction = getTextContent(msg.content) || undefined;
-        continue;
-      }
-
-      if (msg.role === "user") {
-        if (isMultiModal(msg.content)) {
-          contents.push({
-            role: "user",
-            parts: msg.content.map((p) => this.partToGoogle(p)),
-          });
-        } else {
-          contents.push({
-            role: "user",
-            parts: [{ text: msg.content ?? "" }],
-          });
-        }
-        continue;
-      }
-
-      if (msg.role === "assistant") {
-        const parts: unknown[] = [];
-        if (msg.content) parts.push({ text: msg.content });
-        if (msg.toolCalls) {
-          for (const tc of msg.toolCalls) {
-            parts.push({
-              functionCall: { name: tc.name, args: tc.arguments },
-            });
-          }
-        }
-        if (parts.length === 0) parts.push({ text: "" });
-        contents.push({ role: "model", parts });
-        continue;
-      }
-
-      if (msg.role === "tool") {
-        contents.push({
-          role: "function",
-          parts: [
-            {
-              functionResponse: {
-                name: msg.name ?? "unknown",
-                response: { result: msg.content ?? "" },
-              },
-            },
-          ],
-        });
-        continue;
-      }
-    }
-
-    return { systemInstruction, contents };
-  }
-
-  private partToGoogle(part: ContentPart): unknown {
-    switch (part.type) {
-      case "text":
-        return { text: part.text };
-      case "image":
-      case "audio":
-      case "file": {
-        const isUrl =
-          part.data.startsWith("http://") || part.data.startsWith("https://");
-        if (isUrl) {
-          return {
-            fileData: {
-              fileUri: part.data,
-              mimeType:
-                part.mimeType ??
-                (part.type === "image"
-                  ? "image/png"
-                  : "application/octet-stream"),
-            },
-          };
-        }
-        return {
-          inlineData: {
-            data: part.data,
-            mimeType:
-              part.mimeType ??
-              (part.type === "image"
-                ? "image/png"
-                : part.type === "audio"
-                  ? "audio/mp3"
-                  : "application/octet-stream"),
-          },
-        };
-      }
-    }
-  }
-
-  private toGoogleTools(tools: ToolDefinition[]): unknown[] {
-    return tools.map((t) => ({
-      name: t.name,
-      description: t.description,
-      parameters: t.parameters,
-    }));
-  }
-
-  private cleanJsonSchema(
-    schema: Record<string, unknown>
-  ): Record<string, unknown> {
-    const cleaned = { ...schema };
-    delete cleaned["$schema"];
-    delete cleaned["$ref"];
-    delete cleaned["additionalProperties"];
-
-    if (cleaned.properties && typeof cleaned.properties === "object") {
-      const props: Record<string, unknown> = {};
-      for (const [key, val] of Object.entries(
-        cleaned.properties as Record<string, unknown>
-      )) {
-        props[key] =
-          typeof val === "object" && val
-            ? this.cleanJsonSchema(val as Record<string, unknown>)
-            : val;
-      }
-      cleaned.properties = props;
-    }
-
-    if (cleaned.items && typeof cleaned.items === "object") {
-      cleaned.items = this.cleanJsonSchema(
-        cleaned.items as Record<string, unknown>
-      );
-    }
-
-    return cleaned;
-  }
-
-  private normalizeResponse(response: any): ModelResponse & { thinking?: string } {
-    const candidate = response.candidates?.[0];
-    const parts = candidate?.content?.parts ?? [];
-
-    let textContent = "";
-    let thinkingContent = "";
-    const toolCalls: ToolCall[] = [];
-    let toolCallCounter = 0;
-
-    for (const part of parts) {
-      if (part.thought && part.text) {
-        thinkingContent += part.text;
-      } else if (part.text) {
-        textContent += part.text;
-      }
-      if (part.functionCall) {
-        toolCalls.push({
-          id: `vertex_tc_${toolCallCounter++}`,
-          name: part.functionCall.name,
-          arguments: part.functionCall.args ?? {},
-        });
-      }
-    }
-
-    const thinkingTokens = response.usageMetadata?.thoughtsTokenCount ?? 0;
-    const usage: TokenUsage = {
-      promptTokens: response.usageMetadata?.promptTokenCount ?? 0,
-      completionTokens: response.usageMetadata?.candidatesTokenCount ?? 0,
-      totalTokens: response.usageMetadata?.totalTokenCount ?? 0,
-      ...(thinkingTokens > 0 ? { reasoningTokens: thinkingTokens } : {}),
-    };
-
-    let finishReason: ModelResponse["finishReason"] = "stop";
-    if (toolCalls.length > 0) finishReason = "tool_calls";
-    else if (candidate?.finishReason === "MAX_TOKENS")
-      finishReason = "length";
-    else if (candidate?.finishReason === "SAFETY")
-      finishReason = "content_filter";
-
-    const result: ModelResponse & { thinking?: string } = {
-      message: {
-        role: "assistant",
-        content: textContent || null,
-        toolCalls: toolCalls.length > 0 ? toolCalls : undefined,
-      },
-      usage,
-      finishReason,
-      raw: response,
-    };
-
-    if (thinkingContent) {
-      result.thinking = thinkingContent;
-    }
-
-    return result;
-  }
-}

package/src/models/registry.ts

@@ -1,107 +0,0 @@
-import type { ModelProvider } from "./provider.js";
-import { OpenAIProvider } from "./providers/openai.js";
-import { AnthropicProvider } from "./providers/anthropic.js";
-import { GoogleProvider } from "./providers/google.js";
-import { OllamaProvider } from "./providers/ollama.js";
-import { VertexAIProvider } from "./providers/vertex.js";
-
-type ProviderFactory = (
-  modelId: string,
-  config?: Record<string, unknown>
-) => ModelProvider;
-
-export class ModelRegistry {
-  private factories = new Map<string, ProviderFactory>();
-
-  register(providerId: string, factory: ProviderFactory): void {
-    this.factories.set(providerId, factory);
-  }
-
-  resolve(
-    providerId: string,
-    modelId: string,
-    config?: Record<string, unknown>
-  ): ModelProvider {
-    const factory = this.factories.get(providerId);
-    if (!factory) {
-      throw new Error(
-        `Unknown provider "${providerId}". Register it first with registry.register().`
-      );
-    }
-    return factory(modelId, config);
-  }
-
-  has(providerId: string): boolean {
-    return this.factories.has(providerId);
-  }
-}
-
-export const registry = new ModelRegistry();
-
-registry.register(
-  "openai",
-  (modelId, config) =>
-    new OpenAIProvider(modelId, config as { apiKey?: string; baseURL?: string })
-);
-
-registry.register(
-  "anthropic",
-  (modelId, config) =>
-    new AnthropicProvider(modelId, config as { apiKey?: string })
-);
-
-registry.register(
-  "google",
-  (modelId, config) =>
-    new GoogleProvider(modelId, config as { apiKey?: string })
-);
-
-registry.register(
-  "ollama",
-  (modelId, config) =>
-    new OllamaProvider(modelId, config as { host?: string })
-);
-
-export function openai(
-  modelId: string,
-  config?: { apiKey?: string; baseURL?: string }
-): ModelProvider {
-  return registry.resolve("openai", modelId, config);
-}
-
-export function anthropic(
-  modelId: string,
-  config?: { apiKey?: string }
-): ModelProvider {
-  return registry.resolve("anthropic", modelId, config);
-}
-
-export function google(
-  modelId: string,
-  config?: { apiKey?: string }
-): ModelProvider {
-  return registry.resolve("google", modelId, config);
-}
-
-export function ollama(
-  modelId: string,
-  config?: { host?: string }
-): ModelProvider {
-  return registry.resolve("ollama", modelId, config);
-}
-
-registry.register(
-  "vertex",
-  (modelId, config) =>
-    new VertexAIProvider(
-      modelId,
-      config as { project?: string; location?: string; credentials?: string }
-    )
-);
-
-export function vertex(
-  modelId: string,
-  config?: { project?: string; location?: string; credentials?: string }
-): ModelProvider {
-  return registry.resolve("vertex", modelId, config);
-}

package/src/models/types.ts

@@ -1,124 +0,0 @@
-export type MessageRole = "system" | "user" | "assistant" | "tool";
-
-// ── Multi-modal content parts ─────────────────────────────────────────────
-
-export interface TextPart {
-  type: "text";
-  text: string;
-}
-
-export interface ImagePart {
-  type: "image";
-  /** Base64-encoded image data OR a URL. */
-  data: string;
-  mimeType?: "image/png" | "image/jpeg" | "image/gif" | "image/webp";
-}
-
-export interface AudioPart {
-  type: "audio";
-  /** Base64-encoded audio data. */
-  data: string;
-  mimeType?: "audio/mp3" | "audio/wav" | "audio/ogg" | "audio/webm";
-}
-
-export interface FilePart {
-  type: "file";
-  /** Base64-encoded file data OR a URL. */
-  data: string;
-  mimeType: string;
-  filename?: string;
-}
-
-export type ContentPart = TextPart | ImagePart | AudioPart | FilePart;
-
-/** Convenience: plain string, or an array of multi-modal content parts. */
-export type MessageContent = string | ContentPart[];
-
-// ── Chat message ──────────────────────────────────────────────────────────
-
-export interface ChatMessage {
-  role: MessageRole;
-  content: MessageContent | null;
-  toolCalls?: ToolCall[];
-  toolCallId?: string;
-  name?: string;
-}
-
-// ── Tool definitions ──────────────────────────────────────────────────────
-
-export interface ToolCall {
-  id: string;
-  name: string;
-  arguments: Record<string, unknown>;
-}
-
-export interface ToolDefinition {
-  name: string;
-  description: string;
-  parameters: Record<string, unknown>;
-}
-
-// ── Token usage ───────────────────────────────────────────────────────────
-
-export interface TokenUsage {
-  promptTokens: number;
-  completionTokens: number;
-  totalTokens: number;
-  reasoningTokens?: number;
-}
-
-// ── Model response ────────────────────────────────────────────────────────
-
-export interface ModelResponse {
-  message: ChatMessage;
-  usage: TokenUsage;
-  finishReason: "stop" | "tool_calls" | "length" | "content_filter";
-  raw: unknown;
-}
-
-export type StreamChunk =
-  | { type: "text"; text: string }
-  | { type: "thinking"; text: string }
-  | { type: "tool_call_start"; toolCall: { id: string; name: string } }
-  | { type: "tool_call_delta"; toolCallId: string; argumentsDelta: string }
-  | { type: "tool_call_end"; toolCallId: string }
-  | { type: "finish"; finishReason: string; usage?: TokenUsage };
-
-// ── Model config ──────────────────────────────────────────────────────────
-
-export interface ReasoningConfig {
-  enabled: boolean;
-  /** Reasoning effort for OpenAI o-series models. */
-  effort?: "low" | "medium" | "high";
-  /** Token budget for thinking (Anthropic / Gemini). */
-  budgetTokens?: number;
-}
-
-export interface ModelConfig {
-  temperature?: number;
-  maxTokens?: number;
-  topP?: number;
-  stop?: string[];
-  responseFormat?: "text" | "json" | { type: "json_schema"; schema: Record<string, unknown>; name?: string };
-  /** Per-request API key override. When provided, the provider uses this key instead of the one set at construction. */
-  apiKey?: string;
-  /** Enable extended thinking / reasoning. */
-  reasoning?: ReasoningConfig;
-}
-
-// ── Helpers ───────────────────────────────────────────────────────────────
-
-/** Extract the text content from a MessageContent value. */
-export function getTextContent(content: MessageContent | null): string {
-  if (content === null) return "";
-  if (typeof content === "string") return content;
-  return content
-    .filter((p): p is TextPart => p.type === "text")
-    .map((p) => p.text)
-    .join("");
-}
-
-/** Check if content has multi-modal parts. */
-export function isMultiModal(content: MessageContent | null): content is ContentPart[] {
-  return Array.isArray(content);
-}