Tools/Notes
1
0
Fork 0
mirror of https://github.com/TriliumNext/Notes.git synced 2025-07-27 18:12:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(unit): resolve type errors

Browse Source
This commit is contained in:
perf3ct 2025-06-08 21:03:07 +00:00
parent 0c44dd0e63
commit 224cae6db2
No known key found for this signature in database
GPG Key ID: 569C4EEC436F5232
5 changed files with 17 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
apps/server/src/routes/api/llm.spec.ts
View File

@ -323,7 +323,7 @@ describe("LLM API Tests", () => {
const options = (await import("../../services/options.js")).default; const options = (await import("../../services/options.js")).default;
// Setup default mock behaviors // Setup default mock behaviors
options.getOptionBool.mockReturnValue(true); // AI enabled (options.getOptionBool as any).mockReturnValue(true); // AI enabled
mockAiServiceManager.getOrCreateAnyService.mockResolvedValue({}); mockAiServiceManager.getOrCreateAnyService.mockResolvedValue({});
mockGetSelectedModelConfig.mockResolvedValue({ mockGetSelectedModelConfig.mockResolvedValue({
model: 'test-model', model: 'test-model',
@ -466,8 +466,8 @@ describe("LLM API Tests", () => {
getContent: () => 'Root note content for testing' getContent: () => 'Root note content for testing'
}) })
}) })
}; } as any;
vi.mocked(await import('../../becca/becca.js')).default = mockBecca; (await import('../../becca/becca.js') as any).default = mockBecca;
// Setup streaming with mention context // Setup streaming with mention context
mockChatPipelineExecute.mockImplementation(async (input) => { mockChatPipelineExecute.mockImplementation(async (input) => {
@ -628,7 +628,7 @@ describe("LLM API Tests", () => {
it("should handle AI disabled state", async () => { it("should handle AI disabled state", async () => {
// Import options service to access mock // Import options service to access mock
const options = (await import("../../services/options.js")).default; const options = (await import("../../services/options.js")).default;
options.getOptionBool.mockReturnValue(false); // AI disabled (options.getOptionBool as any).mockReturnValue(false); // AI disabled
const response = await supertest(app) const response = await supertest(app)
.post(`/api/llm/chat/${testChatId}/messages/stream`) .post(`/api/llm/chat/${testChatId}/messages/stream`)
@ -740,7 +740,7 @@ describe("LLM API Tests", () => {
const ws = (await import("../../services/ws.js")).default; const ws = (await import("../../services/ws.js")).default;
// Verify multiple chunks were sent // Verify multiple chunks were sent
const streamCalls = ws.sendMessageToAllClients.mock.calls.filter( const streamCalls = (ws.sendMessageToAllClients as any).mock.calls.filter(
call => call[0].type === 'llm-stream' && call[0].content call => call[0].type === 'llm-stream' && call[0].content
); );
expect(streamCalls.length).toBeGreaterThan(5); expect(streamCalls.length).toBeGreaterThan(5);

12
apps/server/src/services/llm/pipeline/chat_pipeline.spec.ts
View File

@ -269,7 +269,7 @@ describe('ChatPipeline', () => {
it('should handle tool calling iterations', async () => { it('should handle tool calling iterations', async () => {
// Mock LLM response to include tool calls // Mock LLM response to include tool calls
pipeline.stages.llmCompletion.execute.mockResolvedValue({ (pipeline.stages.llmCompletion.execute as any).mockResolvedValue({
response: { response: {
text: 'Hello! How can I help you?', text: 'Hello! How can I help you?',
role: 'assistant', role: 'assistant',
@ -279,7 +279,7 @@ describe('ChatPipeline', () => {
}); });
// Mock tool calling to require iteration then stop // Mock tool calling to require iteration then stop
pipeline.stages.toolCalling.execute (pipeline.stages.toolCalling.execute as any)
.mockResolvedValueOnce({ needsFollowUp: true, messages: [] }) .mockResolvedValueOnce({ needsFollowUp: true, messages: [] })
.mockResolvedValueOnce({ needsFollowUp: false, messages: [] }); .mockResolvedValueOnce({ needsFollowUp: false, messages: [] });
@ -290,7 +290,7 @@ describe('ChatPipeline', () => {
it('should respect max tool call iterations', async () => { it('should respect max tool call iterations', async () => {
// Mock LLM response to include tool calls // Mock LLM response to include tool calls
pipeline.stages.llmCompletion.execute.mockResolvedValue({ (pipeline.stages.llmCompletion.execute as any).mockResolvedValue({
response: { response: {
text: 'Hello! How can I help you?', text: 'Hello! How can I help you?',
role: 'assistant', role: 'assistant',
@ -300,7 +300,7 @@ describe('ChatPipeline', () => {
}); });
// Mock tool calling to always require iteration // Mock tool calling to always require iteration
pipeline.stages.toolCalling.execute.mockResolvedValue({ needsFollowUp: true, messages: [] }); (pipeline.stages.toolCalling.execute as any).mockResolvedValue({ needsFollowUp: true, messages: [] });
await pipeline.execute(input); await pipeline.execute(input);
@ -309,7 +309,7 @@ describe('ChatPipeline', () => {
}); });
it('should handle stage errors gracefully', async () => { it('should handle stage errors gracefully', async () => {
pipeline.stages.modelSelection.execute.mockRejectedValueOnce(new Error('Model selection failed')); (pipeline.stages.modelSelection.execute as any).mockRejectedValueOnce(new Error('Model selection failed'));
await expect(pipeline.execute(input)).rejects.toThrow('Model selection failed'); await expect(pipeline.execute(input)).rejects.toThrow('Model selection failed');
}); });
@ -408,7 +408,7 @@ describe('ChatPipeline', () => {
}; };
it('should propagate errors from stages', async () => { it('should propagate errors from stages', async () => {
pipeline.stages.modelSelection.execute.mockRejectedValueOnce(new Error('Model selection failed')); (pipeline.stages.modelSelection.execute as any).mockRejectedValueOnce(new Error('Model selection failed'));
await expect(pipeline.execute(input)).rejects.toThrow('Model selection failed'); await expect(pipeline.execute(input)).rejects.toThrow('Model selection failed');
}); });

2
apps/server/src/services/llm/providers/integration/streaming.spec.ts
View File

@ -255,7 +255,7 @@ describe('Provider Streaming Integration Tests', () => {
// Anthropic format needs conversion to our standard format // Anthropic format needs conversion to our standard format
if (chunk.type === 'content_block_delta') { if (chunk.type === 'content_block_delta') {
yield { yield {
message: { content: chunk.delta.text }, message: { content: chunk.delta?.text || '' },
done: false done: false
}; };
} else if (chunk.type === 'message_stop') { } else if (chunk.type === 'message_stop') {

6
apps/server/src/services/llm/streaming/error_handling.spec.ts
View File

@ -3,7 +3,7 @@ import { processProviderStream, StreamProcessor } from '../providers/stream_hand
import type { ProviderStreamOptions } from '../providers/stream_handler.js'; import type { ProviderStreamOptions } from '../providers/stream_handler.js';
// Mock log service // Mock log service
vi.mock('../log.js', () => ({ vi.mock('../../log.js', () => ({
default: { default: {
info: vi.fn(), info: vi.fn(),
error: vi.fn(), error: vi.fn(),
@ -17,7 +17,7 @@ describe('Streaming Error Handling Tests', () => {
beforeEach(async () => { beforeEach(async () => {
vi.clearAllMocks(); vi.clearAllMocks();
log = (await import('../log.js')).default; log = (await import('../../log.js')).default;
mockOptions = { mockOptions = {
providerName: 'ErrorTestProvider', providerName: 'ErrorTestProvider',
modelName: 'error-test-model' modelName: 'error-test-model'
@ -147,7 +147,7 @@ describe('Streaming Error Handling Tests', () => {
} }
}; };
const hangingCallback = vi.fn(async () => { const hangingCallback = vi.fn(async (): Promise<void> => {
// Never resolves // Never resolves
return new Promise(() => {}); return new Promise(() => {});
}); });

4
apps/server/src/services/llm/streaming/tool_execution.spec.ts
View File

@ -3,7 +3,7 @@ import { processProviderStream, StreamProcessor } from '../providers/stream_hand
import type { ProviderStreamOptions } from '../providers/stream_handler.js'; import type { ProviderStreamOptions } from '../providers/stream_handler.js';
// Mock log service // Mock log service
vi.mock('../log.js', () => ({ vi.mock('../../log.js', () => ({
default: { default: {
info: vi.fn(), info: vi.fn(),
error: vi.fn(), error: vi.fn(),
@ -623,7 +623,7 @@ describe('Tool Execution During Streaming Tests', () => {
describe('Tool Call Logging and Debugging', () => { describe('Tool Call Logging and Debugging', () => {
it('should log tool call detection', async () => { it('should log tool call detection', async () => {
const log = (await import('../log.js')).default; const log = (await import('../../log.js')).default;
const toolChunk = { const toolChunk = {
message: { message: {