continue as new example (#254)

Author: justrdk

agent_chat/pyproject.toml

@@ -10,7 +10,8 @@ dependencies = [
     "watchfiles>=1.0.4",
     "python-dotenv==1.0.1",
     "openai>=1.61.0",
-    "restack-ai>=0.0.94",]
+    "restack-ai>=0.0.100",
+]
 
 [project.scripts]
 dev = "src.services:watch_services"

agent_chat/src/agents/agent.py

@@ -49,4 +49,10 @@ async def end(self, end: EndEvent) -> EndEvent:
     @agent.run
     async def run(self, function_input: dict) -> None:
         log.info("AgentChat function_input", function_input=function_input)
-        await agent.condition(lambda: self.end)
+        await agent.condition(lambda: self.end or agent.should_continue_as_new())
+
+        if self.end is True:
+            log.info("AgentChat end")
+            return
+
+        await agent.agent_continue_as_new()