onyx-kb-mcp 1.0.0

package/build/services/onyxApi.js

@@ -0,0 +1,367 @@
+/**
+ * Onyx API Service
+ * Handles all interactions with the Onyx API
+ */
+import axios from 'axios';
+import { DEBUG } from '../config/index.js';
+/**
+ * OnyxApiService class for interacting with the Onyx API
+ */
+export class OnyxApiService {
+    /**
+     * Constructor
+     * @param config The Onyx configuration
+     */
+    constructor(config) {
+        this.config = config;
+    }
+    /**
+     * Search Onyx for relevant documents
+     * @param query The search query
+     * @param documentSets Optional document sets to search within
+     * @param chunksAbove Number of chunks to include above the matching chunk
+     * @param chunksBelow Number of chunks to include below the matching chunk
+     * @returns Array of search results
+     */
+    async searchOnyx(query, documentSets = [], chunksAbove = 0, chunksBelow = 0) {
+        try {
+            const response = await axios.post(`${this.config.apiUrl}/admin/search`, {
+                query,
+                filters: {
+                    document_set: documentSets.length > 0 ? documentSets : null,
+                    source_type: null,
+                    time_cutoff: null,
+                    tags: null
+                },
+                chunks_above: chunksAbove,
+                chunks_below: chunksBelow,
+                evaluation_type: 'skip' // Disable LLM filtering - return all semantic matches, let Claude filter
+            }, {
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Authorization': `Bearer ${this.config.apiToken}`
+                }
+            });
+            return response.data.top_documents || response.data.documents || [];
+        }
+        catch (error) {
+            console.error('Error searching Onyx:', error);
+            throw new Error(`Failed to search Onyx: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Fetch a specific chunk from a document
+     * @param documentId The document ID
+     * @param chunkId The chunk ID
+     * @returns The chunk content
+     */
+    async fetchDocumentChunk(documentId, chunkId) {
+        try {
+            const encodedDocId = encodeURIComponent(documentId);
+            const response = await axios.get(`${this.config.apiUrl}/document/chunk-info?document_id=${encodedDocId}&chunk_id=${chunkId}`, {
+                headers: {
+                    'Authorization': `Bearer ${this.config.apiToken}`
+                }
+            });
+            return response.data.content || '';
+        }
+        catch (error) {
+            console.error(`Error fetching chunk ${chunkId} for document ${documentId}:`, error);
+            return '';
+        }
+    }
+    /**
+     * Fetch the full content of a document by retrieving all its chunks
+     * @param documentId The document ID
+     * @returns The full document content
+     */
+    async fetchDocumentContent(documentId) {
+        try {
+            const encodedDocId = encodeURIComponent(documentId);
+            // First get document info to know how many chunks there are
+            const infoResponse = await axios.get(`${this.config.apiUrl}/document/document-size-info?document_id=${encodedDocId}`, {
+                headers: {
+                    'Authorization': `Bearer ${this.config.apiToken}`
+                }
+            });
+            const numChunks = infoResponse.data.num_chunks;
+            // Fetch all chunks and combine them
+            let fullContent = '';
+            for (let i = 0; i < numChunks; i++) {
+                const chunkResponse = await axios.get(`${this.config.apiUrl}/document/chunk-info?document_id=${encodedDocId}&chunk_id=${i}`, {
+                    headers: {
+                        'Authorization': `Bearer ${this.config.apiToken}`
+                    }
+                });
+                fullContent += chunkResponse.data.content + '\n\n';
+            }
+            return fullContent;
+        }
+        catch (error) {
+            console.error(`Error fetching full content for document ${documentId}:`, error);
+            return '';
+        }
+    }
+    /**
+     * Create a new chat session
+     * @param personaId The persona ID to use
+     * @returns The chat session ID
+     */
+    async createChatSession(personaId) {
+        console.error(`Creating new chat session with persona_id: ${personaId}`);
+        const createSessionUrl = `${this.config.apiUrl}/chat/create-chat-session`;
+        console.error(`Sending request to: ${createSessionUrl}`);
+        try {
+            const chatSessionResponse = await axios.post(createSessionUrl, {
+                persona_id: personaId,
+                description: 'API Test Chat'
+            }, {
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Authorization': `Bearer ${this.config.apiToken}`
+                }
+            });
+            if (DEBUG) {
+                console.error(`Chat session response: ${JSON.stringify(chatSessionResponse.data)}`);
+                console.error(`Response status: ${chatSessionResponse.status}`);
+                console.error(`Response headers: ${JSON.stringify(chatSessionResponse.headers)}`);
+            }
+            const sessionId = chatSessionResponse.data.chat_session_id;
+            console.error(`Chat session created with ID: ${sessionId}`);
+            return sessionId;
+        }
+        catch (error) {
+            console.error('Error creating chat session:', error);
+            throw new Error(`Failed to create chat session: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Send a message to a chat session
+     * @param sessionId The chat session ID
+     * @param query The message to send
+     * @param documentSets Optional document sets to search within
+     * @returns The chat response
+     */
+    async sendChatMessage(sessionId, query, documentSets = []) {
+        const sendMessageUrl = `${this.config.apiUrl}/chat/send-message`;
+        console.error(`Sending message to: ${sendMessageUrl}`);
+        console.error(`With chat_session_id: ${sessionId}`);
+        // Build filters - only include document_set if explicitly specified
+        // Otherwise, let the persona's configured document sets be used
+        const filters = {
+            source_type: null,
+            time_cutoff: null,
+            tags: null
+        };
+        // Only add document_set filter if user explicitly specified document sets
+        if (documentSets.length > 0) {
+            filters.document_set = documentSets;
+        }
+        const messagePayload = {
+            chat_session_id: sessionId,
+            parent_message_id: null,
+            message: query,
+            search_doc_ids: [],
+            file_descriptors: [],
+            prompt_id: null,
+            retrieval_options: {
+                run_search: 'auto',
+                real_time: true,
+                filters
+            },
+            regenerate: false
+        };
+        if (DEBUG) {
+            console.error('=== DEBUG INFO ===');
+            console.error(`API URL: ${this.config.apiUrl}`);
+            console.error(`Send Message URL: ${sendMessageUrl}`);
+            console.error(`Message Payload: ${JSON.stringify(messagePayload)}`);
+        }
+        try {
+            const messageResponse = await axios.post(sendMessageUrl, messagePayload, {
+                headers: {
+                    'Content-Type': 'application/json',
+                    'Authorization': `Bearer ${this.config.apiToken}`
+                },
+                responseType: 'text'
+            });
+            // Get the response text
+            const responseText = messageResponse.data;
+            if (DEBUG) {
+                console.error(`Response status: ${messageResponse.status}`);
+                console.error(`Response headers: ${JSON.stringify(messageResponse.headers)}`);
+                console.error(`Response text (first 200 chars): ${responseText.substring(0, 200)}...`);
+            }
+            // Initialize variables for answer and documents
+            let answer = '';
+            // eslint-disable-next-line @typescript-eslint/no-explicit-any
+            let documents = [];
+            try {
+                // First try parsing as a single JSON object
+                const jsonResponse = JSON.parse(responseText);
+                if (DEBUG) {
+                    console.error(`Parsed response as single JSON: ${Object.keys(jsonResponse).join(', ')}`);
+                }
+                if (jsonResponse.answer) {
+                    answer = jsonResponse.answer;
+                }
+                if (jsonResponse.documents || jsonResponse.top_documents) {
+                    documents = jsonResponse.documents || jsonResponse.top_documents || [];
+                }
+            }
+            catch (e) {
+                // If that fails, try parsing as JSON lines (streaming response)
+                console.error(`Failed to parse as single JSON, trying JSON lines: ${e}`);
+                const responseLines = responseText.split('\n').filter(line => line.trim());
+                for (const line of responseLines) {
+                    try {
+                        const data = JSON.parse(line);
+                        // Handle legacy format
+                        if (data.answer_piece) {
+                            answer += data.answer_piece;
+                        }
+                        else if (data.top_documents) {
+                            documents = data.top_documents;
+                        }
+                        // Handle new streaming format: {"ind": X, "obj": {...}}
+                        if (data.obj) {
+                            const obj = data.obj;
+                            // Message content delta
+                            if (obj.type === 'message_delta' && obj.content) {
+                                answer += obj.content;
+                            }
+                            // Search results with documents
+                            if (obj.type === 'internal_search_tool_delta' && obj.documents && obj.documents.length > 0) {
+                                documents = obj.documents.map((doc) => ({
+                                    document_id: doc.document_id,
+                                    semantic_identifier: doc.semantic_identifier || '',
+                                    link: doc.link,
+                                    score: doc.score
+                                }));
+                            }
+                        }
+                    }
+                    catch (parseError) {
+                        // Skip invalid JSON lines
+                        if (DEBUG) {
+                            console.error(`Error parsing line: ${parseError}`);
+                        }
+                    }
+                }
+            }
+            return { answer, documents };
+        }
+        catch (error) {
+            console.error('Error sending chat message:', error);
+            if (axios.isAxiosError(error) && error.response) {
+                console.error('=== ERROR DETAILS ===');
+                console.error(`Response status: ${error.response?.status}`);
+                console.error(`Response data: ${JSON.stringify(error.response?.data)}`);
+                if (DEBUG) {
+                    console.error(`Request URL: ${error.config?.url}`);
+                    console.error(`Request method: ${error.config?.method}`);
+                    console.error(`Request headers: ${JSON.stringify(error.config?.headers)}`);
+                    console.error(`Request data: ${JSON.stringify(error.config?.data)}`);
+                }
+            }
+            throw new Error(`Error chatting with Onyx: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Build a comprehensive context from search results and content
+     * @param results The search results
+     * @param contents The content for each result
+     * @returns Formatted context string
+     */
+    buildContext(results, contents) {
+        let contextBuilder = '';
+        // Filter to valid results
+        const validResults = [];
+        for (let i = 0; i < results.length; i++) {
+            if (contents[i]) {
+                validResults.push({ result: results[i], content: contents[i], index: i });
+            }
+        }
+        // START with a summary of ALL sources found
+        contextBuilder += '## Search Results Summary\n';
+        contextBuilder += `**Found ${validResults.length} documents. All sources:**\n\n`;
+        validResults.forEach((item, i) => {
+            const link = item.result.link || item.result.document_id;
+            const name = item.result.semantic_identifier;
+            if (link.startsWith('http')) {
+                contextBuilder += `${i + 1}. [${name}](${link})\n`;
+            }
+            else {
+                contextBuilder += `${i + 1}. ${name}\n`;
+            }
+        });
+        contextBuilder += '\n---\n\n## Full Content\n\n';
+        // Add full content for each result
+        for (let i = 0; i < validResults.length; i++) {
+            const { result, content } = validResults[i];
+            contextBuilder += `### [${i + 1}/${validResults.length}] ${result.semantic_identifier}\n`;
+            contextBuilder += `**Source:** ${result.link || result.document_id}\n\n`;
+            contextBuilder += content;
+            contextBuilder += '\n\n---\n\n';
+        }
+        // END with reminder to check all results
+        contextBuilder += '\n## END OF RESULTS\n';
+        contextBuilder += `Total: ${validResults.length} documents. Make sure you reviewed all of them above.\n`;
+        return contextBuilder;
+    }
+    /**
+     * List all available personas/agents
+     * @returns Array of personas
+     */
+    async listPersonas() {
+        try {
+            const response = await axios.get(`${this.config.apiUrl}/persona`, {
+                headers: {
+                    'Authorization': `Bearer ${this.config.apiToken}`
+                }
+            });
+            return response.data || [];
+        }
+        catch (error) {
+            console.error('Error listing personas:', error);
+            throw new Error(`Failed to list personas: ${error instanceof Error ? error.message : String(error)}`);
+        }
+    }
+    /**
+     * Find a persona by name (case-insensitive)
+     * @param name The persona name to search for
+     * @returns The persona ID and name if found, null otherwise
+     */
+    async findPersonaByName(name) {
+        try {
+            const personas = await this.listPersonas();
+            console.error(`Found ${personas.length} personas. Available: ${personas.map(p => `"${p.name}" (${p.id})`).join(', ')}`);
+            const normalizedName = name.toLowerCase().trim();
+            // Try exact match first (case-insensitive)
+            const exactMatch = personas.find(p => p.name.toLowerCase().trim() === normalizedName);
+            if (exactMatch) {
+                console.error(`Found exact persona match: "${exactMatch.name}" (ID: ${exactMatch.id})`);
+                return { id: exactMatch.id, name: exactMatch.name };
+            }
+            // Try partial match (name contains search term)
+            const partialMatch = personas.find(p => p.name.toLowerCase().includes(normalizedName));
+            if (partialMatch) {
+                console.error(`Found partial persona match: "${partialMatch.name}" (ID: ${partialMatch.id})`);
+                return { id: partialMatch.id, name: partialMatch.name };
+            }
+            // Try if search term contains persona name
+            const reverseMatch = personas.find(p => normalizedName.includes(p.name.toLowerCase().trim()));
+            if (reverseMatch) {
+                console.error(`Found reverse persona match: "${reverseMatch.name}" (ID: ${reverseMatch.id})`);
+                return { id: reverseMatch.id, name: reverseMatch.name };
+            }
+            console.error(`No persona found matching "${name}".`);
+            return null;
+        }
+        catch (error) {
+            console.error('Error finding persona by name:', error);
+            return null;
+        }
+    }
+}
+//# sourceMappingURL=onyxApi.js.map

package/build/services/onyxApi.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"onyxApi.js","sourceRoot":"","sources":["../../src/services/onyxApi.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,KAAK,MAAM,OAAO,CAAC;AAE1B,OAAO,EAAE,KAAK,EAAE,MAAM,oBAAoB,CAAC;AAY3C;;GAEG;AACH,MAAM,OAAO,cAAc;IAGzB;;;OAGG;IACH,YAAY,MAAkB;QAC5B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;IAED;;;;;;;OAOG;IACH,KAAK,CAAC,UAAU,CACd,KAAa,EACb,eAAyB,EAAE,EAC3B,WAAW,GAAG,CAAC,EACf,WAAW,GAAG,CAAC;QAEf,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,IAAI,CAC/B,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,eAAe,EACpC;gBACE,KAAK;gBACL,OAAO,EAAE;oBACP,YAAY,EAAE,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,IAAI;oBAC3D,WAAW,EAAE,IAAI;oBACjB,WAAW,EAAE,IAAI;oBACjB,IAAI,EAAE,IAAI;iBACX;gBACD,YAAY,EAAE,WAAW;gBACzB,YAAY,EAAE,WAAW;gBACzB,eAAe,EAAE,MAAM,CAAC,yEAAyE;aAClG,EACD;gBACE,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;iBAClD;aACF,CACF,CAAC;YAEF,OAAO,QAAQ,CAAC,IAAI,CAAC,aAAa,IAAI,QAAQ,CAAC,IAAI,CAAC,SAAS,IAAI,EAAE,CAAC;QACtE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAK,CAAC,CAAC;YAC9C,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACtG,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,kBAAkB,CAAC,UAAkB,EAAE,OAAe;QAC1D,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YACpD,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAC9B,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,oCAAoC,YAAY,aAAa,OAAO,EAAE,EAC3F;gBACE,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;iBAClD;aACF,CACF,CAAC;YAEF,OAAO,QAAQ,CAAC,IAAI,CAAC,OAAO,IAAI,EAAE,CAAC;QACrC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,wBAAwB,OAAO,iBAAiB,UAAU,GAAG,EAAE,KAAK,CAAC,CAAC;YACpF,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,oBAAoB,CAAC,UAAkB;QAC3C,IAAI,CAAC;YACH,MAAM,YAAY,GAAG,kBAAkB,CAAC,UAAU,CAAC,CAAC;YAEpD,4DAA4D;YAC5D,MAAM,YAAY,GAAG,MAAM,KAAK,CAAC,GAAG,CAClC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,4CAA4C,YAAY,EAAE,EAC/E;gBACE,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;iBAClD;aACF,CACF,CAAC;YAEF,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,UAAU,CAAC;YAE/C,oCAAoC;YACpC,IAAI,WAAW,GAAG,EAAE,CAAC;YACrB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,EAAE,CAAC,EAAE,EAAE,CAAC;gBACnC,MAAM,aAAa,GAAG,MAAM,KAAK,CAAC,GAAG,CACnC,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,oCAAoC,YAAY,aAAa,CAAC,EAAE,EACrF;oBACE,OAAO,EAAE;wBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;qBAClD;iBACF,CACF,CAAC;gBAEF,WAAW,IAAI,aAAa,CAAC,IAAI,CAAC,OAAO,GAAG,MAAM,CAAC;YACrD,CAAC;YAED,OAAO,WAAW,CAAC;QACrB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,4CAA4C,UAAU,GAAG,EAAE,KAAK,CAAC,CAAC;YAChF,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,iBAAiB,CAAC,SAAiB;QACvC,OAAO,CAAC,KAAK,CAAC,8CAA8C,SAAS,EAAE,CAAC,CAAC;QACzE,MAAM,gBAAgB,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,2BAA2B,CAAC;QAC1E,OAAO,CAAC,KAAK,CAAC,uBAAuB,gBAAgB,EAAE,CAAC,CAAC;QAEzD,IAAI,CAAC;YACH,MAAM,mBAAmB,GAAG,MAAM,KAAK,CAAC,IAAI,CAC1C,gBAAgB,EAChB;gBACE,UAAU,EAAE,SAAS;gBACrB,WAAW,EAAE,eAAe;aAC7B,EACD;gBACE,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;iBAClD;aACF,CACF,CAAC;YAEF,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,CAAC,KAAK,CAAC,0BAA0B,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBACpF,OAAO,CAAC,KAAK,CAAC,oBAAoB,mBAAmB,CAAC,MAAM,EAAE,CAAC,CAAC;gBAChE,OAAO,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,SAAS,CAAC,mBAAmB,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;YACpF,CAAC;YAED,MAAM,SAAS,GAAG,mBAAmB,CAAC,IAAI,CAAC,eAAe,CAAC;YAC3D,OAAO,CAAC,KAAK,CAAC,iCAAiC,SAAS,EAAE,CAAC,CAAC;YAC5D,OAAO,SAAS,CAAC;QACnB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;YACrD,MAAM,IAAI,KAAK,CAAC,kCAAkC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9G,CAAC;IACH,CAAC;IAED;;;;;;OAMG;IACH,KAAK,CAAC,eAAe,CAAC,SAAiB,EAAE,KAAa,EAAE,eAAyB,EAAE;QACjF,MAAM,cAAc,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,oBAAoB,CAAC;QACjE,OAAO,CAAC,KAAK,CAAC,uBAAuB,cAAc,EAAE,CAAC,CAAC;QACvD,OAAO,CAAC,KAAK,CAAC,yBAAyB,SAAS,EAAE,CAAC,CAAC;QAEpD,oEAAoE;QACpE,gEAAgE;QAChE,MAAM,OAAO,GAA4B;YACvC,WAAW,EAAE,IAAI;YACjB,WAAW,EAAE,IAAI;YACjB,IAAI,EAAE,IAAI;SACX,CAAC;QAEF,0EAA0E;QAC1E,IAAI,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YAC5B,OAAO,CAAC,YAAY,GAAG,YAAY,CAAC;QACtC,CAAC;QAED,MAAM,cAAc,GAAG;YACrB,eAAe,EAAE,SAAS;YAC1B,iBAAiB,EAAE,IAAI;YACvB,OAAO,EAAE,KAAK;YACd,cAAc,EAAE,EAAE;YAClB,gBAAgB,EAAE,EAAE;YACpB,SAAS,EAAE,IAAI;YACf,iBAAiB,EAAE;gBACjB,UAAU,EAAE,MAAM;gBAClB,SAAS,EAAE,IAAI;gBACf,OAAO;aACR;YACD,UAAU,EAAE,KAAK;SAClB,CAAC;QAEF,IAAI,KAAK,EAAE,CAAC;YACV,OAAO,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACpC,OAAO,CAAC,KAAK,CAAC,YAAY,IAAI,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC;YAChD,OAAO,CAAC,KAAK,CAAC,qBAAqB,cAAc,EAAE,CAAC,CAAC;YACrD,OAAO,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,IAAI,CAAC;YACH,MAAM,eAAe,GAAG,MAAM,KAAK,CAAC,IAAI,CACtC,cAAc,EACd,cAAc,EACd;gBACE,OAAO,EAAE;oBACP,cAAc,EAAE,kBAAkB;oBAClC,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;iBAClD;gBACD,YAAY,EAAE,MAAM;aACrB,CACF,CAAC;YAEF,wBAAwB;YACxB,MAAM,YAAY,GAAG,eAAe,CAAC,IAAI,CAAC;YAC1C,IAAI,KAAK,EAAE,CAAC;gBACV,OAAO,CAAC,KAAK,CAAC,oBAAoB,eAAe,CAAC,MAAM,EAAE,CAAC,CAAC;gBAC5D,OAAO,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,SAAS,CAAC,eAAe,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;gBAC9E,OAAO,CAAC,KAAK,CAAC,oCAAoC,YAAY,CAAC,SAAS,CAAC,CAAC,EAAE,GAAG,CAAC,KAAK,CAAC,CAAC;YACzF,CAAC;YAED,gDAAgD;YAChD,IAAI,MAAM,GAAG,EAAE,CAAC;YAChB,8DAA8D;YAC9D,IAAI,SAAS,GAAmB,EAAE,CAAC;YAEnC,IAAI,CAAC;gBACH,4CAA4C;gBAC5C,MAAM,YAAY,GAAG,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,CAAC;gBAC9C,IAAI,KAAK,EAAE,CAAC;oBACV,OAAO,CAAC,KAAK,CAAC,mCAAmC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;gBAC3F,CAAC;gBAED,IAAI,YAAY,CAAC,MAAM,EAAE,CAAC;oBACxB,MAAM,GAAG,YAAY,CAAC,MAAM,CAAC;gBAC/B,CAAC;gBACD,IAAI,YAAY,CAAC,SAAS,IAAI,YAAY,CAAC,aAAa,EAAE,CAAC;oBACzD,SAAS,GAAG,YAAY,CAAC,SAAS,IAAI,YAAY,CAAC,aAAa,IAAI,EAAE,CAAC;gBACzE,CAAC;YACH,CAAC;YAAC,OAAO,CAAC,EAAE,CAAC;gBACX,gEAAgE;gBAChE,OAAO,CAAC,KAAK,CAAC,sDAAsD,CAAC,EAAE,CAAC,CAAC;gBACzE,MAAM,aAAa,GAAG,YAAY,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;gBAE3E,KAAK,MAAM,IAAI,IAAI,aAAa,EAAE,CAAC;oBACjC,IAAI,CAAC;wBACH,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;wBAE9B,uBAAuB;wBACvB,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;4BACtB,MAAM,IAAI,IAAI,CAAC,YAAY,CAAC;wBAC9B,CAAC;6BAAM,IAAI,IAAI,CAAC,aAAa,EAAE,CAAC;4BAC9B,SAAS,GAAG,IAAI,CAAC,aAAa,CAAC;wBACjC,CAAC;wBAED,wDAAwD;wBACxD,IAAI,IAAI,CAAC,GAAG,EAAE,CAAC;4BACb,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC;4BAErB,wBAAwB;4BACxB,IAAI,GAAG,CAAC,IAAI,KAAK,eAAe,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;gCAChD,MAAM,IAAI,GAAG,CAAC,OAAO,CAAC;4BACxB,CAAC;4BAED,gCAAgC;4BAChC,IAAI,GAAG,CAAC,IAAI,KAAK,4BAA4B,IAAI,GAAG,CAAC,SAAS,IAAI,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gCAC3F,SAAS,GAAG,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,GAK9B,EAAE,EAAE,CAAC,CAAC;oCACL,WAAW,EAAE,GAAG,CAAC,WAAW;oCAC5B,mBAAmB,EAAE,GAAG,CAAC,mBAAmB,IAAI,EAAE;oCAClD,IAAI,EAAE,GAAG,CAAC,IAAI;oCACd,KAAK,EAAE,GAAG,CAAC,KAAK;iCACjB,CAAC,CAAC,CAAC;4BACN,CAAC;wBACH,CAAC;oBACH,CAAC;oBAAC,OAAO,UAAU,EAAE,CAAC;wBACpB,0BAA0B;wBAC1B,IAAI,KAAK,EAAE,CAAC;4BACV,OAAO,CAAC,KAAK,CAAC,uBAAuB,UAAU,EAAE,CAAC,CAAC;wBACrD,CAAC;oBACH,CAAC;gBACH,CAAC;YACH,CAAC;YAED,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,6BAA6B,EAAE,KAAK,CAAC,CAAC;YACpD,IAAI,KAAK,CAAC,YAAY,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,EAAE,CAAC;gBAChD,OAAO,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;gBACvC,OAAO,CAAC,KAAK,CAAC,oBAAoB,KAAK,CAAC,QAAQ,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC5D,OAAO,CAAC,KAAK,CAAC,kBAAkB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,QAAQ,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;gBAExE,IAAI,KAAK,EAAE,CAAC;oBACV,OAAO,CAAC,KAAK,CAAC,gBAAgB,KAAK,CAAC,MAAM,EAAE,GAAG,EAAE,CAAC,CAAC;oBACnD,OAAO,CAAC,KAAK,CAAC,mBAAmB,KAAK,CAAC,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;oBACzD,OAAO,CAAC,KAAK,CAAC,oBAAoB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,CAAC,CAAC;oBAC3E,OAAO,CAAC,KAAK,CAAC,iBAAiB,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,MAAM,EAAE,IAAI,CAAC,EAAE,CAAC,CAAC;gBACvE,CAAC;YACH,CAAC;YACD,MAAM,IAAI,KAAK,CAAC,6BAA6B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACzG,CAAC;IACH,CAAC;IAED;;;;;OAKG;IACH,YAAY,CAAC,OAA2B,EAAE,QAAkB;QAC1D,IAAI,cAAc,GAAG,EAAE,CAAC;QAExB,0BAA0B;QAC1B,MAAM,YAAY,GAAmE,EAAE,CAAC;QACxF,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACxC,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;gBAChB,YAAY,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC;QAED,4CAA4C;QAC5C,cAAc,IAAI,6BAA6B,CAAC;QAChD,cAAc,IAAI,WAAW,YAAY,CAAC,MAAM,gCAAgC,CAAC;QACjF,YAAY,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE;YAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,IAAI,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC;YACzD,MAAM,IAAI,GAAG,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC;YAC7C,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE,CAAC;gBAC5B,cAAc,IAAI,GAAG,CAAC,GAAG,CAAC,MAAM,IAAI,KAAK,IAAI,KAAK,CAAC;YACrD,CAAC;iBAAM,CAAC;gBACN,cAAc,IAAI,GAAG,CAAC,GAAG,CAAC,KAAK,IAAI,IAAI,CAAC;YAC1C,CAAC;QACH,CAAC,CAAC,CAAC;QACH,cAAc,IAAI,8BAA8B,CAAC;QAEjD,mCAAmC;QACnC,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC7C,MAAM,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC;YAE5C,cAAc,IAAI,QAAQ,CAAC,GAAG,CAAC,IAAI,YAAY,CAAC,MAAM,KAAK,MAAM,CAAC,mBAAmB,IAAI,CAAC;YAC1F,cAAc,IAAI,eAAe,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,WAAW,MAAM,CAAC;YACzE,cAAc,IAAI,OAAO,CAAC;YAC1B,cAAc,IAAI,aAAa,CAAC;QAClC,CAAC;QAED,yCAAyC;QACzC,cAAc,IAAI,uBAAuB,CAAC;QAC1C,cAAc,IAAI,UAAU,YAAY,CAAC,MAAM,yDAAyD,CAAC;QAEzG,OAAO,cAAc,CAAC;IACxB,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,YAAY;QAChB,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,KAAK,CAAC,GAAG,CAC9B,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,UAAU,EAC/B;gBACE,OAAO,EAAE;oBACP,eAAe,EAAE,UAAU,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE;iBAClD;aACF,CACF,CAAC;YAEF,OAAO,QAAQ,CAAC,IAAI,IAAI,EAAE,CAAC;QAC7B,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAK,CAAC,CAAC;YAChD,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QACxG,CAAC;IACH,CAAC;IAED;;;;OAIG;IACH,KAAK,CAAC,iBAAiB,CAAC,IAAY;QAClC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,YAAY,EAAE,CAAC;YAC3C,OAAO,CAAC,KAAK,CAAC,SAAS,QAAQ,CAAC,MAAM,yBAAyB,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,CAAC,IAAI,MAAM,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAExH,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC;YAEjD,2CAA2C;YAC3C,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAC9B,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,KAAK,cAAc,CACpD,CAAC;YACF,IAAI,UAAU,EAAE,CAAC;gBACf,OAAO,CAAC,KAAK,CAAC,+BAA+B,UAAU,CAAC,IAAI,UAAU,UAAU,CAAC,EAAE,GAAG,CAAC,CAAC;gBACxF,OAAO,EAAE,EAAE,EAAE,UAAU,CAAC,EAAE,EAAE,IAAI,EAAE,UAAU,CAAC,IAAI,EAAE,CAAC;YACtD,CAAC;YAED,gDAAgD;YAChD,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAChC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,CACnD,CAAC;YACF,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,iCAAiC,YAAY,CAAC,IAAI,UAAU,YAAY,CAAC,EAAE,GAAG,CAAC,CAAC;gBAC9F,OAAO,EAAE,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,CAAC;YAC1D,CAAC;YAED,2CAA2C;YAC3C,MAAM,YAAY,GAAG,QAAQ,CAAC,IAAI,CAChC,CAAC,CAAC,EAAE,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,IAAI,EAAE,CAAC,CAC1D,CAAC;YACF,IAAI,YAAY,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,iCAAiC,YAAY,CAAC,IAAI,UAAU,YAAY,CAAC,EAAE,GAAG,CAAC,CAAC;gBAC9F,OAAO,EAAE,EAAE,EAAE,YAAY,CAAC,EAAE,EAAE,IAAI,EAAE,YAAY,CAAC,IAAI,EAAE,CAAC;YAC1D,CAAC;YAED,OAAO,CAAC,KAAK,CAAC,8BAA8B,IAAI,IAAI,CAAC,CAAC;YACtD,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;YACvD,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;CACF"}

package/build/tools/chatTool.d.ts

@@ -0,0 +1,25 @@
+import { OnyxApiService } from '../services/onyxApi.js';
+/**
+ * Handle the chat_with_onyx tool request
+ * @param args The tool arguments
+ * @param onyxApiService The Onyx API service
+ * @returns The tool response
+ */
+export declare function handleChatWithOnyx(args: unknown, onyxApiService: OnyxApiService): Promise<{
+    content: {
+        type: string;
+        text: string;
+        metadata: {
+            chat_session_id: string;
+            persona_id: number;
+            persona_name: string | null;
+        };
+    }[];
+    isError?: undefined;
+} | {
+    content: {
+        type: string;
+        text: string;
+    }[];
+    isError: boolean;
+}>;

package/build/tools/chatTool.js

@@ -0,0 +1,84 @@
+/**
+ * Chat Tool Handler
+ * Handles the chat_with_onyx tool requests
+ */
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
+const DEFAULT_PERSONA_ID = parseInt(process.env.DEFAULT_PERSONA_ID || '15', 10);
+/**
+ * Handle the chat_with_onyx tool request
+ * @param args The tool arguments
+ * @param onyxApiService The Onyx API service
+ * @returns The tool response
+ */
+export async function handleChatWithOnyx(args, onyxApiService) {
+    try {
+        if (typeof args !== 'object' || args === null) {
+            throw new McpError(ErrorCode.InvalidParams, 'Invalid arguments');
+        }
+        const { query, personaId: providedPersonaId, personaName, documentSets = [], 
+        // Unused parameter removed: enableAutoDetectFilters
+        chatSessionId = null } = args;
+        if (!query || typeof query !== 'string') {
+            throw new McpError(ErrorCode.InvalidParams, 'Query is required');
+        }
+        // Resolve persona ID from name if provided
+        let resolvedPersonaId = providedPersonaId ?? DEFAULT_PERSONA_ID;
+        let resolvedPersonaName = null;
+        let personaLookupMessage = '';
+        if (personaName && typeof personaName === 'string') {
+            console.error(`Looking up persona by name: "${personaName}"`);
+            const result = await onyxApiService.findPersonaByName(personaName);
+            if (result !== null) {
+                resolvedPersonaId = result.id;
+                resolvedPersonaName = result.name;
+                personaLookupMessage = `Persona found: "${resolvedPersonaName}" (ID: ${resolvedPersonaId})`;
+                console.error(personaLookupMessage);
+            }
+            else {
+                personaLookupMessage = `Persona "${personaName}" not found, using default (ID: ${resolvedPersonaId})`;
+                console.error(personaLookupMessage);
+            }
+        }
+        // Variable to store the chat session ID (either existing or new)
+        let sessionId = chatSessionId;
+        // Step 1: Create a chat session if one doesn't exist
+        if (!sessionId) {
+            sessionId = await onyxApiService.createChatSession(resolvedPersonaId);
+        }
+        else {
+            console.error(`Using existing chat session with ID: ${sessionId}`);
+        }
+        // Step 2: Send a message to the chat session
+        const { answer, documents } = await onyxApiService.sendChatMessage(sessionId, query, documentSets);
+        // Build persona info for response
+        const personaInfo = resolvedPersonaName
+            ? `Persona: "${resolvedPersonaName}" (ID: ${resolvedPersonaId})`
+            : `Persona ID: ${resolvedPersonaId}`;
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `${answer}\n\nSources:\n${documents.map(doc => `- ${doc.semantic_identifier || 'Unknown'} (${doc.document_id || 'Unknown ID'})`).join('\n')}\n\n---\n${personaInfo}\nChat Session ID: ${sessionId}`,
+                    metadata: {
+                        chat_session_id: sessionId,
+                        persona_id: resolvedPersonaId,
+                        persona_name: resolvedPersonaName
+                    }
+                }
+            ]
+        };
+    }
+    catch (error) {
+        console.error('Error in handleChatWithOnyx:', error);
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `Error chatting with Onyx: ${error instanceof Error ? error.message : String(error)}`
+                }
+            ],
+            isError: true
+        };
+    }
+}
+//# sourceMappingURL=chatTool.js.map

package/build/tools/chatTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"chatTool.js","sourceRoot":"","sources":["../../src/tools/chatTool.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,oCAAoC,CAAC;AAIzE,MAAM,kBAAkB,GAAG,QAAQ,CAAC,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,IAAI,EAAE,EAAE,CAAC,CAAC;AAEhF;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,kBAAkB,CAAC,IAAa,EAAE,cAA8B;IACpF,IAAI,CAAC;QACH,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAC9C,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,mBAAmB,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,EACJ,KAAK,EACL,SAAS,EAAE,iBAAiB,EAC5B,WAAW,EACX,YAAY,GAAG,EAAE;QACjB,oDAAoD;QACpD,aAAa,GAAG,IAAI,EACrB,GAAG,IAAkB,CAAC;QAEvB,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACxC,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,mBAAmB,CAAC,CAAC;QACnE,CAAC;QAED,2CAA2C;QAC3C,IAAI,iBAAiB,GAAG,iBAAiB,IAAI,kBAAkB,CAAC;QAChE,IAAI,mBAAmB,GAAkB,IAAI,CAAC;QAC9C,IAAI,oBAAoB,GAAG,EAAE,CAAC;QAE9B,IAAI,WAAW,IAAI,OAAO,WAAW,KAAK,QAAQ,EAAE,CAAC;YACnD,OAAO,CAAC,KAAK,CAAC,gCAAgC,WAAW,GAAG,CAAC,CAAC;YAC9D,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,iBAAiB,CAAC,WAAW,CAAC,CAAC;YACnE,IAAI,MAAM,KAAK,IAAI,EAAE,CAAC;gBACpB,iBAAiB,GAAG,MAAM,CAAC,EAAE,CAAC;gBAC9B,mBAAmB,GAAG,MAAM,CAAC,IAAI,CAAC;gBAClC,oBAAoB,GAAG,mBAAmB,mBAAmB,UAAU,iBAAiB,GAAG,CAAC;gBAC5F,OAAO,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACtC,CAAC;iBAAM,CAAC;gBACN,oBAAoB,GAAG,YAAY,WAAW,mCAAmC,iBAAiB,GAAG,CAAC;gBACtG,OAAO,CAAC,KAAK,CAAC,oBAAoB,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QAED,iEAAiE;QACjE,IAAI,SAAS,GAAG,aAAa,CAAC;QAE9B,qDAAqD;QACrD,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,SAAS,GAAG,MAAM,cAAc,CAAC,iBAAiB,CAAC,iBAAiB,CAAC,CAAC;QACxE,CAAC;aAAM,CAAC;YACN,OAAO,CAAC,KAAK,CAAC,wCAAwC,SAAS,EAAE,CAAC,CAAC;QACrE,CAAC;QAED,6CAA6C;QAC7C,MAAM,EAAE,MAAM,EAAE,SAAS,EAAE,GAAG,MAAM,cAAc,CAAC,eAAe,CAAC,SAAS,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC;QAEnG,kCAAkC;QAClC,MAAM,WAAW,GAAG,mBAAmB;YACrC,CAAC,CAAC,aAAa,mBAAmB,UAAU,iBAAiB,GAAG;YAChE,CAAC,CAAC,eAAe,iBAAiB,EAAE,CAAC;QAEvC,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,GAAG,MAAM,iBAAiB,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAClD,KAAK,GAAG,CAAC,mBAAmB,IAAI,SAAS,KAAK,GAAG,CAAC,WAAW,IAAI,YAAY,GAAG,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,WAAW,sBAAsB,SAAS,EAAE;oBACtJ,QAAQ,EAAE;wBACR,eAAe,EAAE,SAAS;wBAC1B,UAAU,EAAE,iBAAiB;wBAC7B,YAAY,EAAE,mBAAmB;qBAClC;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,8BAA8B,EAAE,KAAK,CAAC,CAAC;QACrD,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,6BAA6B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;iBAC5F;aACF;YACD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC"}

package/build/tools/index.d.ts

@@ -0,0 +1,7 @@
+/**
+ * Tool Handlers
+ * Exports all tool handlers for the Arqitech MCP Server
+ */
+export { handleSearchOnyx } from './searchTool.js';
+export { handleChatWithOnyx } from './chatTool.js';
+export { handleSummarizeArq } from './summarizeTool.js';

package/build/tools/index.js

@@ -0,0 +1,8 @@
+/**
+ * Tool Handlers
+ * Exports all tool handlers for the Arqitech MCP Server
+ */
+export { handleSearchOnyx } from './searchTool.js';
+export { handleChatWithOnyx } from './chatTool.js';
+export { handleSummarizeArq } from './summarizeTool.js';
+//# sourceMappingURL=index.js.map

package/build/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,gBAAgB,EAAE,MAAM,iBAAiB,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,eAAe,CAAC;AACnD,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC"}

package/build/tools/searchTool.d.ts

@@ -0,0 +1,20 @@
+import { OnyxApiService } from '../services/onyxApi.js';
+/**
+ * Handle the search_arq tool request
+ * @param args The tool arguments
+ * @param onyxApiService The Onyx API service
+ * @returns The tool response
+ */
+export declare function handleSearchOnyx(args: unknown, onyxApiService: OnyxApiService): Promise<{
+    content: {
+        type: string;
+        text: string;
+    }[];
+    isError?: undefined;
+} | {
+    content: {
+        type: string;
+        text: string;
+    }[];
+    isError: boolean;
+}>;

package/build/tools/searchTool.js

@@ -0,0 +1,55 @@
+/**
+ * Search Tool Handler
+ * Handles the search_arq tool requests
+ */
+import { McpError, ErrorCode } from '@modelcontextprotocol/sdk/types.js';
+/**
+ * Handle the search_arq tool request
+ * @param args The tool arguments
+ * @param onyxApiService The Onyx API service
+ * @returns The tool response
+ */
+export async function handleSearchOnyx(args, onyxApiService) {
+    try {
+        if (typeof args !== 'object' || args === null) {
+            throw new McpError(ErrorCode.InvalidParams, 'Invalid arguments');
+        }
+        const { query, documentSets = [], maxResults = 10, chunksAbove = 1, chunksBelow = 1, retrieveFullDocuments = false } = args;
+        if (!query || typeof query !== 'string') {
+            throw new McpError(ErrorCode.InvalidParams, 'Query is required');
+        }
+        // Step 1: Perform semantic search to find relevant documents
+        const searchResults = await onyxApiService.searchOnyx(query, documentSets, chunksAbove, chunksBelow);
+        // Step 2: Get the most relevant results
+        const relevantResults = searchResults
+            .sort((a, b) => b.score - a.score)
+            .slice(0, maxResults);
+        // Step 3: For each relevant document, fetch either the specific chunk or full document content
+        const relevantContent = await Promise.all(relevantResults.map(result => retrieveFullDocuments
+            ? onyxApiService.fetchDocumentContent(result.document_id)
+            : onyxApiService.fetchDocumentChunk(result.document_id, result.chunk_ind)));
+        // Step 4: Combine into comprehensive context
+        const context = onyxApiService.buildContext(relevantResults, relevantContent);
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: context,
+                },
+            ],
+        };
+    }
+    catch (error) {
+        console.error('Error in handleSearchOnyx:', error);
+        return {
+            content: [
+                {
+                    type: 'text',
+                    text: `Error searching Onyx: ${error instanceof Error ? error.message : String(error)}`,
+                },
+            ],
+            isError: true,
+        };
+    }
+}
+//# sourceMappingURL=searchTool.js.map

package/build/tools/searchTool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"searchTool.js","sourceRoot":"","sources":["../../src/tools/searchTool.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAAE,QAAQ,EAAE,SAAS,EAAE,MAAM,oCAAoC,CAAC;AAIzE;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,gBAAgB,CAAC,IAAa,EAAE,cAA8B;IAClF,IAAI,CAAC;QACH,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,IAAI,EAAE,CAAC;YAC9C,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,mBAAmB,CAAC,CAAC;QACnE,CAAC;QAED,MAAM,EACJ,KAAK,EACL,YAAY,GAAG,EAAE,EACjB,UAAU,GAAG,EAAE,EACf,WAAW,GAAG,CAAC,EACf,WAAW,GAAG,CAAC,EACf,qBAAqB,GAAG,KAAK,EAC9B,GAAG,IAAoB,CAAC;QAEzB,IAAI,CAAC,KAAK,IAAI,OAAO,KAAK,KAAK,QAAQ,EAAE,CAAC;YACxC,MAAM,IAAI,QAAQ,CAAC,SAAS,CAAC,aAAa,EAAE,mBAAmB,CAAC,CAAC;QACnE,CAAC;QAED,6DAA6D;QAC7D,MAAM,aAAa,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,KAAK,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;QAErG,wCAAwC;QACxC,MAAM,eAAe,GAAG,aAAa;aAClC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC;aACjC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC;QAExB,+FAA+F;QAC/F,MAAM,eAAe,GAAG,MAAM,OAAO,CAAC,GAAG,CACvC,eAAe,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAC3B,qBAAqB;YACnB,CAAC,CAAC,cAAc,CAAC,oBAAoB,CAAC,MAAM,CAAC,WAAW,CAAC;YACzD,CAAC,CAAC,cAAc,CAAC,kBAAkB,CAAC,MAAM,CAAC,WAAW,EAAE,MAAM,CAAC,SAAS,CAAC,CAC5E,CACF,CAAC;QAEF,6CAA6C;QAC7C,MAAM,OAAO,GAAG,cAAc,CAAC,YAAY,CAAC,eAAe,EAAE,eAAe,CAAC,CAAC;QAE9E,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,OAAO;iBACd;aACF;SACF,CAAC;IACJ,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,4BAA4B,EAAE,KAAK,CAAC,CAAC;QACnD,OAAO;YACL,OAAO,EAAE;gBACP;oBACE,IAAI,EAAE,MAAM;oBACZ,IAAI,EAAE,yBAAyB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;iBACxF;aACF;YACD,OAAO,EAAE,IAAI;SACd,CAAC;IACJ,CAAC;AACH,CAAC"}

package/build/tools/summarizeTool.d.ts

@@ -0,0 +1,20 @@
+import { OnyxApiService } from '../services/onyxApi.js';
+/**
+ * Handle the summarize_arq tool request
+ * @param args The tool arguments
+ * @param onyxApiService The Onyx API service
+ * @returns The tool response with executive summary
+ */
+export declare function handleSummarizeArq(args: unknown, onyxApiService: OnyxApiService): Promise<{
+    content: {
+        type: string;
+        text: string;
+    }[];
+    isError?: undefined;
+} | {
+    content: {
+        type: string;
+        text: string;
+    }[];
+    isError: boolean;
+}>;