@lobehub/chat 1.61.5 → 1.62.0

package/src/utils/errorResponse.test.ts

@@ -1,4 +1,4 @@
-import { describe, expect, it } from 'vitest';
+import { describe, expect, it, vi } from 'vitest';
 
 import { AgentRuntimeErrorType } from '@/libs/agent-runtime';
 import { ChatErrorType } from '@/types/fetch';
@@ -32,6 +32,30 @@ describe('createErrorResponse', () => {
     expect(response.status).toBe(403);
   });
 
+  it('returns a 429 status for InsufficientQuota error type', () => {
+    const errorType = AgentRuntimeErrorType.InsufficientQuota;
+    const response = createErrorResponse(errorType);
+    expect(response.status).toBe(429);
+  });
+
+  it('returns a 429 status for QuotaLimitReached error type', () => {
+    const errorType = AgentRuntimeErrorType.QuotaLimitReached;
+    const response = createErrorResponse(errorType);
+    expect(response.status).toBe(429);
+  });
+
+  it('returns a 400 status for ExceededContextWindow error type', () => {
+    const errorType = AgentRuntimeErrorType.ExceededContextWindow;
+    const response = createErrorResponse(errorType);
+    expect(response.status).toBe(400);
+  });
+
+  it('returns a 400 status for SystemTimeNotMatchError error type', () => {
+    const errorType = ChatErrorType.SystemTimeNotMatchError;
+    const response = createErrorResponse(errorType);
+    expect(response.status).toBe(400);
+  });
+
   describe('Provider Biz Error', () => {
     it('returns a 471 status for ProviderBizError error type', () => {
       const errorType = AgentRuntimeErrorType.ProviderBizError;
@@ -44,6 +68,18 @@ describe('createErrorResponse', () => {
       const response = createErrorResponse(errorType);
       expect(response.status).toBe(470);
     });
+
+    it('returns a 472 status for OllamaBizError error type', () => {
+      const errorType = AgentRuntimeErrorType.OllamaBizError;
+      const response = createErrorResponse(errorType);
+      expect(response.status).toBe(472);
+    });
+
+    it('returns a 472 status for OllamaServiceUnavailable error type', () => {
+      const errorType = ChatErrorType.OllamaServiceUnavailable;
+      const response = createErrorResponse(errorType);
+      expect(response.status).toBe(472);
+    });
   });
 
   // 测试状态码不在200-599范围内的情况

package/src/utils/fetch/fetchSSE.ts

@@ -6,6 +6,7 @@ import { ChatErrorType } from '@/types/fetch';
 import { SmoothingParams } from '@/types/llm';
 import {
   ChatMessageError,
+  CitationItem,
   MessageToolCall,
   MessageToolCallChunk,
   MessageToolCallSchema,
@@ -20,6 +21,7 @@ type SSEFinishType = 'done' | 'error' | 'abort';
 export type OnFinishHandler = (
   text: string,
   context: {
+    citations?: CitationItem[];
     observationId?: string | null;
     reasoning?: string;
     toolCalls?: MessageToolCall[];
@@ -38,6 +40,11 @@ export interface MessageReasoningChunk {
   type: 'reasoning';
 }
 
+export interface MessageCitationsChunk {
+  citations: CitationItem[];
+  type: 'citations';
+}
+
 interface MessageToolCallsChunk {
   isAnimationActives?: boolean[];
   tool_calls: MessageToolCall[];
@@ -50,7 +57,7 @@ export interface FetchSSEOptions {
   onErrorHandle?: (error: ChatMessageError) => void;
   onFinish?: OnFinishHandler;
   onMessageHandle?: (
-    chunk: MessageTextChunk | MessageToolCallsChunk | MessageReasoningChunk,
+    chunk: MessageTextChunk | MessageToolCallsChunk | MessageReasoningChunk | MessageCitationsChunk,
   ) => void;
   smoothing?: SmoothingParams | boolean;
 }
@@ -279,6 +286,7 @@ export const fetchSSE = async (url: string, options: RequestInit & FetchSSEOptio
     startSpeed: smoothingSpeed,
   });
 
+  let citations: CitationItem[] | undefined = undefined;
   await fetchEventSource(url, {
     body: options.body,
     fetch: options?.fetcher,
@@ -350,6 +358,13 @@ export const fetchSSE = async (url: string, options: RequestInit & FetchSSEOptio
 
           break;
         }
+
+        case 'citations': {
+          citations = data;
+          options.onMessageHandle?.({ citations: data, type: 'citations' });
+          break;
+        }
+
         case 'reasoning': {
           if (textSmoothing) {
             thinkingController.pushToQueue(data);
@@ -419,6 +434,7 @@ export const fetchSSE = async (url: string, options: RequestInit & FetchSSEOptio
       }
 
       await options?.onFinish?.(output, {
+        citations,
         observationId,
         reasoning: !!thinking ? thinking : undefined,
         toolCalls,