greptile-mcp

Instructions

Implementation Reference

• src/server.ts:456-544 (handler)

  Primary handler function for the 'query_repository' MCP tool. Processes input arguments, manages session continuity, handles both streaming and non-streaming queries, and delegates execution to GreptileClient.queryRepositories.
  private async handleQueryRepository( args: unknown ): Promise<{ content: Array<{ type: string; text: string }> }> { if (!this.greptileClient) { return { content: [ { type: 'text', text: createErrorResponse( 'Cannot query repository: Missing environment variables. Use greptile_env_check for setup guidance.', 'Configuration Error', undefined ), }, ], }; } const { query, repositories = [], session_id, stream = false, genius = true, timeout, previous_messages = [], } = args as any; // Generate session ID if not provided const sessionId = session_id || generateSessionId(); // Prepare messages array const messages = [...previous_messages, { role: 'user' as const, content: query }]; if (stream) { // Handle streaming response const streamResults: string[] = []; const streamingResponse = await this.greptileClient.queryRepositories( messages, repositories, sessionId, true, genius, timeout ); for await (const chunk of streamingResponse as AsyncIterable<any>) { if (chunk.type === 'text' && chunk.content) { streamResults.push(chunk.content); } } return { content: [ { type: 'text', text: JSON.stringify( { message: streamResults.join(''), session_id: sessionId, streamed: true, }, null, 2 ), }, ], }; } else { // Handle regular response const result = await this.greptileClient.queryRepositories( messages, repositories, sessionId, false, genius, timeout ); return { content: [ { type: 'text', text: JSON.stringify({ ...result, session_id: sessionId }, null, 2), }, ], }; } }
• src/server.ts:234-235 (registration)

  Dispatch registration in the CallToolRequestSchema handler that routes 'query_repository' calls to the specific handler method.
  case 'query_repository': return await this.handleQueryRepository(request.params.arguments);
• src/server.ts:135-194 (schema)

  Tool registration and input schema definition for 'query_repository' in the ListToolsRequestSchema handler.
  { name: 'query_repository', description: 'Query repositories using natural language to get detailed answers with code references', inputSchema: { type: 'object', properties: { query: { type: 'string', description: 'Natural language query about the codebase', }, repositories: { type: 'array', items: { type: 'object', properties: { remote: { type: 'string', enum: ['github', 'gitlab'] }, repository: { type: 'string' }, branch: { type: 'string' }, }, required: ['remote', 'repository', 'branch'], }, description: 'List of repositories to query', }, session_id: { type: 'string', description: 'Session ID for conversation continuity (auto-generated if not provided)', }, stream: { type: 'boolean', description: 'Enable streaming response', default: false, }, genius: { type: 'boolean', description: 'Use enhanced query capabilities', default: true, }, timeout: { type: 'number', description: 'Request timeout in milliseconds', default: 60000, }, previous_messages: { type: 'array', items: { type: 'object', properties: { role: { type: 'string', enum: ['user', 'assistant'] }, content: { type: 'string' }, }, required: ['role', 'content'], }, description: 'Previous conversation messages for context', }, }, required: ['query'], }, },
• src/types/index.ts:78-86 (schema)

  TypeScript interface defining the input structure for QueryRepository tool.
  export interface QueryRepositoryInput { query: string; repositories?: Repository[]; session_id?: string; stream?: boolean; genius?: boolean; timeout?: number; previous_messages?: QueryMessage[]; }
• src/clients/greptile.ts:73-101 (helper)

  Core GreptileClient method that implements the actual query_repository logic by making HTTP POST request to Greptile API /query endpoint, handling both sync and stream modes.
  async queryRepositories( messages: QueryMessage[], repositories: Repository[], sessionId?: string, stream: boolean = false, genius: boolean = true, timeout?: number ): Promise<QueryResponse | AsyncIterable<StreamingChunk>> { if (stream) { return this.streamQueryRepositories(messages, repositories, sessionId, genius, timeout); } const url = `${this.baseUrl}/query`; const payload: Record<string, unknown> = { messages, stream: false, genius, }; if (repositories.length > 0) { payload.repositories = repositories; } if (sessionId) { payload.sessionId = sessionId; } const response = await this.makeRequest('POST', url, payload, timeout); return response as QueryResponse; }