|
|
@@ -61,7 +61,20 @@ export default function Playground() {
|
|
|
);
|
|
|
|
|
|
const sendMessage = trpc.chat.sendMessage.useMutation({
|
|
|
- onSuccess: () => setIsTyping(false),
|
|
|
+ onSuccess: (data) => {
|
|
|
+ setIsTyping(false);
|
|
|
+ if (data.reply) {
|
|
|
+ setFlowPath(prev => [...prev, data.source ?? data.status ?? "reply"]);
|
|
|
+ setMessages(prev => [...prev, {
|
|
|
+ id: Date.now(),
|
|
|
+ sender: "bot",
|
|
|
+ content: data.reply as string,
|
|
|
+ timestamp: new Date().toISOString(),
|
|
|
+ flowStep: data.source === "flow" ? "Flow: matched" : data.source === "knowledge" ? "KB: matched" : `LLM: ${data.status}`,
|
|
|
+ }]);
|
|
|
+ setShowGreeting(false);
|
|
|
+ }
|
|
|
+ },
|
|
|
onError: () => setIsTyping(false),
|
|
|
});
|
|
|
|