解析工具调用参数,优化事件处理逻辑
This commit is contained in:
+28
-2
@@ -52,6 +52,30 @@ const parseEventBlock = (block: string): { event?: string; data?: string } => {
|
||||
};
|
||||
};
|
||||
|
||||
const isObjectRecord = (value: unknown): value is Record<string, unknown> =>
|
||||
typeof value === "object" && value !== null && !Array.isArray(value);
|
||||
|
||||
const resolveToolParams = (
|
||||
params: unknown,
|
||||
argumentsPayload: unknown,
|
||||
): Record<string, unknown> => {
|
||||
if (isObjectRecord(params) && Object.keys(params).length > 0) {
|
||||
return params;
|
||||
}
|
||||
if (isObjectRecord(argumentsPayload)) {
|
||||
return argumentsPayload;
|
||||
}
|
||||
if (typeof argumentsPayload === "string") {
|
||||
try {
|
||||
const parsed = JSON.parse(argumentsPayload) as unknown;
|
||||
return isObjectRecord(parsed) ? parsed : {};
|
||||
} catch {
|
||||
return {};
|
||||
}
|
||||
}
|
||||
return isObjectRecord(params) ? params : {};
|
||||
};
|
||||
|
||||
export const streamAgentChat = async ({
|
||||
message,
|
||||
sessionId,
|
||||
@@ -125,11 +149,13 @@ export const streamAgentChat = async ({
|
||||
try {
|
||||
const parsed = JSON.parse(data) as {
|
||||
session_id?: string;
|
||||
conversationId?: string;
|
||||
content?: string;
|
||||
message?: string;
|
||||
detail?: string;
|
||||
tool?: string;
|
||||
params?: Record<string, unknown>;
|
||||
arguments?: unknown;
|
||||
id?: string;
|
||||
phase?: string;
|
||||
status?: "running" | "completed" | "error";
|
||||
@@ -166,9 +192,9 @@ export const streamAgentChat = async ({
|
||||
} else if (event === "tool_call") {
|
||||
onEvent({
|
||||
type: "tool_call",
|
||||
sessionId: parsed.session_id ?? "",
|
||||
sessionId: parsed.session_id ?? parsed.conversationId ?? "",
|
||||
tool: parsed.tool ?? "",
|
||||
params: parsed.params ?? {},
|
||||
params: resolveToolParams(parsed.params, parsed.arguments),
|
||||
});
|
||||
}
|
||||
} catch {
|
||||
|
||||
Reference in New Issue
Block a user