@providerprotocol/ai 0.0.11 → 0.0.12

package/src/providers/google/llm.ts

@@ -1,203 +0,0 @@
-import type { LLMHandler, BoundLLMModel, LLMRequest, LLMResponse, LLMStreamResult, LLMCapabilities } from '../../types/llm.ts';
-import type { StreamEvent } from '../../types/stream.ts';
-import type { LLMProvider } from '../../types/provider.ts';
-import { UPPError } from '../../types/errors.ts';
-import { resolveApiKey } from '../../http/keys.ts';
-import { doFetch, doStreamFetch } from '../../http/fetch.ts';
-import { parseSSEStream } from '../../http/sse.ts';
-import { normalizeHttpError } from '../../http/errors.ts';
-import type { GoogleLLMParams, GoogleResponse, GoogleStreamChunk } from './types.ts';
-import {
-  transformRequest,
-  transformResponse,
-  transformStreamChunk,
-  createStreamState,
-  buildResponseFromState,
-} from './transform.ts';
-
-const GOOGLE_API_BASE = 'https://generativelanguage.googleapis.com/v1beta';
-
-/**
- * Google API capabilities
- */
-const GOOGLE_CAPABILITIES: LLMCapabilities = {
-  streaming: true,
-  tools: true,
-  structuredOutput: true,
-  imageInput: true,
-  videoInput: true,
-  audioInput: true,
-};
-
-/**
- * Build Google API URL for a model
- */
-function buildUrl(modelId: string, action: 'generateContent' | 'streamGenerateContent', apiKey: string): string {
-  const base = `${GOOGLE_API_BASE}/models/${modelId}:${action}`;
-  return `${base}?key=${apiKey}`;
-}
-
-/**
- * Create Google LLM handler
- */
-export function createLLMHandler(): LLMHandler<GoogleLLMParams> {
-  // Provider reference injected by createProvider() after construction
-  let providerRef: LLMProvider<GoogleLLMParams> | null = null;
-
-  return {
-    _setProvider(provider: LLMProvider<GoogleLLMParams>) {
-      providerRef = provider;
-    },
-
-    bind(modelId: string): BoundLLMModel<GoogleLLMParams> {
-      // Use the injected provider reference (set by createProvider)
-      if (!providerRef) {
-        throw new UPPError(
-          'Provider reference not set. Handler must be used with createProvider().',
-          'INVALID_REQUEST',
-          'google',
-          'llm'
-        );
-      }
-
-      const model: BoundLLMModel<GoogleLLMParams> = {
-        modelId,
-        capabilities: GOOGLE_CAPABILITIES,
-
-        get provider(): LLMProvider<GoogleLLMParams> {
-          return providerRef!;
-        },
-
-        async complete(request: LLMRequest<GoogleLLMParams>): Promise<LLMResponse> {
-          const apiKey = await resolveApiKey(
-            request.config,
-            'GOOGLE_API_KEY',
-            'google',
-            'llm'
-          );
-
-          const url = request.config.baseUrl
-            ? `${request.config.baseUrl}/models/${modelId}:generateContent?key=${apiKey}`
-            : buildUrl(modelId, 'generateContent', apiKey);
-
-          const body = transformRequest(request, modelId);
-
-          const response = await doFetch(
-            url,
-            {
-              method: 'POST',
-              headers: {
-                'Content-Type': 'application/json',
-              },
-              body: JSON.stringify(body),
-              signal: request.signal,
-            },
-            request.config,
-            'google',
-            'llm'
-          );
-
-          const data = (await response.json()) as GoogleResponse;
-          return transformResponse(data);
-        },
-
-        stream(request: LLMRequest<GoogleLLMParams>): LLMStreamResult {
-          const state = createStreamState();
-          let responseResolve: (value: LLMResponse) => void;
-          let responseReject: (error: Error) => void;
-
-          const responsePromise = new Promise<LLMResponse>((resolve, reject) => {
-            responseResolve = resolve;
-            responseReject = reject;
-          });
-
-          async function* generateEvents(): AsyncGenerator<StreamEvent, void, unknown> {
-            try {
-              const apiKey = await resolveApiKey(
-                request.config,
-                'GOOGLE_API_KEY',
-                'google',
-                'llm'
-              );
-
-              const url = request.config.baseUrl
-                ? `${request.config.baseUrl}/models/${modelId}:streamGenerateContent?alt=sse&key=${apiKey}`
-                : `${buildUrl(modelId, 'streamGenerateContent', apiKey)}&alt=sse`;
-
-              const body = transformRequest(request, modelId);
-
-              const response = await doStreamFetch(
-                url,
-                {
-                  method: 'POST',
-                  headers: {
-                    'Content-Type': 'application/json',
-                  },
-                  body: JSON.stringify(body),
-                  signal: request.signal,
-                },
-                request.config,
-                'google',
-                'llm'
-              );
-
-              if (!response.ok) {
-                const error = await normalizeHttpError(response, 'google', 'llm');
-                responseReject(error);
-                throw error;
-              }
-
-              if (!response.body) {
-                const error = new UPPError(
-                  'No response body for streaming request',
-                  'PROVIDER_ERROR',
-                  'google',
-                  'llm'
-                );
-                responseReject(error);
-                throw error;
-              }
-
-              for await (const data of parseSSEStream(response.body)) {
-                if (typeof data === 'object' && data !== null) {
-                  const chunk = data as GoogleStreamChunk;
-
-                  // Check for error
-                  if ('error' in chunk) {
-                    const error = new UPPError(
-                      (chunk as any).error.message,
-                      'PROVIDER_ERROR',
-                      'google',
-                      'llm'
-                    );
-                    responseReject(error);
-                    throw error;
-                  }
-
-                  const events = transformStreamChunk(chunk, state);
-                  for (const event of events) {
-                    yield event;
-                  }
-                }
-              }
-
-              responseResolve(buildResponseFromState(state));
-            } catch (error) {
-              responseReject(error as Error);
-              throw error;
-            }
-          }
-
-          return {
-            [Symbol.asyncIterator]() {
-              return generateEvents();
-            },
-            response: responsePromise,
-          };
-        },
-      };
-
-      return model;
-    },
-  };
-}

package/src/providers/google/transform.ts

@@ -1,447 +0,0 @@
-import type { LLMRequest, LLMResponse } from '../../types/llm.ts';
-import type { Message } from '../../types/messages.ts';
-import type { StreamEvent } from '../../types/stream.ts';
-import type { Tool, ToolCall } from '../../types/tool.ts';
-import type { TokenUsage } from '../../types/turn.ts';
-import type { ContentBlock, TextBlock, ImageBlock } from '../../types/content.ts';
-import {
-  AssistantMessage,
-  isUserMessage,
-  isAssistantMessage,
-  isToolResultMessage,
-} from '../../types/messages.ts';
-import type {
-  GoogleLLMParams,
-  GoogleRequest,
-  GoogleContent,
-  GooglePart,
-  GoogleTool,
-  GoogleResponse,
-  GoogleStreamChunk,
-  GoogleFunctionCallPart,
-} from './types.ts';
-
-/**
- * Transform UPP request to Google format
- *
- * Params are spread into generationConfig to allow pass-through of any Google API fields,
- * even those not explicitly defined in our type. This enables developers to
- * use new API features without waiting for library updates.
- */
-export function transformRequest<TParams extends GoogleLLMParams>(
-  request: LLMRequest<TParams>,
-  modelId: string
-): GoogleRequest {
-  const params = (request.params ?? {}) as GoogleLLMParams;
-
-  const googleRequest: GoogleRequest = {
-    contents: transformMessages(request.messages),
-  };
-
-  // System instruction (separate from contents in Google)
-  if (request.system) {
-    googleRequest.systemInstruction = {
-      parts: [{ text: request.system }],
-    };
-  }
-
-  // Spread params into generationConfig to pass through all fields
-  const generationConfig: NonNullable<GoogleRequest['generationConfig']> = {
-    ...params,
-  };
-
-  // Protocol-level structured output (overrides provider-specific settings)
-  if (request.structure) {
-    generationConfig.responseMimeType = 'application/json';
-    generationConfig.responseSchema = request.structure as unknown as Record<string, unknown>;
-  }
-
-  if (Object.keys(generationConfig).length > 0) {
-    googleRequest.generationConfig = generationConfig;
-  }
-
-  // Tools come from request, not params
-  if (request.tools && request.tools.length > 0) {
-    googleRequest.tools = [
-      {
-        functionDeclarations: request.tools.map(transformTool),
-      },
-    ];
-  }
-
-  return googleRequest;
-}
-
-/**
- * Filter to only valid content blocks with a type property
- */
-function filterValidContent<T extends { type?: string }>(content: T[]): T[] {
-  return content.filter((c) => c && typeof c.type === 'string');
-}
-
-/**
- * Transform UPP Messages to Google contents
- */
-function transformMessages(messages: Message[]): GoogleContent[] {
-  const contents: GoogleContent[] = [];
-
-  for (const msg of messages) {
-    if (isUserMessage(msg)) {
-      const validContent = filterValidContent(msg.content);
-      const parts = validContent.map(transformContentBlock);
-      // Google requires at least one part - add placeholder if empty
-      if (parts.length === 0) {
-        parts.push({ text: '' });
-      }
-      contents.push({
-        role: 'user',
-        parts,
-      });
-    } else if (isAssistantMessage(msg)) {
-      const validContent = filterValidContent(msg.content);
-      const parts: GooglePart[] = validContent.map(transformContentBlock);
-
-      // Add function calls - use stored parts with thought signatures if available
-      const googleMeta = msg.metadata?.google as {
-        functionCallParts?: Array<{
-          name: string;
-          args: Record<string, unknown>;
-          thoughtSignature?: string;
-        }>;
-      } | undefined;
-
-      if (googleMeta?.functionCallParts && googleMeta.functionCallParts.length > 0) {
-        // Use stored function call parts with thought signatures
-        for (const fc of googleMeta.functionCallParts) {
-          const part: GoogleFunctionCallPart = {
-            functionCall: {
-              name: fc.name,
-              args: fc.args,
-            },
-          };
-          if (fc.thoughtSignature) {
-            part.thoughtSignature = fc.thoughtSignature;
-          }
-          parts.push(part);
-        }
-      } else if (msg.toolCalls) {
-        // Fallback: reconstruct from tool calls (no thought signatures)
-        for (const call of msg.toolCalls) {
-          parts.push({
-            functionCall: {
-              name: call.toolName,
-              args: call.arguments,
-            },
-          });
-        }
-      }
-
-      // Google requires at least one part - add placeholder if empty
-      if (parts.length === 0) {
-        parts.push({ text: '' });
-      }
-
-      contents.push({
-        role: 'model',
-        parts,
-      });
-    } else if (isToolResultMessage(msg)) {
-      // Function results are sent as user messages in Google
-      contents.push({
-        role: 'user',
-        parts: msg.results.map((result) => ({
-          functionResponse: {
-            name: result.toolCallId, // Google uses the function name, but we store it in toolCallId
-            response:
-              typeof result.result === 'object'
-                ? (result.result as Record<string, unknown>)
-                : { result: result.result },
-          },
-        })),
-      });
-    }
-  }
-
-  return contents;
-}
-
-/**
- * Transform a content block to Google format
- */
-function transformContentBlock(block: ContentBlock): GooglePart {
-  switch (block.type) {
-    case 'text':
-      return { text: block.text };
-
-    case 'image': {
-      const imageBlock = block as ImageBlock;
-      let data: string;
-
-      if (imageBlock.source.type === 'base64') {
-        data = imageBlock.source.data;
-      } else if (imageBlock.source.type === 'bytes') {
-        data = btoa(
-          Array.from(imageBlock.source.data)
-            .map((b) => String.fromCharCode(b))
-            .join('')
-        );
-      } else {
-        throw new Error('Google API does not support URL image sources directly');
-      }
-
-      return {
-        inlineData: {
-          mimeType: imageBlock.mimeType,
-          data,
-        },
-      };
-    }
-
-    default:
-      throw new Error(`Unsupported content type: ${block.type}`);
-  }
-}
-
-/**
- * Transform a UPP Tool to Google format
- */
-function transformTool(tool: Tool): GoogleTool['functionDeclarations'][0] {
-  return {
-    name: tool.name,
-    description: tool.description,
-    parameters: {
-      type: 'object',
-      properties: tool.parameters.properties,
-      required: tool.parameters.required,
-    },
-  };
-}
-
-/**
- * Transform Google response to UPP LLMResponse
- */
-export function transformResponse(data: GoogleResponse): LLMResponse {
-  const candidate = data.candidates?.[0];
-  if (!candidate) {
-    throw new Error('No candidates in Google response');
-  }
-
-  const textContent: TextBlock[] = [];
-  const toolCalls: ToolCall[] = [];
-  let structuredData: unknown;
-  // Store original function call parts with thought signatures for echoing back
-  const functionCallParts: Array<{
-    name: string;
-    args: Record<string, unknown>;
-    thoughtSignature?: string;
-  }> = [];
-
-  for (const part of candidate.content.parts) {
-    if ('text' in part) {
-      textContent.push({ type: 'text', text: part.text });
-      // Try to parse as JSON for structured output (native JSON mode)
-      if (structuredData === undefined) {
-        try {
-          structuredData = JSON.parse(part.text);
-        } catch {
-          // Not valid JSON - that's fine, might not be structured output
-        }
-      }
-    } else if ('functionCall' in part) {
-      const fc = part as GoogleFunctionCallPart;
-      toolCalls.push({
-        toolCallId: fc.functionCall.name, // Google doesn't have call IDs, use name
-        toolName: fc.functionCall.name,
-        arguments: fc.functionCall.args,
-      });
-      // Store the full part including thought signature
-      functionCallParts.push({
-        name: fc.functionCall.name,
-        args: fc.functionCall.args,
-        thoughtSignature: fc.thoughtSignature,
-      });
-    }
-  }
-
-  const message = new AssistantMessage(
-    textContent,
-    toolCalls.length > 0 ? toolCalls : undefined,
-    {
-      metadata: {
-        google: {
-          finishReason: candidate.finishReason,
-          safetyRatings: candidate.safetyRatings,
-          // Store function call parts with thought signatures for multi-turn
-          functionCallParts: functionCallParts.length > 0 ? functionCallParts : undefined,
-        },
-      },
-    }
-  );
-
-  const usage: TokenUsage = {
-    inputTokens: data.usageMetadata?.promptTokenCount ?? 0,
-    outputTokens: data.usageMetadata?.candidatesTokenCount ?? 0,
-    totalTokens: data.usageMetadata?.totalTokenCount ?? 0,
-  };
-
-  return {
-    message,
-    usage,
-    stopReason: candidate.finishReason ?? 'STOP',
-    data: structuredData,
-  };
-}
-
-/**
- * State for accumulating streaming response
- */
-export interface StreamState {
-  content: string;
-  toolCalls: Array<{ name: string; args: Record<string, unknown>; thoughtSignature?: string }>;
-  finishReason: string | null;
-  inputTokens: number;
-  outputTokens: number;
-  isFirstChunk: boolean;
-}
-
-/**
- * Create initial stream state
- */
-export function createStreamState(): StreamState {
-  return {
-    content: '',
-    toolCalls: [],
-    finishReason: null,
-    inputTokens: 0,
-    outputTokens: 0,
-    isFirstChunk: true,
-  };
-}
-
-/**
- * Transform Google stream chunk to UPP StreamEvents
- */
-export function transformStreamChunk(
-  chunk: GoogleStreamChunk,
-  state: StreamState
-): StreamEvent[] {
-  const events: StreamEvent[] = [];
-
-  // First chunk - emit message start
-  if (state.isFirstChunk) {
-    events.push({ type: 'message_start', index: 0, delta: {} });
-    state.isFirstChunk = false;
-  }
-
-  // Usage metadata
-  if (chunk.usageMetadata) {
-    state.inputTokens = chunk.usageMetadata.promptTokenCount;
-    state.outputTokens = chunk.usageMetadata.candidatesTokenCount;
-  }
-
-  const candidate = chunk.candidates?.[0];
-  if (!candidate) {
-    return events;
-  }
-
-  // Process parts
-  for (const part of candidate.content?.parts ?? []) {
-    if ('text' in part) {
-      state.content += part.text;
-      events.push({
-        type: 'text_delta',
-        index: 0,
-        delta: { text: part.text },
-      });
-    } else if ('functionCall' in part) {
-      const fc = part as GoogleFunctionCallPart;
-      // Store with thought signature for echoing back
-      state.toolCalls.push({
-        name: fc.functionCall.name,
-        args: fc.functionCall.args,
-        thoughtSignature: fc.thoughtSignature,
-      });
-      events.push({
-        type: 'tool_call_delta',
-        index: state.toolCalls.length - 1,
-        delta: {
-          toolCallId: fc.functionCall.name,
-          toolName: fc.functionCall.name,
-          argumentsJson: JSON.stringify(fc.functionCall.args),
-        },
-      });
-    }
-  }
-
-  // Finish reason
-  if (candidate.finishReason) {
-    state.finishReason = candidate.finishReason;
-    events.push({ type: 'message_stop', index: 0, delta: {} });
-  }
-
-  return events;
-}
-
-/**
- * Build LLMResponse from accumulated stream state
- */
-export function buildResponseFromState(state: StreamState): LLMResponse {
-  const textContent: TextBlock[] = [];
-  const toolCalls: ToolCall[] = [];
-  let structuredData: unknown;
-  const functionCallParts: Array<{
-    name: string;
-    args: Record<string, unknown>;
-    thoughtSignature?: string;
-  }> = [];
-
-  if (state.content) {
-    textContent.push({ type: 'text', text: state.content });
-    // Try to parse as JSON for structured output (native JSON mode)
-    try {
-      structuredData = JSON.parse(state.content);
-    } catch {
-      // Not valid JSON - that's fine, might not be structured output
-    }
-  }
-
-  for (const tc of state.toolCalls) {
-    toolCalls.push({
-      toolCallId: tc.name,
-      toolName: tc.name,
-      arguments: tc.args,
-    });
-    functionCallParts.push({
-      name: tc.name,
-      args: tc.args,
-      thoughtSignature: tc.thoughtSignature,
-    });
-  }
-
-  const message = new AssistantMessage(
-    textContent,
-    toolCalls.length > 0 ? toolCalls : undefined,
-    {
-      metadata: {
-        google: {
-          finishReason: state.finishReason,
-          // Store function call parts with thought signatures for multi-turn
-          functionCallParts: functionCallParts.length > 0 ? functionCallParts : undefined,
-        },
-      },
-    }
-  );
-
-  const usage: TokenUsage = {
-    inputTokens: state.inputTokens,
-    outputTokens: state.outputTokens,
-    totalTokens: state.inputTokens + state.outputTokens,
-  };
-
-  return {
-    message,
-    usage,
-    stopReason: state.finishReason ?? 'STOP',
-    data: structuredData,
-  };
-}