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

Make save/load state for agent async (#4195)

Browse Source
This commit is contained in:
Jack Gerrits
2024-11-15 10:38:01 -05:00
committed by GitHub
parent 88229513e5
commit 2b565713d0
8 changed files with 22 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/packages/autogen-core/samples/common/agents/_chat_completion_agent.py
View File

@@ -250,12 +250,12 @@ class ChatCompletionAgent(RoutedAgent):
result_as_str = f"Error: {str(e)}"
return (result_as_str, call_id)
def save_state(self) -> Mapping[str, Any]:
async def save_state(self) -> Mapping[str, Any]:
return {
"memory": self._model_context.save_state(),
"system_messages": self._system_messages,
}
def load_state(self, state: Mapping[str, Any]) -> None:
async def load_state(self, state: Mapping[str, Any]) -> None:
self._model_context.load_state(state["memory"])
self._system_messages = state["system_messages"]