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

Rename fields in agent metadata (#336)

Browse Source 
* Rename fields in agent metadata

* team one fixes

* another fix
This commit is contained in:
Jack Gerrits
2024-08-07 16:08:13 -04:00
committed by GitHub
parent 437dbefc32
commit 39489ba395
32 changed files with 94 additions and 95 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/samples/tool-use/coding_direct.py
View File

@@ -66,7 +66,7 @@ class ToolUseAgent(TypeRoutedAgent):
session: List[LLMMessage] = []
session.append(UserMessage(content=message.content, source="User"))
response = await self._model_client.create(self._system_messages + session, tools=self._tool_schema)
session.append(AssistantMessage(content=response.content, source=self.metadata["name"]))
session.append(AssistantMessage(content=response.content, source=self.metadata["type"]))
# Keep executing the tools until the response is not a list of function calls.
while isinstance(response.content, list) and all(isinstance(item, FunctionCall) for item in response.content):
@@ -89,7 +89,7 @@ class ToolUseAgent(TypeRoutedAgent):
session.append(FunctionExecutionResultMessage(content=function_results))
# Execute the model again with the new response.
response = await self._model_client.create(self._system_messages + session, tools=self._tool_schema)
session.append(AssistantMessage(content=response.content, source=self.metadata["name"]))
session.append(AssistantMessage(content=response.content, source=self.metadata["type"]))
assert isinstance(response.content, str)
return Message(content=response.content)

4
python/samples/tool-use/coding_pub_sub.py
View File

@@ -121,7 +121,7 @@ class ToolUseAgent(TypeRoutedAgent):
response = await self._model_client.create(
self._system_messages + self._sessions[session_id], tools=self._tools
)
self._sessions[session_id].append(AssistantMessage(content=response.content, source=self.metadata["name"]))
self._sessions[session_id].append(AssistantMessage(content=response.content, source=self.metadata["type"]))
if isinstance(response.content, str):
# If the response is a string, just publish the response.
@@ -163,7 +163,7 @@ class ToolUseAgent(TypeRoutedAgent):
self._system_messages + self._sessions[message.session_id], tools=self._tools
)
self._sessions[message.session_id].append(
AssistantMessage(content=response.content, source=self.metadata["name"])
AssistantMessage(content=response.content, source=self.metadata["type"])
)
# If the response is a string, just publish the response.
if isinstance(response.content, str):