@alumbwe/anvil 1.0.20 → 1.0.22

package/vendor/@anvil/common/src/types/contracts/agent-runtime.ts

@@ -0,0 +1,75 @@
+import type { TrackEventFn } from './analytics'
+import type { ConsumeCreditsWithFallbackFn } from './billing'
+import type {
+  HandleStepsLogChunkFn,
+  RequestFilesFn,
+  RequestMcpToolDataFn,
+  RequestOptionalFileFn,
+  RequestToolCallFn,
+  SendActionFn,
+  SendSubagentChunkFn,
+} from './client'
+import type {
+  AddAgentStepFn,
+  DatabaseAgentCache,
+  FetchAgentFromDatabaseFn,
+  FinishAgentRunFn,
+  GetUserInfoFromApiKeyFn,
+  StartAgentRunFn,
+} from './database'
+import type { ClientEnv, CiEnv } from './env'
+import type {
+  PromptAiSdkFn,
+  PromptAiSdkStreamFn,
+  PromptAiSdkStructuredFn,
+} from './llm'
+import type { Logger } from './logger'
+import type { TraceWriter } from './trace'
+
+/** Shared dependencies */
+export type AgentRuntimeDeps = {
+  // Environment
+  clientEnv: ClientEnv
+  ciEnv: CiEnv
+
+  // Database
+  getUserInfoFromApiKey: GetUserInfoFromApiKeyFn
+  fetchAgentFromDatabase: FetchAgentFromDatabaseFn
+  startAgentRun: StartAgentRunFn
+  finishAgentRun: FinishAgentRunFn
+  addAgentStep: AddAgentStepFn
+
+  // Billing
+  consumeCreditsWithFallback: ConsumeCreditsWithFallbackFn
+
+  // LLM
+  promptAiSdkStream: PromptAiSdkStreamFn
+  promptAiSdk: PromptAiSdkFn
+  promptAiSdkStructured: PromptAiSdkStructuredFn
+
+  // Mutable State
+  databaseAgentCache: DatabaseAgentCache
+
+  // Analytics
+  trackEvent: TrackEventFn
+
+  // Other
+  logger: Logger
+  /** Optional debug trace of agent message histories (see TraceWriter) */
+  traceWriter?: TraceWriter
+  fetch: typeof globalThis.fetch
+}
+
+/** Per-run dependencies */
+export type AgentRuntimeScopedDeps = {
+  // Client (WebSocket)
+  handleStepsLogChunk: HandleStepsLogChunkFn
+  requestToolCall: RequestToolCallFn
+  requestMcpToolData: RequestMcpToolDataFn
+  requestFiles: RequestFilesFn
+  requestOptionalFile: RequestOptionalFileFn
+  sendAction: SendActionFn
+  sendSubagentChunk: SendSubagentChunkFn
+
+  apiKey: string
+}

package/vendor/@anvil/common/src/types/contracts/analytics.ts

@@ -0,0 +1,9 @@
+import type { Logger } from './logger'
+import type { AnalyticsEvent } from '../../constants/analytics-events'
+
+export type TrackEventFn = (params: {
+  event: AnalyticsEvent
+  userId: string
+  properties?: Record<string, any>
+  logger: Logger
+}) => void

package/vendor/@anvil/common/src/types/contracts/bigquery.ts

@@ -0,0 +1,55 @@
+import type { Logger } from './logger'
+
+export type MessageRow = {
+  id: string
+  user_id: string
+  finished_at: Date
+  created_at: Date
+  request: unknown
+  reasoning_text: string
+  response: string
+  output_tokens?: number | null
+  reasoning_tokens?: number | null
+  cost?: number | null
+  upstream_inference_cost?: number | null
+  input_tokens?: number | null
+  cache_read_input_tokens?: number | null
+}
+
+export type InsertMessageBigqueryFn = (params: {
+  row: MessageRow
+  dataset?: string
+  logger: Logger
+}) => Promise<boolean>
+
+export type ChatCompletionTraceRow = {
+  id: string
+  user_id: string
+  client_id?: string | null
+  trace_session_id: string
+  trace_lineage_id: string
+  run_id: string
+  agent_id: string
+  created_at: Date
+  model: string
+  cost_mode?: string | null
+  request: unknown
+  message_count: number
+  message_start_index: number
+  message_delta_count: number
+  previous_message_count?: number | null
+  common_prefix_length: number
+  cache_hit: boolean
+  full_snapshot: boolean
+  messages: unknown[]
+  delta_message_hashes: string[]
+  tool_count: number
+  tools?: unknown[] | null
+  tools_omitted: boolean
+}
+
+export type InsertChatCompletionTraceBigqueryFn = (params: {
+  row: ChatCompletionTraceRow
+  dataset?: string
+  logger: Logger
+}) => Promise<boolean>

package/vendor/@anvil/common/src/types/contracts/billing.ts

@@ -0,0 +1,46 @@
+import type { Logger } from './logger'
+import type { ErrorOr } from '../../util/error'
+
+export type GetUserUsageDataFn = (params: {
+  userId: string
+  logger: Logger
+  includeSubscriptionCredits?: boolean
+}) => Promise<{
+  usageThisCycle: number
+  balance: {
+    totalRemaining: number
+    totalDebt: number
+    netBalance: number
+    breakdown: Record<string, number>
+    principals: Record<string, number>
+  }
+  nextQuotaReset: string
+  autoTopupTriggered?: boolean
+  autoTopupEnabled?: boolean
+}>
+
+export type ConsumeCreditsWithFallbackFn = (params: {
+  userId: string
+  creditsToCharge: number
+  repoUrl?: string | null
+  context: string // Description of what the credits are for (e.g., 'web search', 'documentation lookup')
+  logger: Logger
+}) => Promise<ErrorOr<CreditFallbackResult>>
+
+export type CreditFallbackResult = {
+  organizationId?: string
+  organizationName?: string
+  chargedToOrganization: boolean
+}
+
+export type GetOrganizationUsageResponseFn = (params: {
+  organizationId: string
+  userId: string
+  logger: Logger
+}) => Promise<{
+  type: 'usage-response'
+  usage: number
+  remainingBalance: number
+  balanceBreakdown: Record<string, never>
+  next_quota_reset: null
+}>

package/vendor/@anvil/common/src/types/contracts/client.ts

@@ -0,0 +1,53 @@
+import type { ServerAction } from '../../actions'
+import type { MCPConfig } from '../mcp'
+import type { ToolResultOutput } from '../messages/content-part'
+
+export type RequestToolCallFn = (params: {
+  userInputId: string
+  toolName: string
+  input: Record<string, any> & { timeout_seconds?: number }
+  mcpConfig?: MCPConfig
+}) => Promise<{
+  output: ToolResultOutput[]
+}>
+
+export type RequestMcpToolDataFn = (params: {
+  mcpConfig: MCPConfig
+  toolNames: string[] | null
+}) => Promise<
+  {
+    name: string
+    description?: string
+    inputSchema: unknown
+  }[]
+>
+
+export type FileReadWindow = { offset?: number; limit?: number }
+
+export type RequestFilesFn = (params: {
+  filePaths: string[]
+  fileWindows?: Record<string, FileReadWindow[]>
+}) => Promise<Record<string, string | null>>
+
+export type RequestOptionalFileFn = (params: {
+  filePath: string
+}) => Promise<string | null>
+
+export type SendSubagentChunkFn = (params: {
+  userInputId: string
+  agentId: string
+  agentType: string
+  chunk: string
+  prompt?: string | undefined
+  forwardToPrompt?: boolean
+}) => void
+
+export type HandleStepsLogChunkFn = (params: {
+  userInputId: string
+  runId: string
+  level: 'debug' | 'info' | 'warn' | 'error'
+  data: unknown
+  message?: string
+}) => void
+
+export type SendActionFn = (params: { action: ServerAction }) => void

package/vendor/@anvil/common/src/types/contracts/database.ts

@@ -0,0 +1,112 @@
+import type { AgentTemplate } from '@anvil/common/types/agent-template'
+import type { Logger } from '@anvil/common/types/contracts/logger'
+
+type User = {
+  id: string
+  email: string
+  discord_id: string | null
+  stripe_customer_id: string | null
+  banned: boolean
+  created_at: Date
+}
+export const userColumns = [
+  'id',
+  'email',
+  'discord_id',
+  'stripe_customer_id',
+  'banned',
+  'created_at',
+] as const
+export type UserColumn = keyof User
+export type GetUserInfoFromApiKeyInput<T extends UserColumn> = {
+  apiKey: string
+  fields: readonly T[]
+  logger: Logger
+}
+export type GetUserInfoFromApiKeyOutput<T extends UserColumn> = Promise<
+  | {
+      [K in T]: User[K]
+    }
+  | null
+>
+export type GetUserInfoFromApiKeyFn = <T extends UserColumn>(
+  params: GetUserInfoFromApiKeyInput<T>,
+) => GetUserInfoFromApiKeyOutput<T>
+
+export type GetUserInfoByIdInput<T extends UserColumn> = {
+  userId: string
+  fields: readonly T[]
+}
+export type GetUserInfoByIdFn = <T extends UserColumn>(
+  params: GetUserInfoByIdInput<T>,
+) => GetUserInfoFromApiKeyOutput<T>
+
+type AgentRun = {
+  agent_id: string
+  ancestor_run_ids: string[]
+  status: 'running' | 'completed' | 'failed' | 'cancelled'
+}
+export type AgentRunColumn = keyof AgentRun
+export type GetAgentRunFromIdInput<T extends AgentRunColumn> = {
+  runId: string
+  userId: string
+  fields: readonly T[]
+}
+export type GetAgentRunFromIdOutput<T extends AgentRunColumn> = Promise<
+  | {
+      [K in T]: AgentRun[K]
+    }
+  | null
+>
+export type GetAgentRunFromIdFn = <T extends AgentRunColumn>(
+  params: GetAgentRunFromIdInput<T>,
+) => GetAgentRunFromIdOutput<T>
+
+/**
+ * Fetch and validate an agent from the database by `publisher/agent-id[@version]` format
+ */
+export type FetchAgentFromDatabaseFn = (params: {
+  apiKey: string
+  parsedAgentId: {
+    publisherId: string
+    agentId: string
+    version?: string
+  }
+  logger: Logger
+}) => Promise<AgentTemplate | null>
+
+export type StartAgentRunFn = (params: {
+  apiKey: string
+  userId?: string
+  agentId: string
+  ancestorRunIds: string[]
+  logger: Logger
+}) => Promise<string | null>
+
+export type FinishAgentRunFn = (params: {
+  apiKey: string
+  userId: string | undefined
+  runId: string
+  status: 'completed' | 'failed' | 'cancelled'
+  totalSteps: number
+  directCredits: number
+  totalCredits: number
+  errorMessage?: string
+  logger: Logger
+}) => Promise<void>
+
+export type AddAgentStepFn = (params: {
+  apiKey: string
+  userId: string | undefined
+  agentRunId: string
+  stepNumber: number
+  credits?: number
+  childRunIds?: string[]
+  messageId: string | null
+  status?: 'running' | 'completed' | 'skipped'
+  errorMessage?: string
+  startTime: Date
+  logger: Logger
+}) => Promise<string | null>
+
+export type DatabaseAgentCache = Map<string, AgentTemplate | null>

package/vendor/@anvil/common/src/types/contracts/env.ts

@@ -0,0 +1,203 @@
+/**
+ * Environment variable contract types for dependency injection.
+ *
+ * ARCHITECTURE:
+ * =============
+ * Base types (defined here in common) that packages extend:
+ *
+ * ┌─────────────────────────────────────────────────────────────────┐
+ * │                      BASE TYPES (common)                        │
+ * ├─────────────────────────────────────────────────────────────────┤
+ * │  BaseProcessEnv  - OS/runtime vars (SHELL, HOME, TERM, etc.)   │
+ * │  BaseCiEnv       - CI vars (CI, GITHUB_ACTIONS, etc.)          │
+ * │  ClientEnv       - Public vars (NEXT_PUBLIC_*) from env-schema │
+ * └─────────────────────────────────────────────────────────────────┘
+ *                               │
+ *                     extends   │
+ *                               ▼
+ * ┌─────────────────────────────────────────────────────────────────┐
+ * │                    PACKAGE-SPECIFIC ENVS                        │
+ * ├─────────────────────────────────────────────────────────────────┤
+ * │  CLI:    CliProcessEnv = BaseProcessEnv & { CLI-specific }     │
+ * │  SDK:    SdkProcessEnv = BaseProcessEnv & { SDK-specific }     │
+ * │  Server: ServerEnv = ClientEnv & { server secrets }            │
+ * │  Evals:  EvalsCiEnv = BaseCiEnv & { eval-specific }             │
+ * └─────────────────────────────────────────────────────────────────┘
+ *
+ * Each package imports the base types and extends them with package-specific vars.
+ */
+
+import type { ClientEnv } from '../../env-schema'
+
+// Re-export the env types for convenience
+export type { ClientEnv } from '../../env-schema'
+
+// =============================================================================
+// BASE TYPES - Extended by packages
+// =============================================================================
+
+/**
+ * Base CI environment variables.
+ * Used in CI/CD pipelines across all packages.
+ */
+export type BaseCiEnv = {
+  CI?: string
+  GITHUB_ACTIONS?: string
+  RENDER?: string
+  IS_PULL_REQUEST?: string
+}
+
+/**
+ * Base runtime environment variables.
+ * These are OS-level env vars common across all packages.
+ */
+export type BaseEnv = {
+  // Shell detection
+  SHELL?: string
+  COMSPEC?: string // Windows command processor
+
+  // Home directory
+  HOME?: string
+  USERPROFILE?: string // Windows home
+  APPDATA?: string // Windows app data
+  XDG_CONFIG_HOME?: string // Linux config home
+
+  // Terminal detection
+  TERM?: string
+  TERM_PROGRAM?: string
+  TERM_BACKGROUND?: string
+  TERMINAL_EMULATOR?: string
+  COLORFGBG?: string
+
+  // Node/runtime
+  NODE_ENV?: string
+  NODE_PATH?: string
+  PATH?: string
+}
+
+// =============================================================================
+// EXTENDED TYPES - For packages that need more than base
+// =============================================================================
+
+/**
+ * Extended CI env with Anvil-specific CI vars.
+ * Used by agent-runtime and server code.
+ */
+export type CiEnv = BaseCiEnv & {
+  ANVIL_GITHUB_TOKEN?: string
+  ANVIL_API_KEY?: string
+  EVAL_RESULTS_EMAIL?: string
+}
+
+/**
+ * Extended process env with terminal/IDE detection.
+ * This is the full ProcessEnv used by CLI and SDK.
+ * Packages can import this or create their own extensions of BaseProcessEnv.
+ */
+export type ProcessEnv = BaseEnv & {
+  // Terminal-specific
+  KITTY_WINDOW_ID?: string
+  SIXEL_SUPPORT?: string
+  ZED_NODE_ENV?: string
+
+  // VS Code family detection
+  VSCODE_THEME_KIND?: string
+  VSCODE_COLOR_THEME_KIND?: string
+  VSCODE_GIT_IPC_HANDLE?: string
+  VSCODE_PID?: string
+  VSCODE_CWD?: string
+  VSCODE_NLS_CONFIG?: string
+
+  // Cursor editor detection
+  CURSOR_PORT?: string
+  CURSOR?: string
+
+  // JetBrains IDE detection
+  JETBRAINS_REMOTE_RUN?: string
+  IDEA_INITIAL_DIRECTORY?: string
+  IDE_CONFIG_DIR?: string
+  JB_IDE_CONFIG_DIR?: string
+
+  // Editor preferences
+  VISUAL?: string
+  EDITOR?: string
+  ANVIL_CLI_EDITOR?: string
+  ANVIL_EDITOR?: string
+
+  // Theme preferences
+  OPEN_TUI_THEME?: string
+  OPENTUI_THEME?: string
+
+  // Anvil CLI-specific (set during binary build)
+  ANVIL_IS_BINARY?: string
+  ANVIL_CLI_VERSION?: string
+  ANVIL_CLI_TARGET?: string
+  ANVIL_RG_PATH?: string
+  ANVIL_WASM_DIR?: string
+
+  // Build/CI flags
+  VERBOSE?: string
+  OVERRIDE_TARGET?: string
+  OVERRIDE_PLATFORM?: string
+  OVERRIDE_ARCH?: string
+}
+
+// =============================================================================
+// FUNCTION TYPES - For dependency injection
+// =============================================================================
+
+/**
+ * Function type for getting a client env value.
+ */
+export type GetClientEnvFn = () => ClientEnv
+
+/**
+ * Function type for getting base env values.
+ */
+export type GetBaseEnvFn = () => BaseEnv
+
+/**
+ * Function type for getting full process env values.
+ */
+export type GetProcessEnvFn = () => ProcessEnv
+
+/**
+ * Function type for getting base CI env values.
+ */
+export type GetBaseCiEnvFn = () => BaseCiEnv
+
+/**
+ * Function type for getting extended CI env values.
+ */
+export type GetCiEnvFn = () => CiEnv
+
+// =============================================================================
+// COMBINED DEPS - For functions that need multiple env types
+// =============================================================================
+
+/**
+ * Base env deps - minimal set for most functions.
+ */
+export type BaseEnvDeps = {
+  clientEnv: ClientEnv
+  env: BaseEnv
+}
+
+/**
+ * Combined env deps with full process env.
+ * Used in CLI and client-side code.
+ */
+export type EnvDeps = {
+  clientEnv: ClientEnv
+  processEnv: ProcessEnv
+}
+
+/**
+ * Full env deps including CI vars.
+ * Note: ServerEnv is defined in packages/internal and should be imported from there.
+ */
+export type FullEnvDeps = {
+  clientEnv: ClientEnv
+  processEnv: ProcessEnv
+  ciEnv: CiEnv
+}