@iflow-mcp/lupuletic-onyx-mcp-server 1.2.2

package/build/__tests__/services/onyxApi.test.js

@@ -0,0 +1,317 @@
+/**
+ * Unit tests for the OnyxApiService
+ */
+import nock from 'nock';
+import { OnyxApiService } from '../../services/onyxApi.js';
+// Mock config for testing
+const mockConfig = {
+    apiUrl: 'http://test-api.com/api',
+    apiToken: 'test-token',
+};
+describe('OnyxApiService', () => {
+    let onyxApiService;
+    beforeEach(() => {
+        // Create service instance
+        onyxApiService = new OnyxApiService(mockConfig);
+        // Disable real HTTP requests
+        nock.disableNetConnect();
+    });
+    afterEach(() => {
+        // Clean up nock after each test
+        nock.cleanAll();
+        nock.enableNetConnect();
+    });
+    describe('searchOnyx', () => {
+        it('should make a POST request to the search endpoint with correct parameters', async () => {
+            // Mock the search API response
+            const mockResponse = {
+                documents: [
+                    {
+                        document_id: 'doc1',
+                        chunk_ind: 1,
+                        semantic_identifier: 'Test Document',
+                        blurb: 'Test blurb',
+                        source_type: 'test',
+                        score: 0.95,
+                    },
+                ],
+            };
+            // Set up the mock
+            nock('http://test-api.com')
+                .post('/api/admin/search')
+                .reply(200, mockResponse);
+            // Call the method
+            const result = await onyxApiService.searchOnyx('test query', [], 1, 2);
+            // Verify the result
+            expect(result).toEqual(mockResponse.documents);
+        });
+        it('should include document sets in the request when provided', async () => {
+            // Mock the search API response
+            const mockResponse = {
+                documents: [
+                    {
+                        document_id: 'doc1',
+                        chunk_ind: 1,
+                        semantic_identifier: 'Test Document',
+                        blurb: 'Test blurb',
+                        source_type: 'test',
+                        score: 0.95,
+                    },
+                ],
+            };
+            // Set up the mock
+            nock('http://test-api.com')
+                .post('/api/admin/search')
+                .reply(200, mockResponse);
+            // Call the method with document sets
+            const result = await onyxApiService.searchOnyx('test query', ['Test Set']);
+            // Verify the result
+            expect(result).toEqual(mockResponse.documents);
+        });
+        it('should throw an error when the API request fails', async () => {
+            // Set up the mock to return an error
+            nock('http://test-api.com')
+                .post('/api/admin/search')
+                .reply(500, { error: 'Internal server error' });
+            // Call the method and expect it to throw
+            await expect(onyxApiService.searchOnyx('test query')).rejects.toThrow('Failed to search Onyx');
+        });
+    });
+    describe('fetchDocumentChunk', () => {
+        it('should make a GET request to the chunk-info endpoint with correct parameters', async () => {
+            // Mock the chunk-info API response
+            const mockResponse = {
+                content: 'Test chunk content',
+                num_tokens: 10,
+            };
+            // Set up the mock
+            nock('http://test-api.com')
+                .get('/api/document/chunk-info')
+                .query({
+                document_id: 'doc1',
+                chunk_id: 1,
+            })
+                .reply(200, mockResponse);
+            // Call the method
+            const result = await onyxApiService.fetchDocumentChunk('doc1', 1);
+            // Verify the result
+            expect(result).toEqual(mockResponse.content);
+        });
+        it('should return empty string when the API request fails', async () => {
+            // Set up the mock to return an error
+            nock('http://test-api.com')
+                .get('/api/document/chunk-info')
+                .query({
+                document_id: 'doc1',
+                chunk_id: 1,
+            })
+                .reply(500, { error: 'Internal server error' });
+            // Call the method
+            const result = await onyxApiService.fetchDocumentChunk('doc1', 1);
+            // Verify the result
+            expect(result).toEqual('');
+        });
+    });
+    describe('fetchDocumentContent', () => {
+        it('should fetch document size info and then all chunks', async () => {
+            // Mock the document-size-info API response
+            const mockSizeResponse = {
+                num_chunks: 2,
+                total_tokens: 20
+            };
+            // Mock the chunk-info API responses
+            const mockChunk1Response = {
+                content: 'Chunk 1 content',
+                num_tokens: 10
+            };
+            const mockChunk2Response = {
+                content: 'Chunk 2 content',
+                num_tokens: 10
+            };
+            // Set up the mocks
+            nock('http://test-api.com')
+                .get('/api/document/document-size-info')
+                .query({
+                document_id: 'doc1'
+            })
+                .reply(200, mockSizeResponse);
+            nock('http://test-api.com')
+                .get('/api/document/chunk-info')
+                .query({
+                document_id: 'doc1',
+                chunk_id: 0
+            })
+                .reply(200, mockChunk1Response);
+            nock('http://test-api.com')
+                .get('/api/document/chunk-info')
+                .query({
+                document_id: 'doc1',
+                chunk_id: 1
+            })
+                .reply(200, mockChunk2Response);
+            // Call the method
+            const result = await onyxApiService.fetchDocumentContent('doc1');
+            // Verify the result
+            expect(result).toContain(mockChunk1Response.content);
+            expect(result).toContain(mockChunk2Response.content);
+        });
+        it('should return empty string when the API request fails', async () => {
+            // Set up the mock to return an error
+            nock('http://test-api.com')
+                .get('/api/document/document-size-info')
+                .query({
+                document_id: 'doc1'
+            })
+                .reply(500, { error: 'Internal server error' });
+            // Call the method
+            const result = await onyxApiService.fetchDocumentContent('doc1');
+            // Verify the result
+            expect(result).toEqual('');
+        });
+    });
+    describe('createChatSession', () => {
+        it('should make a POST request to create a chat session', async () => {
+            // Mock the create-chat-session API response
+            const mockResponse = {
+                chat_session_id: 'test-session-id'
+            };
+            // Set up the mock
+            nock('http://test-api.com')
+                .post('/api/chat/create-chat-session')
+                .reply(200, mockResponse);
+            // Call the method
+            const result = await onyxApiService.createChatSession(15);
+            // Verify the result
+            expect(result).toEqual(mockResponse.chat_session_id);
+        });
+        it('should throw an error when the API request fails', async () => {
+            // Set up the mock to return an error
+            nock('http://test-api.com')
+                .post('/api/chat/create-chat-session')
+                .reply(500, { error: 'Internal server error' });
+            // Call the method and expect it to throw
+            await expect(onyxApiService.createChatSession(15)).rejects.toThrow('Failed to create chat session');
+        });
+    });
+    describe('sendChatMessage', () => {
+        it('should make a POST request to send a chat message and parse JSON response', async () => {
+            // Mock the send-message API response (single JSON)
+            const mockResponse = JSON.stringify({
+                answer: 'Test answer',
+                documents: [
+                    { document_id: 'doc1', semantic_identifier: 'Test Document 1' },
+                    { document_id: 'doc2', semantic_identifier: 'Test Document 2' }
+                ]
+            });
+            // Set up the mock
+            nock('http://test-api.com')
+                .post('/api/chat/send-message')
+                .reply(200, mockResponse);
+            // Call the method
+            const result = await onyxApiService.sendChatMessage('test-session-id', 'test query');
+            // Verify the result
+            expect(result.answer).toEqual('Test answer');
+            expect(result.documents).toHaveLength(2);
+            expect(result.documents[0].document_id).toEqual('doc1');
+        });
+        it('should handle JSON lines format response', async () => {
+            // Mock the send-message API response (JSON lines)
+            const mockResponse = '{"answer_piece":"Part 1 of answer"}\n' +
+                '{"answer_piece":"Part 2 of answer"}\n' +
+                '{"top_documents":[{"document_id":"doc1","semantic_identifier":"Test Document 1"}]}';
+            // Set up the mock
+            nock('http://test-api.com')
+                .post('/api/chat/send-message')
+                .reply(200, mockResponse);
+            // Call the method
+            const result = await onyxApiService.sendChatMessage('test-session-id', 'test query');
+            // Verify the result
+            expect(result.answer).toEqual('Part 1 of answerPart 2 of answer');
+            expect(result.documents).toHaveLength(1);
+            expect(result.documents[0].document_id).toEqual('doc1');
+        });
+        it('should throw an error when the API request fails', async () => {
+            // Set up the mock to return an error
+            nock('http://test-api.com')
+                .post('/api/chat/send-message')
+                .reply(500, { error: 'Internal server error' });
+            // Call the method and expect it to throw
+            await expect(onyxApiService.sendChatMessage('test-session-id', 'test query')).rejects.toThrow('Error chatting with Onyx');
+        });
+    });
+    describe('buildContext', () => {
+        it('should build a formatted context string from search results and contents', () => {
+            // Test data
+            const results = [
+                {
+                    document_id: 'doc1',
+                    chunk_ind: 1,
+                    semantic_identifier: 'Test Document 1',
+                    score: 0.95,
+                    link: 'http://example.com/doc1',
+                    blurb: 'Test blurb 1',
+                    source_type: 'test'
+                },
+                {
+                    document_id: 'doc2',
+                    chunk_ind: 2,
+                    semantic_identifier: 'Test Document 2',
+                    score: 0.85,
+                    link: undefined,
+                    blurb: 'Test blurb 2',
+                    source_type: 'test'
+                }
+            ];
+            const contents = [
+                'Content for document 1',
+                'Content for document 2'
+            ];
+            // Call the method
+            const result = onyxApiService.buildContext(results, contents);
+            // Verify the result
+            expect(result).toContain('# Test Document 1');
+            expect(result).toContain('Source: http://example.com/doc1');
+            expect(result).toContain('Relevance Score: 0.95');
+            expect(result).toContain('Content for document 1');
+            expect(result).toContain('# Test Document 2');
+            expect(result).toContain('Source: doc2');
+            expect(result).toContain('Relevance Score: 0.85');
+            expect(result).toContain('Content for document 2');
+        });
+        it('should skip results with no content', () => {
+            // Test data
+            const results = [
+                {
+                    document_id: 'doc1',
+                    chunk_ind: 1,
+                    semantic_identifier: 'Test Document 1',
+                    score: 0.95,
+                    link: 'http://example.com/doc1',
+                    blurb: 'Test blurb 1',
+                    source_type: 'test'
+                },
+                {
+                    document_id: 'doc2',
+                    chunk_ind: 2,
+                    semantic_identifier: 'Test Document 2',
+                    score: 0.85,
+                    link: 'http://example.com/doc2',
+                    blurb: 'Test blurb 2',
+                    source_type: 'test'
+                }
+            ];
+            const contents = [
+                'Content for document 1',
+                '' // Empty content for second document
+            ];
+            // Call the method
+            const result = onyxApiService.buildContext(results, contents);
+            // Verify the result
+            expect(result).toContain('# Test Document 1');
+            expect(result).toContain('Content for document 1');
+            // Should not contain document 2 info
+            expect(result).not.toContain('# Test Document 2');
+        });
+    });
+});
+//# sourceMappingURL=onyxApi.test.js.map

package/build/__tests__/services/onyxApi.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"onyxApi.test.js","sourceRoot":"","sources":["../../../src/__tests__/services/onyxApi.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,cAAc,EAAE,MAAM,2BAA2B,CAAC;AAG3D,0BAA0B;AAC1B,MAAM,UAAU,GAAe;IAC7B,MAAM,EAAE,yBAAyB;IACjC,QAAQ,EAAE,YAAY;CACvB,CAAC;AAEF,QAAQ,CAAC,gBAAgB,EAAE,GAAG,EAAE;IAC9B,IAAI,cAA8B,CAAC;IAEnC,UAAU,CAAC,GAAG,EAAE;QACd,0BAA0B;QAC1B,cAAc,GAAG,IAAI,cAAc,CAAC,UAAU,CAAC,CAAC;QAEhD,6BAA6B;QAC7B,IAAI,CAAC,iBAAiB,EAAE,CAAC;IAC3B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,gCAAgC;QAChC,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChB,IAAI,CAAC,gBAAgB,EAAE,CAAC;IAC1B,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,+BAA+B;YAC/B,MAAM,YAAY,GAAG;gBACnB,SAAS,EAAE;oBACT;wBACE,WAAW,EAAE,MAAM;wBACnB,SAAS,EAAE,CAAC;wBACZ,mBAAmB,EAAE,eAAe;wBACpC,KAAK,EAAE,YAAY;wBACnB,WAAW,EAAE,MAAM;wBACnB,KAAK,EAAE,IAAI;qBACZ;iBACF;aACF,CAAC;YAEF,kBAAkB;YAClB,IAAI,CAAC,qBAAqB,CAAC;iBACxB,IAAI,CAAC,mBAAmB,CAAC;iBACzB,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YAE5B,kBAAkB;YAClB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,YAAY,EAAE,EAAE,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAEvE,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2DAA2D,EAAE,KAAK,IAAI,EAAE;YACzE,+BAA+B;YAC/B,MAAM,YAAY,GAAG;gBACnB,SAAS,EAAE;oBACT;wBACE,WAAW,EAAE,MAAM;wBACnB,SAAS,EAAE,CAAC;wBACZ,mBAAmB,EAAE,eAAe;wBACpC,KAAK,EAAE,YAAY;wBACnB,WAAW,EAAE,MAAM;wBACnB,KAAK,EAAE,IAAI;qBACZ;iBACF;aACF,CAAC;YAEF,kBAAkB;YAClB,IAAI,CAAC,qBAAqB,CAAC;iBACxB,IAAI,CAAC,mBAAmB,CAAC;iBACzB,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YAE5B,qCAAqC;YACrC,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,UAAU,CAAC,YAAY,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;YAE3E,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QACjD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,qCAAqC;YACrC,IAAI,CAAC,qBAAqB,CAAC;iBACxB,IAAI,CAAC,mBAAmB,CAAC;iBACzB,KAAK,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;YAElD,yCAAyC;YACzC,MAAM,MAAM,CAAC,cAAc,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CACnE,uBAAuB,CACxB,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,EAAE,CAAC,8EAA8E,EAAE,KAAK,IAAI,EAAE;YAC5F,mCAAmC;YACnC,MAAM,YAAY,GAAG;gBACnB,OAAO,EAAE,oBAAoB;gBAC7B,UAAU,EAAE,EAAE;aACf,CAAC;YAEF,kBAAkB;YAClB,IAAI,CAAC,qBAAqB,CAAC;iBACxB,GAAG,CAAC,0BAA0B,CAAC;iBAC/B,KAAK,CAAC;gBACL,WAAW,EAAE,MAAM;gBACnB,QAAQ,EAAE,CAAC;aACZ,CAAC;iBACD,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YAE5B,kBAAkB;YAClB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAElE,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,qCAAqC;YACrC,IAAI,CAAC,qBAAqB,CAAC;iBACxB,GAAG,CAAC,0BAA0B,CAAC;iBAC/B,KAAK,CAAC;gBACL,WAAW,EAAE,MAAM;gBACnB,QAAQ,EAAE,CAAC;aACZ,CAAC;iBACD,KAAK,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;YAElD,kBAAkB;YAClB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,kBAAkB,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;YAElE,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,sBAAsB,EAAE,GAAG,EAAE;QACpC,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,2CAA2C;YAC3C,MAAM,gBAAgB,GAAG;gBACvB,UAAU,EAAE,CAAC;gBACb,YAAY,EAAE,EAAE;aACjB,CAAC;YAEF,oCAAoC;YACpC,MAAM,kBAAkB,GAAG;gBACzB,OAAO,EAAE,iBAAiB;gBAC1B,UAAU,EAAE,EAAE;aACf,CAAC;YAEF,MAAM,kBAAkB,GAAG;gBACzB,OAAO,EAAE,iBAAiB;gBAC1B,UAAU,EAAE,EAAE;aACf,CAAC;YAEF,mBAAmB;YACnB,IAAI,CAAC,qBAAqB,CAAC;iBACxB,GAAG,CAAC,kCAAkC,CAAC;iBACvC,KAAK,CAAC;gBACL,WAAW,EAAE,MAAM;aACpB,CAAC;iBACD,KAAK,CAAC,GAAG,EAAE,gBAAgB,CAAC,CAAC;YAEhC,IAAI,CAAC,qBAAqB,CAAC;iBACxB,GAAG,CAAC,0BAA0B,CAAC;iBAC/B,KAAK,CAAC;gBACL,WAAW,EAAE,MAAM;gBACnB,QAAQ,EAAE,CAAC;aACZ,CAAC;iBACD,KAAK,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;YAElC,IAAI,CAAC,qBAAqB,CAAC;iBACxB,GAAG,CAAC,0BAA0B,CAAC;iBAC/B,KAAK,CAAC;gBACL,WAAW,EAAE,MAAM;gBACnB,QAAQ,EAAE,CAAC;aACZ,CAAC;iBACD,KAAK,CAAC,GAAG,EAAE,kBAAkB,CAAC,CAAC;YAElC,kBAAkB;YAClB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEjE,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,uDAAuD,EAAE,KAAK,IAAI,EAAE;YACrE,qCAAqC;YACrC,IAAI,CAAC,qBAAqB,CAAC;iBACxB,GAAG,CAAC,kCAAkC,CAAC;iBACvC,KAAK,CAAC;gBACL,WAAW,EAAE,MAAM;aACpB,CAAC;iBACD,KAAK,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;YAElD,kBAAkB;YAClB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,oBAAoB,CAAC,MAAM,CAAC,CAAC;YAEjE,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE;QACjC,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,4CAA4C;YAC5C,MAAM,YAAY,GAAG;gBACnB,eAAe,EAAE,iBAAiB;aACnC,CAAC;YAEF,kBAAkB;YAClB,IAAI,CAAC,qBAAqB,CAAC;iBACxB,IAAI,CAAC,+BAA+B,CAAC;iBACrC,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YAE5B,kBAAkB;YAClB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;YAE1D,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,YAAY,CAAC,eAAe,CAAC,CAAC;QACvD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,qCAAqC;YACrC,IAAI,CAAC,qBAAqB,CAAC;iBACxB,IAAI,CAAC,+BAA+B,CAAC;iBACrC,KAAK,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;YAElD,yCAAyC;YACzC,MAAM,MAAM,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAChE,+BAA+B,CAChC,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,iBAAiB,EAAE,GAAG,EAAE;QAC/B,EAAE,CAAC,2EAA2E,EAAE,KAAK,IAAI,EAAE;YACzF,mDAAmD;YACnD,MAAM,YAAY,GAAG,IAAI,CAAC,SAAS,CAAC;gBAClC,MAAM,EAAE,aAAa;gBACrB,SAAS,EAAE;oBACT,EAAE,WAAW,EAAE,MAAM,EAAE,mBAAmB,EAAE,iBAAiB,EAAE;oBAC/D,EAAE,WAAW,EAAE,MAAM,EAAE,mBAAmB,EAAE,iBAAiB,EAAE;iBAChE;aACF,CAAC,CAAC;YAEH,kBAAkB;YAClB,IAAI,CAAC,qBAAqB,CAAC;iBACxB,IAAI,CAAC,wBAAwB,CAAC;iBAC9B,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YAE5B,kBAAkB;YAClB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,eAAe,CAAC,iBAAiB,EAAE,YAAY,CAAC,CAAC;YAErF,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC;YAC7C,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0CAA0C,EAAE,KAAK,IAAI,EAAE;YACxD,kDAAkD;YAClD,MAAM,YAAY,GAChB,uCAAuC;gBACvC,uCAAuC;gBACvC,oFAAoF,CAAC;YAEvF,kBAAkB;YAClB,IAAI,CAAC,qBAAqB,CAAC;iBACxB,IAAI,CAAC,wBAAwB,CAAC;iBAC9B,KAAK,CAAC,GAAG,EAAE,YAAY,CAAC,CAAC;YAE5B,kBAAkB;YAClB,MAAM,MAAM,GAAG,MAAM,cAAc,CAAC,eAAe,CAAC,iBAAiB,EAAE,YAAY,CAAC,CAAC;YAErF,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,kCAAkC,CAAC,CAAC;YAClE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QAC1D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;YAChE,qCAAqC;YACrC,IAAI,CAAC,qBAAqB,CAAC;iBACxB,IAAI,CAAC,wBAAwB,CAAC;iBAC9B,KAAK,CAAC,GAAG,EAAE,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;YAElD,yCAAyC;YACzC,MAAM,MAAM,CAAC,cAAc,CAAC,eAAe,CAAC,iBAAiB,EAAE,YAAY,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAC3F,0BAA0B,CAC3B,CAAC;QACJ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,cAAc,EAAE,GAAG,EAAE;QAC5B,EAAE,CAAC,0EAA0E,EAAE,GAAG,EAAE;YAClF,YAAY;YACZ,MAAM,OAAO,GAAG;gBACd;oBACE,WAAW,EAAE,MAAM;oBACnB,SAAS,EAAE,CAAC;oBACZ,mBAAmB,EAAE,iBAAiB;oBACtC,KAAK,EAAE,IAAI;oBACX,IAAI,EAAE,yBAAyB;oBAC/B,KAAK,EAAE,cAAc;oBACrB,WAAW,EAAE,MAAM;iBACpB;gBACD;oBACE,WAAW,EAAE,MAAM;oBACnB,SAAS,EAAE,CAAC;oBACZ,mBAAmB,EAAE,iBAAiB;oBACtC,KAAK,EAAE,IAAI;oBACX,IAAI,EAAE,SAAS;oBACf,KAAK,EAAE,cAAc;oBACrB,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG;gBACf,wBAAwB;gBACxB,wBAAwB;aACzB,CAAC;YAEF,kBAAkB;YAClB,MAAM,MAAM,GAAG,cAAc,CAAC,YAAY,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAE9D,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,iCAAiC,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;YAEnD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;YACzC,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;YAClD,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;QACrD,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,GAAG,EAAE;YAC7C,YAAY;YACZ,MAAM,OAAO,GAAG;gBACd;oBACE,WAAW,EAAE,MAAM;oBACnB,SAAS,EAAE,CAAC;oBACZ,mBAAmB,EAAE,iBAAiB;oBACtC,KAAK,EAAE,IAAI;oBACX,IAAI,EAAE,yBAAyB;oBAC/B,KAAK,EAAE,cAAc;oBACrB,WAAW,EAAE,MAAM;iBACpB;gBACD;oBACE,WAAW,EAAE,MAAM;oBACnB,SAAS,EAAE,CAAC;oBACZ,mBAAmB,EAAE,iBAAiB;oBACtC,KAAK,EAAE,IAAI;oBACX,IAAI,EAAE,yBAAyB;oBAC/B,KAAK,EAAE,cAAc;oBACrB,WAAW,EAAE,MAAM;iBACpB;aACF,CAAC;YAEF,MAAM,QAAQ,GAAG;gBACf,wBAAwB;gBACxB,EAAE,CAAC,oCAAoC;aACxC,CAAC;YAEF,kBAAkB;YAClB,MAAM,MAAM,GAAG,cAAc,CAAC,YAAY,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC;YAE9D,oBAAoB;YACpB,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAC9C,MAAM,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,wBAAwB,CAAC,CAAC;YAEnD,qCAAqC;YACrC,MAAM,CAAC,MAAM,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACpD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AAEL,CAAC,CAAC,CAAC"}

package/build/__tests__/tools/tools.test.d.ts

@@ -0,0 +1 @@
+export {};

package/build/__tests__/tools/tools.test.js

@@ -0,0 +1,177 @@
+/**
+ * Unit tests for the tool handlers
+ */
+import { handleChatWithOnyx } from '../../tools/chatTool.js';
+import { handleSearchOnyx } from '../../tools/searchTool.js';
+import { jest } from '@jest/globals';
+describe('Tool Handlers', () => {
+    let mockOnyxApiService;
+    beforeEach(() => {
+        // Create a mock instance of OnyxApiService
+        mockOnyxApiService = {
+            apiUrl: 'http://test-api.com/api',
+            apiToken: 'test-token',
+            createChatSession: jest.fn(),
+            sendChatMessage: jest.fn(),
+            searchOnyx: jest.fn(),
+            fetchDocumentChunk: jest.fn(),
+            fetchDocumentContent: jest.fn(),
+            buildContext: jest.fn()
+        };
+    });
+    describe('handleChatWithOnyx', () => {
+        beforeEach(() => {
+            // Set up the mock implementations
+            mockOnyxApiService.createChatSession.mockImplementation(() => {
+                return Promise.resolve('test-session-id');
+            });
+            mockOnyxApiService.sendChatMessage.mockImplementation(() => {
+                return Promise.resolve({
+                    answer: 'Test answer',
+                    documents: [
+                        { document_id: 'doc1', semantic_identifier: 'Test Document 1' },
+                        { document_id: 'doc2', semantic_identifier: 'Test Document 2' },
+                    ],
+                });
+            });
+        });
+        it('should throw an error for invalid arguments', async () => {
+            // Test with null arguments
+            const result1 = await handleChatWithOnyx(null, mockOnyxApiService);
+            expect(result1.isError).toBe(true);
+            expect(result1.content[0].text).toContain('Invalid arguments');
+            // Test with non-object arguments
+            const result2 = await handleChatWithOnyx('not an object', mockOnyxApiService);
+            expect(result2.isError).toBe(true);
+            expect(result2.content[0].text).toContain('Invalid arguments');
+        });
+        it('should throw an error for missing query', async () => {
+            const result = await handleChatWithOnyx({ personaId: 15 }, mockOnyxApiService);
+            expect(result.isError).toBe(true);
+            expect(result.content[0].text).toContain('Query is required');
+        });
+        it('should create a new chat session when chatSessionId is not provided', async () => {
+            await handleChatWithOnyx({ query: 'test query' }, mockOnyxApiService);
+            expect(mockOnyxApiService.createChatSession).toHaveBeenCalledWith(15); // Default personaId
+        });
+        it('should use existing chat session when chatSessionId is provided', async () => {
+            await handleChatWithOnyx({ query: 'test query', chatSessionId: 'existing-session' }, mockOnyxApiService);
+            expect(mockOnyxApiService.createChatSession).not.toHaveBeenCalled();
+            expect(mockOnyxApiService.sendChatMessage).toHaveBeenCalledWith('existing-session', 'test query', []);
+        });
+        it('should send a chat message and return formatted response', async () => {
+            const result = await handleChatWithOnyx({ query: 'test query' }, mockOnyxApiService);
+            expect(mockOnyxApiService.sendChatMessage).toHaveBeenCalledWith('test-session-id', 'test query', []);
+            expect(result.content[0].type).toBe('text');
+            expect(result.content[0].text).toContain('Test answer');
+            expect(result.content[0].text).toContain('Test Document 1');
+            expect(result.content[0].text).toContain('doc2');
+            // Use type assertion to access metadata
+            const content = result.content[0];
+            expect(content.metadata?.chat_session_id).toBe('test-session-id');
+        });
+        it('should handle API errors gracefully', async () => {
+            // Mock the API to throw an error
+            mockOnyxApiService.createChatSession.mockImplementation(() => {
+                return Promise.reject(new Error('API error'));
+            });
+            const result = await handleChatWithOnyx({ query: 'test query' }, mockOnyxApiService);
+            expect(result.isError).toBe(true);
+            expect(result.content[0].text).toContain('Error chatting with Onyx');
+            expect(result.content[0].text).toContain('API error');
+        });
+    });
+    describe('handleSearchOnyx', () => {
+        beforeEach(() => {
+            // Set up the mock implementations
+            mockOnyxApiService.searchOnyx.mockImplementation(() => {
+                const results = [
+                    {
+                        document_id: 'doc1',
+                        chunk_ind: 1,
+                        semantic_identifier: 'Test Document 1',
+                        score: 0.95,
+                        link: 'http://example.com/doc1',
+                        blurb: 'Test blurb 1',
+                        source_type: 'test'
+                    },
+                    {
+                        document_id: 'doc2',
+                        chunk_ind: 2,
+                        semantic_identifier: 'Test Document 2',
+                        score: 0.85,
+                        link: 'http://example.com/doc2',
+                        blurb: 'Test blurb 2',
+                        source_type: 'test'
+                    }
+                ];
+                return Promise.resolve(results);
+            });
+            // @ts-expect-error - TypeScript errors for mock implementation
+            mockOnyxApiService.fetchDocumentChunk = jest.fn().mockImplementation(() => Promise.resolve('Mocked chunk content'));
+            // @ts-expect-error - TypeScript errors for mock implementation
+            mockOnyxApiService.fetchDocumentContent = jest.fn().mockImplementation(() => Promise.resolve('Mocked document content'));
+            // @ts-expect-error - TypeScript errors for mock implementation
+            mockOnyxApiService.buildContext = jest.fn().mockReturnValue('Formatted context');
+        });
+        it('should throw an error for invalid arguments', async () => {
+            // Test with null arguments
+            const result1 = await handleSearchOnyx(null, mockOnyxApiService);
+            expect(result1.isError).toBe(true);
+            expect(result1.content[0].text).toContain('Invalid arguments');
+            // Test with non-object arguments
+            const result2 = await handleSearchOnyx('not an object', mockOnyxApiService);
+            expect(result2.isError).toBe(true);
+            expect(result2.content[0].text).toContain('Invalid arguments');
+        });
+        it('should throw an error for missing query', async () => {
+            const result = await handleSearchOnyx({ maxResults: 5 }, mockOnyxApiService);
+            expect(result.isError).toBe(true);
+            expect(result.content[0].text).toContain('Query is required');
+        });
+        it('should search with default parameters when only query is provided', async () => {
+            await handleSearchOnyx({ query: 'test query' }, mockOnyxApiService);
+            expect(mockOnyxApiService.searchOnyx).toHaveBeenCalledWith('test query', [], 1, 1);
+        });
+        it('should search with custom parameters when provided', async () => {
+            await handleSearchOnyx({
+                query: 'test query',
+                documentSets: ['set1', 'set2'],
+                maxResults: 10,
+                chunksAbove: 2,
+                chunksBelow: 3,
+            }, mockOnyxApiService);
+            expect(mockOnyxApiService.searchOnyx).toHaveBeenCalledWith('test query', ['set1', 'set2'], 2, 3);
+        });
+        it('should fetch document chunks by default', async () => {
+            await handleSearchOnyx({ query: 'test query' }, mockOnyxApiService);
+            expect(mockOnyxApiService.fetchDocumentChunk).toHaveBeenCalledTimes(2);
+            expect(mockOnyxApiService.fetchDocumentContent).not.toHaveBeenCalled();
+        });
+        it('should fetch full documents when retrieveFullDocuments is true', async () => {
+            await handleSearchOnyx({
+                query: 'test query',
+                retrieveFullDocuments: true,
+            }, mockOnyxApiService);
+            expect(mockOnyxApiService.fetchDocumentContent).toHaveBeenCalledTimes(2);
+            expect(mockOnyxApiService.fetchDocumentChunk).not.toHaveBeenCalled();
+        });
+        it('should build and return context from search results', async () => {
+            const result = await handleSearchOnyx({ query: 'test query' }, mockOnyxApiService);
+            expect(mockOnyxApiService.buildContext).toHaveBeenCalled();
+            expect(result.content[0].type).toBe('text');
+            expect(result.content[0].text).toBe('Formatted context');
+        });
+        it('should handle API errors gracefully', async () => {
+            // Mock the API to throw an error
+            mockOnyxApiService.searchOnyx.mockImplementation(() => {
+                return Promise.reject(new Error('API error'));
+            });
+            const result = await handleSearchOnyx({ query: 'test query' }, mockOnyxApiService);
+            expect(result.isError).toBe(true);
+            expect(result.content[0].text).toContain('Error searching Onyx');
+            expect(result.content[0].text).toContain('API error');
+        });
+    });
+});
+//# sourceMappingURL=tools.test.js.map

package/build/__tests__/tools/tools.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tools.test.js","sourceRoot":"","sources":["../../../src/__tests__/tools/tools.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,kBAAkB,EAAE,MAAM,yBAAyB,CAAC;AAC7D,OAAO,EAAE,gBAAgB,EAAE,MAAM,2BAA2B,CAAC;AAG7D,OAAO,EAAE,IAAI,EAAE,MAAM,eAAe,CAAC;AAErC,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,IAAI,kBAAkC,CAAC;IAEvC,UAAU,CAAC,GAAG,EAAE;QACd,2CAA2C;QAC3C,kBAAkB,GAAG;YACnB,MAAM,EAAE,yBAAyB;YACjC,QAAQ,EAAE,YAAY;YACtB,iBAAiB,EAAE,IAAI,CAAC,EAAE,EAAyB;YACnD,eAAe,EAAE,IAAI,CAAC,EAAE,EAAE;YAC1B,UAAU,EAAE,IAAI,CAAC,EAAE,EAAE;YACrB,kBAAkB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC7B,oBAAoB,EAAE,IAAI,CAAC,EAAE,EAAE;YAC/B,YAAY,EAAE,IAAI,CAAC,EAAE,EAAE;SACK,CAAC;IACjC,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,oBAAoB,EAAE,GAAG,EAAE;QAClC,UAAU,CAAC,GAAG,EAAE;YACd,kCAAkC;YACjC,kBAAkB,CAAC,iBAA+B,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC1E,OAAO,OAAO,CAAC,OAAO,CAAC,iBAAiB,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC;YAEF,kBAAkB,CAAC,eAA6B,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACxE,OAAO,OAAO,CAAC,OAAO,CAAC;oBACrB,MAAM,EAAE,aAAa;oBACrB,SAAS,EAAE;wBACT,EAAE,WAAW,EAAE,MAAM,EAAE,mBAAmB,EAAE,iBAAiB,EAAE;wBAC/D,EAAE,WAAW,EAAE,MAAM,EAAE,mBAAmB,EAAE,iBAAiB,EAAE;qBAChE;iBACF,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,2BAA2B;YAC3B,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;YACnE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAE/D,iCAAiC;YACjC,MAAM,OAAO,GAAG,MAAM,kBAAkB,CAAC,eAAe,EAAE,kBAAkB,CAAC,CAAC;YAC9E,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAC/E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qEAAqE,EAAE,KAAK,IAAI,EAAE;YACnF,MAAM,kBAAkB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,kBAAkB,CAAC,CAAC;YACtE,MAAM,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,CAAC,oBAAoB,CAAC,EAAE,CAAC,CAAC,CAAC,oBAAoB;QAC7F,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;YAC/E,MAAM,kBAAkB,CACtB,EAAE,KAAK,EAAE,YAAY,EAAE,aAAa,EAAE,kBAAkB,EAAE,EAC1D,kBAAkB,CACnB,CAAC;YACF,MAAM,CAAC,kBAAkB,CAAC,iBAAiB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;YACpE,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC7D,kBAAkB,EAClB,YAAY,EACZ,EAAE,CACH,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,0DAA0D,EAAE,KAAK,IAAI,EAAE;YACxE,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAErF,MAAM,CAAC,kBAAkB,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAC7D,iBAAiB,EACjB,YAAY,EACZ,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YACxD,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,iBAAiB,CAAC,CAAC;YAC5D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;YAEjD,wCAAwC;YACxC,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC,CAAwB,CAAC;YACzD,MAAM,CAAC,OAAO,CAAC,QAAQ,EAAE,eAAe,CAAC,CAAC,IAAI,CAAC,iBAAiB,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,iCAAiC;YAChC,kBAAkB,CAAC,iBAA+B,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBAC1E,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,kBAAkB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAErF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;YACrE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;QAChC,UAAU,CAAC,GAAG,EAAE;YACd,kCAAkC;YACjC,kBAAkB,CAAC,UAAwB,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACnE,MAAM,OAAO,GAAuB;oBAClC;wBACE,WAAW,EAAE,MAAM;wBACnB,SAAS,EAAE,CAAC;wBACZ,mBAAmB,EAAE,iBAAiB;wBACtC,KAAK,EAAE,IAAI;wBACX,IAAI,EAAE,yBAAyB;wBAC/B,KAAK,EAAE,cAAc;wBACrB,WAAW,EAAE,MAAM;qBACpB;oBACD;wBACE,WAAW,EAAE,MAAM;wBACnB,SAAS,EAAE,CAAC;wBACZ,mBAAmB,EAAE,iBAAiB;wBACtC,KAAK,EAAE,IAAI;wBACX,IAAI,EAAE,yBAAyB;wBAC/B,KAAK,EAAE,cAAc;wBACrB,WAAW,EAAE,MAAM;qBACpB;iBACF,CAAC;gBACF,OAAO,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;YAClC,CAAC,CAAC,CAAC;YAEH,+DAA+D;YAC/D,kBAAkB,CAAC,kBAAkB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CAClE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,sBAAsB,CAAC,CAC9C,CAAC;YAEF,+DAA+D;YAC/D,kBAAkB,CAAC,oBAAoB,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,kBAAkB,CACpE,GAAG,EAAE,CAAC,OAAO,CAAC,OAAO,CAAC,yBAAyB,CAAC,CACjD,CAAC;YAEF,+DAA+D;YAC/D,kBAAkB,CAAC,YAAY,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,eAAe,CAAC,mBAAmB,CAAC,CAAC;QACnF,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,6CAA6C,EAAE,KAAK,IAAI,EAAE;YAC3D,2BAA2B;YAC3B,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,IAAI,EAAE,kBAAkB,CAAC,CAAC;YACjE,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;YAE/D,iCAAiC;YACjC,MAAM,OAAO,GAAG,MAAM,gBAAgB,CAAC,eAAe,EAAE,kBAAkB,CAAC,CAAC;YAC5E,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACnC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QACjE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAC7E,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,mBAAmB,CAAC,CAAC;QAChE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,mEAAmE,EAAE,KAAK,IAAI,EAAE;YACjF,MAAM,gBAAgB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAEpE,MAAM,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACxD,YAAY,EACZ,EAAE,EACF,CAAC,EACD,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;YAClE,MAAM,gBAAgB,CAAC;gBACrB,KAAK,EAAE,YAAY;gBACnB,YAAY,EAAE,CAAC,MAAM,EAAE,MAAM,CAAC;gBAC9B,UAAU,EAAE,EAAE;gBACd,WAAW,EAAE,CAAC;gBACd,WAAW,EAAE,CAAC;aACf,EAAE,kBAAkB,CAAC,CAAC;YAEvB,MAAM,CAAC,kBAAkB,CAAC,UAAU,CAAC,CAAC,oBAAoB,CACxD,YAAY,EACZ,CAAC,MAAM,EAAE,MAAM,CAAC,EAChB,CAAC,EACD,CAAC,CACF,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,yCAAyC,EAAE,KAAK,IAAI,EAAE;YACvD,MAAM,gBAAgB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAEpE,MAAM,CAAC,kBAAkB,CAAC,kBAAkB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACvE,MAAM,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACzE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;YAC9E,MAAM,gBAAgB,CAAC;gBACrB,KAAK,EAAE,YAAY;gBACnB,qBAAqB,EAAE,IAAI;aAC5B,EAAE,kBAAkB,CAAC,CAAC;YAEvB,MAAM,CAAC,kBAAkB,CAAC,oBAAoB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;YACzE,MAAM,CAAC,kBAAkB,CAAC,kBAAkB,CAAC,CAAC,GAAG,CAAC,gBAAgB,EAAE,CAAC;QACvE,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qDAAqD,EAAE,KAAK,IAAI,EAAE;YACnE,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAEnF,MAAM,CAAC,kBAAkB,CAAC,YAAY,CAAC,CAAC,gBAAgB,EAAE,CAAC;YAC3D,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,mBAAmB,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,qCAAqC,EAAE,KAAK,IAAI,EAAE;YACnD,iCAAiC;YAChC,kBAAkB,CAAC,UAAwB,CAAC,kBAAkB,CAAC,GAAG,EAAE;gBACnE,OAAO,OAAO,CAAC,MAAM,CAAC,IAAI,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;YAChD,CAAC,CAAC,CAAC;YAEH,MAAM,MAAM,GAAG,MAAM,gBAAgB,CAAC,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,kBAAkB,CAAC,CAAC;YAEnF,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YAClC,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,sBAAsB,CAAC,CAAC;YACjE,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;QACxD,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/build/config/index.d.ts

@@ -0,0 +1,20 @@
+/**
+ * Configuration for the Onyx MCP Server
+ */
+import { OnyxConfig } from '../types/index.js';
+/**
+ * Load configuration from environment variables
+ * @returns The Onyx configuration
+ */
+export declare function loadConfig(): OnyxConfig;
+/**
+ * Server configuration
+ */
+export declare const SERVER_CONFIG: {
+    name: string;
+    version: string;
+};
+/**
+ * Debug mode flag
+ */
+export declare const DEBUG: boolean;

package/build/config/index.js

@@ -0,0 +1,26 @@
+/**
+ * Load configuration from environment variables
+ * @returns The Onyx configuration
+ */
+export function loadConfig() {
+    const config = {
+        apiUrl: process.env.ONYX_API_URL || 'http://localhost:8080/api',
+        apiToken: process.env.ONYX_API_TOKEN || '',
+    };
+    if (!config.apiToken) {
+        console.error('ONYX_API_TOKEN environment variable is required');
+    }
+    return config;
+}
+/**
+ * Server configuration
+ */
+export const SERVER_CONFIG = {
+    name: 'onyx-mcp-server',
+    version: '1.0.0',
+};
+/**
+ * Debug mode flag
+ */
+export const DEBUG = process.env.DEBUG === 'true';
+//# sourceMappingURL=index.js.map

package/build/config/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/config/index.ts"],"names":[],"mappings":"AAKA;;;GAGG;AACH,MAAM,UAAU,UAAU;IACxB,MAAM,MAAM,GAAe;QACzB,MAAM,EAAE,OAAO,CAAC,GAAG,CAAC,YAAY,IAAI,2BAA2B;QAC/D,QAAQ,EAAE,OAAO,CAAC,GAAG,CAAC,cAAc,IAAI,EAAE;KAC3C,CAAC;IAEF,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QACrB,OAAO,CAAC,KAAK,CAAC,iDAAiD,CAAC,CAAC;IACnE,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B,IAAI,EAAE,iBAAiB;IACvB,OAAO,EAAE,OAAO;CACjB,CAAC;AAEF;;GAEG;AACH,MAAM,CAAC,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,KAAK,MAAM,CAAC"}

package/build/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/build/index.js

@@ -0,0 +1,24 @@
+#!/usr/bin/env node
+/**
+ * Onyx MCP Server
+ * Entry point for the Onyx MCP Server
+ */
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import { OnyxMcpServer } from './server.js';
+/**
+ * Main function to start the server
+ */
+async function main() {
+    try {
+        const server = new OnyxMcpServer();
+        const transport = new StdioServerTransport();
+        await server.run(transport);
+    }
+    catch (error) {
+        console.error('Failed to start Onyx MCP Server:', error);
+        process.exit(1);
+    }
+}
+// Start the server
+main().catch(console.error);
+//# sourceMappingURL=index.js.map

package/build/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":";AACA;;;GAGG;AACH,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,aAAa,EAAE,MAAM,aAAa,CAAC;AAE5C;;GAEG;AACH,KAAK,UAAU,IAAI;IACjB,IAAI,CAAC;QACH,MAAM,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;QACnC,MAAM,SAAS,GAAG,IAAI,oBAAoB,EAAE,CAAC;QAC7C,MAAM,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;IAC9B,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAK,CAAC,CAAC;QACzD,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClB,CAAC;AACH,CAAC;AAED,mBAAmB;AACnB,IAAI,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC"}

package/build/server.d.ts

@@ -0,0 +1,21 @@
+/**
+ * OnyxMcpServer class
+ * Main server class for the Onyx MCP Server
+ */
+export declare class OnyxMcpServer {
+    private server;
+    private onyxApiService;
+    /**
+     * Constructor
+     */
+    constructor();
+    /**
+     * Set up tool handlers
+     */
+    private setupToolHandlers;
+    /**
+     * Run the server
+     * @param transport The server transport
+     */
+    run(transport: any): Promise<void>;
+}