beamlit 0.0.59rc123__tar.gz → 0.0.59rc124__tar.gz

{beamlit-0.0.59rc123 → beamlit-0.0.59rc124}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.59rc123
+Version: 0.0.59rc124
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 License-File: LICENSE

{beamlit-0.0.59rc123 → beamlit-0.0.59rc124}/docs/beamlit/agents/decorator.md

@@ -7,14 +7,14 @@ Defines decorators for agent functionalities.
 Functions
 ---------
 
-`agent(agent: beamlit.models.agent.Agent | dict = None, override_model=None, override_agent=None, override_functions=None, remote_functions=None, local_functions=None) ‑> Callable`
+`agent(agent: beamlit.models.agent.Agent | dict = None, override_model=None, override_agent: Any | Callable = None, override_functions=None, remote_functions=None, local_functions=None) ‑> Callable`
 :   A decorator factory that configures and wraps functions to integrate with Beamlit agents.
     Handles model initialization, function retrieval, and agent setup.
     
     Parameters:
         agent (Agent | dict, optional): An `Agent` instance or a dictionary containing agent metadata and specifications.
         override_model (Any, optional): An optional model to override the default agent model.
-        override_agent (Any, optional): An optional agent instance to override the default agent.
+        override_agent (Any, Callable, optional): An optional agent instance to override the default agent, or a function which returns an agent instance.
         mcp_hub (Any, optional): An optional MCP hub configuration.
         remote_functions (Any, optional): An optional list of remote functions to be integrated.
         local_functions (Any, optional): An optional list of local functions to be integrated.

{beamlit-0.0.59rc123 → beamlit-0.0.59rc124}/docs/beamlit/agents/index.md

@@ -12,14 +12,14 @@ Sub-modules
 Functions
 ---------
 
-`agent(agent: beamlit.models.agent.Agent | dict = None, override_model=None, override_agent=None, override_functions=None, remote_functions=None, local_functions=None) ‑> Callable`
+`agent(agent: beamlit.models.agent.Agent | dict = None, override_model=None, override_agent: Any | Callable = None, override_functions=None, remote_functions=None, local_functions=None) ‑> Callable`
 :   A decorator factory that configures and wraps functions to integrate with Beamlit agents.
     Handles model initialization, function retrieval, and agent setup.
     
     Parameters:
         agent (Agent | dict, optional): An `Agent` instance or a dictionary containing agent metadata and specifications.
         override_model (Any, optional): An optional model to override the default agent model.
-        override_agent (Any, optional): An optional agent instance to override the default agent.
+        override_agent (Any, Callable, optional): An optional agent instance to override the default agent, or a function which returns an agent instance.
         mcp_hub (Any, optional): An optional MCP hub configuration.
         remote_functions (Any, optional): An optional list of remote functions to be integrated.
         local_functions (Any, optional): An optional list of local functions to be integrated.

{beamlit-0.0.59rc123 → beamlit-0.0.59rc124}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beamlit"
-version = "0.0.59.pre-123"
+version = "0.0.59.pre-124"
 description = "Add your description here"
 readme = "README.md"
 authors = [

{beamlit-0.0.59rc123 → beamlit-0.0.59rc124}/src/beamlit/agents/decorator.py

@@ -8,7 +8,7 @@ import asyncio
 import functools
 import inspect
 from logging import getLogger
-from typing import Callable
+from typing import Any, Callable
 
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.prebuilt import create_react_agent
@@ -37,6 +37,9 @@ async def initialize_agent(
     client = new_client()
     chat_model = override_model or None
 
+    if asyncio.iscoroutinefunction(override_agent):
+        override_agent = await override_agent()
+
     if agent is not None:
         metadata = Metadata(**agent.get("metadata", {}))
         spec = AgentSpec(**agent.get("spec", {}))
@@ -125,7 +128,7 @@ async def initialize_agent(
 def agent(
     agent: Agent | dict = None,
     override_model=None,
-    override_agent=None,
+    override_agent: Any | Callable = None,
     override_functions=None,
     remote_functions=None,
     local_functions=None,
@@ -137,7 +140,7 @@ def agent(
     Parameters:
         agent (Agent | dict, optional): An `Agent` instance or a dictionary containing agent metadata and specifications.
         override_model (Any, optional): An optional model to override the default agent model.
-        override_agent (Any, optional): An optional agent instance to override the default agent.
+        override_agent (Any, Callable, optional): An optional agent instance to override the default agent, or a function which returns an agent instance.
         mcp_hub (Any, optional): An optional MCP hub configuration.
         remote_functions (Any, optional): An optional list of remote functions to be integrated.
         local_functions (Any, optional): An optional list of local functions to be integrated.