beamlit 0.0.57rc113__py3-none-any.whl → 0.0.57rc115__py3-none-any.whl

beamlit/agents/thread.py

@@ -18,6 +18,8 @@ def get_default_thread(request: Request) -> str:
     Returns:
         str: The extracted thread identifier. Returns an empty string if no valid identifier is found.
     """
+    if request.headers.get("X-Beamlit-Thread-Id"):
+        return request.headers.get("X-Beamlit-Thread-Id")
     if request.headers.get("X-Beamlit-Sub"):
         return request.headers.get("X-Beamlit-Sub")
     authorization = request.headers.get("Authorization", request.headers.get("X-Beamlit-Authorization"))

beamlit/deploy/deploy.py

@@ -14,21 +14,13 @@ from pathlib import Path
 from typing import Literal
 
 import yaml
-
 from beamlit.api.agents import get_agent
 from beamlit.authentication import new_client
 from beamlit.client import AuthenticatedClient
 from beamlit.common import slugify
 from beamlit.common.settings import Settings, get_settings, init
-from beamlit.models import (
-    Agent,
-    AgentSpec,
-    Flavor,
-    Function,
-    FunctionSpec,
-    Metadata,
-    MetadataLabels,
-)
+from beamlit.models import (Agent, AgentSpec, Flavor, Function, FunctionSpec,
+                            Metadata, MetadataLabels)
 
 from .format import arg_to_dict
 from .parser import Resource, get_description, get_parameters, get_resources
@@ -55,6 +47,14 @@ def set_default_values(resource: Resource, deployment: Agent | Function):
         deployment.metadata.display_name = deployment.metadata.name
     if not deployment.spec.description:
         deployment.spec.description = get_description(None, resource)
+    if isinstance(deployment, Agent):
+        deployment.spec.functions = []
+        for arg in resource.decorator.keywords:
+            if arg.arg == "remote_functions":
+                if isinstance(arg.value, ast.List):
+                    for value in arg.value.elts:
+                        if isinstance(value, ast.Constant):
+                            deployment.spec.functions.append(slugify(value.value))
     return deployment
 
 def get_beamlit_deployment_from_resource(
@@ -131,7 +131,8 @@ def get_agent_yaml(
         agent.spec.repository = agent_response.spec.repository
     except Exception:
         pass
-    agent.spec.functions = [slugify(function.metadata.name) for (_, function) in functions]
+    agent.spec.functions = agent.spec.functions or []
+    agent.spec.functions = agent.spec.functions + [slugify(function.metadata.name) for (_, function) in functions]
     agent.metadata.labels = agent.metadata.labels and MetadataLabels.from_dict(agent.metadata.labels) or MetadataLabels()
     agent.metadata.labels["x-beamlit-auto-generated"] = "true"
     agent_yaml = yaml.dump(agent.to_dict())

beamlit/functions/mcp/mcp.py

@@ -12,14 +12,13 @@ import pydantic
 import pydantic_core
 import requests
 import typing_extensions as t
-from langchain_core.tools.base import BaseTool, BaseToolkit, ToolException
-from mcp import ClientSession
-from mcp.types import CallToolResult, ListToolsResult
-
 from beamlit.authentication import get_authentication_headers
 from beamlit.authentication.authentication import AuthenticatedClient
 from beamlit.common.settings import get_settings
 from beamlit.functions.mcp.client import websocket_client
+from langchain_core.tools.base import BaseTool, BaseToolkit, ToolException
+from mcp import ClientSession
+from mcp.types import CallToolResult, ListToolsResult
 
 from .utils import create_schema_model
 
@@ -48,7 +47,7 @@ class MCPClient:
                     logger.debug(f"WebSocket tools: {response}")
                     return response
         except Exception as e:
-            logger.error(f"Error listing SSE tools: {e}")
+            logger.error(f"Error listing tools: {e}")
             logger.debug("WebSocket not available, trying HTTP")
             return None  # Signal to list_tools() to try HTTP instead
 

{beamlit-0.0.57rc113.dist-info → beamlit-0.0.57rc115.dist-info}/METADATA

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

{beamlit-0.0.57rc113.dist-info → beamlit-0.0.57rc115.dist-info}/RECORD

@@ -8,7 +8,7 @@ beamlit/agents/__init__.py,sha256=bWsFaXUbAps3IsL3Prti89m1s714vICXodbQi77h3vY,20
 beamlit/agents/chain.py,sha256=JsinjAYBr3oaM4heouZaiaV2jMmi779LHAMtD_4P59s,4867
 beamlit/agents/chat.py,sha256=ufuydptucLNe_Jyr7lQO1WfQ5pe0I5YKh-y0smwWABM,8301
 beamlit/agents/decorator.py,sha256=yRxzrLOMhXW0S4ThRD7cm6UqUarw6gdB_ZP2jnZanvM,8311
-beamlit/agents/thread.py,sha256=PVP9Gey8fMZHA-Cs8WbfmdSlD7g-n4HKuk1sTVf8yOQ,1087
+beamlit/agents/thread.py,sha256=fbQUsnEOPqr-E-Gx-YhmP-c26XCpjqDeVdbNbqdGpDE,1196
 beamlit/agents/voice/openai.py,sha256=-RDBwl16i4TbUhFo5-77Ci3zmI3Y8U2yf2MmvXR2haQ,9479
 beamlit/agents/voice/utils.py,sha256=tQidyM40Ewuy12wKqpvJLvfJgneQ0sZf50dqnerPGHg,836
 beamlit/api/__init__.py,sha256=zTSiG_ujSjAqWPyc435YXaX9XTlpMjiJWBbV-f-YtdA,45
@@ -127,7 +127,7 @@ beamlit/common/settings.py,sha256=9LshW8yVXmX9ZeNhfYH-nBnQV5nKSpi1RTZDNUOxe2s,52
 beamlit/common/slugify.py,sha256=QPJqa1VrSjRle931RN37t24rUjAbyZEg1UFAp2fLgac,679
 beamlit/common/utils.py,sha256=eG201z9gMRnhoHkaZGNtfFUbCzfg_Y59JR4ciMgidW8,1465
 beamlit/deploy/__init__.py,sha256=uRsI_-gTbbki59LlvubeTfG6wfI3o2XqZODW0QXA-Ao,292
-beamlit/deploy/deploy.py,sha256=qpNXHvLj712Uo34MPT8z0AkPNsG03lEPe08p1XpGQhc,11162
+beamlit/deploy/deploy.py,sha256=xt6MY8J3jxMEEA9NCCZXYiy9eGQ5NxdPK0u_z3wW24w,11650
 beamlit/deploy/format.py,sha256=W3ESUHyFv-iZDjVnHOf9YFDDXZSXYIFFbwCoL1GInE0,1162
 beamlit/deploy/parser.py,sha256=gjRUhOVtfKnc1UNc_FhXsEfj9zrMNuq8W93pNsJBpo0,7586
 beamlit/functions/__init__.py,sha256=Mnoqpa1dm7TXwjodBbF_40JyD78aXsOYWmqjDSnA1lU,317
@@ -135,7 +135,7 @@ beamlit/functions/common.py,sha256=v4nmLP9Wotpb5e6hV30XbItgZjr5lwXcF0EOotxRngI,1
 beamlit/functions/decorator.py,sha256=iQbLwUo0K83DFJ3ub8O5jKtkbSINnku6GZcKJ9h7-5E,2292
 beamlit/functions/local/local.py,sha256=F7b_xYDytJIZc0fM1sYMtJgfBCF1cLjquQm83VchTLs,1891
 beamlit/functions/mcp/client.py,sha256=enLo0dzWMBHJEQf6as3UWM8tN3CjUN1YO3UPn67DLac,4072
-beamlit/functions/mcp/mcp.py,sha256=1KMOufh0b8SSXi7BbQd7vu9OClEXDCHUt_nuEOxu8Rc,5950
+beamlit/functions/mcp/mcp.py,sha256=BXNq2i1R1eWQytfXDUR2TUmfUkvg14XJxD4XytXkM4g,5945
 beamlit/functions/mcp/utils.py,sha256=V7bah6cymdtjJ_LJUrNcHDeApDHA6uXvaGVeFJGKj2U,1850
 beamlit/functions/remote/remote.py,sha256=bkDUFiZI8YokqX8Fo76AnKLZF9PcjcDBr37hhxLevs8,6728
 beamlit/models/__init__.py,sha256=042wT7sG_YUmJJAb9edrui-DesMxLjOGVvnEtqc92CY,8916
@@ -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.57rc113.dist-info/METADATA,sha256=Hwx6ZX-alRukrSehva8xagCSY_JA3GhJ3RcSnvY0uWY,3547
-beamlit-0.0.57rc113.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.57rc113.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.57rc113.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.57rc113.dist-info/RECORD,,
+beamlit-0.0.57rc115.dist-info/METADATA,sha256=-4QXAnnZZ9N1D6GkBoJnpIbzthjSFdu9TLQY_NpJraI,3547
+beamlit-0.0.57rc115.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.57rc115.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
+beamlit-0.0.57rc115.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.57rc115.dist-info/RECORD,,