onyx-kb-mcp 1.0.0

package/LICENSE

@@ -0,0 +1 @@
+MIT License

package/README.md

@@ -0,0 +1,166 @@
+# onyx-kb-mcp
+
+A [Model Context Protocol (MCP)](https://modelcontextprotocol.io/) server that provides seamless integration with [Onyx](https://github.com/onyx-dot-app/onyx) AI knowledge bases. Enable any MCP-compatible client (Claude Desktop, Claude Code, Cline, etc.) to search, query, and summarize your organization's knowledge base.
+
+[![npm version](https://badge.fury.io/js/onyx-kb-mcp.svg)](https://www.npmjs.com/package/onyx-kb-mcp)
+
+## Features
+
+- **Semantic Search** (`search_arq`) - Search your Onyx document sets with configurable context retrieval
+- **Q&A with RAG** (`ask_arq`) - Get comprehensive answers with sources using Onyx's LLM + RAG capabilities
+- **Executive Summaries** (`summarize_arq`) - Generate concise bullet-point summaries on any topic
+
+## Quick Start
+
+### Prerequisites
+
+- Node.js v16 or higher
+- An Onyx instance with API access
+- An Onyx API token
+
+### Installation
+
+```bash
+npm install -g onyx-kb-mcp
+```
+
+Or use directly with npx (no installation required):
+
+```bash
+npx onyx-kb-mcp
+```
+
+## MCP Client Configuration
+
+### Claude Desktop
+
+Add to your Claude Desktop config file (`~/Library/Application Support/Claude/claude_desktop_config.json` on macOS):
+
+```json
+{
+  "mcpServers": {
+    "onyx": {
+      "command": "npx",
+      "args": ["-y", "onyx-kb-mcp"],
+      "env": {
+        "ONYX_API_TOKEN": "your-api-token-here",
+        "ONYX_API_URL": "http://localhost:8080/api"
+      }
+    }
+  }
+}
+```
+
+### Claude Code
+
+Add to your Claude Code MCP settings (`~/.claude/settings.json`):
+
+```json
+{
+  "mcpServers": {
+    "onyx": {
+      "command": "npx",
+      "args": ["-y", "onyx-kb-mcp"],
+      "env": {
+        "ONYX_API_TOKEN": "your-api-token-here",
+        "ONYX_API_URL": "http://localhost:8080/api"
+      }
+    }
+  }
+}
+```
+
+### Smithery
+
+This server is compatible with [Smithery](https://smithery.ai/). See `smithery.yaml` for configuration details.
+
+## Environment Variables
+
+| Variable | Required | Default | Description |
+|----------|----------|---------|-------------|
+| `ONYX_API_TOKEN` | Yes | - | Your Onyx API authentication token |
+| `ONYX_API_URL` | No | `http://localhost:8080/api` | URL of your Onyx API endpoint |
+| `DEFAULT_PERSONA_ID` | No | `15` | Default persona ID for chat interactions |
+| `DEBUG` | No | `false` | Enable debug logging |
+
+## Available Tools
+
+### search_arq
+
+Search your knowledge base for relevant documents with configurable context retrieval.
+
+| Parameter | Type | Required | Default | Description |
+|-----------|------|----------|---------|-------------|
+| `query` | string | Yes | - | Search query |
+| `documentSets` | string[] | No | all | Document sets to search |
+| `maxResults` | integer | No | 10 | Maximum results (1-20) |
+| `chunksAbove` | integer | No | 1 | Context chunks above match |
+| `chunksBelow` | integer | No | 1 | Context chunks below match |
+| `retrieveFullDocuments` | boolean | No | false | Return full documents |
+
+### ask_arq
+
+Ask questions and receive comprehensive answers with source citations.
+
+| Parameter | Type | Required | Default | Description |
+|-----------|------|----------|---------|-------------|
+| `query` | string | Yes | - | Question to ask |
+| `personaId` | integer | No | 15 | Persona ID to use |
+| `personaName` | string | No | - | Persona name (alternative to ID) |
+| `documentSets` | string[] | No | all | Document sets to search |
+| `chatSessionId` | string | No | - | Session ID for conversation continuity |
+| `enableAutoDetectFilters` | boolean | No | true | Auto-detect search filters |
+
+### summarize_arq
+
+Generate executive summaries with bullet points and source links.
+
+| Parameter | Type | Required | Default | Description |
+|-----------|------|----------|---------|-------------|
+| `topic` | string | Yes | - | Topic to summarize |
+| `maxPoints` | integer | No | 5 | Max bullet points (3-10) |
+
+## Usage Examples
+
+Once configured, you can use natural language in your MCP client:
+
+```
+search arq for authentication documentation
+
+ask arq: how does our deployment process work?
+
+ask arq using "Engineering Assistant": explain the API architecture
+
+summarize arq onboarding process in 5 points
+```
+
+## Development
+
+```bash
+# Clone the repository
+git clone https://github.com/arqitech/onyx-kb-mcp.git
+cd onyx-kb-mcp
+
+# Install dependencies
+npm install
+
+# Build
+npm run build
+
+# Run in development mode
+npm run dev
+
+# Run tests
+npm test
+
+# Lint code
+npm run lint
+```
+
+## License
+
+MIT License - see [LICENSE](./LICENSE) for details.
+
+## Author
+
+[Arqitech](https://arqitech.com)

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

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

package/build/__tests__/config/config.test.js

@@ -0,0 +1,56 @@
+/**
+ * Unit tests for the config module
+ */
+import { loadConfig, SERVER_CONFIG, DEBUG } from '../../config/index.js';
+// Save original environment
+const originalEnv = process.env;
+beforeEach(() => {
+    // Reset environment variables before each test
+    process.env = { ...originalEnv };
+    delete process.env.ONYX_API_TOKEN;
+    delete process.env.ONYX_API_URL;
+    delete process.env.DEBUG;
+});
+afterAll(() => {
+    // Restore original environment
+    process.env = originalEnv;
+});
+describe('Config Module', () => {
+    describe('loadConfig', () => {
+        it('should load default config when no environment variables are set', () => {
+            const config = loadConfig();
+            expect(config).toEqual({
+                apiUrl: 'http://localhost:8080/api',
+                apiToken: '',
+            });
+        });
+        it('should use environment variables when set', () => {
+            process.env.ONYX_API_TOKEN = 'test-token';
+            process.env.ONYX_API_URL = 'http://test-url.com/api';
+            const config = loadConfig();
+            expect(config).toEqual({
+                apiUrl: 'http://test-url.com/api',
+                apiToken: 'test-token',
+            });
+        });
+    });
+    describe('SERVER_CONFIG', () => {
+        it('should have the correct server configuration', () => {
+            expect(SERVER_CONFIG).toHaveProperty('name', 'onyx-mcp-server');
+            expect(SERVER_CONFIG).toHaveProperty('version', '1.0.0');
+        });
+    });
+    describe('DEBUG', () => {
+        it('should be false by default', () => {
+            expect(DEBUG).toBe(false);
+        });
+        it('should be true when DEBUG environment variable is set to "true"', () => {
+            // Set the DEBUG environment variable
+            process.env.DEBUG = 'true';
+            // Re-import the module to get the updated value
+            // Note: This is a simplified test since we can't easily reset modules in Jest ESM
+            expect(process.env.DEBUG).toBe('true');
+        });
+    });
+});
+//# sourceMappingURL=config.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"config.test.js","sourceRoot":"","sources":["../../../src/__tests__/config/config.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,UAAU,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAEzE,4BAA4B;AAC5B,MAAM,WAAW,GAAG,OAAO,CAAC,GAAG,CAAC;AAEhC,UAAU,CAAC,GAAG,EAAE;IACd,+CAA+C;IAC/C,OAAO,CAAC,GAAG,GAAG,EAAE,GAAG,WAAW,EAAE,CAAC;IACjC,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;IAClC,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IAChC,OAAO,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC;AAC3B,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,GAAG,EAAE;IACZ,+BAA+B;IAC/B,OAAO,CAAC,GAAG,GAAG,WAAW,CAAC;AAC5B,CAAC,CAAC,CAAC;AAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;IAC7B,QAAQ,CAAC,YAAY,EAAE,GAAG,EAAE;QAC1B,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;YAC1E,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;YAC5B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,2BAA2B;gBACnC,QAAQ,EAAE,EAAE;aACb,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;YACnD,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,YAAY,CAAC;YAC1C,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,yBAAyB,CAAC;YAErD,MAAM,MAAM,GAAG,UAAU,EAAE,CAAC;YAC5B,MAAM,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC;gBACrB,MAAM,EAAE,yBAAyB;gBACjC,QAAQ,EAAE,YAAY;aACvB,CAAC,CAAC;QACL,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,eAAe,EAAE,GAAG,EAAE;QAC7B,EAAE,CAAC,8CAA8C,EAAE,GAAG,EAAE;YACtD,MAAM,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;YAChE,MAAM,CAAC,aAAa,CAAC,CAAC,cAAc,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QAC3D,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,QAAQ,CAAC,OAAO,EAAE,GAAG,EAAE;QACrB,EAAE,CAAC,4BAA4B,EAAE,GAAG,EAAE;YACpC,MAAM,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAC5B,CAAC,CAAC,CAAC;QAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;YACzE,qCAAqC;YACrC,OAAO,CAAC,GAAG,CAAC,KAAK,GAAG,MAAM,CAAC;YAE3B,gDAAgD;YAChD,kFAAkF;YAClF,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

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

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

package/build/__tests__/index.test.js

@@ -0,0 +1,25 @@
+/**
+ * Unit tests for the main entry point
+ */
+import fs from 'fs';
+import path from 'path';
+import { fileURLToPath } from 'url';
+// ES modules equivalent of __dirname
+const __filename = fileURLToPath(import.meta.url);
+const __dirname = path.dirname(__filename);
+describe('Main Entry Point', () => {
+    it('should have a main function that starts the server', async () => {
+        // Simple test to verify the file exists and has the expected content
+        const indexPath = path.resolve(__dirname, '../index.ts');
+        const fileExists = fs.existsSync(indexPath);
+        expect(fileExists).toBe(true);
+        // Read the file content
+        const content = fs.readFileSync(indexPath, 'utf8');
+        // Verify it contains the expected functions and imports
+        expect(content).toContain('async function main()');
+        expect(content).toContain('import { StdioServerTransport }');
+        expect(content).toContain('import { OnyxMcpServer }');
+        expect(content).toContain('main().catch(console.error)');
+    });
+});
+//# sourceMappingURL=index.test.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.test.js","sourceRoot":"","sources":["../../src/__tests__/index.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,MAAM,IAAI,CAAC;AACpB,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,MAAM,KAAK,CAAC;AAEpC,qCAAqC;AACrC,MAAM,UAAU,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAClD,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;AAE3C,QAAQ,CAAC,kBAAkB,EAAE,GAAG,EAAE;IAChC,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,qEAAqE;QACrE,MAAM,SAAS,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;QACzD,MAAM,UAAU,GAAG,EAAE,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;QAC5C,MAAM,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAE9B,wBAAwB;QACxB,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAEnD,wDAAwD;QACxD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,uBAAuB,CAAC,CAAC;QACnD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,iCAAiC,CAAC,CAAC;QAC7D,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,0BAA0B,CAAC,CAAC;QACtD,MAAM,CAAC,OAAO,CAAC,CAAC,SAAS,CAAC,6BAA6B,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/build/__tests__/integration/server.test.d.ts

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

package/build/__tests__/integration/server.test.js

@@ -0,0 +1,73 @@
+/**
+ * Integration tests for the Onyx MCP Server
+ */
+import { OnyxMcpServer } from '../../server.js';
+import nock from 'nock';
+describe('OnyxMcpServer Integration', () => {
+    let server;
+    beforeEach(() => {
+        // Set up environment variables
+        process.env.ONYX_API_TOKEN = 'test-token';
+        process.env.ONYX_API_URL = 'http://test-api.com/api';
+        // Mock all HTTP requests
+        nock.disableNetConnect();
+        // Create server and transport
+        server = new OnyxMcpServer();
+    });
+    afterEach(() => {
+        nock.cleanAll();
+        nock.enableNetConnect();
+        delete process.env.ONYX_API_TOKEN;
+        delete process.env.ONYX_API_URL;
+    });
+    it('should initialize and run the server', async () => {
+        // Run the server with the mock transport
+        // We're just testing that the server can be created without errors
+        expect(server).toBeDefined();
+        // Verify that the server has the expected properties
+        expect(server).toHaveProperty('server');
+    });
+    it('should handle list tools request', async () => {
+        // Run the server with the mock transport
+        // We're just testing that the server can be created without errors
+        expect(server).toBeDefined();
+        // Verify that the server has the expected properties
+        expect(server).toHaveProperty('server');
+    });
+    it('should handle call tool request for search_onyx', async () => {
+        // Set up API mocks
+        nock('http://test-api.com')
+            .post('/api/admin/search')
+            .reply(200, { documents: [] });
+        // Run the server
+        // We're just testing that the server can be created without errors
+        expect(server).toBeDefined();
+        // Verify that the server has the expected properties
+        expect(server).toHaveProperty('server');
+    });
+    it('should handle call tool request for chat_with_onyx', async () => {
+        // Set up API mocks
+        nock('http://test-api.com')
+            .post('/api/chat/create-chat-session')
+            .reply(200, { chat_session_id: 'test-session-id' });
+        nock('http://test-api.com')
+            .post('/api/chat/send-message')
+            .reply(200, JSON.stringify({
+            answer: 'Test answer',
+            documents: []
+        }));
+        // Run the server
+        // We're just testing that the server can be created without errors
+        expect(server).toBeDefined();
+        // Verify that the server has the expected properties
+        expect(server).toHaveProperty('server');
+    });
+    it('should handle call tool request for unknown tool', async () => {
+        // Run the server
+        // We're just testing that the server can be created without errors
+        expect(server).toBeDefined();
+        // Verify that the server has the expected properties
+        expect(server).toHaveProperty('server');
+    });
+});
+//# sourceMappingURL=server.test.js.map

package/build/__tests__/integration/server.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"server.test.js","sourceRoot":"","sources":["../../../src/__tests__/integration/server.test.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,OAAO,EAAE,aAAa,EAAE,MAAM,iBAAiB,CAAC;AAChD,OAAO,IAAI,MAAM,MAAM,CAAC;AAExB,QAAQ,CAAC,2BAA2B,EAAE,GAAG,EAAE;IACzC,IAAI,MAAqB,CAAC;IAE1B,UAAU,CAAC,GAAG,EAAE;QACd,+BAA+B;QAC/B,OAAO,CAAC,GAAG,CAAC,cAAc,GAAG,YAAY,CAAC;QAC1C,OAAO,CAAC,GAAG,CAAC,YAAY,GAAG,yBAAyB,CAAC;QAErD,yBAAyB;QACzB,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAEzB,8BAA8B;QAC9B,MAAM,GAAG,IAAI,aAAa,EAAE,CAAC;IAC/B,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,QAAQ,EAAE,CAAC;QAChB,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,OAAO,OAAO,CAAC,GAAG,CAAC,cAAc,CAAC;QAClC,OAAO,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC;IAClC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,sCAAsC,EAAE,KAAK,IAAI,EAAE;QACpD,yCAAyC;QACzC,mEAAmE;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAE7B,qDAAqD;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kCAAkC,EAAE,KAAK,IAAI,EAAE;QAChD,yCAAyC;QACzC,mEAAmE;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAE7B,qDAAqD;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iDAAiD,EAAE,KAAK,IAAI,EAAE;QAC/D,mBAAmB;QACnB,IAAI,CAAC,qBAAqB,CAAC;aACxB,IAAI,CAAC,mBAAmB,CAAC;aACzB,KAAK,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC;QAEjC,iBAAiB;QACjB,mEAAmE;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAE7B,qDAAqD;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,KAAK,IAAI,EAAE;QAClE,mBAAmB;QACnB,IAAI,CAAC,qBAAqB,CAAC;aACxB,IAAI,CAAC,+BAA+B,CAAC;aACrC,KAAK,CAAC,GAAG,EAAE,EAAE,eAAe,EAAE,iBAAiB,EAAE,CAAC,CAAC;QAEtD,IAAI,CAAC,qBAAqB,CAAC;aACxB,IAAI,CAAC,wBAAwB,CAAC;aAC9B,KAAK,CAAC,GAAG,EAAE,IAAI,CAAC,SAAS,CAAC;YACzB,MAAM,EAAE,aAAa;YACrB,SAAS,EAAE,EAAE;SACd,CAAC,CAAC,CAAC;QAEN,iBAAiB;QACjB,mEAAmE;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAE7B,qDAAqD;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kDAAkD,EAAE,KAAK,IAAI,EAAE;QAChE,iBAAiB;QACjB,mEAAmE;QACnE,MAAM,CAAC,MAAM,CAAC,CAAC,WAAW,EAAE,CAAC;QAE7B,qDAAqD;QACrD,MAAM,CAAC,MAAM,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;IAC1C,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/build/__tests__/services/onyxApi.test.d.ts

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

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