how to use AgentInstantiationContext?

[bio-mlhui]

I am using 0.4.6, I want the factory function to take runtime/agent_key as params, how to implement?

[bio-mlhui]

Can I use the code like this to pass runtime/agent to the factory?

        async def factory_wrapper_2(runtime, agent_id) -> T:
            maybe_agent_instance = agent_factory(runtime, agent_id)
            if inspect.isawaitable(maybe_agent_instance):
                agent_instance = await maybe_agent_instance
            else:
                agent_instance = maybe_agent_instance

            if type_func_alias(agent_instance) != expected_class:
                raise ValueError("Factory registered using the wrong type.")

            return agent_instance

        self._agent_factories[type.type] = factory_wrapper_2

[ekzhu]

Why do you need to pass the runtime and key? Inside the agent constructor you can already get the runtime and key from the AgentInstantiationContext:

from autogen_core import AgentInstantiationContext

# The following code must be inside the factory method. 
agent_id = AgentInstantiationContext.current_agent_id()
runtime = AgentInstantiationContext.current_runtime()

[bio-mlhui]

Sorry for interruption. There seems no documentation to AgentInstantiationContext. Thanks for your help!

[ekzhu]

Let's add this to the API doc. Issue #5519