Update examples

Author: RussellLuo

examples/opencsg/README.md

@@ -40,7 +40,7 @@ export AZURE_API_KEY=<YOUR API KEY>
 python examples/opencsg/csghub.py team1.csghub
 ```
 
-Then search in the client terminal and you will get a professional answer from CSGHub:
+Then search in the client terminal, then you will get a professional answer from CSGHub:
 
 ```bash
 python examples/rich_client.py opencsg

examples/opencsg/csghub.py

@@ -2,7 +2,7 @@
 import asyncio
 from enum import Enum
 
-from coagent.agents import ChatAgent, confirm, RunContext, tool
+from coagent.agents import ChatAgent, confirm, tool
 from coagent.agents.messages import ChatMessage
 from coagent.core import idle_loop, new, set_stderr_logger
 from coagent.runtimes import NATSRuntime
@@ -19,15 +19,6 @@ class Language(str, Enum):
     de = "de"
 
 
-class Task(str, Enum):
-    unknown = ""
-    text_generation = "text-generation"
-    text_to_image = "text-to-image"
-    image_to_image = "image-to-image"
-    text_to_speech = "text-to-speech"
-    conversational = "conversational"
-
-
 class CSGHub(ChatAgent):
     """An agent that help users deal with tasks related to models."""
 
@@ -37,41 +28,33 @@ class CSGHub(ChatAgent):
     @confirm("About to search model {name}, are you sure?")
     async def search_model(
         self,
-        ctx: RunContext,
         name: str = Field(description="The name of the model"),
         language: Language = Field(description="The language that the model supports"),
-        task: Task = Field(description="The task that the model supports"),
+        limit: int = Field(
+            default=10, description="The maximum number of models to return"
+        ),
     ) -> str | ChatMessage:
         """Search models by the given name and language."""
-        print(f"RunContext: {ctx}")
-
         params = dict()
         if name:
             params["search"] = name
         if language:
             params["language"] = language
-        if task:
-            params["task"] = task
+        if limit:
+            params["per"] = limit
 
         async with httpx.AsyncClient() as client:
             resp = await client.get(
-                "https://hub.opencsg-stg.com/api/v1/models",
+                "https://hub.opencsg.com/api/v1/models",
                 params=params,
             )
+            if resp.status_code != 200:
+                return f"Error: {resp.text}"
             models = resp.json()["data"]
             if not models:
                 return "Sorry, no models found."
             return ", ".join([m["name"] for m in models])
 
-    @tool
-    async def upload_model(
-        self,
-        name: str = Field(description="The name of the model"),
-        addr: str = Field(description="The address to upload the model"),
-    ) -> str:
-        """Upload the model to the given address."""
-        return "Success"
-
 
 async def main(name: str):
     async with NATSRuntime.from_servers() as runtime:

examples/opencsg/dataflow.py

@@ -21,9 +21,11 @@ async def search_dataset(
         """Search datasets by the given name."""
         async with httpx.AsyncClient() as client:
             resp = await client.get(
-                "https://hub.opencsg-stg.com/api/v1/datasets",
+                "https://hub.opencsg.com/api/v1/datasets",
                 params=dict(search=name),
             )
+            if resp.status_code != 200:
+                return f"Error: {resp.text}"
             datasets = resp.json()["data"]
             if not datasets:
                 return "Sorry, no datasets found."