drizzle-cube 0.4.19 → 0.4.21

package/dist/server/index.d.ts

@@ -25,13 +25,28 @@ export declare interface AgentChatRequest {
 
 /**
  * Agent configuration options (provided in adapter options).
- * Uses `@anthropic-ai/sdk` (the standard Anthropic API client) for direct
- * HTTP API calls — no subprocess spawning, fully edge-runtime compatible.
+ * Supports multiple LLM providers — defaults to Anthropic for backward compatibility.
+ * All providers use direct HTTP API calls — no subprocess spawning, fully edge-runtime compatible.
  */
 export declare interface AgentConfig {
-    /** Server-side Anthropic API key */
+    /**
+     * LLM provider to use (default: 'anthropic').
+     * - 'anthropic': Claude models via @anthropic-ai/sdk
+     * - 'openai': OpenAI models via openai SDK (also supports Groq, Together, Mistral, Ollama via baseURL)
+     * - 'google': Gemini models via @google/generative-ai
+     */
+    provider?: 'anthropic' | 'openai' | 'google';
+    /**
+     * Base URL for OpenAI-compatible providers.
+     * Only used when provider is 'openai'. Examples:
+     * - Groq: 'https://api.groq.com/openai/v1'
+     * - Together: 'https://api.together.xyz/v1'
+     * - Ollama: 'http://localhost:11434/v1'
+     */
+    baseURL?: string;
+    /** Server-side API key for the selected provider */
     apiKey?: string;
-    /** Model to use (default: 'claude-sonnet-4-6') */
+    /** Model to use (default depends on provider: 'claude-sonnet-4-6' for Anthropic, 'gpt-4.1-mini' for OpenAI, 'gemini-3-flash-preview' for Google) */
     model?: string;
     /** Maximum agentic turns per request (default: 25) */
     maxTurns?: number;
@@ -41,6 +56,16 @@ export declare interface AgentConfig {
     allowClientApiKey?: boolean;
     /** Optional observability hooks for tracing */
     observability?: AgentObservabilityHooks;
+    /**
+     * Build per-request system context for the LLM prompt from the authenticated security context.
+     * Called on every agent chat request with the resolved security context.
+     * The returned string is appended to the system prompt under "## User Context".
+     *
+     * @example
+     * buildSystemContext: (securityContext) =>
+     *   `User: ${securityContext.userName}, Role: ${securityContext.role}`
+     */
+    buildSystemContext?: (securityContext: Record<string, unknown>) => string | undefined;
 }
 
 /**
@@ -76,7 +101,7 @@ export declare interface AgentObservabilityHooks {
         model: string;
         historyLength: number;
     }) => void;
-    /** Called after each Anthropic API call completes */
+    /** Called after each LLM API call completes */
     onGenerationEnd?: (event: {
         traceId: string;
         turn: number;
@@ -85,6 +110,10 @@ export declare interface AgentObservabilityHooks {
         inputTokens?: number;
         outputTokens?: number;
         durationMs: number;
+        /** Messages array sent to the API */
+        input?: unknown;
+        /** Assistant content blocks from the response */
+        output?: unknown;
     }) => void;
     /** Called after each tool execution */
     onToolEnd?: (event: {
@@ -2227,10 +2256,9 @@ export declare function getToolDefinitions(): ToolDefinition[];
 /**
  * Handle an agent chat request, yielding SSE events as the agent works.
  *
- * Uses the Anthropic Messages API (`@anthropic-ai/sdk`) with `stream: true`
- * (raw async iterable) for Cloudflare Workers / edge-runtime compatibility.
+ * Uses the configured LLM provider with streaming for edge-runtime compatibility.
  * Implements a manual agentic loop: send messages → stream response → execute
- * tool calls → append results → repeat until stop_reason !== 'tool_use'.
+ * tool calls → append results → repeat until the model stops requesting tools.
  */
 export declare function handleAgentChat(options: {
     message: string;
@@ -2240,6 +2268,12 @@ export declare function handleAgentChat(options: {
     securityContext: SecurityContext;
     agentConfig: AgentConfig;
     apiKey: string;
+    /** Per-request context appended to the system prompt (e.g. user info, tenant context) */
+    systemContext?: string;
+    /** Per-request overrides from client headers (take precedence over agentConfig) */
+    providerOverride?: string;
+    modelOverride?: string;
+    baseURLOverride?: string;
 }): AsyncGenerator<AgentSSEEvent>;
 
 /**
@@ -4814,12 +4848,17 @@ declare interface TimeDimensionRef {
 export declare type TimeGranularity = 'second' | 'minute' | 'hour' | 'day' | 'week' | 'month' | 'quarter' | 'year';
 
 /**
- * Anthropic API tool definition (JSON schema format)
+ * LLM Provider Abstraction Types
+ * Provider-agnostic interfaces for multi-provider LLM support in the agentic notebook.
+ */
+/**
+ * Generic tool definition (provider-agnostic).
+ * Each provider's formatTools() wraps this in its own format.
  */
 declare interface ToolDefinition {
     name: string;
     description: string;
-    input_schema: {
+    parameters: {
         type: 'object';
         properties: Record<string, unknown>;
         required?: string[];