yujunjun/autogen
1
0
Fork 0
mirror of https://github.com/microsoft/autogen.git synced 2025-08-06 07:43:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix namespace registration (#92)

Browse Source
This commit is contained in:
Jack Gerrits 2024-06-19 10:49:08 -04:00 committed by GitHub
parent 4cebf7257b
commit 1c69c3ad0b
1 changed files with 15 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
src/agnext/application/_single_threaded_agent_runtime.py
View File

@ -121,8 +121,6 @@ class SingleThreadedAgentRuntime(AgentRuntime):
# )
# )
if recipient.namespace not in self._known_namespaces:
self._prepare_namespace(recipient.namespace)
future = asyncio.get_event_loop().create_future()
if recipient.name not in self._known_agent_names:
@ -131,6 +129,8 @@ class SingleThreadedAgentRuntime(AgentRuntime):
if sender is not None and sender.namespace != recipient.namespace:
raise ValueError("Sender and recipient must be in the same namespace to communicate.")
self._process_seen_namespace(recipient.namespace)
logger.info(f"Sending message of type {type(message).__name__} to {recipient.name}: {message.__dict__}")
self._message_queue.append(
@ -180,8 +180,8 @@ class SingleThreadedAgentRuntime(AgentRuntime):
assert explicit_namespace is not None or sender_namespace is not None
namespace = cast(str, explicit_namespace or sender_namespace)
if namespace not in self._known_namespaces:
self._prepare_namespace(namespace)
self._process_seen_namespace(namespace)
self._message_queue.append(
PublishMessageEnvelope(
@ -389,6 +389,11 @@ class SingleThreadedAgentRuntime(AgentRuntime):
else:
self._valid_namespaces[name] = []
# For all already prepared namespaces we need to prepare this agent
for namespace in self._known_namespaces:
if self._type_valid_for_namespace(AgentId(name=name, namespace=namespace)):
self._get_agent(AgentId(name=name, namespace=namespace))
def _invoke_agent_factory(
self, agent_factory: Callable[[], T] | Callable[[AgentRuntime, AgentId], T], agent_id: AgentId
) -> T:
@ -419,13 +424,13 @@ class SingleThreadedAgentRuntime(AgentRuntime):
return agent_id.namespace in valid_namespaces
def _get_agent(self, agent_id: AgentId) -> Agent:
self._process_seen_namespace(agent_id.namespace)
if agent_id in self._instantiated_agents:
return self._instantiated_agents[agent_id]
if not self._type_valid_for_namespace(agent_id):
raise ValueError(f"Agent with name {agent_id.name} not valid for namespace {agent_id.namespace}.")
self._known_namespaces.add(agent_id.namespace)
if agent_id.name not in self._agent_factories:
raise ValueError(f"Agent with name {agent_id.name} not found.")
@ -446,7 +451,11 @@ class SingleThreadedAgentRuntime(AgentRuntime):
# Hydrate the agent instances in a namespace. The primary reason for this is
# to ensure message type subscriptions are set up.
def _prepare_namespace(self, namespace: str) -> None:
def _process_seen_namespace(self, namespace: str) -> None:
if namespace in self._known_namespaces:
return
self._known_namespaces.add(namespace)
for name in self._known_agent_names:
if self._type_valid_for_namespace(AgentId(name=name, namespace=namespace)):
self._get_agent(AgentId(name=name, namespace=namespace))