UNPKG

@restnfeel/agentc-starter-kit

Version:

한국어 기업용 CMS 모듈 - Task Master AI와 함께 빠르게 웹사이트를 구현할 수 있는 재사용 가능한 컴포넌트 시스템

105 lines (102 loc) 4.07 kB
import { OpenAIError, LengthFinishReasonError, ContentFilterFinishReasonError } from '../error.js'; function isAutoParsableResponseFormat(response_format) { return response_format?.['$brand'] === 'auto-parseable-response-format'; } function isAutoParsableTool(tool) { return tool?.['$brand'] === 'auto-parseable-tool'; } function maybeParseChatCompletion(completion, params) { if (!params || !hasAutoParseableInput(params)) { return { ...completion, choices: completion.choices.map((choice) => ({ ...choice, message: { ...choice.message, parsed: null, ...(choice.message.tool_calls ? { tool_calls: choice.message.tool_calls, } : undefined), }, })), }; } return parseChatCompletion(completion, params); } function parseChatCompletion(completion, params) { const choices = completion.choices.map((choice) => { if (choice.finish_reason === 'length') { throw new LengthFinishReasonError(); } if (choice.finish_reason === 'content_filter') { throw new ContentFilterFinishReasonError(); } return { ...choice, message: { ...choice.message, ...(choice.message.tool_calls ? { tool_calls: choice.message.tool_calls?.map((toolCall) => parseToolCall(params, toolCall)) ?? undefined, } : undefined), parsed: choice.message.content && !choice.message.refusal ? parseResponseFormat(params, choice.message.content) : null, }, }; }); return { ...completion, choices }; } function parseResponseFormat(params, content) { if (params.response_format?.type !== 'json_schema') { return null; } if (params.response_format?.type === 'json_schema') { if ('$parseRaw' in params.response_format) { const response_format = params.response_format; return response_format.$parseRaw(content); } return JSON.parse(content); } return null; } function parseToolCall(params, toolCall) { const inputTool = params.tools?.find((inputTool) => inputTool.function?.name === toolCall.function.name); return { ...toolCall, function: { ...toolCall.function, parsed_arguments: isAutoParsableTool(inputTool) ? inputTool.$parseRaw(toolCall.function.arguments) : inputTool?.function.strict ? JSON.parse(toolCall.function.arguments) : null, }, }; } function shouldParseToolCall(params, toolCall) { if (!params) { return false; } const inputTool = params.tools?.find((inputTool) => inputTool.function?.name === toolCall.function.name); return isAutoParsableTool(inputTool) || inputTool?.function.strict || false; } function hasAutoParseableInput(params) { if (isAutoParsableResponseFormat(params.response_format)) { return true; } return (params.tools?.some((t) => isAutoParsableTool(t) || (t.type === 'function' && t.function.strict === true)) ?? false); } function validateInputTools(tools) { for (const tool of tools ?? []) { if (tool.type !== 'function') { throw new OpenAIError(`Currently only \`function\` tool types support auto-parsing; Received \`${tool.type}\``); } if (tool.function.strict !== true) { throw new OpenAIError(`The \`${tool.function.name}\` tool is not marked with \`strict: true\`. Only strict function tools can be auto-parsed`); } } } export { hasAutoParseableInput, isAutoParsableResponseFormat, isAutoParsableTool, maybeParseChatCompletion, parseChatCompletion, shouldParseToolCall, validateInputTools }; //# sourceMappingURL=parser.js.map