beamlit 0.0.58__py3-none-any.whl → 0.0.59__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/authentication/clientcredentials.py

@@ -96,7 +96,7 @@ class ClientCredentials(Auth):
         Raises:
             Exception: If token refresh fails.
         """
-        err = self.refresh_if_needed()
+        err = self.do_refresh()
         if err:
             return err
 

beamlit/authentication/credentials.py

@@ -10,7 +10,6 @@ from pathlib import Path
 from typing import List
 
 import yaml
-
 from beamlit.common.settings import Settings
 
 logger = getLogger(__name__)
@@ -132,7 +131,7 @@ def load_config() -> Config:
                             workspaces.append(WorkspaceConfig(name=ws["name"], credentials=creds))
                         config.workspaces = workspaces
                         if "context" in data:
-                            config.context = ContextConfig(**data["context"])
+                            config.context = ContextConfig(workspace=data["context"].get("workspace", ""))
             except yaml.YAMLError:
                 # Invalid YAML, use empty config
                 pass

{beamlit-0.0.58.dist-info → beamlit-0.0.59.dist-info}/METADATA

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

{beamlit-0.0.58.dist-info → beamlit-0.0.59.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
@@ -115,8 +115,8 @@ beamlit/api/workspaces/update_workspace_user_role.py,sha256=Yn9iuJ4tKtauzBiJyU4-
 beamlit/authentication/__init__.py,sha256=wiXqRbc7E-ulrH_ueA9duOGFvXeo7-RvhSD1XbFogMo,1020
 beamlit/authentication/apikey.py,sha256=Q3Kz4Zbc_Lr6lYwiwiLo6AhggKcdqsd6xJjDi-nERiI,1557
 beamlit/authentication/authentication.py,sha256=OgqS6YciBnh0MomeEaL2YuZt1fHPOfQUkRhsIu6tcvk,5812
-beamlit/authentication/clientcredentials.py,sha256=a3KNT3x3c2izS1EQXbL-bO9Xm-6T16qY7cLf97eE8u8,5198
-beamlit/authentication/credentials.py,sha256=Xb2Ajj-72FdU1l_pI8OMDRSXruCJHUQjME1NxTBmkTU,8713
+beamlit/authentication/clientcredentials.py,sha256=oKopOql7nugarWwcuJ_2mGaPODuIsixnxyX2I-yRTGM,5191
+beamlit/authentication/credentials.py,sha256=YNMCRpNa2Cs7lN647Sz4-8LxfZtJPhSyyxCVwNzTAxo,8741
 beamlit/authentication/device_mode.py,sha256=gBJqJpWQXEG8gZbWXhlvQyk8FfHH95-aqURUnudil-4,6526
 beamlit/common/__init__.py,sha256=saX5X3hRCJ9erSlXuSkZ2VGgquvpgdcofAU_9sM4bCE,354
 beamlit/common/error.py,sha256=eOftZvd6XsAcQIYl9V7B1msGnTZf-HO0hGxBw1uIxaA,763
@@ -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.58.dist-info/METADATA,sha256=1JuqbE0ek155mK2KttjVURNzVtGu6efMKMX_4aj59FU,3542
-beamlit-0.0.58.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.58.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.58.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.58.dist-info/RECORD,,
+beamlit-0.0.59.dist-info/METADATA,sha256=yrp2SCE0IZUOM1-gJkdGJAK0nY9G2DHi2HQvM9QVLNQ,3542
+beamlit-0.0.59.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.59.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
+beamlit-0.0.59.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.59.dist-info/RECORD,,