Taylor Mullen
2025-05-06 22:11:29 -07:00
committed by N. Taylor Mullen
parent 201eb38479
commit 782686bcf3
+24 -2
View File
@@ -424,8 +424,30 @@ export const useGeminiStream = (
updateFunctionResponseUI(responseInfo, ToolCallStatus.Error); updateFunctionResponseUI(responseInfo, ToolCallStatus.Error);
setStreamingState(StreamingState.Idle); setStreamingState(StreamingState.Idle);
} else { } else {
// If accepted, set state back to Responding to wait for server execution/response const tool = toolRegistry.getTool(request.name);
setStreamingState(StreamingState.Responding); if (!tool) {
throw new Error(
`Tool "${request.name}" not found or is not registered.`,
);
}
const result = await tool.execute(request.args);
const functionResponse: Part = {
functionResponse: {
name: request.name,
id: request.callId,
response: { output: result.llmContent },
},
};
const responseInfo: ToolCallResponseInfo = {
callId: request.callId,
responsePart: functionResponse,
resultDisplay: result.returnDisplay,
error: undefined,
};
updateFunctionResponseUI(responseInfo, ToolCallStatus.Success);
setStreamingState(StreamingState.Idle);
await submitQuery(functionResponse);
} }
}; };