@firstlovecenter/ai-chat 0.2.2 → 0.5.0

package/dist/server/index.d.cts

@@ -1,7 +1,8 @@
-import { S as SystemBlock, T as ToolSchema, a as ToolContext, b as ToolDefinition, P as PresentPayload, c as PersistencePort, A as AuthPort, d as ScopePort, e as ToolsPort, V as VertexPort, L as LoggerPort } from '../types-DNwFvL-C.cjs';
-export { f as AiSettings, g as AiSettingsPatch, h as AppendMessageInput, i as AuthFail, j as AuthOk, k as AuthResult, B as Block, C as ChartSpec, l as ChatMessage, m as ChatMessageRole, n as ChatSession, o as CreateSessionInput, p as ListSessionsOpts, q as TERMINAL_TOOL_NAME, r as ToolResult, s as err, t as ok } from '../types-DNwFvL-C.cjs';
+import { S as SystemBlock, T as ToolSchema, a as ToolContext, b as ToolDefinition, P as PresentPayload, c as PersistencePort, A as AuthPort, d as ScopePort, e as ToolsPort, V as VertexPort, L as LoggerPort } from '../types-CDKxdzQc.cjs';
+export { f as AiSettings, g as AiSettingsPatch, h as AppendMessageInput, i as AuthFail, j as AuthOk, k as AuthResult, B as Block, C as ChartSpec, l as ChatMessage, m as ChatMessageRole, n as ChatSession, o as CreateSessionInput, p as ListSessionsOpts, q as TERMINAL_TOOL_NAME, r as ToolResult, s as err, t as ok } from '../types-CDKxdzQc.cjs';
 import { GoogleAuth } from 'google-auth-library';
 export { GoogleAuth } from 'google-auth-library';
+import 'zod';
 
 /**
  * Provider-agnostic tool-calling abstraction.
@@ -302,6 +303,34 @@ declare function createAgentCustomRoutes<S>(ctx: AgentCustomRouteCtx<S>): {
     POST: (req: Request) => Promise<Response>;
 };
 
+/**
+ * Streaming-route hooks for the Vercel AI SDK chat.
+ *
+ * Identical to `AgentCustomHooks` — the lifecycle (pre-auth → auth →
+ * post-auth → resolve session → onSessionStart → stream → onSessionEnd)
+ * is the same. We re-alias rather than introducing a new type so consumers
+ * can plug a single hook bag into both routes.
+ */
+type AgentVercelHooks$1<S> = AgentCustomHooks$1<S>;
+type AgentVercelRouteCtx<S> = {
+    persistence: PersistencePort;
+    auth: AuthPort<S>;
+    scope: ScopePort<S>;
+    tools: ToolsPort;
+    vertex: VertexPort;
+    logger?: LoggerPort;
+    /**
+     * Optional lifecycle hooks. See `AgentCustomHooks` for the available
+     * extension points (shutdown gating, rate limiting, per-request
+     * resource setup/teardown).
+     */
+    hooks?: AgentVercelHooks$1<S>;
+};
+declare function createAgentVercelRoutes<S>(ctx: AgentVercelRouteCtx<S>): {
+    /** Next.js-compatible POST handler. */
+    POST: (req: Request) => Promise<Response>;
+};
+
 /**
  * `chat-sessions` route factory — host-agnostic CRUD for chat sessions.
  *
@@ -501,6 +530,7 @@ type AiChatRuntime<S = unknown> = {
     }) => Promise<AgentResult>;
     routes: {
         agentCustom: ReturnType<typeof createAgentCustomRoutes<S>>;
+        agentVercel: ReturnType<typeof createAgentVercelRoutes<S>>;
         chatSessions: ReturnType<typeof createChatSessionsRoutes<S>>;
         adminSettings: ReturnType<typeof createAdminSettingsRoutes<S>>;
     };
@@ -513,5 +543,12 @@ declare function configureAiChat<S = unknown>(opts: ConfigureAiChatOpts<S>): AiC
 
 type RouteHooks<S> = RouteHooks$1<S>;
 type AgentCustomHooks<S> = AgentCustomHooks$1<S>;
+/**
+ * Alias of `AgentCustomHooks` — the Vercel AI SDK chat reuses the same
+ * lifecycle (pre-auth → auth → post-auth → resolve session → onSessionStart
+ * → stream → onSessionEnd). Re-exported for callers that prefer the more
+ * specific name.
+ */
+type AgentVercelHooks<S> = AgentVercelHooks$1<S>;
 
-export { type AgentCustomHooks, type AgentInput, type AgentResult, type AiChatRuntime, AuthPort, BUILTIN_CHAT_INTERFACE_IDS, type ChatInterfaceRegistryEntry, type ConfigureAiChatOpts, DEFAULT_MAX_OUTPUT_TOKENS, DEFAULT_MAX_TOOL_TURNS, LoggerPort, PersistencePort, PresentPayload, type ProviderInitOpts, type RouteHooks, ScopePort, SystemBlock, ToolContext, ToolDefinition, type ToolProviderDef, ToolSchema, ToolsPort, type TranscriptEntry, VertexPort, configureAiChat, getToolProvider, runAgent, toolProviders };
+export { type AgentCustomHooks, type AgentInput, type AgentResult, type AgentVercelHooks, type AiChatRuntime, AuthPort, BUILTIN_CHAT_INTERFACE_IDS, type ChatInterfaceRegistryEntry, type ConfigureAiChatOpts, DEFAULT_MAX_OUTPUT_TOKENS, DEFAULT_MAX_TOOL_TURNS, LoggerPort, PersistencePort, PresentPayload, type ProviderInitOpts, type RouteHooks, ScopePort, SystemBlock, ToolContext, ToolDefinition, type ToolProviderDef, ToolSchema, ToolsPort, type TranscriptEntry, VertexPort, configureAiChat, getToolProvider, runAgent, toolProviders };

package/dist/server/index.d.ts

@@ -1,7 +1,8 @@
-import { S as SystemBlock, T as ToolSchema, a as ToolContext, b as ToolDefinition, P as PresentPayload, c as PersistencePort, A as AuthPort, d as ScopePort, e as ToolsPort, V as VertexPort, L as LoggerPort } from '../types-DNwFvL-C.js';
-export { f as AiSettings, g as AiSettingsPatch, h as AppendMessageInput, i as AuthFail, j as AuthOk, k as AuthResult, B as Block, C as ChartSpec, l as ChatMessage, m as ChatMessageRole, n as ChatSession, o as CreateSessionInput, p as ListSessionsOpts, q as TERMINAL_TOOL_NAME, r as ToolResult, s as err, t as ok } from '../types-DNwFvL-C.js';
+import { S as SystemBlock, T as ToolSchema, a as ToolContext, b as ToolDefinition, P as PresentPayload, c as PersistencePort, A as AuthPort, d as ScopePort, e as ToolsPort, V as VertexPort, L as LoggerPort } from '../types-CDKxdzQc.js';
+export { f as AiSettings, g as AiSettingsPatch, h as AppendMessageInput, i as AuthFail, j as AuthOk, k as AuthResult, B as Block, C as ChartSpec, l as ChatMessage, m as ChatMessageRole, n as ChatSession, o as CreateSessionInput, p as ListSessionsOpts, q as TERMINAL_TOOL_NAME, r as ToolResult, s as err, t as ok } from '../types-CDKxdzQc.js';
 import { GoogleAuth } from 'google-auth-library';
 export { GoogleAuth } from 'google-auth-library';
+import 'zod';
 
 /**
  * Provider-agnostic tool-calling abstraction.
@@ -302,6 +303,34 @@ declare function createAgentCustomRoutes<S>(ctx: AgentCustomRouteCtx<S>): {
     POST: (req: Request) => Promise<Response>;
 };
 
+/**
+ * Streaming-route hooks for the Vercel AI SDK chat.
+ *
+ * Identical to `AgentCustomHooks` — the lifecycle (pre-auth → auth →
+ * post-auth → resolve session → onSessionStart → stream → onSessionEnd)
+ * is the same. We re-alias rather than introducing a new type so consumers
+ * can plug a single hook bag into both routes.
+ */
+type AgentVercelHooks$1<S> = AgentCustomHooks$1<S>;
+type AgentVercelRouteCtx<S> = {
+    persistence: PersistencePort;
+    auth: AuthPort<S>;
+    scope: ScopePort<S>;
+    tools: ToolsPort;
+    vertex: VertexPort;
+    logger?: LoggerPort;
+    /**
+     * Optional lifecycle hooks. See `AgentCustomHooks` for the available
+     * extension points (shutdown gating, rate limiting, per-request
+     * resource setup/teardown).
+     */
+    hooks?: AgentVercelHooks$1<S>;
+};
+declare function createAgentVercelRoutes<S>(ctx: AgentVercelRouteCtx<S>): {
+    /** Next.js-compatible POST handler. */
+    POST: (req: Request) => Promise<Response>;
+};
+
 /**
  * `chat-sessions` route factory — host-agnostic CRUD for chat sessions.
  *
@@ -501,6 +530,7 @@ type AiChatRuntime<S = unknown> = {
     }) => Promise<AgentResult>;
     routes: {
         agentCustom: ReturnType<typeof createAgentCustomRoutes<S>>;
+        agentVercel: ReturnType<typeof createAgentVercelRoutes<S>>;
         chatSessions: ReturnType<typeof createChatSessionsRoutes<S>>;
         adminSettings: ReturnType<typeof createAdminSettingsRoutes<S>>;
     };
@@ -513,5 +543,12 @@ declare function configureAiChat<S = unknown>(opts: ConfigureAiChatOpts<S>): AiC
 
 type RouteHooks<S> = RouteHooks$1<S>;
 type AgentCustomHooks<S> = AgentCustomHooks$1<S>;
+/**
+ * Alias of `AgentCustomHooks` — the Vercel AI SDK chat reuses the same
+ * lifecycle (pre-auth → auth → post-auth → resolve session → onSessionStart
+ * → stream → onSessionEnd). Re-exported for callers that prefer the more
+ * specific name.
+ */
+type AgentVercelHooks<S> = AgentVercelHooks$1<S>;
 
-export { type AgentCustomHooks, type AgentInput, type AgentResult, type AiChatRuntime, AuthPort, BUILTIN_CHAT_INTERFACE_IDS, type ChatInterfaceRegistryEntry, type ConfigureAiChatOpts, DEFAULT_MAX_OUTPUT_TOKENS, DEFAULT_MAX_TOOL_TURNS, LoggerPort, PersistencePort, PresentPayload, type ProviderInitOpts, type RouteHooks, ScopePort, SystemBlock, ToolContext, ToolDefinition, type ToolProviderDef, ToolSchema, ToolsPort, type TranscriptEntry, VertexPort, configureAiChat, getToolProvider, runAgent, toolProviders };
+export { type AgentCustomHooks, type AgentInput, type AgentResult, type AgentVercelHooks, type AiChatRuntime, AuthPort, BUILTIN_CHAT_INTERFACE_IDS, type ChatInterfaceRegistryEntry, type ConfigureAiChatOpts, DEFAULT_MAX_OUTPUT_TOKENS, DEFAULT_MAX_TOOL_TURNS, LoggerPort, PersistencePort, PresentPayload, type ProviderInitOpts, type RouteHooks, ScopePort, SystemBlock, ToolContext, ToolDefinition, type ToolProviderDef, ToolSchema, ToolsPort, type TranscriptEntry, VertexPort, configureAiChat, getToolProvider, runAgent, toolProviders };

package/dist/server/index.js

@@ -1,5 +1,8 @@
 import { AnthropicVertex } from '@anthropic-ai/vertex-sdk';
 import { randomUUID } from 'crypto';
+import { StreamData, streamText, tool } from 'ai';
+import { createVertex } from '@ai-sdk/google-vertex';
+import { createVertexAnthropic } from '@ai-sdk/google-vertex/anthropic';
 export { GoogleAuth } from 'google-auth-library';
 
 // src/server/tools/types.ts
@@ -70,8 +73,8 @@ async function runAgent(input) {
     const toolResults = [];
     for (const tc of response.toolCalls) {
       transcript.push({ kind: "tool_use", name: tc.name, input: tc.input });
-      const tool = input.tools[tc.name];
-      if (!tool) {
+      const tool2 = input.tools[tc.name];
+      if (!tool2) {
         const errResult = {
           ok: false,
           error: { code: "UNKNOWN_TOOL", message: `Unknown tool: ${tc.name}` }
@@ -85,7 +88,7 @@ async function runAgent(input) {
         });
         continue;
       }
-      const result = await tool.execute(tc.input, {
+      const result = await tool2.execute(tc.input, {
         ...input.ctx,
         toolCallCount
       });
@@ -128,8 +131,46 @@ async function runAgent(input) {
       transcript
     };
   }
+  if (allBlocksEmpty(presentPayload.blocks)) {
+    const lastAssistantText = [...transcript].reverse().find((e) => e.kind === "assistant_text");
+    const text = lastAssistantText?.text.trim();
+    if (text) {
+      const topic = input.question.length > 80 ? input.question.slice(0, 77) + "..." : input.question;
+      presentPayload = {
+        blocks: [
+          {
+            kind: "paragraph_brief",
+            topic,
+            key_facts: [text]
+          }
+        ],
+        raw_numbers: presentPayload.raw_numbers ?? {}
+      };
+    }
+  }
   return { ok: true, structured: presentPayload, toolCallCount, transcript };
 }
+function allBlocksEmpty(blocks) {
+  if (blocks.length === 0) return true;
+  return blocks.every((b) => {
+    if (b.kind === "paragraph_brief") {
+      return b.key_facts.length === 0 || b.key_facts.every((f) => !f.trim());
+    }
+    if (b.kind === "list") {
+      return b.items.length === 0 || b.items.every((i) => !i.trim());
+    }
+    if (b.kind === "callout") {
+      return !b.text.trim();
+    }
+    if (b.kind === "chart") {
+      return b.data.length === 0;
+    }
+    if (b.kind === "table") {
+      return b.rows.length === 0;
+    }
+    return false;
+  });
+}
 var ClaudeToolProvider = class {
   id = "claude";
   client;
@@ -1097,6 +1138,297 @@ data: {}
     }
   };
 }
+function buildVercelTools(tools, ctx, data, onPresent) {
+  const result = {};
+  let toolCallCount = 0;
+  for (const [name, def] of Object.entries(tools)) {
+    if (!def.zodSchema) {
+      throw new Error(
+        `Tool '${name}' has no zodSchema; required for the Vercel AI SDK chat. Add a Zod schema to the tool definition (or remove it from the registry if the host only uses the custom SSE chat).`
+      );
+    }
+    if (name === TERMINAL_TOOL_NAME) {
+      result[name] = tool({
+        description: def.schema.description,
+        // The Zod schema doubles as the runtime parameter validator the SDK
+        // hands the model. We accept whatever Zod shape the host registered;
+        // the SDK uses it to validate the tool-call arguments before dispatch.
+        parameters: def.zodSchema,
+        execute: async (input) => {
+          if (toolCallCount < 2) {
+            return {
+              error: {
+                code: "SELF_VERIFY_REQUIRED",
+                message: "Per FR-8.3 you must run at least one CROSS-CHECK tool call (a different metric, a different period, or a run_sql sanity-check) before present. Make that extra call now, then call present again."
+              }
+            };
+          }
+          const payload = input;
+          for (let i = 0; i < payload.blocks.length; i++) {
+            data.append({
+              type: "block",
+              value: { index: i, ...payload.blocks[i] }
+            });
+          }
+          onPresent(payload);
+          return { ok: true };
+        }
+      });
+      continue;
+    }
+    result[name] = tool({
+      description: def.schema.description,
+      parameters: def.zodSchema,
+      execute: async (input) => {
+        const res = await def.execute(input, { ...ctx, toolCallCount });
+        toolCallCount += 1;
+        if (res.ok) return res.data;
+        return { error: res.error };
+      }
+    });
+  }
+  return result;
+}
+
+// src/server/routes/agent-vercel.ts
+var VALID_MODELS = /* @__PURE__ */ new Set(["claude", "gemini"]);
+function jsonError2(status, code, message) {
+  return new Response(JSON.stringify({ error: { code, message } }), {
+    status,
+    headers: { "Content-Type": "application/json" }
+  });
+}
+function defaultGenerateSessionId2() {
+  return randomUUID().replace(/-/g, "").slice(0, 16);
+}
+function createAgentVercelRoutes(ctx) {
+  const { persistence, auth, scope, tools, vertex, logger, hooks } = ctx;
+  return {
+    /** Next.js-compatible POST handler. */
+    POST: async (req) => {
+      if (hooks?.onRequest) {
+        const short = await hooks.onRequest(req);
+        if (short) return short;
+      }
+      const authResult = await auth.requireAuth(req);
+      if (!authResult.ok) return authResult.response;
+      const { scope: callerScope, userId } = authResult;
+      if (hooks?.onAuthenticated) {
+        const short = await hooks.onAuthenticated({
+          req,
+          scope: callerScope,
+          userId
+        });
+        if (short) return short;
+      }
+      const body = await req.json().catch(() => null);
+      const question = typeof body?.question === "string" ? body.question.trim() : "";
+      if (!question) {
+        return jsonError2(
+          400,
+          "VALIDATION_FAILED",
+          "question must be a non-empty string."
+        );
+      }
+      const rawChatSessionId = body?.chatSessionId;
+      const incomingChatSessionId = typeof rawChatSessionId === "number" && Number.isInteger(rawChatSessionId) ? rawChatSessionId : null;
+      const rawModel = body?.model;
+      const requestedModel = typeof rawModel === "string" && VALID_MODELS.has(rawModel) ? rawModel : null;
+      const aiSettings = await persistence.getAiSettings();
+      let chatSessionId;
+      if (incomingChatSessionId !== null) {
+        const owned = await persistence.getSession(
+          incomingChatSessionId,
+          userId
+        );
+        if (!owned) {
+          return jsonError2(404, "NOT_FOUND", "Chat session not found.");
+        }
+        chatSessionId = owned.id;
+      } else {
+        const created = await persistence.createSession({
+          userId,
+          title: question.slice(0, 200)
+        });
+        chatSessionId = created.id;
+      }
+      await persistence.appendMessage({
+        sessionId: chatSessionId,
+        role: "user",
+        question
+      });
+      const sessionId = hooks?.generateSessionId ? await hooks.generateSessionId({
+        scope: callerScope,
+        userId,
+        chatSessionId: incomingChatSessionId
+      }) : defaultGenerateSessionId2();
+      const scopeSummary = await scope.buildScopeSummary(callerScope);
+      const scopeLabel = await scope.resolveScopeLabel(callerScope);
+      const toolContext = {
+        scope: callerScope,
+        sessionId,
+        scopeSummary,
+        toolCallCount: 0
+      };
+      const systemBlocks = await tools.buildSystemBlocks(toolContext);
+      const provider = requestedModel ?? aiSettings.toolProvider;
+      if (!VALID_MODELS.has(provider)) {
+        return jsonError2(
+          400,
+          "INVALID_PROVIDER",
+          `Vercel chat only supports 'claude' or 'gemini'; got '${provider}'.`
+        );
+      }
+      const data = new StreamData();
+      let presentPayload = null;
+      let persistedError = null;
+      let sessionStarted = false;
+      try {
+        if (hooks?.onSessionStart) {
+          await hooks.onSessionStart({
+            scope: callerScope,
+            sessionId,
+            userId
+          });
+        }
+        sessionStarted = true;
+        const vercelTools = buildVercelTools(
+          tools.tools,
+          toolContext,
+          data,
+          (p) => {
+            presentPayload = p;
+          }
+        );
+        data.append({
+          type: "meta",
+          value: { chatSessionId, scopeLabel }
+        });
+        const system = systemBlocks.map((b) => b.text).join("\n\n");
+        const model = provider === "claude" ? createVertexAnthropic({
+          project: vertex.projectId,
+          location: vertex.defaultLocation,
+          googleAuthOptions: {}
+        })(vertex.modelIds.claude) : createVertex({
+          project: vertex.projectId,
+          location: aiSettings.gcpLocation,
+          googleAuthOptions: {}
+        })(vertex.modelIds.gemini);
+        const result = streamText({
+          model,
+          system,
+          messages: [{ role: "user", content: question }],
+          tools: vercelTools,
+          maxSteps: 12,
+          maxTokens: 4096,
+          onFinish: async ({ text }) => {
+            try {
+              let blocks = presentPayload?.blocks ?? [];
+              const prose = {};
+              const trimmed = (text ?? "").trim();
+              if (presentPayload === null && trimmed) {
+                const topic = question.length > 80 ? question.slice(0, 77) + "..." : question;
+                const synthetic = {
+                  kind: "paragraph_brief",
+                  topic,
+                  key_facts: [trimmed]
+                };
+                blocks = [synthetic];
+                prose[0] = trimmed;
+                data.append({
+                  type: "block",
+                  value: { index: 0, ...synthetic }
+                });
+              } else if (text) {
+                const firstPbIdx = blocks.findIndex(
+                  (b) => b.kind === "paragraph_brief"
+                );
+                if (firstPbIdx >= 0) prose[firstPbIdx] = text;
+              }
+              await persistence.appendMessage({
+                sessionId: chatSessionId,
+                role: "assistant",
+                blocks: blocks.length ? blocks : null,
+                prose: Object.keys(prose).length ? prose : null,
+                errorJson: persistedError
+              });
+            } catch (err2) {
+              logger?.warn?.(
+                {
+                  chatSessionId,
+                  sessionId,
+                  err: err2.message
+                },
+                "[agent-vercel] failed to persist assistant turn"
+              );
+            } finally {
+              try {
+                await data.close();
+              } catch {
+              }
+            }
+          }
+        });
+        return result.toDataStreamResponse({ data });
+      } catch (e) {
+        const message = e.message ?? "Internal error";
+        persistedError = { code: "INTERNAL", message };
+        logger?.error?.(
+          { chatSessionId, sessionId, err: message },
+          "[agent-vercel] route errored"
+        );
+        try {
+          data.append({
+            type: "error",
+            value: { code: "INTERNAL", message }
+          });
+        } catch {
+        }
+        try {
+          await data.close();
+        } catch {
+        }
+        try {
+          await persistence.appendMessage({
+            sessionId: chatSessionId,
+            role: "assistant",
+            blocks: null,
+            prose: null,
+            errorJson: persistedError
+          });
+        } catch (err2) {
+          logger?.warn?.(
+            { chatSessionId, sessionId, err: err2.message },
+            "[agent-vercel] failed to persist error turn"
+          );
+        }
+        return jsonError2(500, "INTERNAL", message);
+      } finally {
+        if (hooks?.onSessionEnd) {
+          const cause = req.signal.aborted ? "abort" : persistedError ? "error" : "complete";
+          try {
+            await hooks.onSessionEnd({
+              scope: callerScope,
+              sessionId,
+              userId,
+              cause
+            });
+          } catch (err2) {
+            logger?.warn?.(
+              {
+                chatSessionId,
+                sessionId,
+                sessionStarted,
+                err: err2.message
+              },
+              "[agent-vercel] onSessionEnd hook failed"
+            );
+          }
+        }
+      }
+    }
+  };
+}
 
 // src/server/routes/chat-sessions.ts
 var DEFAULT_TITLE = "New chat";
@@ -1457,6 +1789,15 @@ function configureAiChat(opts) {
     resolveNarratorId: opts.resolveNarratorId,
     hooks: opts.hooks
   });
+  const agentVercel = createAgentVercelRoutes({
+    persistence: opts.persistence,
+    auth: opts.auth,
+    scope: opts.scope,
+    tools,
+    vertex: opts.vertex,
+    logger: opts.logger,
+    hooks: opts.hooks
+  });
   const chatSessions = createChatSessionsRoutes({
     persistence: opts.persistence,
     auth: opts.auth,
@@ -1473,7 +1814,7 @@ function configureAiChat(opts) {
   });
   return {
     runAgent: runAgentBound,
-    routes: { agentCustom, chatSessions, adminSettings },
+    routes: { agentCustom, agentVercel, chatSessions, adminSettings },
     registries: { toolProviders: toolProviders2, chatInterfaces }
   };
 }