Update examples

Author: RussellLuo

examples/app-builder/auto_team.py

@@ -1,11 +1,11 @@
 import asyncio
 
-from coagent.agents import StreamChatAgent, tool
+from coagent.agents import ChatAgent, tool
 from coagent.core import AgentSpec, idle_loop, new, set_stderr_logger
 from coagent.runtimes import NATSRuntime
 
 
-class AutoTeam(StreamChatAgent):
+class AutoTeam(ChatAgent):
     system = """You are an manager who manages a team that consists of a dev agent and a qa agent.
     
     Your team is responsible for build software for users, and you should follow these rules:

examples/opencsg/csghub.py

@@ -2,7 +2,7 @@
 import asyncio
 from enum import Enum
 
-from coagent.agents import StreamChatAgent, confirm, RunContext, tool
+from coagent.agents import ChatAgent, confirm, RunContext, tool
 from coagent.agents.messages import ChatMessage
 from coagent.core import AgentSpec, idle_loop, new, set_stderr_logger
 from coagent.runtimes import NATSRuntime
@@ -19,7 +19,7 @@ class Language(str, Enum):
     de = "de"
 
 
-class CSGHub(StreamChatAgent):
+class CSGHub(ChatAgent):
     """An agent that help users deal with tasks related to models."""
 
     system = "You are an assistant that help users deal with tasks related to models."

examples/opencsg/dataflow.py

@@ -1,15 +1,15 @@
 import argparse
 import asyncio
 
-from coagent.agents import StreamChatAgent, RunContext, submit, tool
+from coagent.agents import ChatAgent, RunContext, submit, tool
 from coagent.core import AgentSpec, idle_loop, new, set_stderr_logger
 from coagent.runtimes import NATSRuntime
 
 import httpx
 from pydantic import Field
 
 
-class DataFlow(StreamChatAgent):
+class DataFlow(ChatAgent):
     """An agent that help users deal with tasks related to datasets."""
 
     system = "You are an agent that help users deal with tasks related to datasets."

examples/patterns/chaining.py

@@ -1,7 +1,7 @@
 import asyncio
 import os
 
-from coagent.agents import ChatAgent, StreamChatAgent, Sequential, ModelClient
+from coagent.agents import ChatAgent, Sequential, ModelClient
 from coagent.agents.messages import ChatMessage
 from coagent.core import AgentSpec, new, set_stderr_logger
 from coagent.runtimes import LocalRuntime
@@ -62,7 +62,7 @@
 formatter = AgentSpec(
     "formatter",
     new(
-        StreamChatAgent,
+        ChatAgent,
         system="""\
 Format the sorted data as a markdown table with columns:
 | Metric | Value |

examples/patterns/triaging.py

@@ -1,7 +1,7 @@
 import asyncio
 import os
 
-from coagent.agents import ChatMessage, DynamicTriage, ModelClient, StreamChatAgent
+from coagent.agents import ChatAgent, ChatMessage, DynamicTriage, ModelClient
 from coagent.core import AgentSpec, new, set_stderr_logger
 from coagent.runtimes import LocalRuntime
 
@@ -15,7 +15,7 @@
 billing = AgentSpec(
     "team.billing",  # Under the team namespace
     new(
-        StreamChatAgent,
+        ChatAgent,
         system="""\
 You are a billing support specialist. Follow these guidelines:
 1. Always start with "Billing Support Response:"
@@ -33,7 +33,7 @@
 account = AgentSpec(
     "team.account",  # Under the team namespace
     new(
-        StreamChatAgent,
+        ChatAgent,
         system="""\
 You are an account security specialist. Follow these guidelines:
 1. Always start with "Account Support Response:"

examples/translator/translator.py

@@ -1,14 +1,14 @@
 import asyncio
 
-from coagent.agents import StreamChatAgent
+from coagent.agents import ChatAgent
 from coagent.core import AgentSpec, idle_loop, new, set_stderr_logger
 from coagent.runtimes import NATSRuntime
 
 
 translator = AgentSpec(
     "translator",
     new(
-        StreamChatAgent,
+        ChatAgent,
         system="""You are a professional translator that can translate Chinese to English.""",
     ),
 )