@agi-cli/server 0.1.59 → 0.1.62

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@agi-cli/server",
-	"version": "0.1.59",
+	"version": "0.1.62",
 	"description": "HTTP API server for AGI CLI",
 	"type": "module",
 	"main": "./src/index.ts",
@@ -29,8 +29,8 @@
 		"typecheck": "tsc --noEmit"
 	},
 	"dependencies": {
-		"@agi-cli/sdk": "0.1.59",
-		"@agi-cli/database": "0.1.59",
+		"@agi-cli/sdk": "0.1.62",
+		"@agi-cli/database": "0.1.62",
 		"drizzle-orm": "^0.44.5",
 		"hono": "^4.9.9"
 	},

package/src/openapi/spec.ts

@@ -14,6 +14,8 @@ export function getOpenAPISpec() {
 			{ name: 'messages' },
 			{ name: 'stream' },
 			{ name: 'ask' },
+			{ name: 'config' },
+			{ name: 'git' },
 		],
 		paths: {
 			'/v1/ask': {
@@ -237,6 +239,498 @@ export function getOpenAPISpec() {
 					},
 				},
 			},
+			'/v1/sessions/{sessionId}/abort': {
+				delete: {
+					tags: ['sessions'],
+					operationId: 'abortSession',
+					summary: 'Abort a running session',
+					description:
+						'Aborts any currently running assistant generation for the session',
+					parameters: [
+						{
+							in: 'path',
+							name: 'sessionId',
+							required: true,
+							schema: { type: 'string' },
+							description: 'Session ID to abort',
+						},
+					],
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: { success: { type: 'boolean' } },
+										required: ['success'],
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+			'/v1/config': {
+				get: {
+					tags: ['config'],
+					operationId: 'getConfig',
+					summary: 'Get full configuration',
+					description:
+						'Returns agents, authorized providers, models, and defaults',
+					parameters: [projectQueryParam()],
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: { $ref: '#/components/schemas/Config' },
+								},
+							},
+						},
+					},
+				},
+			},
+			'/v1/config/cwd': {
+				get: {
+					tags: ['config'],
+					operationId: 'getCwd',
+					summary: 'Get current working directory info',
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											cwd: { type: 'string' },
+											dirName: { type: 'string' },
+										},
+										required: ['cwd', 'dirName'],
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+			'/v1/config/agents': {
+				get: {
+					tags: ['config'],
+					operationId: 'getAgents',
+					summary: 'Get available agents',
+					parameters: [projectQueryParam()],
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											agents: {
+												type: 'array',
+												items: { type: 'string' },
+											},
+											default: { type: 'string' },
+										},
+										required: ['agents', 'default'],
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+			'/v1/config/providers': {
+				get: {
+					tags: ['config'],
+					operationId: 'getProviders',
+					summary: 'Get available providers',
+					description: 'Returns only providers that have been authorized',
+					parameters: [projectQueryParam()],
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											providers: {
+												type: 'array',
+												items: { $ref: '#/components/schemas/Provider' },
+											},
+											default: { $ref: '#/components/schemas/Provider' },
+										},
+										required: ['providers', 'default'],
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+			'/v1/config/providers/{provider}/models': {
+				get: {
+					tags: ['config'],
+					operationId: 'getProviderModels',
+					summary: 'Get available models for a provider',
+					parameters: [
+						projectQueryParam(),
+						{
+							in: 'path',
+							name: 'provider',
+							required: true,
+							schema: { $ref: '#/components/schemas/Provider' },
+						},
+					],
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											models: {
+												type: 'array',
+												items: { $ref: '#/components/schemas/Model' },
+											},
+											default: { type: 'string', nullable: true },
+										},
+										required: ['models'],
+									},
+								},
+							},
+						},
+						403: {
+							description: 'Provider not authorized',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: { error: { type: 'string' } },
+										required: ['error'],
+									},
+								},
+							},
+						},
+						404: {
+							description: 'Provider not found',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: { error: { type: 'string' } },
+										required: ['error'],
+									},
+								},
+							},
+						},
+					},
+				},
+			},
+			'/v1/git/status': {
+				get: {
+					tags: ['git'],
+					operationId: 'getGitStatus',
+					summary: 'Get git status',
+					description:
+						'Returns current git status including staged, unstaged, and untracked files',
+					parameters: [projectQueryParam()],
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											status: { type: 'string', enum: ['ok'] },
+											data: { $ref: '#/components/schemas/GitStatus' },
+										},
+										required: ['status', 'data'],
+									},
+								},
+							},
+						},
+						400: gitErrorResponse(),
+						500: gitErrorResponse(),
+					},
+				},
+			},
+			'/v1/git/diff': {
+				get: {
+					tags: ['git'],
+					operationId: 'getGitDiff',
+					summary: 'Get git diff for a file',
+					parameters: [
+						projectQueryParam(),
+						{
+							in: 'query',
+							name: 'file',
+							required: true,
+							schema: { type: 'string' },
+							description: 'File path to get diff for',
+						},
+						{
+							in: 'query',
+							name: 'staged',
+							required: false,
+							schema: { type: 'string', enum: ['true', 'false'] },
+							description: 'Show staged diff (default: unstaged)',
+						},
+					],
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											status: { type: 'string', enum: ['ok'] },
+											data: { $ref: '#/components/schemas/GitDiff' },
+										},
+										required: ['status', 'data'],
+									},
+								},
+							},
+						},
+						400: gitErrorResponse(),
+						500: gitErrorResponse(),
+					},
+				},
+			},
+			'/v1/git/branch': {
+				get: {
+					tags: ['git'],
+					operationId: 'getGitBranch',
+					summary: 'Get git branch information',
+					parameters: [projectQueryParam()],
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											status: { type: 'string', enum: ['ok'] },
+											data: { $ref: '#/components/schemas/GitBranch' },
+										},
+										required: ['status', 'data'],
+									},
+								},
+							},
+						},
+						400: gitErrorResponse(),
+						500: gitErrorResponse(),
+					},
+				},
+			},
+			'/v1/git/stage': {
+				post: {
+					tags: ['git'],
+					operationId: 'stageFiles',
+					summary: 'Stage files',
+					requestBody: {
+						required: true,
+						content: {
+							'application/json': {
+								schema: {
+									type: 'object',
+									properties: {
+										project: { type: 'string' },
+										files: {
+											type: 'array',
+											items: { type: 'string' },
+										},
+									},
+									required: ['files'],
+								},
+							},
+						},
+					},
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											status: { type: 'string', enum: ['ok'] },
+											data: {
+												type: 'object',
+												properties: {
+													staged: {
+														type: 'array',
+														items: { type: 'string' },
+													},
+													failed: {
+														type: 'array',
+														items: { type: 'string' },
+													},
+												},
+												required: ['staged', 'failed'],
+											},
+										},
+										required: ['status', 'data'],
+									},
+								},
+							},
+						},
+						500: gitErrorResponse(),
+					},
+				},
+			},
+			'/v1/git/unstage': {
+				post: {
+					tags: ['git'],
+					operationId: 'unstageFiles',
+					summary: 'Unstage files',
+					requestBody: {
+						required: true,
+						content: {
+							'application/json': {
+								schema: {
+									type: 'object',
+									properties: {
+										project: { type: 'string' },
+										files: {
+											type: 'array',
+											items: { type: 'string' },
+										},
+									},
+									required: ['files'],
+								},
+							},
+						},
+					},
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											status: { type: 'string', enum: ['ok'] },
+											data: {
+												type: 'object',
+												properties: {
+													unstaged: {
+														type: 'array',
+														items: { type: 'string' },
+													},
+													failed: {
+														type: 'array',
+														items: { type: 'string' },
+													},
+												},
+												required: ['unstaged', 'failed'],
+											},
+										},
+										required: ['status', 'data'],
+									},
+								},
+							},
+						},
+						500: gitErrorResponse(),
+					},
+				},
+			},
+			'/v1/git/commit': {
+				post: {
+					tags: ['git'],
+					operationId: 'commitChanges',
+					summary: 'Commit staged changes',
+					requestBody: {
+						required: true,
+						content: {
+							'application/json': {
+								schema: {
+									type: 'object',
+									properties: {
+										project: { type: 'string' },
+										message: { type: 'string', minLength: 1 },
+									},
+									required: ['message'],
+								},
+							},
+						},
+					},
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											status: { type: 'string', enum: ['ok'] },
+											data: { $ref: '#/components/schemas/GitCommit' },
+										},
+										required: ['status', 'data'],
+									},
+								},
+							},
+						},
+						400: gitErrorResponse(),
+						500: gitErrorResponse(),
+					},
+				},
+			},
+			'/v1/git/generate-commit-message': {
+				post: {
+					tags: ['git'],
+					operationId: 'generateCommitMessage',
+					summary: 'Generate AI-powered commit message',
+					description:
+						'Uses AI to generate a commit message based on staged changes',
+					requestBody: {
+						required: false,
+						content: {
+							'application/json': {
+								schema: {
+									type: 'object',
+									properties: {
+										project: { type: 'string' },
+									},
+								},
+							},
+						},
+					},
+					responses: {
+						200: {
+							description: 'OK',
+							content: {
+								'application/json': {
+									schema: {
+										type: 'object',
+										properties: {
+											status: { type: 'string', enum: ['ok'] },
+											data: {
+												type: 'object',
+												properties: {
+													message: { type: 'string' },
+												},
+												required: ['message'],
+											},
+										},
+										required: ['status', 'data'],
+									},
+								},
+							},
+						},
+						400: gitErrorResponse(),
+						500: gitErrorResponse(),
+					},
+				},
+			},
 		},
 		components: {
 			schemas: {
@@ -421,6 +915,134 @@ export function getOpenAPISpec() {
 					},
 					required: ['kind', 'path'],
 				},
+				Config: {
+					type: 'object',
+					properties: {
+						agents: {
+							type: 'array',
+							items: { type: 'string' },
+						},
+						providers: {
+							type: 'array',
+							items: { $ref: '#/components/schemas/Provider' },
+						},
+						defaults: {
+							type: 'object',
+							properties: {
+								agent: { type: 'string' },
+								provider: { $ref: '#/components/schemas/Provider' },
+								model: { type: 'string' },
+							},
+							required: ['agent', 'provider', 'model'],
+						},
+					},
+					required: ['agents', 'providers', 'defaults'],
+				},
+				Model: {
+					type: 'object',
+					properties: {
+						id: { type: 'string' },
+						label: { type: 'string' },
+						toolCall: { type: 'boolean' },
+						reasoning: { type: 'boolean' },
+					},
+					required: ['id', 'label'],
+				},
+				GitStatus: {
+					type: 'object',
+					properties: {
+						branch: { type: 'string' },
+						ahead: { type: 'integer' },
+						behind: { type: 'integer' },
+						staged: {
+							type: 'array',
+							items: { $ref: '#/components/schemas/GitFile' },
+						},
+						unstaged: {
+							type: 'array',
+							items: { $ref: '#/components/schemas/GitFile' },
+						},
+						untracked: {
+							type: 'array',
+							items: { $ref: '#/components/schemas/GitFile' },
+						},
+						hasChanges: { type: 'boolean' },
+					},
+					required: [
+						'branch',
+						'ahead',
+						'behind',
+						'staged',
+						'unstaged',
+						'untracked',
+						'hasChanges',
+					],
+				},
+				GitFile: {
+					type: 'object',
+					properties: {
+						path: { type: 'string' },
+						status: {
+							type: 'string',
+							enum: ['modified', 'added', 'deleted', 'renamed', 'untracked'],
+						},
+						staged: { type: 'boolean' },
+						insertions: { type: 'integer' },
+						deletions: { type: 'integer' },
+						oldPath: { type: 'string' },
+					},
+					required: ['path', 'status', 'staged'],
+				},
+				GitDiff: {
+					type: 'object',
+					properties: {
+						file: { type: 'string' },
+						diff: { type: 'string' },
+						insertions: { type: 'integer' },
+						deletions: { type: 'integer' },
+						language: { type: 'string' },
+						binary: { type: 'boolean' },
+					},
+					required: [
+						'file',
+						'diff',
+						'insertions',
+						'deletions',
+						'language',
+						'binary',
+					],
+				},
+				GitBranch: {
+					type: 'object',
+					properties: {
+						current: { type: 'string' },
+						upstream: { type: 'string' },
+						ahead: { type: 'integer' },
+						behind: { type: 'integer' },
+						all: {
+							type: 'array',
+							items: { type: 'string' },
+						},
+					},
+					required: ['current', 'upstream', 'ahead', 'behind', 'all'],
+				},
+				GitCommit: {
+					type: 'object',
+					properties: {
+						hash: { type: 'string' },
+						message: { type: 'string' },
+						filesChanged: { type: 'integer' },
+						insertions: { type: 'integer' },
+						deletions: { type: 'integer' },
+					},
+					required: [
+						'hash',
+						'message',
+						'filesChanged',
+						'insertions',
+						'deletions',
+					],
+				},
 			},
 		},
 	} as const;
@@ -472,3 +1094,22 @@ function errorResponse() {
 		},
 	} as const;
 }
+
+function gitErrorResponse() {
+	return {
+		description: 'Error',
+		content: {
+			'application/json': {
+				schema: {
+					type: 'object',
+					properties: {
+						status: { type: 'string', enum: ['error'] },
+						error: { type: 'string' },
+						code: { type: 'string' },
+					},
+					required: ['status', 'error'],
+				},
+			},
+		},
+	} as const;
+}

package/src/runtime/context-optimizer.ts

@@ -10,6 +10,17 @@ interface FileRead {
 	path: string;
 }
 
+interface ToolPart {
+	type: string;
+	input?: {
+		path?: string;
+		filePattern?: string;
+		pattern?: string;
+	};
+	output?: unknown;
+	[key: string]: unknown;
+}
+
 /**
  * Deduplicates file read results, keeping only the latest version of each file.
  *
@@ -38,7 +49,8 @@ export function deduplicateFileReads(messages: ModelMessage[]): ModelMessage[] {
 			if (!['read', 'grep', 'glob'].includes(toolName)) return;
 
 			// Extract file path from input
-			const input = (part as any).input;
+			const toolPart = part as ToolPart;
+			const input = toolPart.input;
 			if (!input) return;
 
 			const path = input.path || input.filePattern || input.pattern;
@@ -49,8 +61,8 @@ export function deduplicateFileReads(messages: ModelMessage[]): ModelMessage[] {
 				fileReads.set(path, []);
 			}
 			fileReads
-				.get(path)!
-				.push({ messageIndex: msgIdx, partIndex: partIdx, path });
+				.get(path)
+				?.push({ messageIndex: msgIdx, partIndex: partIdx, path });
 		});
 	});
 
@@ -112,7 +124,8 @@ export function pruneToolResults(
 			if (!toolType.startsWith('tool-')) return;
 
 			// Check if this has output
-			const hasOutput = (part as any).output !== undefined;
+			const toolPart = part as ToolPart;
+			const hasOutput = toolPart.output !== undefined;
 			if (!hasOutput) return;
 
 			toolResults.push({ messageIndex: msgIdx, partIndex: partIdx });
@@ -142,11 +155,12 @@ export function pruneToolResults(
 			if (!part || typeof part !== 'object') return part;
 			if (!('type' in part)) return part;
 
-			const toolType = (part as any).type as string;
+			const toolPart = part as ToolPart;
+			const toolType = toolPart.type;
 			if (!toolType.startsWith('tool-')) return part;
 
 			const key = `${msgIdx}-${partIdx}`;
-			const hasOutput = (part as any).output !== undefined;
+			const hasOutput = toolPart.output !== undefined;
 
 			// If this tool result should be pruned, remove its output
 			if (hasOutput && !toKeep.has(key)) {

package/src/runtime/history-truncator.ts

@@ -0,0 +1,26 @@
+import type { ModelMessage } from 'ai';
+
+/**
+ * Truncates conversation history to keep only the most recent messages.
+ * This helps manage context window size and improves performance.
+ *
+ * Strategy:
+ * - Keep only the last N messages
+ * - Preserve message pairs (assistant + user responses) when possible
+ * - Always keep at least the system message if present
+ */
+export function truncateHistory(
+	messages: ModelMessage[],
+	maxMessages: number,
+): ModelMessage[] {
+	if (messages.length <= maxMessages) {
+		return messages;
+	}
+
+	// Calculate how many messages to keep
+	const keepCount = Math.min(maxMessages, messages.length);
+	const startIndex = messages.length - keepCount;
+
+	// Return the most recent messages
+	return messages.slice(startIndex);
+}

package/src/runtime/runner.ts

@@ -27,7 +27,6 @@ import {
 	type RunnerToolContext,
 } from './tool-context-setup.ts';
 import {
-	updateSessionTokens,
 	updateSessionTokensIncremental,
 	updateMessageTokensIncremental,
 	completeAssistantMessage,
@@ -269,6 +268,7 @@ async function runAssistant(opts: RunOpts) {
 	);
 
 	try {
+		// @ts-expect-error this is fine 🔥
 		const result = streamText({
 			model,
 			tools: toolset,