Update tests

Author: RussellLuo

coagent/agents/aswarm/util.py

@@ -1,7 +1,6 @@
+from datetime import datetime
 import inspect
 import typing
-from datetime import datetime
-from typing import Any
 
 from pydantic import Field, create_model
 from pydantic.fields import FieldInfo
@@ -237,7 +236,7 @@ def transfer_back_to_triage():
             agent.functions.append(transfer_back_to_triage)
 
 
-def normalize_function_result(result: Any) -> ChatMessage:
+def normalize_function_result(result: typing.Any) -> ChatMessage:
     if isinstance(result, ChatMessage):
         return result
     return ChatMessage(role="assistant", content=str(result))

tests/agents/test_util.py

@@ -56,7 +56,10 @@ def func(a: int, b: str = "ok") -> None:
 
 
 def test_function_to_jsonschema_annotated():
-    def func(a: Annotated[int, "Param a"], b: Annotated[str, "Param b"] = "ok") -> None:
+    def func(
+        a: Annotated[int, "The description for parameter a"],
+        b: Annotated[str, "The description for parameter b"] = "ok",
+    ) -> None:
         """This is a test function."""
         pass
 
@@ -67,10 +70,14 @@ def func(a: Annotated[int, "Param a"], b: Annotated[str, "Param b"] = "ok") -> N
             "name": "func",
             "parameters": {
                 "properties": {
-                    "a": {"description": "Param a", "title": "A", "type": "integer"},
+                    "a": {
+                        "description": "The description for parameter a",
+                        "title": "A",
+                        "type": "integer",
+                    },
                     "b": {
                         "default": "ok",
-                        "description": "Param b",
+                        "description": "The description for parameter b",
                         "title": "B",
                         "type": "string",
                     },
@@ -86,8 +93,8 @@ def func(a: Annotated[int, "Param a"], b: Annotated[str, "Param b"] = "ok") -> N
 
 def test_function_to_jsonschema_pydantic_field():
     def func(
-        a: int = Field(description="Param a"),
-        b: str = Field(default="ok", description="Param b"),
+        a: int = Field(description="The description for parameter a"),
+        b: str = Field(default="ok", description="The description for parameter b"),
     ) -> None:
         """This is a test function."""
         pass
@@ -99,10 +106,14 @@ def func(
             "name": "func",
             "parameters": {
                 "properties": {
-                    "a": {"description": "Param a", "title": "A", "type": "integer"},
+                    "a": {
+                        "description": "The description for parameter a",
+                        "title": "A",
+                        "type": "integer",
+                    },
                     "b": {
                         "default": "ok",
-                        "description": "Param b",
+                        "description": "The description for parameter b",
                         "title": "B",
                         "type": "string",
                     },