|
|
@@ -1,1344 +0,0 @@
|
|
|
-import { z } from "zod"
|
|
|
-
|
|
|
-/**
|
|
|
- * TS
|
|
|
- */
|
|
|
-
|
|
|
-export type Keys<T> = keyof T
|
|
|
-
|
|
|
-export type Values<T> = T[keyof T]
|
|
|
-
|
|
|
-export type Equals<X, Y> = (<T>() => T extends X ? 1 : 2) extends <T>() => T extends Y ? 1 : 2 ? true : false
|
|
|
-
|
|
|
-export type AssertEqual<T extends true> = T
|
|
|
-
|
|
|
-/**
|
|
|
- * CodeAction
|
|
|
- */
|
|
|
-
|
|
|
-export const codeActionIds = ["explainCode", "fixCode", "improveCode", "addToContext", "newTask"] as const
|
|
|
-
|
|
|
-export type CodeActionId = (typeof codeActionIds)[number]
|
|
|
-
|
|
|
-export type CodeActionName = "EXPLAIN" | "FIX" | "IMPROVE" | "ADD_TO_CONTEXT" | "NEW_TASK"
|
|
|
-
|
|
|
-/**
|
|
|
- * TerminalAction
|
|
|
- */
|
|
|
-
|
|
|
-export const terminalActionIds = ["terminalAddToContext", "terminalFixCommand", "terminalExplainCommand"] as const
|
|
|
-
|
|
|
-export type TerminalActionId = (typeof terminalActionIds)[number]
|
|
|
-
|
|
|
-export type TerminalActionName = "ADD_TO_CONTEXT" | "FIX" | "EXPLAIN"
|
|
|
-
|
|
|
-export type TerminalActionPromptType = `TERMINAL_${TerminalActionName}`
|
|
|
-
|
|
|
-/**
|
|
|
- * Command
|
|
|
- */
|
|
|
-
|
|
|
-export const commandIds = [
|
|
|
- "activationCompleted",
|
|
|
-
|
|
|
- "plusButtonClicked",
|
|
|
- "promptsButtonClicked",
|
|
|
- "mcpButtonClicked",
|
|
|
- "historyButtonClicked",
|
|
|
- "popoutButtonClicked",
|
|
|
- "settingsButtonClicked",
|
|
|
-
|
|
|
- "openInNewTab",
|
|
|
-
|
|
|
- "showHumanRelayDialog",
|
|
|
- "registerHumanRelayCallback",
|
|
|
- "unregisterHumanRelayCallback",
|
|
|
- "handleHumanRelayResponse",
|
|
|
-
|
|
|
- "newTask",
|
|
|
-
|
|
|
- "setCustomStoragePath",
|
|
|
-
|
|
|
- "focusInput",
|
|
|
- "acceptInput",
|
|
|
-] as const
|
|
|
-
|
|
|
-export type CommandId = (typeof commandIds)[number]
|
|
|
-
|
|
|
-/**
|
|
|
- * ProviderName
|
|
|
- */
|
|
|
-
|
|
|
-export const providerNames = [
|
|
|
- "anthropic",
|
|
|
- "glama",
|
|
|
- "openrouter",
|
|
|
- "bedrock",
|
|
|
- "vertex",
|
|
|
- "openai",
|
|
|
- "ollama",
|
|
|
- "vscode-lm",
|
|
|
- "lmstudio",
|
|
|
- "gemini",
|
|
|
- "openai-native",
|
|
|
- "mistral",
|
|
|
- "deepseek",
|
|
|
- "unbound",
|
|
|
- "requesty",
|
|
|
- "human-relay",
|
|
|
- "fake-ai",
|
|
|
- "xai",
|
|
|
- "groq",
|
|
|
- "chutes",
|
|
|
- "litellm",
|
|
|
-] as const
|
|
|
-
|
|
|
-export const providerNamesSchema = z.enum(providerNames)
|
|
|
-
|
|
|
-export type ProviderName = z.infer<typeof providerNamesSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ToolGroup
|
|
|
- */
|
|
|
-
|
|
|
-export const toolGroups = ["read", "edit", "browser", "command", "mcp", "modes"] as const
|
|
|
-
|
|
|
-export const toolGroupsSchema = z.enum(toolGroups)
|
|
|
-
|
|
|
-export type ToolGroup = z.infer<typeof toolGroupsSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * Language
|
|
|
- */
|
|
|
-
|
|
|
-export const languages = [
|
|
|
- "ca",
|
|
|
- "de",
|
|
|
- "en",
|
|
|
- "es",
|
|
|
- "fr",
|
|
|
- "hi",
|
|
|
- "it",
|
|
|
- "ja",
|
|
|
- "ko",
|
|
|
- "nl",
|
|
|
- "pl",
|
|
|
- "pt-BR",
|
|
|
- "ru",
|
|
|
- "tr",
|
|
|
- "vi",
|
|
|
- "zh-CN",
|
|
|
- "zh-TW",
|
|
|
-] as const
|
|
|
-
|
|
|
-export const languagesSchema = z.enum(languages)
|
|
|
-
|
|
|
-export type Language = z.infer<typeof languagesSchema>
|
|
|
-
|
|
|
-export const isLanguage = (value: string): value is Language => languages.includes(value as Language)
|
|
|
-
|
|
|
-/**
|
|
|
- * TelemetrySetting
|
|
|
- */
|
|
|
-
|
|
|
-export const telemetrySettings = ["unset", "enabled", "disabled"] as const
|
|
|
-
|
|
|
-export const telemetrySettingsSchema = z.enum(telemetrySettings)
|
|
|
-
|
|
|
-export type TelemetrySetting = z.infer<typeof telemetrySettingsSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ReasoningEffort
|
|
|
- */
|
|
|
-
|
|
|
-export const reasoningEfforts = ["low", "medium", "high"] as const
|
|
|
-
|
|
|
-export const reasoningEffortsSchema = z.enum(reasoningEfforts)
|
|
|
-
|
|
|
-export type ReasoningEffort = z.infer<typeof reasoningEffortsSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ModelParameter
|
|
|
- */
|
|
|
-
|
|
|
-export const modelParameters = ["max_tokens", "temperature", "reasoning", "include_reasoning"] as const
|
|
|
-
|
|
|
-export const modelParametersSchema = z.enum(modelParameters)
|
|
|
-
|
|
|
-export type ModelParameter = z.infer<typeof modelParametersSchema>
|
|
|
-
|
|
|
-export const isModelParameter = (value: string): value is ModelParameter =>
|
|
|
- modelParameters.includes(value as ModelParameter)
|
|
|
-
|
|
|
-/**
|
|
|
- * ModelInfo
|
|
|
- */
|
|
|
-
|
|
|
-export const modelInfoSchema = z.object({
|
|
|
- maxTokens: z.number().nullish(),
|
|
|
- maxThinkingTokens: z.number().nullish(),
|
|
|
- contextWindow: z.number(),
|
|
|
- supportsImages: z.boolean().optional(),
|
|
|
- supportsComputerUse: z.boolean().optional(),
|
|
|
- supportsPromptCache: z.boolean(),
|
|
|
- supportsReasoningBudget: z.boolean().optional(),
|
|
|
- requiredReasoningBudget: z.boolean().optional(),
|
|
|
- supportsReasoningEffort: z.boolean().optional(),
|
|
|
- supportedParameters: z.array(modelParametersSchema).optional(),
|
|
|
- inputPrice: z.number().optional(),
|
|
|
- outputPrice: z.number().optional(),
|
|
|
- cacheWritesPrice: z.number().optional(),
|
|
|
- cacheReadsPrice: z.number().optional(),
|
|
|
- description: z.string().optional(),
|
|
|
- reasoningEffort: reasoningEffortsSchema.optional(),
|
|
|
- minTokensPerCachePoint: z.number().optional(),
|
|
|
- maxCachePoints: z.number().optional(),
|
|
|
- cachableFields: z.array(z.string()).optional(),
|
|
|
- tiers: z
|
|
|
- .array(
|
|
|
- z.object({
|
|
|
- contextWindow: z.number(),
|
|
|
- inputPrice: z.number().optional(),
|
|
|
- outputPrice: z.number().optional(),
|
|
|
- cacheWritesPrice: z.number().optional(),
|
|
|
- cacheReadsPrice: z.number().optional(),
|
|
|
- }),
|
|
|
- )
|
|
|
- .optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type ModelInfo = z.infer<typeof modelInfoSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * Codebase Index Config
|
|
|
- */
|
|
|
-export const codebaseIndexConfigSchema = z.object({
|
|
|
- codebaseIndexEnabled: z.boolean().optional(),
|
|
|
- codebaseIndexQdrantUrl: z.string().optional(),
|
|
|
- codebaseIndexEmbedderProvider: z.enum(["openai", "ollama"]).optional(),
|
|
|
- codebaseIndexEmbedderBaseUrl: z.string().optional(),
|
|
|
- codebaseIndexEmbedderModelId: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type CodebaseIndexConfig = z.infer<typeof codebaseIndexConfigSchema>
|
|
|
-
|
|
|
-export const codebaseIndexModelsSchema = z.object({
|
|
|
- openai: z.record(z.string(), z.object({ dimension: z.number() })).optional(),
|
|
|
- ollama: z.record(z.string(), z.object({ dimension: z.number() })).optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type CodebaseIndexModels = z.infer<typeof codebaseIndexModelsSchema>
|
|
|
-
|
|
|
-export const codebaseIndexProviderSchema = z.object({
|
|
|
- codeIndexOpenAiKey: z.string().optional(),
|
|
|
- codeIndexQdrantApiKey: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-/**
|
|
|
- * HistoryItem
|
|
|
- */
|
|
|
-
|
|
|
-export const historyItemSchema = z.object({
|
|
|
- id: z.string(),
|
|
|
- number: z.number(),
|
|
|
- ts: z.number(),
|
|
|
- task: z.string(),
|
|
|
- tokensIn: z.number(),
|
|
|
- tokensOut: z.number(),
|
|
|
- cacheWrites: z.number().optional(),
|
|
|
- cacheReads: z.number().optional(),
|
|
|
- totalCost: z.number(),
|
|
|
- size: z.number().optional(),
|
|
|
- workspace: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type HistoryItem = z.infer<typeof historyItemSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * GroupOptions
|
|
|
- */
|
|
|
-
|
|
|
-export const groupOptionsSchema = z.object({
|
|
|
- fileRegex: z
|
|
|
- .string()
|
|
|
- .optional()
|
|
|
- .refine(
|
|
|
- (pattern) => {
|
|
|
- if (!pattern) {
|
|
|
- return true // Optional, so empty is valid.
|
|
|
- }
|
|
|
-
|
|
|
- try {
|
|
|
- new RegExp(pattern)
|
|
|
- return true
|
|
|
- } catch {
|
|
|
- return false
|
|
|
- }
|
|
|
- },
|
|
|
- { message: "Invalid regular expression pattern" },
|
|
|
- ),
|
|
|
- description: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type GroupOptions = z.infer<typeof groupOptionsSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * GroupEntry
|
|
|
- */
|
|
|
-
|
|
|
-export const groupEntrySchema = z.union([toolGroupsSchema, z.tuple([toolGroupsSchema, groupOptionsSchema])])
|
|
|
-
|
|
|
-export type GroupEntry = z.infer<typeof groupEntrySchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ModeConfig
|
|
|
- */
|
|
|
-
|
|
|
-const groupEntryArraySchema = z.array(groupEntrySchema).refine(
|
|
|
- (groups) => {
|
|
|
- const seen = new Set()
|
|
|
-
|
|
|
- return groups.every((group) => {
|
|
|
- // For tuples, check the group name (first element).
|
|
|
- const groupName = Array.isArray(group) ? group[0] : group
|
|
|
-
|
|
|
- if (seen.has(groupName)) {
|
|
|
- return false
|
|
|
- }
|
|
|
-
|
|
|
- seen.add(groupName)
|
|
|
- return true
|
|
|
- })
|
|
|
- },
|
|
|
- { message: "Duplicate groups are not allowed" },
|
|
|
-)
|
|
|
-
|
|
|
-export const modeConfigSchema = z.object({
|
|
|
- slug: z.string().regex(/^[a-zA-Z0-9-]+$/, "Slug must contain only letters numbers and dashes"),
|
|
|
- name: z.string().min(1, "Name is required"),
|
|
|
- roleDefinition: z.string().min(1, "Role definition is required"),
|
|
|
- whenToUse: z.string().optional(),
|
|
|
- customInstructions: z.string().optional(),
|
|
|
- groups: groupEntryArraySchema,
|
|
|
- source: z.enum(["global", "project"]).optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type ModeConfig = z.infer<typeof modeConfigSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * CustomModesSettings
|
|
|
- */
|
|
|
-
|
|
|
-export const customModesSettingsSchema = z.object({
|
|
|
- customModes: z.array(modeConfigSchema).refine(
|
|
|
- (modes) => {
|
|
|
- const slugs = new Set()
|
|
|
-
|
|
|
- return modes.every((mode) => {
|
|
|
- if (slugs.has(mode.slug)) {
|
|
|
- return false
|
|
|
- }
|
|
|
-
|
|
|
- slugs.add(mode.slug)
|
|
|
- return true
|
|
|
- })
|
|
|
- },
|
|
|
- {
|
|
|
- message: "Duplicate mode slugs are not allowed",
|
|
|
- },
|
|
|
- ),
|
|
|
-})
|
|
|
-
|
|
|
-export type CustomModesSettings = z.infer<typeof customModesSettingsSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * PromptComponent
|
|
|
- */
|
|
|
-
|
|
|
-export const promptComponentSchema = z.object({
|
|
|
- roleDefinition: z.string().optional(),
|
|
|
- whenToUse: z.string().optional(),
|
|
|
- customInstructions: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type PromptComponent = z.infer<typeof promptComponentSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * CustomModePrompts
|
|
|
- */
|
|
|
-
|
|
|
-export const customModePromptsSchema = z.record(z.string(), promptComponentSchema.optional())
|
|
|
-
|
|
|
-export type CustomModePrompts = z.infer<typeof customModePromptsSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * CustomSupportPrompts
|
|
|
- */
|
|
|
-
|
|
|
-export const customSupportPromptsSchema = z.record(z.string(), z.string().optional())
|
|
|
-
|
|
|
-export type CustomSupportPrompts = z.infer<typeof customSupportPromptsSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * CommandExecutionStatus
|
|
|
- */
|
|
|
-
|
|
|
-export const commandExecutionStatusSchema = z.discriminatedUnion("status", [
|
|
|
- z.object({
|
|
|
- executionId: z.string(),
|
|
|
- status: z.literal("started"),
|
|
|
- pid: z.number().optional(),
|
|
|
- command: z.string(),
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- executionId: z.string(),
|
|
|
- status: z.literal("output"),
|
|
|
- output: z.string(),
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- executionId: z.string(),
|
|
|
- status: z.literal("exited"),
|
|
|
- exitCode: z.number().optional(),
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- executionId: z.string(),
|
|
|
- status: z.literal("fallback"),
|
|
|
- }),
|
|
|
-])
|
|
|
-
|
|
|
-export type CommandExecutionStatus = z.infer<typeof commandExecutionStatusSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ExperimentId
|
|
|
- */
|
|
|
-
|
|
|
-export const experimentIds = ["autoCondenseContext", "powerSteering"] as const
|
|
|
-
|
|
|
-export const experimentIdsSchema = z.enum(experimentIds)
|
|
|
-
|
|
|
-export type ExperimentId = z.infer<typeof experimentIdsSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * Experiments
|
|
|
- */
|
|
|
-
|
|
|
-const experimentsSchema = z.object({
|
|
|
- autoCondenseContext: z.boolean(),
|
|
|
- powerSteering: z.boolean(),
|
|
|
-})
|
|
|
-
|
|
|
-export type Experiments = z.infer<typeof experimentsSchema>
|
|
|
-
|
|
|
-type _AssertExperiments = AssertEqual<Equals<ExperimentId, Keys<Experiments>>>
|
|
|
-
|
|
|
-/**
|
|
|
- * ProviderSettingsEntry
|
|
|
- */
|
|
|
-
|
|
|
-export const providerSettingsEntrySchema = z.object({
|
|
|
- id: z.string(),
|
|
|
- name: z.string(),
|
|
|
- apiProvider: providerNamesSchema.optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type ProviderSettingsEntry = z.infer<typeof providerSettingsEntrySchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ProviderSettings
|
|
|
- */
|
|
|
-
|
|
|
-const baseProviderSettingsSchema = z.object({
|
|
|
- includeMaxTokens: z.boolean().optional(),
|
|
|
- diffEnabled: z.boolean().optional(),
|
|
|
- fuzzyMatchThreshold: z.number().optional(),
|
|
|
- modelTemperature: z.number().nullish(),
|
|
|
- rateLimitSeconds: z.number().optional(),
|
|
|
-
|
|
|
- // Model reasoning.
|
|
|
- enableReasoningEffort: z.boolean().optional(),
|
|
|
- reasoningEffort: reasoningEffortsSchema.optional(),
|
|
|
- modelMaxTokens: z.number().optional(),
|
|
|
- modelMaxThinkingTokens: z.number().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-// Several of the providers share common model config properties.
|
|
|
-const apiModelIdProviderModelSchema = baseProviderSettingsSchema.extend({
|
|
|
- apiModelId: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const anthropicSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- apiKey: z.string().optional(),
|
|
|
- anthropicBaseUrl: z.string().optional(),
|
|
|
- anthropicUseAuthToken: z.boolean().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const glamaSchema = baseProviderSettingsSchema.extend({
|
|
|
- glamaModelId: z.string().optional(),
|
|
|
- glamaApiKey: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const openRouterSchema = baseProviderSettingsSchema.extend({
|
|
|
- openRouterApiKey: z.string().optional(),
|
|
|
- openRouterModelId: z.string().optional(),
|
|
|
- openRouterBaseUrl: z.string().optional(),
|
|
|
- openRouterSpecificProvider: z.string().optional(),
|
|
|
- openRouterUseMiddleOutTransform: z.boolean().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const bedrockSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- awsAccessKey: z.string().optional(),
|
|
|
- awsSecretKey: z.string().optional(),
|
|
|
- awsSessionToken: z.string().optional(),
|
|
|
- awsRegion: z.string().optional(),
|
|
|
- awsUseCrossRegionInference: z.boolean().optional(),
|
|
|
- awsUsePromptCache: z.boolean().optional(),
|
|
|
- awsProfile: z.string().optional(),
|
|
|
- awsUseProfile: z.boolean().optional(),
|
|
|
- awsCustomArn: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const vertexSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- vertexKeyFile: z.string().optional(),
|
|
|
- vertexJsonCredentials: z.string().optional(),
|
|
|
- vertexProjectId: z.string().optional(),
|
|
|
- vertexRegion: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const openAiSchema = baseProviderSettingsSchema.extend({
|
|
|
- openAiBaseUrl: z.string().optional(),
|
|
|
- openAiApiKey: z.string().optional(),
|
|
|
- openAiLegacyFormat: z.boolean().optional(),
|
|
|
- openAiR1FormatEnabled: z.boolean().optional(),
|
|
|
- openAiModelId: z.string().optional(),
|
|
|
- openAiCustomModelInfo: modelInfoSchema.nullish(),
|
|
|
- openAiUseAzure: z.boolean().optional(),
|
|
|
- azureApiVersion: z.string().optional(),
|
|
|
- openAiStreamingEnabled: z.boolean().optional(),
|
|
|
- openAiHostHeader: z.string().optional(), // Keep temporarily for backward compatibility during migration.
|
|
|
- openAiHeaders: z.record(z.string(), z.string()).optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const ollamaSchema = baseProviderSettingsSchema.extend({
|
|
|
- ollamaModelId: z.string().optional(),
|
|
|
- ollamaBaseUrl: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const vsCodeLmSchema = baseProviderSettingsSchema.extend({
|
|
|
- vsCodeLmModelSelector: z
|
|
|
- .object({
|
|
|
- vendor: z.string().optional(),
|
|
|
- family: z.string().optional(),
|
|
|
- version: z.string().optional(),
|
|
|
- id: z.string().optional(),
|
|
|
- })
|
|
|
- .optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const lmStudioSchema = baseProviderSettingsSchema.extend({
|
|
|
- lmStudioModelId: z.string().optional(),
|
|
|
- lmStudioBaseUrl: z.string().optional(),
|
|
|
- lmStudioDraftModelId: z.string().optional(),
|
|
|
- lmStudioSpeculativeDecodingEnabled: z.boolean().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const geminiSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- geminiApiKey: z.string().optional(),
|
|
|
- googleGeminiBaseUrl: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const openAiNativeSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- openAiNativeApiKey: z.string().optional(),
|
|
|
- openAiNativeBaseUrl: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const mistralSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- mistralApiKey: z.string().optional(),
|
|
|
- mistralCodestralUrl: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const deepSeekSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- deepSeekBaseUrl: z.string().optional(),
|
|
|
- deepSeekApiKey: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const unboundSchema = baseProviderSettingsSchema.extend({
|
|
|
- unboundApiKey: z.string().optional(),
|
|
|
- unboundModelId: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const requestySchema = baseProviderSettingsSchema.extend({
|
|
|
- requestyApiKey: z.string().optional(),
|
|
|
- requestyModelId: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const humanRelaySchema = baseProviderSettingsSchema
|
|
|
-
|
|
|
-const fakeAiSchema = baseProviderSettingsSchema.extend({
|
|
|
- fakeAi: z.unknown().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const xaiSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- xaiApiKey: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const groqSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- groqApiKey: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const chutesSchema = apiModelIdProviderModelSchema.extend({
|
|
|
- chutesApiKey: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const litellmSchema = baseProviderSettingsSchema.extend({
|
|
|
- litellmBaseUrl: z.string().optional(),
|
|
|
- litellmApiKey: z.string().optional(),
|
|
|
- litellmModelId: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-const defaultSchema = z.object({
|
|
|
- apiProvider: z.undefined(),
|
|
|
-})
|
|
|
-
|
|
|
-export const providerSettingsSchemaDiscriminated = z.discriminatedUnion("apiProvider", [
|
|
|
- anthropicSchema.merge(z.object({ apiProvider: z.literal("anthropic") })),
|
|
|
- glamaSchema.merge(z.object({ apiProvider: z.literal("glama") })),
|
|
|
- openRouterSchema.merge(z.object({ apiProvider: z.literal("openrouter") })),
|
|
|
- bedrockSchema.merge(z.object({ apiProvider: z.literal("bedrock") })),
|
|
|
- vertexSchema.merge(z.object({ apiProvider: z.literal("vertex") })),
|
|
|
- openAiSchema.merge(z.object({ apiProvider: z.literal("openai") })),
|
|
|
- ollamaSchema.merge(z.object({ apiProvider: z.literal("ollama") })),
|
|
|
- vsCodeLmSchema.merge(z.object({ apiProvider: z.literal("vscode-lm") })),
|
|
|
- lmStudioSchema.merge(z.object({ apiProvider: z.literal("lmstudio") })),
|
|
|
- geminiSchema.merge(z.object({ apiProvider: z.literal("gemini") })),
|
|
|
- openAiNativeSchema.merge(z.object({ apiProvider: z.literal("openai-native") })),
|
|
|
- mistralSchema.merge(z.object({ apiProvider: z.literal("mistral") })),
|
|
|
- deepSeekSchema.merge(z.object({ apiProvider: z.literal("deepseek") })),
|
|
|
- unboundSchema.merge(z.object({ apiProvider: z.literal("unbound") })),
|
|
|
- requestySchema.merge(z.object({ apiProvider: z.literal("requesty") })),
|
|
|
- humanRelaySchema.merge(z.object({ apiProvider: z.literal("human-relay") })),
|
|
|
- fakeAiSchema.merge(z.object({ apiProvider: z.literal("fake-ai") })),
|
|
|
- xaiSchema.merge(z.object({ apiProvider: z.literal("xai") })),
|
|
|
- groqSchema.merge(z.object({ apiProvider: z.literal("groq") })),
|
|
|
- chutesSchema.merge(z.object({ apiProvider: z.literal("chutes") })),
|
|
|
- litellmSchema.merge(z.object({ apiProvider: z.literal("litellm") })),
|
|
|
- defaultSchema,
|
|
|
-])
|
|
|
-
|
|
|
-export const providerSettingsSchema = z.object({
|
|
|
- apiProvider: providerNamesSchema.optional(),
|
|
|
- ...anthropicSchema.shape,
|
|
|
- ...glamaSchema.shape,
|
|
|
- ...openRouterSchema.shape,
|
|
|
- ...bedrockSchema.shape,
|
|
|
- ...vertexSchema.shape,
|
|
|
- ...openAiSchema.shape,
|
|
|
- ...ollamaSchema.shape,
|
|
|
- ...vsCodeLmSchema.shape,
|
|
|
- ...lmStudioSchema.shape,
|
|
|
- ...geminiSchema.shape,
|
|
|
- ...openAiNativeSchema.shape,
|
|
|
- ...mistralSchema.shape,
|
|
|
- ...deepSeekSchema.shape,
|
|
|
- ...unboundSchema.shape,
|
|
|
- ...requestySchema.shape,
|
|
|
- ...humanRelaySchema.shape,
|
|
|
- ...fakeAiSchema.shape,
|
|
|
- ...xaiSchema.shape,
|
|
|
- ...groqSchema.shape,
|
|
|
- ...chutesSchema.shape,
|
|
|
- ...litellmSchema.shape,
|
|
|
- ...codebaseIndexProviderSchema.shape,
|
|
|
-})
|
|
|
-
|
|
|
-export type ProviderSettings = z.infer<typeof providerSettingsSchema>
|
|
|
-
|
|
|
-type ProviderSettingsRecord = Record<Keys<ProviderSettings>, undefined>
|
|
|
-
|
|
|
-const providerSettingsRecord: ProviderSettingsRecord = {
|
|
|
- apiProvider: undefined,
|
|
|
- // Anthropic
|
|
|
- apiModelId: undefined,
|
|
|
- apiKey: undefined,
|
|
|
- anthropicBaseUrl: undefined,
|
|
|
- anthropicUseAuthToken: undefined,
|
|
|
- // Glama
|
|
|
- glamaModelId: undefined,
|
|
|
- glamaApiKey: undefined,
|
|
|
- // OpenRouter
|
|
|
- openRouterApiKey: undefined,
|
|
|
- openRouterModelId: undefined,
|
|
|
- openRouterBaseUrl: undefined,
|
|
|
- openRouterSpecificProvider: undefined,
|
|
|
- openRouterUseMiddleOutTransform: undefined,
|
|
|
- // Amazon Bedrock
|
|
|
- awsAccessKey: undefined,
|
|
|
- awsSecretKey: undefined,
|
|
|
- awsSessionToken: undefined,
|
|
|
- awsRegion: undefined,
|
|
|
- awsUseCrossRegionInference: undefined,
|
|
|
- awsUsePromptCache: undefined,
|
|
|
- awsProfile: undefined,
|
|
|
- awsUseProfile: undefined,
|
|
|
- awsCustomArn: undefined,
|
|
|
- // Google Vertex
|
|
|
- vertexKeyFile: undefined,
|
|
|
- vertexJsonCredentials: undefined,
|
|
|
- vertexProjectId: undefined,
|
|
|
- vertexRegion: undefined,
|
|
|
- // OpenAI
|
|
|
- openAiBaseUrl: undefined,
|
|
|
- openAiApiKey: undefined,
|
|
|
- openAiLegacyFormat: undefined,
|
|
|
- openAiR1FormatEnabled: undefined,
|
|
|
- openAiModelId: undefined,
|
|
|
- openAiCustomModelInfo: undefined,
|
|
|
- openAiUseAzure: undefined,
|
|
|
- azureApiVersion: undefined,
|
|
|
- openAiStreamingEnabled: undefined,
|
|
|
- openAiHostHeader: undefined, // Keep temporarily for backward compatibility during migration
|
|
|
- openAiHeaders: undefined,
|
|
|
- // Ollama
|
|
|
- ollamaModelId: undefined,
|
|
|
- ollamaBaseUrl: undefined,
|
|
|
- // VS Code LM
|
|
|
- vsCodeLmModelSelector: undefined,
|
|
|
- lmStudioModelId: undefined,
|
|
|
- lmStudioBaseUrl: undefined,
|
|
|
- lmStudioDraftModelId: undefined,
|
|
|
- lmStudioSpeculativeDecodingEnabled: undefined,
|
|
|
- // Gemini
|
|
|
- geminiApiKey: undefined,
|
|
|
- googleGeminiBaseUrl: undefined,
|
|
|
- // OpenAI Native
|
|
|
- openAiNativeApiKey: undefined,
|
|
|
- openAiNativeBaseUrl: undefined,
|
|
|
- // Mistral
|
|
|
- mistralApiKey: undefined,
|
|
|
- mistralCodestralUrl: undefined,
|
|
|
- // DeepSeek
|
|
|
- deepSeekBaseUrl: undefined,
|
|
|
- deepSeekApiKey: undefined,
|
|
|
- // Unbound
|
|
|
- unboundApiKey: undefined,
|
|
|
- unboundModelId: undefined,
|
|
|
- // Requesty
|
|
|
- requestyApiKey: undefined,
|
|
|
- requestyModelId: undefined,
|
|
|
- // Code Index
|
|
|
- codeIndexOpenAiKey: undefined,
|
|
|
- codeIndexQdrantApiKey: undefined,
|
|
|
- // Reasoning
|
|
|
- enableReasoningEffort: undefined,
|
|
|
- reasoningEffort: undefined,
|
|
|
- modelMaxTokens: undefined,
|
|
|
- modelMaxThinkingTokens: undefined,
|
|
|
- // Generic
|
|
|
- includeMaxTokens: undefined,
|
|
|
- diffEnabled: undefined,
|
|
|
- fuzzyMatchThreshold: undefined,
|
|
|
- modelTemperature: undefined,
|
|
|
- rateLimitSeconds: undefined,
|
|
|
- // Fake AI
|
|
|
- fakeAi: undefined,
|
|
|
- // X.AI (Grok)
|
|
|
- xaiApiKey: undefined,
|
|
|
- // Groq
|
|
|
- groqApiKey: undefined,
|
|
|
- // Chutes AI
|
|
|
- chutesApiKey: undefined,
|
|
|
- // LiteLLM
|
|
|
- litellmBaseUrl: undefined,
|
|
|
- litellmApiKey: undefined,
|
|
|
- litellmModelId: undefined,
|
|
|
-}
|
|
|
-
|
|
|
-export const PROVIDER_SETTINGS_KEYS = Object.keys(providerSettingsRecord) as Keys<ProviderSettings>[]
|
|
|
-
|
|
|
-/**
|
|
|
- * GlobalSettings
|
|
|
- */
|
|
|
-
|
|
|
-export const globalSettingsSchema = z.object({
|
|
|
- currentApiConfigName: z.string().optional(),
|
|
|
- listApiConfigMeta: z.array(providerSettingsEntrySchema).optional(),
|
|
|
- pinnedApiConfigs: z.record(z.string(), z.boolean()).optional(),
|
|
|
-
|
|
|
- lastShownAnnouncementId: z.string().optional(),
|
|
|
- customInstructions: z.string().optional(),
|
|
|
- taskHistory: z.array(historyItemSchema).optional(),
|
|
|
-
|
|
|
- condensingApiConfigId: z.string().optional(),
|
|
|
- customCondensingPrompt: z.string().optional(),
|
|
|
-
|
|
|
- autoApprovalEnabled: z.boolean().optional(),
|
|
|
- alwaysAllowReadOnly: z.boolean().optional(),
|
|
|
- alwaysAllowReadOnlyOutsideWorkspace: z.boolean().optional(),
|
|
|
- codebaseIndexModels: codebaseIndexModelsSchema.optional(),
|
|
|
- codebaseIndexConfig: codebaseIndexConfigSchema.optional(),
|
|
|
- alwaysAllowWrite: z.boolean().optional(),
|
|
|
- alwaysAllowWriteOutsideWorkspace: z.boolean().optional(),
|
|
|
- writeDelayMs: z.number().optional(),
|
|
|
- alwaysAllowBrowser: z.boolean().optional(),
|
|
|
- alwaysApproveResubmit: z.boolean().optional(),
|
|
|
- requestDelaySeconds: z.number().optional(),
|
|
|
- alwaysAllowMcp: z.boolean().optional(),
|
|
|
- alwaysAllowModeSwitch: z.boolean().optional(),
|
|
|
- alwaysAllowSubtasks: z.boolean().optional(),
|
|
|
- alwaysAllowExecute: z.boolean().optional(),
|
|
|
- allowedCommands: z.array(z.string()).optional(),
|
|
|
- allowedMaxRequests: z.number().nullish(),
|
|
|
- autoCondenseContextPercent: z.number().optional(),
|
|
|
-
|
|
|
- browserToolEnabled: z.boolean().optional(),
|
|
|
- browserViewportSize: z.string().optional(),
|
|
|
- screenshotQuality: z.number().optional(),
|
|
|
- remoteBrowserEnabled: z.boolean().optional(),
|
|
|
- remoteBrowserHost: z.string().optional(),
|
|
|
- cachedChromeHostUrl: z.string().optional(),
|
|
|
-
|
|
|
- enableCheckpoints: z.boolean().optional(),
|
|
|
-
|
|
|
- ttsEnabled: z.boolean().optional(),
|
|
|
- ttsSpeed: z.number().optional(),
|
|
|
- soundEnabled: z.boolean().optional(),
|
|
|
- soundVolume: z.number().optional(),
|
|
|
-
|
|
|
- maxOpenTabsContext: z.number().optional(),
|
|
|
- maxWorkspaceFiles: z.number().optional(),
|
|
|
- showRooIgnoredFiles: z.boolean().optional(),
|
|
|
- maxReadFileLine: z.number().optional(),
|
|
|
-
|
|
|
- terminalOutputLineLimit: z.number().optional(),
|
|
|
- terminalShellIntegrationTimeout: z.number().optional(),
|
|
|
- terminalShellIntegrationDisabled: z.boolean().optional(),
|
|
|
- terminalCommandDelay: z.number().optional(),
|
|
|
- terminalPowershellCounter: z.boolean().optional(),
|
|
|
- terminalZshClearEolMark: z.boolean().optional(),
|
|
|
- terminalZshOhMy: z.boolean().optional(),
|
|
|
- terminalZshP10k: z.boolean().optional(),
|
|
|
- terminalZdotdir: z.boolean().optional(),
|
|
|
- terminalCompressProgressBar: z.boolean().optional(),
|
|
|
-
|
|
|
- rateLimitSeconds: z.number().optional(),
|
|
|
- diffEnabled: z.boolean().optional(),
|
|
|
- fuzzyMatchThreshold: z.number().optional(),
|
|
|
- experiments: experimentsSchema.optional(),
|
|
|
-
|
|
|
- language: languagesSchema.optional(),
|
|
|
-
|
|
|
- telemetrySetting: telemetrySettingsSchema.optional(),
|
|
|
-
|
|
|
- mcpEnabled: z.boolean().optional(),
|
|
|
- enableMcpServerCreation: z.boolean().optional(),
|
|
|
-
|
|
|
- mode: z.string().optional(),
|
|
|
- modeApiConfigs: z.record(z.string(), z.string()).optional(),
|
|
|
- customModes: z.array(modeConfigSchema).optional(),
|
|
|
- customModePrompts: customModePromptsSchema.optional(),
|
|
|
- customSupportPrompts: customSupportPromptsSchema.optional(),
|
|
|
- enhancementApiConfigId: z.string().optional(),
|
|
|
- historyPreviewCollapsed: z.boolean().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type GlobalSettings = z.infer<typeof globalSettingsSchema>
|
|
|
-
|
|
|
-type GlobalSettingsRecord = Record<Keys<GlobalSettings>, undefined>
|
|
|
-
|
|
|
-const globalSettingsRecord: GlobalSettingsRecord = {
|
|
|
- codebaseIndexModels: undefined,
|
|
|
- codebaseIndexConfig: undefined,
|
|
|
- currentApiConfigName: undefined,
|
|
|
- listApiConfigMeta: undefined,
|
|
|
- pinnedApiConfigs: undefined,
|
|
|
-
|
|
|
- lastShownAnnouncementId: undefined,
|
|
|
- customInstructions: undefined,
|
|
|
- taskHistory: undefined,
|
|
|
-
|
|
|
- condensingApiConfigId: undefined,
|
|
|
- customCondensingPrompt: undefined,
|
|
|
-
|
|
|
- autoApprovalEnabled: undefined,
|
|
|
- alwaysAllowReadOnly: undefined,
|
|
|
- alwaysAllowReadOnlyOutsideWorkspace: undefined,
|
|
|
- alwaysAllowWrite: undefined,
|
|
|
- alwaysAllowWriteOutsideWorkspace: undefined,
|
|
|
- writeDelayMs: undefined,
|
|
|
- alwaysAllowBrowser: undefined,
|
|
|
- alwaysApproveResubmit: undefined,
|
|
|
- requestDelaySeconds: undefined,
|
|
|
- alwaysAllowMcp: undefined,
|
|
|
- alwaysAllowModeSwitch: undefined,
|
|
|
- alwaysAllowSubtasks: undefined,
|
|
|
- alwaysAllowExecute: undefined,
|
|
|
- allowedCommands: undefined,
|
|
|
- allowedMaxRequests: undefined,
|
|
|
- autoCondenseContextPercent: undefined,
|
|
|
-
|
|
|
- browserToolEnabled: undefined,
|
|
|
- browserViewportSize: undefined,
|
|
|
- screenshotQuality: undefined,
|
|
|
- remoteBrowserEnabled: undefined,
|
|
|
- remoteBrowserHost: undefined,
|
|
|
-
|
|
|
- enableCheckpoints: undefined,
|
|
|
-
|
|
|
- ttsEnabled: undefined,
|
|
|
- ttsSpeed: undefined,
|
|
|
- soundEnabled: undefined,
|
|
|
- soundVolume: undefined,
|
|
|
-
|
|
|
- maxOpenTabsContext: undefined,
|
|
|
- maxWorkspaceFiles: undefined,
|
|
|
- showRooIgnoredFiles: undefined,
|
|
|
- maxReadFileLine: undefined,
|
|
|
-
|
|
|
- terminalOutputLineLimit: undefined,
|
|
|
- terminalShellIntegrationTimeout: undefined,
|
|
|
- terminalShellIntegrationDisabled: undefined,
|
|
|
- terminalCommandDelay: undefined,
|
|
|
- terminalPowershellCounter: undefined,
|
|
|
- terminalZshClearEolMark: undefined,
|
|
|
- terminalZshOhMy: undefined,
|
|
|
- terminalZshP10k: undefined,
|
|
|
- terminalZdotdir: undefined,
|
|
|
- terminalCompressProgressBar: undefined,
|
|
|
-
|
|
|
- rateLimitSeconds: undefined,
|
|
|
- diffEnabled: undefined,
|
|
|
- fuzzyMatchThreshold: undefined,
|
|
|
- experiments: undefined,
|
|
|
-
|
|
|
- language: undefined,
|
|
|
-
|
|
|
- telemetrySetting: undefined,
|
|
|
-
|
|
|
- mcpEnabled: undefined,
|
|
|
- enableMcpServerCreation: undefined,
|
|
|
-
|
|
|
- mode: undefined,
|
|
|
- modeApiConfigs: undefined,
|
|
|
- customModes: undefined,
|
|
|
- customModePrompts: undefined,
|
|
|
- customSupportPrompts: undefined,
|
|
|
- enhancementApiConfigId: undefined,
|
|
|
- cachedChromeHostUrl: undefined,
|
|
|
- historyPreviewCollapsed: undefined,
|
|
|
-}
|
|
|
-
|
|
|
-export const GLOBAL_SETTINGS_KEYS = Object.keys(globalSettingsRecord) as Keys<GlobalSettings>[]
|
|
|
-
|
|
|
-/**
|
|
|
- * RooCodeSettings
|
|
|
- */
|
|
|
-
|
|
|
-export const rooCodeSettingsSchema = providerSettingsSchema.merge(globalSettingsSchema)
|
|
|
-
|
|
|
-export type RooCodeSettings = GlobalSettings & ProviderSettings
|
|
|
-
|
|
|
-/**
|
|
|
- * SecretState
|
|
|
- */
|
|
|
-
|
|
|
-export type SecretState = Pick<
|
|
|
- ProviderSettings,
|
|
|
- | "apiKey"
|
|
|
- | "glamaApiKey"
|
|
|
- | "openRouterApiKey"
|
|
|
- | "awsAccessKey"
|
|
|
- | "awsSecretKey"
|
|
|
- | "awsSessionToken"
|
|
|
- | "openAiApiKey"
|
|
|
- | "geminiApiKey"
|
|
|
- | "openAiNativeApiKey"
|
|
|
- | "deepSeekApiKey"
|
|
|
- | "mistralApiKey"
|
|
|
- | "unboundApiKey"
|
|
|
- | "requestyApiKey"
|
|
|
- | "xaiApiKey"
|
|
|
- | "groqApiKey"
|
|
|
- | "chutesApiKey"
|
|
|
- | "litellmApiKey"
|
|
|
- | "codeIndexOpenAiKey"
|
|
|
- | "codeIndexQdrantApiKey"
|
|
|
->
|
|
|
-
|
|
|
-export type CodeIndexSecrets = "codeIndexOpenAiKey" | "codeIndexQdrantApiKey"
|
|
|
-
|
|
|
-type SecretStateRecord = Record<Keys<SecretState>, undefined>
|
|
|
-
|
|
|
-const secretStateRecord: SecretStateRecord = {
|
|
|
- apiKey: undefined,
|
|
|
- glamaApiKey: undefined,
|
|
|
- openRouterApiKey: undefined,
|
|
|
- awsAccessKey: undefined,
|
|
|
- awsSecretKey: undefined,
|
|
|
- awsSessionToken: undefined,
|
|
|
- openAiApiKey: undefined,
|
|
|
- geminiApiKey: undefined,
|
|
|
- openAiNativeApiKey: undefined,
|
|
|
- deepSeekApiKey: undefined,
|
|
|
- mistralApiKey: undefined,
|
|
|
- unboundApiKey: undefined,
|
|
|
- requestyApiKey: undefined,
|
|
|
- xaiApiKey: undefined,
|
|
|
- groqApiKey: undefined,
|
|
|
- chutesApiKey: undefined,
|
|
|
- litellmApiKey: undefined,
|
|
|
- codeIndexOpenAiKey: undefined,
|
|
|
- codeIndexQdrantApiKey: undefined,
|
|
|
-}
|
|
|
-
|
|
|
-export const SECRET_STATE_KEYS = Object.keys(secretStateRecord) as Keys<SecretState>[]
|
|
|
-
|
|
|
-export const isSecretStateKey = (key: string): key is Keys<SecretState> =>
|
|
|
- SECRET_STATE_KEYS.includes(key as Keys<SecretState>)
|
|
|
-
|
|
|
-/**
|
|
|
- * GlobalState
|
|
|
- */
|
|
|
-
|
|
|
-export type GlobalState = Omit<RooCodeSettings, Keys<SecretState>>
|
|
|
-
|
|
|
-export const GLOBAL_STATE_KEYS = [...GLOBAL_SETTINGS_KEYS, ...PROVIDER_SETTINGS_KEYS].filter(
|
|
|
- (key: Keys<RooCodeSettings>) => !SECRET_STATE_KEYS.includes(key as Keys<SecretState>),
|
|
|
-) as Keys<GlobalState>[]
|
|
|
-
|
|
|
-export const isGlobalStateKey = (key: string): key is Keys<GlobalState> =>
|
|
|
- GLOBAL_STATE_KEYS.includes(key as Keys<GlobalState>)
|
|
|
-
|
|
|
-/**
|
|
|
- * ClineAsk
|
|
|
- */
|
|
|
-
|
|
|
-export const clineAsks = [
|
|
|
- "followup",
|
|
|
- "command",
|
|
|
- "command_output",
|
|
|
- "completion_result",
|
|
|
- "tool",
|
|
|
- "api_req_failed",
|
|
|
- "resume_task",
|
|
|
- "resume_completed_task",
|
|
|
- "mistake_limit_reached",
|
|
|
- "browser_action_launch",
|
|
|
- "use_mcp_server",
|
|
|
- "auto_approval_max_req_reached",
|
|
|
-] as const
|
|
|
-
|
|
|
-export const clineAskSchema = z.enum(clineAsks)
|
|
|
-
|
|
|
-export type ClineAsk = z.infer<typeof clineAskSchema>
|
|
|
-
|
|
|
-// ClineSay
|
|
|
-
|
|
|
-export const clineSays = [
|
|
|
- "error",
|
|
|
- "api_req_started",
|
|
|
- "api_req_finished",
|
|
|
- "api_req_retried",
|
|
|
- "api_req_retry_delayed",
|
|
|
- "api_req_deleted",
|
|
|
- "text",
|
|
|
- "reasoning",
|
|
|
- "completion_result",
|
|
|
- "user_feedback",
|
|
|
- "user_feedback_diff",
|
|
|
- "command_output",
|
|
|
- "shell_integration_warning",
|
|
|
- "browser_action",
|
|
|
- "browser_action_result",
|
|
|
- "mcp_server_request_started",
|
|
|
- "mcp_server_response",
|
|
|
- "subtask_result",
|
|
|
- "checkpoint_saved",
|
|
|
- "rooignore_error",
|
|
|
- "diff_error",
|
|
|
- "condense_context",
|
|
|
- "codebase_search_result",
|
|
|
-] as const
|
|
|
-
|
|
|
-export const clineSaySchema = z.enum(clineSays)
|
|
|
-
|
|
|
-export type ClineSay = z.infer<typeof clineSaySchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ToolProgressStatus
|
|
|
- */
|
|
|
-
|
|
|
-export const toolProgressStatusSchema = z.object({
|
|
|
- icon: z.string().optional(),
|
|
|
- text: z.string().optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type ToolProgressStatus = z.infer<typeof toolProgressStatusSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ContextCondense
|
|
|
- */
|
|
|
-
|
|
|
-export const contextCondenseSchema = z.object({
|
|
|
- cost: z.number(),
|
|
|
- prevContextTokens: z.number(),
|
|
|
- newContextTokens: z.number(),
|
|
|
- summary: z.string(),
|
|
|
-})
|
|
|
-
|
|
|
-export type ContextCondense = z.infer<typeof contextCondenseSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ClineMessage
|
|
|
- */
|
|
|
-
|
|
|
-export const clineMessageSchema = z.object({
|
|
|
- ts: z.number(),
|
|
|
- type: z.union([z.literal("ask"), z.literal("say")]),
|
|
|
- ask: clineAskSchema.optional(),
|
|
|
- say: clineSaySchema.optional(),
|
|
|
- text: z.string().optional(),
|
|
|
- images: z.array(z.string()).optional(),
|
|
|
- partial: z.boolean().optional(),
|
|
|
- reasoning: z.string().optional(),
|
|
|
- conversationHistoryIndex: z.number().optional(),
|
|
|
- checkpoint: z.record(z.string(), z.unknown()).optional(),
|
|
|
- progressStatus: toolProgressStatusSchema.optional(),
|
|
|
- contextCondense: contextCondenseSchema.optional(),
|
|
|
-})
|
|
|
-
|
|
|
-export type ClineMessage = z.infer<typeof clineMessageSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * TokenUsage
|
|
|
- */
|
|
|
-
|
|
|
-export const tokenUsageSchema = z.object({
|
|
|
- totalTokensIn: z.number(),
|
|
|
- totalTokensOut: z.number(),
|
|
|
- totalCacheWrites: z.number().optional(),
|
|
|
- totalCacheReads: z.number().optional(),
|
|
|
- totalCost: z.number(),
|
|
|
- contextTokens: z.number(),
|
|
|
-})
|
|
|
-
|
|
|
-export type TokenUsage = z.infer<typeof tokenUsageSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ToolName
|
|
|
- */
|
|
|
-
|
|
|
-export const toolNames = [
|
|
|
- "execute_command",
|
|
|
- "read_file",
|
|
|
- "write_to_file",
|
|
|
- "apply_diff",
|
|
|
- "insert_content",
|
|
|
- "search_and_replace",
|
|
|
- "search_files",
|
|
|
- "list_files",
|
|
|
- "list_code_definition_names",
|
|
|
- "browser_action",
|
|
|
- "use_mcp_tool",
|
|
|
- "access_mcp_resource",
|
|
|
- "ask_followup_question",
|
|
|
- "attempt_completion",
|
|
|
- "switch_mode",
|
|
|
- "new_task",
|
|
|
- "fetch_instructions",
|
|
|
- "codebase_search",
|
|
|
-] as const
|
|
|
-
|
|
|
-export const toolNamesSchema = z.enum(toolNames)
|
|
|
-
|
|
|
-export type ToolName = z.infer<typeof toolNamesSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * ToolUsage
|
|
|
- */
|
|
|
-
|
|
|
-export const toolUsageSchema = z.record(
|
|
|
- toolNamesSchema,
|
|
|
- z.object({
|
|
|
- attempts: z.number(),
|
|
|
- failures: z.number(),
|
|
|
- }),
|
|
|
-)
|
|
|
-
|
|
|
-export type ToolUsage = z.infer<typeof toolUsageSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * RooCodeEvent
|
|
|
- */
|
|
|
-
|
|
|
-export enum RooCodeEventName {
|
|
|
- Message = "message",
|
|
|
- TaskCreated = "taskCreated",
|
|
|
- TaskStarted = "taskStarted",
|
|
|
- TaskModeSwitched = "taskModeSwitched",
|
|
|
- TaskPaused = "taskPaused",
|
|
|
- TaskUnpaused = "taskUnpaused",
|
|
|
- TaskAskResponded = "taskAskResponded",
|
|
|
- TaskAborted = "taskAborted",
|
|
|
- TaskSpawned = "taskSpawned",
|
|
|
- TaskCompleted = "taskCompleted",
|
|
|
- TaskTokenUsageUpdated = "taskTokenUsageUpdated",
|
|
|
- TaskToolFailed = "taskToolFailed",
|
|
|
-}
|
|
|
-
|
|
|
-export const rooCodeEventsSchema = z.object({
|
|
|
- [RooCodeEventName.Message]: z.tuple([
|
|
|
- z.object({
|
|
|
- taskId: z.string(),
|
|
|
- action: z.union([z.literal("created"), z.literal("updated")]),
|
|
|
- message: clineMessageSchema,
|
|
|
- }),
|
|
|
- ]),
|
|
|
- [RooCodeEventName.TaskCreated]: z.tuple([z.string()]),
|
|
|
- [RooCodeEventName.TaskStarted]: z.tuple([z.string()]),
|
|
|
- [RooCodeEventName.TaskModeSwitched]: z.tuple([z.string(), z.string()]),
|
|
|
- [RooCodeEventName.TaskPaused]: z.tuple([z.string()]),
|
|
|
- [RooCodeEventName.TaskUnpaused]: z.tuple([z.string()]),
|
|
|
- [RooCodeEventName.TaskAskResponded]: z.tuple([z.string()]),
|
|
|
- [RooCodeEventName.TaskAborted]: z.tuple([z.string()]),
|
|
|
- [RooCodeEventName.TaskSpawned]: z.tuple([z.string(), z.string()]),
|
|
|
- [RooCodeEventName.TaskCompleted]: z.tuple([z.string(), tokenUsageSchema, toolUsageSchema]),
|
|
|
- [RooCodeEventName.TaskTokenUsageUpdated]: z.tuple([z.string(), tokenUsageSchema]),
|
|
|
- [RooCodeEventName.TaskToolFailed]: z.tuple([z.string(), toolNamesSchema, z.string()]),
|
|
|
-})
|
|
|
-
|
|
|
-export type RooCodeEvents = z.infer<typeof rooCodeEventsSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * Ack
|
|
|
- */
|
|
|
-
|
|
|
-export const ackSchema = z.object({
|
|
|
- clientId: z.string(),
|
|
|
- pid: z.number(),
|
|
|
- ppid: z.number(),
|
|
|
-})
|
|
|
-
|
|
|
-export type Ack = z.infer<typeof ackSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * TaskCommand
|
|
|
- */
|
|
|
-
|
|
|
-export enum TaskCommandName {
|
|
|
- StartNewTask = "StartNewTask",
|
|
|
- CancelTask = "CancelTask",
|
|
|
- CloseTask = "CloseTask",
|
|
|
-}
|
|
|
-
|
|
|
-export const taskCommandSchema = z.discriminatedUnion("commandName", [
|
|
|
- z.object({
|
|
|
- commandName: z.literal(TaskCommandName.StartNewTask),
|
|
|
- data: z.object({
|
|
|
- configuration: rooCodeSettingsSchema,
|
|
|
- text: z.string(),
|
|
|
- images: z.array(z.string()).optional(),
|
|
|
- newTab: z.boolean().optional(),
|
|
|
- }),
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- commandName: z.literal(TaskCommandName.CancelTask),
|
|
|
- data: z.string(),
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- commandName: z.literal(TaskCommandName.CloseTask),
|
|
|
- data: z.string(),
|
|
|
- }),
|
|
|
-])
|
|
|
-
|
|
|
-export type TaskCommand = z.infer<typeof taskCommandSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * TaskEvent
|
|
|
- */
|
|
|
-
|
|
|
-export const taskEventSchema = z.discriminatedUnion("eventName", [
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.Message),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.Message],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskCreated),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskCreated],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskStarted),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskStarted],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskModeSwitched),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskModeSwitched],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskPaused),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskPaused],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskUnpaused),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskUnpaused],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskAskResponded),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskAskResponded],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskAborted),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskAborted],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskSpawned),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskSpawned],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskCompleted),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskCompleted],
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- eventName: z.literal(RooCodeEventName.TaskTokenUsageUpdated),
|
|
|
- payload: rooCodeEventsSchema.shape[RooCodeEventName.TaskTokenUsageUpdated],
|
|
|
- }),
|
|
|
-])
|
|
|
-
|
|
|
-export type TaskEvent = z.infer<typeof taskEventSchema>
|
|
|
-
|
|
|
-/**
|
|
|
- * IpcMessage
|
|
|
- */
|
|
|
-
|
|
|
-export enum IpcMessageType {
|
|
|
- Connect = "Connect",
|
|
|
- Disconnect = "Disconnect",
|
|
|
- Ack = "Ack",
|
|
|
- TaskCommand = "TaskCommand",
|
|
|
- TaskEvent = "TaskEvent",
|
|
|
-}
|
|
|
-
|
|
|
-export enum IpcOrigin {
|
|
|
- Client = "client",
|
|
|
- Server = "server",
|
|
|
-}
|
|
|
-
|
|
|
-export const ipcMessageSchema = z.discriminatedUnion("type", [
|
|
|
- z.object({
|
|
|
- type: z.literal(IpcMessageType.Ack),
|
|
|
- origin: z.literal(IpcOrigin.Server),
|
|
|
- data: ackSchema,
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- type: z.literal(IpcMessageType.TaskCommand),
|
|
|
- origin: z.literal(IpcOrigin.Client),
|
|
|
- clientId: z.string(),
|
|
|
- data: taskCommandSchema,
|
|
|
- }),
|
|
|
- z.object({
|
|
|
- type: z.literal(IpcMessageType.TaskEvent),
|
|
|
- origin: z.literal(IpcOrigin.Server),
|
|
|
- relayClientId: z.string().optional(),
|
|
|
- data: taskEventSchema,
|
|
|
- }),
|
|
|
-])
|
|
|
-
|
|
|
-export type IpcMessage = z.infer<typeof ipcMessageSchema>
|
Chris Estreich