pi-commandcode-provider 0.1.0

package/src/types.ts

@@ -0,0 +1,156 @@
+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
+}