@jsonstudio/llms 0.4.2

package/dist/sse/json-to-sse/event-generators/chat.js

@@ -0,0 +1,233 @@
+/**
+ * Chat协议事件生成器
+ * 提供纯函数，将Chat响应数据转换为SSE事件对象，不处理流写入
+ */
+import { IdUtils, TimeUtils } from '../../shared/utils.js';
+// 默认配置
+export const DEFAULT_CHAT_EVENT_GENERATOR_CONFIG = {
+    chunkSize: 12,
+    chunkDelayMs: 8,
+    enableIdGeneration: true,
+    enableTimestampGeneration: true
+};
+/**
+ * 创建默认上下文
+ */
+export function createDefaultContext(model, requestId) {
+    return {
+        model,
+        requestId,
+        choiceIndex: 0,
+        toolCallIndexCounter: 0,
+        contentIndexCounter: new Map()
+    };
+}
+/**
+ * 生成基础chunk数据
+ */
+function createBaseChunk(context, config) {
+    return {
+        id: config.enableIdGeneration ? IdUtils.generateRequestId() : context.requestId,
+        object: 'chat.completion.chunk',
+        created: config.enableTimestampGeneration ? Math.floor(TimeUtils.now() / 1000) : 0,
+        model: context.model
+    };
+}
+/**
+ * 构建role delta事件
+ */
+export function buildRoleDelta(role, context, config = DEFAULT_CHAT_EVENT_GENERATOR_CONFIG) {
+    const baseChunk = createBaseChunk(context, config);
+    const chunk = {
+        ...baseChunk,
+        choices: [{
+                index: context.choiceIndex,
+                delta: { role: role },
+                logprobs: null,
+                finish_reason: null
+            }]
+    };
+    return [{
+            event: 'chat_chunk',
+            type: 'chat_chunk',
+            timestamp: TimeUtils.now(),
+            data: JSON.stringify(chunk),
+            sequenceNumber: 0,
+            protocol: 'chat',
+            direction: 'json_to_sse'
+        }];
+}
+/**
+ * 构建reasoning delta事件
+ */
+export function* buildReasoningDeltas(reasoning, context, config = DEFAULT_CHAT_EVENT_GENERATOR_CONFIG) {
+    if (!reasoning)
+        return;
+    const baseChunk = createBaseChunk(context, config);
+    const chunk = {
+        ...baseChunk,
+        choices: [{
+                index: context.choiceIndex,
+                delta: { reasoning },
+                logprobs: null,
+                finish_reason: null
+            }]
+    };
+    yield {
+        event: 'chat_chunk',
+        type: 'chat_chunk',
+        timestamp: TimeUtils.now(),
+        data: JSON.stringify(chunk),
+        sequenceNumber: 0,
+        protocol: 'chat',
+        direction: 'json_to_sse'
+    };
+}
+/**
+ * 构建content delta事件
+ */
+export function* buildContentDeltas(content, context, config = DEFAULT_CHAT_EVENT_GENERATOR_CONFIG) {
+    if (!content)
+        return;
+    const baseChunk = createBaseChunk(context, config);
+    const chunk = {
+        ...baseChunk,
+        choices: [{
+                index: context.choiceIndex,
+                delta: { content },
+                logprobs: null,
+                finish_reason: null
+            }]
+    };
+    yield {
+        event: 'chat_chunk',
+        type: 'chat_chunk',
+        timestamp: TimeUtils.now(),
+        data: JSON.stringify(chunk),
+        sequenceNumber: 0,
+        protocol: 'chat',
+        direction: 'json_to_sse'
+    };
+}
+/**
+ * 构建tool_call开始事件
+ */
+export function buildToolCallStart(toolCall, toolCallIndex, context, config = DEFAULT_CHAT_EVENT_GENERATOR_CONFIG) {
+    const baseChunk = createBaseChunk(context, config);
+    const chunk = {
+        ...baseChunk,
+        choices: [{
+                index: context.choiceIndex,
+                delta: {
+                    tool_calls: [{
+                            index: toolCallIndex,
+                            id: toolCall.id,
+                            type: toolCall.type || 'function',
+                            function: {
+                                name: toolCall.function.name
+                            }
+                        }]
+                },
+                logprobs: null,
+                finish_reason: null
+            }]
+    };
+    return {
+        event: 'chat_chunk',
+        type: 'chat_chunk',
+        timestamp: TimeUtils.now(),
+        data: JSON.stringify(chunk),
+        sequenceNumber: 0,
+        protocol: 'chat',
+        direction: 'json_to_sse'
+    };
+}
+/**
+ * 构建tool_call arguments delta事件
+ */
+export function* buildToolCallArgsDeltas(args, toolCallIndex, context, config = DEFAULT_CHAT_EVENT_GENERATOR_CONFIG) {
+    if (!args)
+        return;
+    const baseChunk = createBaseChunk(context, config);
+    const chunk = {
+        ...baseChunk,
+        choices: [{
+                index: context.choiceIndex,
+                delta: {
+                    tool_calls: [{
+                            index: toolCallIndex,
+                            function: { arguments: args }
+                        }]
+                },
+                logprobs: null,
+                finish_reason: null
+            }]
+    };
+    yield {
+        event: 'chat_chunk',
+        type: 'chat_chunk',
+        timestamp: TimeUtils.now(),
+        data: JSON.stringify(chunk),
+        sequenceNumber: 0,
+        protocol: 'chat',
+        direction: 'json_to_sse'
+    };
+}
+/**
+ * 构建finish_reason事件
+ */
+export function buildFinishEvent(finishReason, context, config = DEFAULT_CHAT_EVENT_GENERATOR_CONFIG) {
+    const baseChunk = createBaseChunk(context, config);
+    const chunk = {
+        ...baseChunk,
+        choices: [{
+                index: context.choiceIndex,
+                delta: {},
+                logprobs: null,
+                finish_reason: finishReason
+            }]
+    };
+    return {
+        event: 'chat_chunk',
+        type: 'chat_chunk',
+        timestamp: TimeUtils.now(),
+        data: JSON.stringify(chunk),
+        sequenceNumber: 0,
+        protocol: 'chat',
+        direction: 'json_to_sse'
+    };
+}
+/**
+ * 构建done事件
+ */
+export function buildDoneEvent(context, config = DEFAULT_CHAT_EVENT_GENERATOR_CONFIG) {
+    return {
+        event: 'chat.done',
+        type: 'chat.done',
+        timestamp: TimeUtils.now(),
+        data: '[DONE]',
+        sequenceNumber: 0,
+        protocol: 'chat',
+        direction: 'json_to_sse'
+    };
+}
+/**
+ * 构建error事件
+ */
+export function buildErrorEvent(error, context, config = DEFAULT_CHAT_EVENT_GENERATOR_CONFIG) {
+    return {
+        event: 'error',
+        type: 'error',
+        timestamp: TimeUtils.now(),
+        data: JSON.stringify({
+            error: {
+                message: error.message,
+                type: 'internal_error',
+                code: 'generation_error'
+            }
+        }),
+        sequenceNumber: 0,
+        protocol: 'chat',
+        direction: 'json_to_sse'
+    };
+}

package/dist/sse/json-to-sse/event-generators/responses.d.ts

@@ -0,0 +1,99 @@
+/**
+ * Responses协议事件生成器
+ * 提供纯函数，将Responses响应数据转换为SSE事件对象，不处理流写入
+ */
+import { ResponsesSseEvent, ResponsesResponse, ResponsesOutputItem, ResponsesFunctionCallItem, ResponsesReasoningItem, ResponsesContent } from '../../types/index.js';
+export interface ResponsesEventGeneratorConfig {
+    chunkSize: number;
+    chunkDelayMs: number;
+    enableIdGeneration: boolean;
+    enableTimestampGeneration: boolean;
+    enableSequenceNumbers: boolean;
+    enableDelay: boolean;
+}
+export interface ResponsesEventGeneratorContext {
+    requestId: string;
+    model: string;
+    outputIndexCounter: number;
+    contentIndexCounter: Map<string, number>;
+    sequenceCounter: number;
+}
+export declare const DEFAULT_RESPONSES_EVENT_GENERATOR_CONFIG: ResponsesEventGeneratorConfig;
+/**
+ * 创建默认上下文
+ */
+export declare function createDefaultResponsesContext(requestId: string, model: string): ResponsesEventGeneratorContext;
+/**
+ * 构建response.created + response.in_progress事件
+ * 映射：response.start → response.created + response.in_progress
+ */
+export declare function buildResponseStartEvents(response: ResponsesResponse, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): Generator<ResponsesSseEvent>;
+/**
+ * 构建response.output_item.added事件
+ * 映射：output_item.start → response.output_item.added
+ */
+export declare function buildOutputItemStartEvent(outputItem: ResponsesOutputItem, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建response.content_part.added事件
+ * 映射：content_part.start → response.content_part.added
+ */
+export declare function buildContentPartStartEvent(outputItemId: string, contentIndex: number, content: ResponsesContent, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建response.output_text.delta事件流
+ * 映射：content_part.delta (input_text/output_text/…) → response.output_text.delta
+ * 对于 reasoning 使用 response.reasoning_text.delta
+ */
+export declare function buildContentPartDeltas(outputItemId: string, contentIndex: number, text: string, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): Generator<ResponsesSseEvent>;
+/**
+ * 构建content_part.done事件
+ */
+export declare function buildContentPartDoneEvent(outputItemId: string, contentIndex: number, content: ResponsesContent | undefined, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+export declare function buildOutputTextDoneEvent(outputItemId: string, contentIndex: number, fullText: string, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建response.function_call_arguments.delta/done的start部分
+ * 映射：function_call.start → 隐含在第一个 delta 中
+ */
+/**
+ * 构建response.function_call_arguments.delta事件流
+ * 映射：function_call.delta → response.function_call_arguments.delta
+ */
+export declare function buildFunctionCallArgsDeltas(functionCall: ResponsesFunctionCallItem, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): Generator<ResponsesSseEvent>;
+/**
+ * 构建response.function_call_arguments.done事件
+ * 映射：function_call.done → response.function_call_arguments.done
+ */
+export declare function buildFunctionCallDoneEvent(functionCall: ResponsesFunctionCallItem, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建reasoning.start事件
+ */
+export declare function buildReasoningStartEvent(reasoning: ResponsesReasoningItem, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建response.reasoning_text.delta事件流
+ * 映射：reasoning.delta → response.reasoning_text.delta
+ */
+export declare function buildReasoningDeltas(reasoning: ResponsesReasoningItem, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): Generator<ResponsesSseEvent>;
+/**
+ * 构建reasoning.done事件
+ */
+export declare function buildReasoningDoneEvent(reasoning: ResponsesReasoningItem, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建output_item.done事件
+ */
+export declare function buildOutputItemDoneEvent(outputItem: ResponsesOutputItem, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建required_action事件
+ */
+export declare function buildRequiredActionEvent(response: ResponsesResponse, requiredAction: any, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建response.completed事件
+ * 新增：在 done 之前增加 response.completed
+ */
+export declare function buildResponseCompletedEvent(response: ResponsesResponse, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建response.done事件
+ */
+export declare function buildResponseDoneEvent(response: ResponsesResponse, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;
+/**
+ * 构建error事件
+ */
+export declare function buildErrorEvent(error: Error, context: ResponsesEventGeneratorContext, config?: ResponsesEventGeneratorConfig): ResponsesSseEvent;