beamlit 0.0.58__tar.gz → 0.0.59__tar.gz

{beamlit-0.0.58 → beamlit-0.0.59}/PKG-INFO

@@ -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 → beamlit-0.0.59}/docs/beamlit/agents/chat.md

@@ -65,6 +65,9 @@ Functions
 `get_deepseek_chat_model(**kwargs)`
 :   
 
+`get_gemini_chat_model(**kwargs)`
+:   
+
 `get_mistral_chat_model(**kwargs) ‑> langchain_core.language_models.chat_models.BaseChatModel`
 :   Initializes and returns a MistralAI chat model with the provided keyword arguments.
     

{beamlit-0.0.58 → beamlit-0.0.59}/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.
@@ -31,4 +31,7 @@ Functions
     
     Raises:
         ValueError: If required configurations such as the model are missing.
-        Re-raises exceptions encountered during model retrieval and agent setup.
+        Re-raises exceptions encountered during model retrieval and agent setup.
+
+`initialize_agent(settings: beamlit.common.settings.Settings, agent: beamlit.models.agent.Agent | dict = None, override_model=None, override_agent=None, override_functions=None, remote_functions=None, local_functions=None)`
+:

{beamlit-0.0.58 → beamlit-0.0.59}/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.58 → beamlit-0.0.59}/docs/beamlit/authentication/clientcredentials.md

@@ -50,12 +50,15 @@ Classes
         Raises:
             Exception: If token refresh fails.
 
-    `refresh_if_needed(self) ‑> Exception | None`
+    `get_token(self) ‑> Exception | None`
     :   Checks if the access token needs to be refreshed and performs the refresh if necessary.
         
         Returns:
             Optional[Exception]: An exception if refreshing fails, otherwise None.
 
+    `need_token(self)`
+    :
+
 `DeviceLoginFinalizeResponse(access_token: str, expires_in: int, refresh_token: str, token_type: str)`
 :   DeviceLoginFinalizeResponse(access_token: str, expires_in: int, refresh_token: str, token_type: str)
 

beamlit-0.0.59/docs/beamlit/functions/mcp/client.md

@@ -0,0 +1,13 @@
+Module beamlit.functions.mcp.client
+===================================
+
+Functions
+---------
+
+`remove_request_params(url: str) ‑> str`
+:   
+
+`websocket_client(url: str, headers: dict[str, typing.Any] | None = None, timeout: float = 5)`
+:   Client transport for WebSocket.
+    
+    The `timeout` parameter controls connection timeout.

{beamlit-0.0.58 → beamlit-0.0.59}/docs/beamlit/functions/mcp/index.md

@@ -3,5 +3,6 @@ Namespace beamlit.functions.mcp
 
 Sub-modules
 -----------
+* beamlit.functions.mcp.client
 * beamlit.functions.mcp.mcp
 * beamlit.functions.mcp.utils

{beamlit-0.0.58 → beamlit-0.0.59}/docs/beamlit/functions/mcp/mcp.md

@@ -6,18 +6,18 @@ It includes classes for managing MCP clients, creating dynamic schemas, and inte
 Classes
 -------
 
-`MCPClient(client: beamlit.client.AuthenticatedClient, url: str, sse: bool = False)`
+`MCPClient(client: beamlit.client.AuthenticatedClient, url: str, fallback_url: str | None = None)`
 :   
 
     ### Methods
 
-    `call_tool(self, tool_name: str, arguments: dict[str, typing.Any] = None) ‑> requests.models.Response | AsyncIterator[mcp.types.CallToolResult]`
+    `call_tool(self, tool_name: str, arguments: dict[str, typing.Any] = None, is_fallback: bool = False) ‑> requests.models.Response | AsyncIterator[mcp.types.CallToolResult]`
     :
 
-    `list_sse_tools(self) ‑> mcp.types.ListToolsResult`
+    `list_tools(self) ‑> mcp.types.ListToolsResult`
     :
 
-    `list_tools(self) ‑> mcp.types.ListToolsResult`
+    `list_ws_tools(self, is_fallback: bool = False) ‑> mcp.types.ListToolsResult`
     :
 
 `MCPTool(**kwargs: Any)`
@@ -26,7 +26,6 @@ Classes
     Attributes:
         client (MCPClient): The MCP client instance.
         handle_tool_error (bool | str | Callable[[ToolException], str] | None): Error handling strategy.
-        sse (bool): Whether to use SSE streaming for responses.
     
     Initialize the tool.
 

beamlit-0.0.59/docs/beamlit/serve/app.md

@@ -0,0 +1,30 @@
+Module beamlit.serve.app
+========================
+Module: app
+
+This module sets up and runs the Beamlit server using FastAPI.
+It configures middleware, handles server lifespan events, and defines endpoints.
+
+Functions
+---------
+
+`health()`
+:   Health check endpoint.
+    
+    Returns:
+        dict: A simple status message indicating the server is running.
+
+`import_module()`
+:   Dynamically imports the main server module based on settings.
+    
+    Returns:
+        Callable: The main function to run the server.
+
+`lifespan(app: fastapi.applications.FastAPI)`
+:   Manages the lifespan events of the FastAPI application.
+    
+    Args:
+        app (FastAPI): The FastAPI application instance.
+
+`root(request: starlette.requests.Request)`
+:

{beamlit-0.0.58 → beamlit-0.0.59}/pyproject.toml

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

{beamlit-0.0.58 → beamlit-0.0.59}/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.

{beamlit-0.0.58 → beamlit-0.0.59}/src/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-0.0.58 → beamlit-0.0.59}/src/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/docs/beamlit/serve/app.md

@@ -1,2 +0,0 @@
-Module beamlit.serve.app
-========================