Fix mypy package resolution for AgentKit tests

Author: digitallysavvy

src/agora_agent/core/pydantic_utilities.py

@@ -209,12 +209,15 @@ class V2RootModel(UniversalBaseModel, pydantic.RootModel):  # type: ignore[misc,
 
 
 def encode_by_type(o: Any) -> Any:
-    encoders_by_class_tuples: Dict[Callable[[Any], Any], Tuple[Any, ...]] = defaultdict(tuple)
+    encoders_by_class_tuples: Dict[Callable[[Any], Any], Tuple[type[Any], ...]] = {}
     for type_, encoder in encoders_by_type.items():
-        encoders_by_class_tuples[encoder] += (type_,)
+        typed_encoder = cast(Callable[[Any], Any], encoder)
+        typed_type = cast(type[Any], type_)
+        encoders_by_class_tuples[typed_encoder] = encoders_by_class_tuples.get(typed_encoder, ()) + (typed_type,)
 
     if type(o) in encoders_by_type:
-        return encoders_by_type[type(o)](o)
+        encoder = cast(Callable[[Any], Any], encoders_by_type[type(o)])
+        return encoder(o)
     for encoder, classes_tuple in encoders_by_class_tuples.items():
         if isinstance(o, classes_tuple):
             return encoder(o)

tests/__init__.py

@@ -0,0 +1 @@
+# This file makes the test suite an explicit package for mypy module resolution.

tests/agentkit/__init__.py

@@ -0,0 +1 @@
+# This file makes the AgentKit test suite an explicit package for mypy module resolution.

tests/agentkit/test_agentkit_parity.py

@@ -1,3 +1,4 @@
+from typing import Any, Dict, List, Tuple
 from types import SimpleNamespace
 import unittest
 
@@ -16,8 +17,8 @@
 
 class DummyAgents:
     def __init__(self) -> None:
-        self.start_calls = []
-        self.turn_calls = []
+        self.start_calls: List[Tuple[Any, Dict[str, Any]]] = []
+        self.turn_calls: List[Tuple[Any, Any, Any]] = []
 
     def start(self, app_id, **kwargs):
         self.start_calls.append((app_id, kwargs))