beamlit 0.0.59rc122__py3-none-any.whl → 0.0.59rc124__py3-none-any.whl

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.

{beamlit-0.0.59rc122.dist-info → beamlit-0.0.59rc124.dist-info}/METADATA

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

{beamlit-0.0.59rc122.dist-info → beamlit-0.0.59rc124.dist-info}/RECORD

@@ -7,7 +7,7 @@ beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
 beamlit/agents/__init__.py,sha256=bWsFaXUbAps3IsL3Prti89m1s714vICXodbQi77h3vY,206
 beamlit/agents/chain.py,sha256=JsinjAYBr3oaM4heouZaiaV2jMmi779LHAMtD_4P59s,4867
 beamlit/agents/chat.py,sha256=elnJbS28aSRXwmIsAlS0guVK7swRpbY61yFL1QVeGbI,8943
-beamlit/agents/decorator.py,sha256=VtFtMu8rNIS5fAUr39MnUIW8EGrY_84fUnkz2859yK4,8338
+beamlit/agents/decorator.py,sha256=YUhvdZfnfK33OuuWu66MWgxrN_8bcLQDe9iEUNGT2es,8519
 beamlit/agents/thread.py,sha256=kMLqMlWZIHNJdsQHvtlh3C99LuMsS0XrMks-vJaV9U4,787
 beamlit/agents/voice/openai.py,sha256=-RDBwl16i4TbUhFo5-77Ci3zmI3Y8U2yf2MmvXR2haQ,9479
 beamlit/agents/voice/utils.py,sha256=tQidyM40Ewuy12wKqpvJLvfJgneQ0sZf50dqnerPGHg,836
@@ -254,8 +254,8 @@ beamlit/serve/app.py,sha256=5XZci-R95Zjl97wMtQd1BRtonnkJJ2AeoTVFPKGAOfA,4283
 beamlit/serve/middlewares/__init__.py,sha256=O7fyfE1DIYmajFY9WWdzxCgeAQWZzJfeUjzHGbpWaAk,309
 beamlit/serve/middlewares/accesslog.py,sha256=lcu33j4epFSHRBaeTpyt8deNb3kaM3K91-andw4fp80,1112
 beamlit/serve/middlewares/processtime.py,sha256=3x5w1yQexB0xFNKK6fgLbINxT-eLLunfZ6UDV0bIIF4,944
-beamlit-0.0.59rc122.dist-info/METADATA,sha256=m_SrV1-nCvObekAwovLP11nDfiDg1oV8yL497RpLxoM,3547
-beamlit-0.0.59rc122.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.59rc122.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.59rc122.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.59rc122.dist-info/RECORD,,
+beamlit-0.0.59rc124.dist-info/METADATA,sha256=FIAPsgRStlEEjUucK5ousxjERRUxWQsbbmkaiVQstxc,3547
+beamlit-0.0.59rc124.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.59rc124.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
+beamlit-0.0.59rc124.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.59rc124.dist-info/RECORD,,