Merge pull request #90 from restackio/simplifyChat

Simplify agent chat

Author: aboutphilippe

agent_chat/chat_put.png

@@ -10,7 +10,7 @@ async function eventWorkflow(input: EventInput) {
     await client.sendWorkflowEvent({
       event: {
         name: "message",
-        input: { content: "Telle ma another one" },
+        input: { content: "Tell me a joke" },
       },
       workflow: {
         workflowId: input.workflowId,

agent_chat/eventWorkflow.ts

@@ -10,7 +10,7 @@
     "event-workflow": "tsx eventWorkflow.ts"
   },
   "dependencies": {
-    "@restackio/ai": "^0.0.97",
+    "@restackio/ai": "^0.0.99",
     "@temporalio/workflow": "1.11.3",
     "openai": "^4.80.1"
   },

agent_chat/package.json

@@ -6,21 +6,14 @@ export type EndEvent = {
   end: boolean;
 };
 
-export const messageEvent = defineEvent<functions.Message>("message");
-
+export const messageEvent = defineEvent<functions.Message[]>("message");
 export const endEvent = defineEvent("end");
 
-type AgentChatInput = {
-  message: string;
-};
-
 type AgentChatOutput = {
   messages: functions.Message[];
 };
 
-export async function agentChat({
-  message = "Hello, can you tell me a joke?",
-}: AgentChatInput): Promise<AgentChatOutput> {
+export async function agentChat(): Promise<AgentChatOutput> {
   let endReceived = false;
   let messages: functions.Message[] = [];
 
@@ -30,18 +23,13 @@ export async function agentChat({
       messages,
     });
     messages.push(result);
-    return result;
+    return messages;
   });
 
   onEvent(endEvent, async () => {
     endReceived = true;
   });
 
-  const result = await step<typeof functions>({}).llmChat({
-    messages: [{ role: "user", content: message }],
-  });
-  messages.push(result);
-
   // We use the `condition` function to wait for the event goodbyeReceived to return `True`.
   await condition(() => endReceived);