Extract Command Code stream core

src/types.ts

@@ -0,0 +1,123 @@
+export type StopReason = "stop" | "length" | "toolUse";
+export type ErrorReason = "error" | "aborted";
+export type TerminalReason = StopReason | ErrorReason;
+
+export interface UsageCost {
+  input: number;
+  output: number;
+  cacheRead: number;
+  cacheWrite: number;
+  total: number;
+}
+
+export interface Usage {
+  input: number;
+  output: number;
+  cacheRead: number;
+  cacheWrite: number;
+  totalTokens: number;
+  cost: UsageCost;
+}
+
+export interface TextContent {
+  type: "text";
+  text: string;
+}
+
+export interface ThinkingContent {
+  type: "thinking";
+  thinking: string;
+}
+
+export interface ToolCallContent {
+  type: "toolCall";
+  id: string;
+  name: string;
+  arguments: Record<string, unknown>;
+}
+
+export type AssistantContent = TextContent | ThinkingContent | ToolCallContent;
+
+export interface AssistantMessageLike {
+  role: "assistant";
+  content: AssistantContent[];
+  api: unknown;
+  provider: string;
+  model: string;
+  usage: Usage;
+  stopReason: TerminalReason;
+  errorMessage?: string;
+  timestamp: number;
+}
+
+export interface ModelLike {
+  id: string;
+  api: unknown;
+  provider: string;
+  maxTokens: number;
+}
+
+export interface MessageLike {
+  role: string;
+  content?: unknown;
+  toolCallId?: string;
+  toolName?: string;
+  isError?: boolean;
+}
+
+export interface ToolLike {
+  name: string;
+  description?: string;
+  parameters?: unknown;
+}
+
+export interface ContextLike {
+  systemPrompt?: string;
+  messages?: readonly MessageLike[];
+  tools?: readonly ToolLike[];
+}
+
+export interface ProviderResponseInfo {
+  status: number;
+  headers: Record<string, string>;
+}
+
+export interface StreamOptions {
+  apiKey?: string;
+  signal?: AbortSignal;
+  headers?: Record<string, string>;
+  maxTokens?: number;
+  onPayload?: (payload: unknown, model: ModelLike) => unknown | Promise<unknown>;
+  onResponse?: (response: ProviderResponseInfo, model: ModelLike) => void | Promise<void>;
+}
+
+export type AssistantMessageEvent =
+  | { type: "start"; partial: AssistantMessageLike }
+  | { type: "text_start"; contentIndex: number; partial: AssistantMessageLike }
+  | { type: "text_delta"; contentIndex: number; delta: string; partial: AssistantMessageLike }
+  | { type: "text_end"; contentIndex: number; content: string; partial: AssistantMessageLike }
+  | { type: "thinking_start"; contentIndex: number; partial: AssistantMessageLike }
+  | { type: "thinking_delta"; contentIndex: number; delta: string; partial: AssistantMessageLike }
+  | { type: "thinking_end"; contentIndex: number; content: string; partial: AssistantMessageLike }
+  | { type: "toolcall_start"; contentIndex: number; partial: AssistantMessageLike }
+  | { type: "toolcall_end"; contentIndex: number; toolCall: ToolCallContent; partial: AssistantMessageLike }
+  | { type: "done"; reason: StopReason; message: AssistantMessageLike }
+  | { type: "error"; reason: ErrorReason; error: AssistantMessageLike };
+
+export interface AssistantMessageEventStreamLike extends AsyncIterable<AssistantMessageEvent> {
+  push(event: AssistantMessageEvent): void;
+  end(): void;
+}
+
+export interface CoreDependencies {
+  createStream: () => AssistantMessageEventStreamLike;
+  calculateCost: (model: ModelLike, usage: Usage) => void;
+  apiBase?: string;
+  fetchImpl?: typeof fetch;
+  authPaths?: readonly string[];
+  env?: NodeJS.ProcessEnv;
+  cwd?: () => string;
+  now?: () => number;
+  uuid?: () => string;
+  homeDir?: () => string;
+}