yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-12-04 10:59:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix pydantic warnings when saving state (#4801)

Browse Source
This commit is contained in:
Sachin Joglekar 2024-12-23 15:17:40 -08:00 committed by GitHub
parent 9c8877ed66
commit 8e116fd86d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/packages/autogen-agentchat/src/autogen_agentchat/state/_states.py
View File

@ -1,4 +1,4 @@
from typing import Any, List, Mapping, Optional
from typing import Annotated, Any, List, Mapping, Optional
from pydantic import BaseModel, Field
@ -7,6 +7,9 @@ from ..messages import (
ChatMessage,
)
# Ensures pydantic can distinguish between types of events & messages.
_AgentMessage = Annotated[AgentEvent | ChatMessage, Field(discriminator="type")]
class BaseState(BaseModel):
"""Base class for all saveable state"""
@ -33,7 +36,7 @@ class TeamState(BaseState):
class BaseGroupChatManagerState(BaseState):
"""Base state for all group chat managers."""
message_thread: List[AgentEvent | ChatMessage] = Field(default_factory=list)
message_thread: List[_AgentMessage] = Field(default_factory=list)
current_turn: int = Field(default=0)
type: str = Field(default="BaseGroupChatManagerState")