github/autogen
1
1
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2026-04-20 03:02:16 -04: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
1 changed files with 5 additions and 2 deletions
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")