Testcases

Author: UtkarshMishra-Microsoft

src/backend/__init__.py

@@ -12,8 +12,8 @@
                                             UserMessage)
 from azure.cosmos.partition_key import PartitionKey
 
-from config import Config
-from models.messages import BaseDataModel, Plan, Session, Step, AgentMessage
+from src.backend.config import Config
+from src.backend.models.messages import BaseDataModel, Plan, Session, Step, AgentMessage
 
 
 class CosmosBufferedChatCompletionContext(BufferedChatCompletionContext):
@@ -244,13 +244,13 @@ async def get_messages(self) -> List[LLMMessage]:
                 content = item.get("content", {})
                 message_type = content.get("type")
                 if message_type == "SystemMessage":
-                    message = SystemMessage.model_validate(content)
+                    message = SystemMessage(**content)
                 elif message_type == "UserMessage":
-                    message = UserMessage.model_validate(content)
+                    message = UserMessage(**content)
                 elif message_type == "AssistantMessage":
-                    message = AssistantMessage.model_validate(content)
+                    message = AssistantMessage(**content)
                 elif message_type == "FunctionExecutionResultMessage":
-                    message = FunctionExecutionResultMessage.model_validate(content)
+                    message = FunctionExecutionResultMessage(**content)
                 else:
                     continue
                 messages.append(message)

src/backend/agents/__init__.py

@@ -3,12 +3,14 @@
 from autogen_core.base import AgentId
 from autogen_core.base.intervention import DefaultInterventionHandler
 
-from models.messages import GetHumanInputMessage, GroupChatMessage
+from src.backend.models.messages import GroupChatMessage
+
+from src.backend.models.messages import GetHumanInputMessage
 
 
 class NeedsUserInputHandler(DefaultInterventionHandler):
     def __init__(self):
-        self.question_for_human: Optional[GetHumanInputMessage] = None
+        self.question_for_human: Optional[GetHumanInputMessage] = None # type: ignore
         self.messages: List[Dict[str, Any]] = []
 
     async def on_publish(self, message: Any, *, sender: AgentId | None) -> Any:
@@ -17,7 +19,7 @@ async def on_publish(self, message: Any, *, sender: AgentId | None) -> Any:
         print(
             f"NeedsUserInputHandler received message: {message} from sender: {sender}"
         )
-        if isinstance(message, GetHumanInputMessage):
+        if isinstance(message, GetHumanInputMessage): # type: ignore
             self.question_for_human = message
             self.messages.append(
                 {

src/backend/auth/utils.py

@@ -265,13 +265,13 @@ def from_dict(data: dict) -> "GroupChatMessage":
         body_type = body_data.pop("type")
 
         if body_type == "SystemMessage":
-            body = SystemMessage.from_dict(body_data)
+            body = SystemMessage(**body_data)
         elif body_type == "UserMessage":
-            body = UserMessage.from_dict(body_data)
+            body = UserMessage(**body_data)
         elif body_type == "AssistantMessage":
-            body = AssistantMessage.from_dict(body_data)
+            body = AssistantMessage(**body_data)
         elif body_type == "FunctionExecutionResultMessage":
-            body = FunctionExecutionResultMessage.from_dict(body_data)
+            body = FunctionExecutionResultMessage(**body_data)
         else:
             raise ValueError(f"Unknown message type: {body_type}")
 
@@ -289,3 +289,10 @@ class RequestToSpeak(BaseModel):
 
     def to_dict(self):
         return self.model_dump()
+
+class GetHumanInputMessage:
+    def __init__(self, message):
+        self.message = message
+
+    def __str__(self):
+        return f"GetHumanInputMessage: {self.message}"