autosubmit mocked tool responses in editor

This commit is contained in:
ramnique 2025-01-22 21:31:37 +05:30
parent 17d119f105
commit 4db69aea24
5 changed files with 51 additions and 19 deletions

View file

@ -186,6 +186,7 @@ export const WorkflowTool = z.object({
name: z.string(),
description: z.string(),
mockInPlayground: z.boolean().default(false).optional(),
autoSubmitMockedResponse: z.boolean().default(false).optional(),
parameters: z.object({
type: z.literal('object'),
properties: z.record(z.object({
@ -216,6 +217,7 @@ export const AgenticAPIPrompt = WorkflowPrompt;
export const AgenticAPITool = WorkflowTool.omit({
mockInPlayground: true,
autoSubmitMockedResponse: true,
});
export const Workflow = z.object({
@ -433,7 +435,7 @@ export function convertWorkflowToAgenticAPI(workflow: z.infer<typeof Workflow>):
controlType: agent.controlType,
})),
tools: workflow.tools.map(tool => {
const { mockInPlayground, ...rest } = tool;
const { mockInPlayground, autoSubmitMockedResponse, ...rest } = tool;
return {
...rest,
};

View file

@ -177,6 +177,7 @@ export async function getAgenticApiResponse(
rawAPIResponse: unknown,
}> {
// call agentic api
console.log(`agentic request`, JSON.stringify(request, null, 2));
const response = await fetch(process.env.AGENTIC_API_URL + '/chat', {
method: 'POST',
body: JSON.stringify(request),

View file

@ -1,6 +1,6 @@
'use client';
import { Button, Spinner, Textarea } from "@nextui-org/react";
import { useEffect, useRef, useState } from "react";
import { useCallback, useEffect, useRef, useState } from "react";
import z from "zod";
import { GetInformationToolResult, WebpageCrawlResponse, Workflow, WorkflowTool } from "@/app/lib/types";
import { executeClientTool, getInformationTool, scrapeWebpage, suggestToolResponse } from "@/app/actions";
@ -201,6 +201,7 @@ function ToolCall({
projectId={projectId}
messages={messages}
sender={sender}
autoSubmit={matchingWorkflowTool?.autoSubmitMockedResponse}
/>;
}
}
@ -504,6 +505,7 @@ function MockToolCall({
projectId,
messages,
sender,
autoSubmit = false,
}: {
toolCall: z.infer<typeof apiV1.AssistantMessageWithToolCalls>['tool_calls'][number];
result: z.infer<typeof apiV1.ToolMessage> | undefined;
@ -511,11 +513,30 @@ function MockToolCall({
projectId: string;
messages: z.infer<typeof apiV1.ChatMessage>[];
sender: string | null | undefined;
autoSubmit?: boolean;
}) {
const [result, setResult] = useState<z.infer<typeof apiV1.ToolMessage> | undefined>(availableResult);
const [response, setResponse] = useState('');
const [generatingResponse, setGeneratingResponse] = useState(false);
const handleSubmit = useCallback(() => {
let parsed;
try {
parsed = JSON.parse(response);
} catch (e) {
alert('Invalid JSON');
return;
}
const result: z.infer<typeof apiV1.ToolMessage> = {
role: 'tool',
tool_call_id: toolCall.id,
tool_name: toolCall.function.name,
content: JSON.stringify(parsed),
};
setResult(result);
handleResult(result);
}, [toolCall.id, toolCall.function.name, handleResult, response]);
useEffect(() => {
if (result) {
return;
@ -549,23 +570,18 @@ function MockToolCall({
};
}, [result, response, toolCall.id, projectId, messages]);
function handleSubmit() {
let parsed;
try {
parsed = JSON.parse(response);
} catch (e) {
alert('Invalid JSON');
// auto submit if autoSubmitMockedResponse is true
useEffect(() => {
if (!autoSubmit) {
return;
}
const result: z.infer<typeof apiV1.ToolMessage> = {
role: 'tool',
tool_call_id: toolCall.id,
tool_name: toolCall.function.name,
content: JSON.stringify(parsed),
};
setResult(result);
handleResult(result);
}
if (result) {
return;
}
if (response) {
handleSubmit();
}
}, [autoSubmit, response, handleSubmit, result]);
return <div className="flex flex-col gap-1">
{sender && <div className='text-gray-500 text-sm ml-3'>{sender}</div>}

View file

@ -226,6 +226,7 @@ export function ToolConfig({
<Divider />
<Checkbox
key="mockInPlayground"
size="sm"
isSelected={tool.mockInPlayground ?? false}
onValueChange={(value) => handleUpdate({
@ -235,6 +236,17 @@ export function ToolConfig({
>
Mock tool in Playground
</Checkbox>
{tool.mockInPlayground && <Checkbox
key="autoSubmitMockedResponse"
size="sm"
isSelected={tool.autoSubmitMockedResponse ?? false}
onValueChange={(value) => handleUpdate({
...tool,
autoSubmitMockedResponse: value
})}
>
Auto-submit mocked response
</Checkbox>}
<Divider />

View file

@ -273,6 +273,7 @@ function reducer(state: State, action: Action): State {
description: "",
parameters: undefined,
mockInPlayground: true,
autoSubmitMockedResponse: true,
...action.tool
});
draft.selection = {