deepspace 0.2.5 → 0.2.6

package/dist/worker.d.ts

@@ -1,7 +1,5 @@
 import * as Y from 'yjs';
-import { AnthropicProvider } from '@ai-sdk/anthropic';
-import { OpenAIProvider } from '@ai-sdk/openai';
-import { OpenAICompatibleProvider } from '@ai-sdk/openai-compatible';
+import { LanguageModel, ModelMessage } from 'ai';
 import * as better_auth from 'better-auth';
 import * as better_auth_plugins from 'better-auth/plugins';
 
@@ -1653,6 +1651,16 @@ type DOBindings<T extends readonly DOManifestEntry[]> = {
 /** Default manifest for apps that don't declare one */
 declare const DEFAULT_DO_MANIFEST: DOManifest;
 
+/**
+ * AI Chat Schemas
+ *
+ * Pre-built collection schemas for DO-backed AI chat history.
+ * The worker is the only writer; the client reads via `useQuery`.
+ */
+
+declare const AI_CHATS_SCHEMA: CollectionSchema;
+declare const AI_MESSAGES_SCHEMA: CollectionSchema;
+
 /**
  * Messaging Schemas
  *
@@ -2221,12 +2229,13 @@ declare function authWorkerFetch(env: AuthWorkerEnv, path: string, init?: Reques
  *   const anthropic = createDeepSpaceAI(c.env, 'anthropic', { authToken: jwt })
  */
 
+/**
+ * Model factory: `(modelId) => LanguageModel`. The explicit return type
+ * keeps tsup's DTS build from leaking unportable `.pnpm/@ai-sdk+provider/...`
+ * paths into the published `dist/index.d.ts`.
+ */
+type DeepSpaceModelFactory = (modelId: string) => LanguageModel;
 type Provider = 'anthropic' | 'openai' | 'cerebras';
-type ProviderReturn = {
-    anthropic: AnthropicProvider;
-    openai: OpenAIProvider;
-    cerebras: OpenAICompatibleProvider;
-};
 interface DeepSpaceAIEnv extends ApiWorkerEnv {
     /**
      * Long-lived owner-scoped JWT minted at deploy time (or by `deepspace dev`).
@@ -2254,7 +2263,176 @@ interface DeepSpaceAIOptions {
  * (explicit or `env.APP_OWNER_JWT`) automatically. Throws a clear error if
  * either is unconfigured.
  */
-declare function createDeepSpaceAI<P extends Provider>(env: DeepSpaceAIEnv, provider: P, options?: DeepSpaceAIOptions): ProviderReturn[P];
+declare function createDeepSpaceAI(env: DeepSpaceAIEnv, provider: Provider, options?: DeepSpaceAIOptions): DeepSpaceModelFactory;
+
+/**
+ * Chat context pipeline — keeps the per-request payload to the LLM bounded.
+ *
+ * `prepareMessagesWithCompaction` runs before `streamText`: truncate old tool
+ * results, apply a cached summary if available, otherwise summarize the older
+ * half of history when over budget. Falls back to a sliding window if
+ * summarization fails. `capToolResultSize` caps individual tool calls.
+ */
+
+interface ChatTurn {
+    id?: string;
+    role: 'user' | 'assistant' | 'system';
+    content: string;
+    parts?: unknown[];
+}
+type Summarizer = (messages: ChatTurn[]) => Promise<string>;
+interface ChatContextConfig {
+    contextBudget: number;
+    toolResultCap: number;
+    keepRecentToolResults: number;
+    minKept: number;
+}
+declare const DEFAULT_CONTEXT_CONFIG: ChatContextConfig;
+declare function totalChars(messages: ChatTurn[]): number;
+/**
+ * Replace older tool-result payloads with a small marker. Keeps the last
+ * `keepRecent` tool results intact. Errors (`success: false`) are preserved —
+ * they're small and the agent needs them for reasoning.
+ */
+declare function truncateOldToolResults(messages: ChatTurn[], keepRecent: number): ChatTurn[];
+/**
+ * Drop oldest messages until total character count is under `charCap`,
+ * never going below `minKept` messages. System messages (e.g. compaction
+ * summaries) are pinned — dropping them would discard the most condensed
+ * context first.
+ */
+declare function applySlidingWindow(messages: ChatTurn[], charCap: number, minKept: number): ChatTurn[];
+/**
+ * Replace oversized tool results with an error telling the agent to narrow
+ * its query. Keeps a small preview so the agent can see what it got.
+ */
+declare function capToolResultSize(result: unknown, byteCap: number): unknown;
+/**
+ * Convert persisted ChatTurns into AI SDK ModelMessages.
+ *
+ * Persisted assistant rows store `parts` in UI shape (text + tool-invocation,
+ * each invocation carrying its own `result`). When fed back to the LLM, the
+ * shape MUST match the original multi-step flow: an assistant message
+ * containing a `tool_use` block must end with that block, the IMMEDIATELY
+ * NEXT message must be a tool/user message containing the matching
+ * `tool_result`, and any text the model produced AFTER seeing the tool
+ * result belongs in a SEPARATE assistant message after the tool message.
+ *
+ * Anthropic specifically rejects an assistant message of the form
+ * `[text, tool_use, text]` — the trailing text breaks its `tool_use` →
+ * `tool_result` pairing check. So we walk the parts in order and split at
+ * each tool-invocation boundary, emitting a fresh assistant + tool pair per
+ * tool call, and a final trailing assistant message for any post-tool text.
+ *
+ * Tool-invocation entries with `state: 'call'` (no result — typically an
+ * interrupted stream) are dropped on both sides.
+ */
+declare function turnsToCoreMessages(turns: ChatTurn[]): ModelMessage[];
+/**
+ * Convert AI SDK response messages into our persisted UI shape (text +
+ * tool-invocation parts), pairing each assistant tool-call with its tool-
+ * result from the following tool message. Order is chronological.
+ *
+ * Inverse of `turnsToCoreMessages`: takes the v5 `ModelMessage[]` returned
+ * from `streamText`'s `onFinish` and produces the flat `parts` array we
+ * persist on `ai-messages` rows. Reads `c.input` / `c.output` (v5 wire
+ * names) and unwraps `output`'s tagged-union via `unwrapToolOutput`.
+ */
+declare function buildUiParts(responseMessages: ModelMessage[]): unknown[];
+/**
+ * Unwrap v5's tagged tool-result `output` to the flat shape we persist.
+ * Errors get remapped to `{ success: false, error }` because
+ * `truncateOldToolResults` preserves entries with that shape across turns —
+ * without the remap, error context would get truncated like a normal result.
+ */
+declare function unwrapToolOutput(output: unknown): unknown;
+/**
+ * Pre-stream pipeline with compaction.
+ *
+ * 1. Truncate old tool results.
+ * 2. If a cached summary covers a known message id, replace prior turns with it.
+ * 3. If still over budget, summarize the older half of `working` and return a
+ *    `newSummary` for persistence — runs even after cached-summary application
+ *    so a long-running chat can re-summarize on subsequent turns.
+ * 4. On summarizer error or missing ids, fall back to a sliding window
+ *    (which preserves system messages — see `applySlidingWindow`).
+ */
+declare function prepareMessagesWithCompaction(messages: ChatTurn[], config: ChatContextConfig, options: {
+    summarizer: Summarizer;
+    cachedSummary?: {
+        text: string;
+        throughId: string;
+    };
+}): Promise<{
+    messages: ChatTurn[];
+    newSummary?: {
+        text: string;
+        throughId: string;
+    };
+}>;
+/**
+ * Build a default summarizer backed by Claude Haiku.
+ *
+ * Billing: defaults to the app owner via `APP_OWNER_JWT` — summarization is
+ * usually infrastructure, not user work. Pass `{ authToken }` to bill a
+ * specific user (e.g. the caller's JWT) instead.
+ */
+declare function makeDefaultSummarizer(env: DeepSpaceAIEnv, options?: {
+    authToken?: string;
+}): Summarizer;
+
+/**
+ * Chat history helpers — wrap RecordRoom's tools API for ai-chats / ai-messages.
+ *
+ * Trust model: every helper sends `X-App-Action: 'true'`, which bypasses
+ * RecordRoom's per-record RBAC. The worker is the trust boundary, not
+ * RecordRoom. Callers MUST verify ownership before invoking write helpers
+ * (`updateChat`, `appendMessage`, `deleteChatCascade`); the worker's
+ * `/api/ai/chat`, `PATCH /api/ai/chats/:id`, and `DELETE /api/ai/chats/:id`
+ * routes do this via a `getChat()` precheck that 404s when the row is
+ * missing or owned by another user. Read helpers (`getChat`, `loadMessages`)
+ * filter by `chatId` against userBound rows, so cross-user reads return
+ * empty — but new consumers should still consider an explicit ownership
+ * check before exposing data.
+ *
+ * The tools API returns records as `{ recordId, data, createdAt, updatedAt }`
+ * envelopes; helpers below flatten them into ChatRow / ChatMessageRow.
+ */
+type ChatRow = {
+    id: string;
+    userId: string;
+    title: string;
+    model?: string;
+    compactedSummary?: string;
+    compactedThroughId?: string;
+    createdAt: string;
+    updatedAt: string;
+};
+type ChatMessageRow = {
+    id: string;
+    chatId: string;
+    userId: string;
+    role: 'user' | 'assistant' | 'system';
+    content: string;
+    parts?: unknown[];
+    createdAt: string;
+};
+declare function getChat(stub: DurableObjectStub, chatId: string, userId: string): Promise<ChatRow | null>;
+declare function createChat(stub: DurableObjectStub, userId: string, opts?: {
+    title?: string;
+    model?: string;
+}): Promise<ChatRow>;
+declare function updateChat(stub: DurableObjectStub, chatId: string, userId: string, patch: Partial<Pick<ChatRow, 'title' | 'model' | 'compactedSummary' | 'compactedThroughId'>>): Promise<void>;
+declare function deleteChatCascade(stub: DurableObjectStub, chatId: string, userId: string): Promise<void>;
+declare function loadMessages(stub: DurableObjectStub, chatId: string, userId: string): Promise<ChatMessageRow[]>;
+declare function appendMessage(stub: DurableObjectStub, msg: {
+    id: string;
+    chatId: string;
+    userId: string;
+    role: 'user' | 'assistant' | 'system';
+    content: string;
+    parts?: unknown[];
+}): Promise<void>;
 
 /**
  * Auth types for the DeepSpace SDK.
@@ -3209,4 +3387,4 @@ type DeepSpaceAuth = ReturnType<typeof createDeepSpaceAuth>;
 /** Collections that bypass schema checks (permissive access) */
 declare const SYSTEM_COLLECTIONS: Set<string>;
 
-export { type ActionContext, type ActionHandler, type ActionResult, type ActionTools, type ApiWorkerEnv, type AuthWorkerEnv, Awareness, type AwarenessState, type AwarenessStates, BASE_USERS_SCHEMA, BUILT_IN_TOOLS, type BaseMessage, BaseRoom, CHANNELS_SCHEMA, CHANNEL_INVITATIONS_SCHEMA, CHANNEL_MEMBERS_SCHEMA, CONVERSATION_SCHEMAS, CanvasRoom, type CanvasShape, type ClientMessage, type CollectionPermissionSummary, type CollectionSchema, type ColumnDefinition, type ColumnInterpretation, type ConnectionAttachment, type ConvMemberData, type ConvMessageData, type ConvReactionData, type ConvReadCursorData, type ConvVoteData, type ConversationStateData, type CronContext, type CronExecution, CronRoom, type CronRoomConfig, type CronTask, DEFAULT_DO_MANIFEST, DEFAULT_MAX_SKEW_MS, DIRECTORY_SCHEMAS, type DOBindings, type DOManifest, type DOManifestEntry, type DeepSpaceAIEnv, type DeepSpaceAIOptions, type DeepSpaceAuth, type DeepSpaceAuthConfig, type DirectoryCommunityData, type DirectoryConversationData, type DirectoryMembershipData, type DirectoryPostData, GLOBAL_DO_TYPES, GLOBAL_DO_TYPE_NAMES, type GameInput, GameRoom, type GameRoomConfig, type GlobalDOType, type HandlerContext, type InternalSignature, type JwtClaims, type JwtVerifierConfig, MESSAGES_SCHEMA, MSG, MSG_AWARENESS, MSG_SYNC, MSG_SYNC_STEP1, MSG_SYNC_STEP2, MSG_SYNC_UPDATE, MSG_YJS_AWARENESS, MSG_YJS_SYNC, type MessageHandlers, type MsgType, type PermissionAnalysis, type PermissionContext, type PermissionLevel, type PermissionSource, type PlatformWorkerEnv, type Player, type PrefixResult, type PresencePeer, PresenceRoom, type ProtocolMessage, REACTIONS_SCHEMA, READ_RECEIPTS_SCHEMA, RESERVED_COLLECTION_NAMES, ROLE_ADMIN, ROLE_ANONYMOUS, ROLE_DEFAULT, RecordRoom, type RecordRoomConfig, type ResolvedColumn, type ResolvedPermission, type RolePermissions, SYSTEM_COLLECTIONS, SYSTEM_MANAGED_COLUMNS, SchemaRegistry, type ScopeContext, type ScopedR2Auth, type ScopedR2Config, type ScopedR2Handler, type ServerMessage, type SharedConnection, type SignInternalPayloadInput, type StoredRecord, type SyncResult, type TokenDebugInfo, type ToolResult, type ToolSchema, USERS_COLUMNS, type User, type UserAttachment, VOTING_SCHEMAS, type VerifiedAuth, type VerifyInternalSignatureInput, type VerifyOutcome, type VerifyResult, type Viewport, WORKSPACE_SCHEMAS, YjsRoom, analyzePermissions, apiWorkerFetch, authWorkerFetch, buildCronContext, buildInternalPayload, buildTableSelect, canCreate, canDelete, canRead, canUpdate, checkFieldPermissions, clientBuild, coerceValue, collectionTableName, columnId, computeHmacHex, createDecoder, createDeepSpaceAI, createDeepSpaceAuth, createEncoder, createScopedR2Handler, dataToColumnValues, decodeJwtPayload, dispatch, encode, encodeAwarenessMessage, encodeSyncStep1, encodeSyncStep2, encodeUpdate, getGlobalDOSchemas, getGlobalDOType, getMessageType, getRolePermissions, handleAwarenessMessage, handleSyncMessage, hasContent, isOwner, noopPermissionContext, platformWorkerFetch, readVarUint, readVarUint8Array, resolveColumn, rowToData, serverBuild, signInternalPayload, timingSafeEqualHex, toUint8Array, verifyInternalSignature, verifyJwt, workspaceAccountsSchema, workspaceContentSharesSchema, workspaceEmailHandlesSchema, workspaceFormResponsesSchema, workspacePeopleSchema, workspaceProjectsSchema, workspaceTagsSchema, workspaceTasksSchema, workspaceTeamMembersSchema, workspaceTeamsSchema, workspaceTransactionsSchema, writeVarUint, writeVarUint8Array };
+export { AI_CHATS_SCHEMA, AI_MESSAGES_SCHEMA, type ActionContext, type ActionHandler, type ActionResult, type ActionTools, type ApiWorkerEnv, type AuthWorkerEnv, Awareness, type AwarenessState, type AwarenessStates, BASE_USERS_SCHEMA, BUILT_IN_TOOLS, type BaseMessage, BaseRoom, CHANNELS_SCHEMA, CHANNEL_INVITATIONS_SCHEMA, CHANNEL_MEMBERS_SCHEMA, CONVERSATION_SCHEMAS, CanvasRoom, type CanvasShape, type ChatContextConfig, type ChatMessageRow, type ChatRow, type ChatTurn, type ClientMessage, type CollectionPermissionSummary, type CollectionSchema, type ColumnDefinition, type ColumnInterpretation, type ConnectionAttachment, type ConvMemberData, type ConvMessageData, type ConvReactionData, type ConvReadCursorData, type ConvVoteData, type ConversationStateData, type CronContext, type CronExecution, CronRoom, type CronRoomConfig, type CronTask, DEFAULT_CONTEXT_CONFIG, DEFAULT_DO_MANIFEST, DEFAULT_MAX_SKEW_MS, DIRECTORY_SCHEMAS, type DOBindings, type DOManifest, type DOManifestEntry, type DeepSpaceAIEnv, type DeepSpaceAIOptions, type DeepSpaceAuth, type DeepSpaceAuthConfig, type DirectoryCommunityData, type DirectoryConversationData, type DirectoryMembershipData, type DirectoryPostData, GLOBAL_DO_TYPES, GLOBAL_DO_TYPE_NAMES, type GameInput, GameRoom, type GameRoomConfig, type GlobalDOType, type HandlerContext, type InternalSignature, type JwtClaims, type JwtVerifierConfig, MESSAGES_SCHEMA, MSG, MSG_AWARENESS, MSG_SYNC, MSG_SYNC_STEP1, MSG_SYNC_STEP2, MSG_SYNC_UPDATE, MSG_YJS_AWARENESS, MSG_YJS_SYNC, type MessageHandlers, type MsgType, type PermissionAnalysis, type PermissionContext, type PermissionLevel, type PermissionSource, type PlatformWorkerEnv, type Player, type PrefixResult, type PresencePeer, PresenceRoom, type ProtocolMessage, REACTIONS_SCHEMA, READ_RECEIPTS_SCHEMA, RESERVED_COLLECTION_NAMES, ROLE_ADMIN, ROLE_ANONYMOUS, ROLE_DEFAULT, RecordRoom, type RecordRoomConfig, type ResolvedColumn, type ResolvedPermission, type RolePermissions, SYSTEM_COLLECTIONS, SYSTEM_MANAGED_COLUMNS, SchemaRegistry, type ScopeContext, type ScopedR2Auth, type ScopedR2Config, type ScopedR2Handler, type ServerMessage, type SharedConnection, type SignInternalPayloadInput, type StoredRecord, type Summarizer, type SyncResult, type TokenDebugInfo, type ToolResult, type ToolSchema, USERS_COLUMNS, type User, type UserAttachment, VOTING_SCHEMAS, type VerifiedAuth, type VerifyInternalSignatureInput, type VerifyOutcome, type VerifyResult, type Viewport, WORKSPACE_SCHEMAS, YjsRoom, analyzePermissions, apiWorkerFetch, appendMessage, applySlidingWindow, authWorkerFetch, buildCronContext, buildInternalPayload, buildTableSelect, buildUiParts, canCreate, canDelete, canRead, canUpdate, capToolResultSize, checkFieldPermissions, clientBuild, coerceValue, collectionTableName, columnId, computeHmacHex, createChat, createDecoder, createDeepSpaceAI, createDeepSpaceAuth, createEncoder, createScopedR2Handler, dataToColumnValues, decodeJwtPayload, deleteChatCascade, dispatch, encode, encodeAwarenessMessage, encodeSyncStep1, encodeSyncStep2, encodeUpdate, getChat, getGlobalDOSchemas, getGlobalDOType, getMessageType, getRolePermissions, handleAwarenessMessage, handleSyncMessage, hasContent, isOwner, loadMessages, makeDefaultSummarizer, noopPermissionContext, platformWorkerFetch, prepareMessagesWithCompaction, readVarUint, readVarUint8Array, resolveColumn, rowToData, serverBuild, signInternalPayload, timingSafeEqualHex, toUint8Array, totalChars, truncateOldToolResults, turnsToCoreMessages, unwrapToolOutput, updateChat, verifyInternalSignature, verifyJwt, workspaceAccountsSchema, workspaceContentSharesSchema, workspaceEmailHandlesSchema, workspaceFormResponsesSchema, workspacePeopleSchema, workspaceProjectsSchema, workspaceTagsSchema, workspaceTasksSchema, workspaceTeamMembersSchema, workspaceTeamsSchema, workspaceTransactionsSchema, writeVarUint, writeVarUint8Array };