btca-server 1.0.50 → 1.0.60

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "btca-server",
-	"version": "1.0.50",
+	"version": "1.0.60",
 	"description": "BTCA server for answering questions about your codebase using OpenCode AI",
 	"author": "Ben Davis",
 	"license": "MIT",
@@ -30,7 +30,9 @@
 	"exports": {
 		".": "./src/index.ts",
 		"./stream": "./src/stream/index.ts",
-		"./stream/types": "./src/stream/types.ts"
+		"./stream/types": "./src/stream/types.ts",
+		"./config/remote": "./src/config/remote.ts",
+		"./resources/schema": "./src/resources/schema.ts"
 	},
 	"files": [
 		"src",
@@ -49,9 +51,26 @@
 		"prettier": "^3.7.4"
 	},
 	"dependencies": {
+		"@ai-sdk/amazon-bedrock": "^4.0.30",
+		"@ai-sdk/anthropic": "^3.0.23",
+		"@ai-sdk/azure": "^3.0.18",
+		"@ai-sdk/cerebras": "^2.0.20",
+		"@ai-sdk/cohere": "^3.0.11",
+		"@ai-sdk/deepinfra": "^2.0.19",
+		"@ai-sdk/google": "^3.0.13",
+		"@ai-sdk/google-vertex": "^4.0.28",
+		"@ai-sdk/groq": "^3.0.15",
+		"@ai-sdk/mistral": "^3.0.12",
+		"@ai-sdk/openai": "^3.0.18",
+		"@ai-sdk/openai-compatible": "^2.0.18",
+		"@ai-sdk/perplexity": "^3.0.11",
+		"@ai-sdk/togetherai": "^2.0.20",
+		"@ai-sdk/xai": "^3.0.34",
 		"@btca/shared": "workspace:*",
 		"@opencode-ai/sdk": "^1.1.28",
+		"ai": "^6.0.49",
 		"hono": "^4.7.11",
+		"opencode-ai": "^1.1.36",
 		"zod": "^3.25.76"
 	}
 }

package/src/agent/agent.test.ts

@@ -103,8 +103,8 @@ describe('Agent', () => {
 			}
 
 			expect(events.length).toBeGreaterThan(0);
-			// Should have received some message.part.updated events
-			const textEvents = events.filter((e) => e.type === 'message.part.updated');
+			// Should have received some text-delta events
+			const textEvents = events.filter((e) => e.type === 'text-delta');
 			expect(textEvents.length).toBeGreaterThan(0);
 		}, 60000);
 	});

package/src/agent/index.ts

@@ -1,2 +1,3 @@
 export { Agent } from './service.ts';
+export { AgentLoop } from './loop.ts';
 export type { AgentResult, OcEvent, SessionState } from './types.ts';

package/src/agent/loop.ts

@@ -0,0 +1,295 @@
+/**
+ * Custom Agent Loop
+ * Uses AI SDK's streamText with custom tools
+ */
+import { streamText, tool, stepCountIs, type ModelMessage } from 'ai';
+
+import { Model } from '../providers/index.ts';
+import { ReadTool, GrepTool, GlobTool, ListTool } from '../tools/index.ts';
+
+export namespace AgentLoop {
+	// Event types for streaming
+	export type AgentEvent =
+		| { type: 'text-delta'; text: string }
+		| { type: 'tool-call'; toolName: string; input: unknown }
+		| { type: 'tool-result'; toolName: string; output: string }
+		| {
+				type: 'finish';
+				finishReason: string;
+				usage?: { inputTokens?: number; outputTokens?: number };
+		  }
+		| { type: 'error'; error: Error };
+
+	// Options for the agent loop
+	export type Options = {
+		providerId: string;
+		modelId: string;
+		collectionPath: string;
+		agentInstructions: string;
+		question: string;
+		maxSteps?: number;
+	};
+
+	// Result type
+	export type Result = {
+		answer: string;
+		model: { provider: string; model: string };
+		events: AgentEvent[];
+	};
+
+	/**
+	 * Build the system prompt for the agent
+	 */
+	function buildSystemPrompt(agentInstructions: string): string {
+		return [
+			'You are btca, an expert documentation search agent.',
+			'Your job is to answer questions by searching through the collection of resources.',
+			'',
+			'You have access to the following tools:',
+			'- read: Read file contents with line numbers',
+			'- grep: Search file contents using regex patterns',
+			'- glob: Find files matching glob patterns',
+			'- list: List directory contents',
+			'',
+			'Guidelines:',
+			'- Use glob to find relevant files first, then read them',
+			'- Use grep to search for specific code patterns or text',
+			'- Always cite the source files in your answers',
+			'- Be concise but thorough in your responses',
+			'- If you cannot find the answer, say so clearly',
+			'',
+			agentInstructions
+		].join('\n');
+	}
+
+	/**
+	 * Create the tools for the agent
+	 */
+	function createTools(basePath: string) {
+		return {
+			read: tool({
+				description: 'Read the contents of a file. Returns the file contents with line numbers.',
+				inputSchema: ReadTool.Parameters,
+				execute: async (params: ReadTool.ParametersType) => {
+					const result = await ReadTool.execute(params, { basePath });
+					return result.output;
+				}
+			}),
+
+			grep: tool({
+				description:
+					'Search for a regex pattern in file contents. Returns matching lines with file paths and line numbers.',
+				inputSchema: GrepTool.Parameters,
+				execute: async (params: GrepTool.ParametersType) => {
+					const result = await GrepTool.execute(params, { basePath });
+					return result.output;
+				}
+			}),
+
+			glob: tool({
+				description:
+					'Find files matching a glob pattern (e.g. "**/*.ts", "src/**/*.js"). Returns a list of matching file paths sorted by modification time.',
+				inputSchema: GlobTool.Parameters,
+				execute: async (params: GlobTool.ParametersType) => {
+					const result = await GlobTool.execute(params, { basePath });
+					return result.output;
+				}
+			}),
+
+			list: tool({
+				description:
+					'List the contents of a directory. Returns files and subdirectories with their types.',
+				inputSchema: ListTool.Parameters,
+				execute: async (params: ListTool.ParametersType) => {
+					const result = await ListTool.execute(params, { basePath });
+					return result.output;
+				}
+			})
+		};
+	}
+
+	/**
+	 * Get initial context by listing the collection directory
+	 */
+	async function getInitialContext(collectionPath: string): Promise<string> {
+		const result = await ListTool.execute({ path: '.' }, { basePath: collectionPath });
+		return `Collection contents:\n${result.output}`;
+	}
+
+	/**
+	 * Run the agent loop and return the final answer
+	 */
+	export async function run(options: Options): Promise<Result> {
+		const {
+			providerId,
+			modelId,
+			collectionPath,
+			agentInstructions,
+			question,
+			maxSteps = 40
+		} = options;
+
+		// Get the model
+		const model = await Model.getModel(providerId, modelId);
+
+		// Get initial context
+		const initialContext = await getInitialContext(collectionPath);
+
+		// Build messages
+		const messages: ModelMessage[] = [
+			{
+				role: 'user',
+				content: `${initialContext}\n\nQuestion: ${question}`
+			}
+		];
+
+		// Create tools
+		const tools = createTools(collectionPath);
+
+		// Collect events
+		const events: AgentEvent[] = [];
+		let fullText = '';
+
+		// Run streamText with tool execution
+		const result = streamText({
+			model,
+			system: buildSystemPrompt(agentInstructions),
+			messages,
+			tools,
+			stopWhen: stepCountIs(maxSteps)
+		});
+
+		// Process the stream
+		for await (const part of result.fullStream) {
+			switch (part.type) {
+				case 'text-delta':
+					fullText += part.text;
+					events.push({ type: 'text-delta', text: part.text });
+					break;
+
+				case 'tool-call':
+					events.push({
+						type: 'tool-call',
+						toolName: part.toolName,
+						input: part.input
+					});
+					break;
+
+				case 'tool-result':
+					events.push({
+						type: 'tool-result',
+						toolName: part.toolName,
+						output: typeof part.output === 'string' ? part.output : JSON.stringify(part.output)
+					});
+					break;
+
+				case 'finish':
+					events.push({
+						type: 'finish',
+						finishReason: part.finishReason ?? 'unknown',
+						usage: {
+							inputTokens: part.totalUsage?.inputTokens,
+							outputTokens: part.totalUsage?.outputTokens
+						}
+					});
+					break;
+
+				case 'error':
+					events.push({
+						type: 'error',
+						error: part.error instanceof Error ? part.error : new Error(String(part.error))
+					});
+					break;
+			}
+		}
+
+		return {
+			answer: fullText.trim(),
+			model: { provider: providerId, model: modelId },
+			events
+		};
+	}
+
+	/**
+	 * Run the agent loop and stream events
+	 */
+	export async function* stream(options: Options): AsyncGenerator<AgentEvent> {
+		const {
+			providerId,
+			modelId,
+			collectionPath,
+			agentInstructions,
+			question,
+			maxSteps = 40
+		} = options;
+
+		// Get the model
+		const model = await Model.getModel(providerId, modelId);
+
+		// Get initial context
+		const initialContext = await getInitialContext(collectionPath);
+
+		// Build messages
+		const messages: ModelMessage[] = [
+			{
+				role: 'user',
+				content: `${initialContext}\n\nQuestion: ${question}`
+			}
+		];
+
+		// Create tools
+		const tools = createTools(collectionPath);
+
+		// Run streamText with tool execution
+		const result = streamText({
+			model,
+			system: buildSystemPrompt(agentInstructions),
+			messages,
+			tools,
+			stopWhen: stepCountIs(maxSteps)
+		});
+
+		// Stream events
+		for await (const part of result.fullStream) {
+			switch (part.type) {
+				case 'text-delta':
+					yield { type: 'text-delta', text: part.text };
+					break;
+
+				case 'tool-call':
+					yield {
+						type: 'tool-call',
+						toolName: part.toolName,
+						input: part.input
+					};
+					break;
+
+				case 'tool-result':
+					yield {
+						type: 'tool-result',
+						toolName: part.toolName,
+						output: typeof part.output === 'string' ? part.output : JSON.stringify(part.output)
+					};
+					break;
+
+				case 'finish':
+					yield {
+						type: 'finish',
+						finishReason: part.finishReason ?? 'unknown',
+						usage: {
+							inputTokens: part.totalUsage?.inputTokens,
+							outputTokens: part.totalUsage?.outputTokens
+						}
+					};
+					break;
+
+				case 'error':
+					yield {
+						type: 'error',
+						error: part.error instanceof Error ? part.error : new Error(String(part.error))
+					};
+					break;
+			}
+		}
+	}
+}