beamlit 0.0.39__tar.gz → 0.0.40__tar.gz

{beamlit-0.0.39 → beamlit-0.0.40}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beamlit
-Version: 0.0.39
+Version: 0.0.40
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12

{beamlit-0.0.39 → beamlit-0.0.40}/pyproject.toml

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

{beamlit-0.0.39 → beamlit-0.0.40}/src/beamlit/agents/decorator.py

@@ -3,14 +3,15 @@ import functools
 import inspect
 from logging import getLogger
 
+from langgraph.checkpoint.memory import MemorySaver
+from langgraph.prebuilt import create_react_agent
+
 from beamlit.api.models import get_model, list_models
 from beamlit.authentication import new_client
 from beamlit.common.settings import init
 from beamlit.errors import UnexpectedStatus
 from beamlit.functions import get_functions
 from beamlit.models import Agent, AgentMetadata, AgentSpec
-from langgraph.checkpoint.memory import MemorySaver
-from langgraph.prebuilt import create_react_agent
 
 from .chat import get_chat_model_full
 
@@ -20,7 +21,6 @@ def agent(
     override_model=None,
     override_agent=None,
     override_functions=None,
-    mcp_hub=None,
     remote_functions=None,
 ):
     logger = getLogger(__name__)
@@ -95,7 +95,6 @@ def agent(
             functions = get_functions(
                 client=client,
                 dir=settings.agent.functions_directory,
-                mcp_hub=mcp_hub,
                 remote_functions=remote_functions,
                 chain=agent.spec.agent_chain,
             remote_functions_empty=not remote_functions,

{beamlit-0.0.39 → beamlit-0.0.40}/src/beamlit/common/settings.py

@@ -4,12 +4,8 @@ from typing import Tuple, Type, Union
 from langchain_core.language_models.chat_models import BaseChatModel
 from langgraph.graph.graph import CompiledGraph
 from pydantic import Field
-from pydantic_settings import (
-    BaseSettings,
-    PydanticBaseSettingsSource,
-    SettingsConfigDict,
-    YamlConfigSettingsSource,
-)
+from pydantic_settings import (BaseSettings, PydanticBaseSettingsSource,
+                               SettingsConfigDict, YamlConfigSettingsSource)
 
 from beamlit.common.logger import init as init_logger
 from beamlit.models import Agent, Function, Model
@@ -59,7 +55,6 @@ class Settings(BaseSettings):
     base_url: str = Field(default="https://api.beamlit.com/v0")
     app_url: str = Field(default="https://app.beamlit.com")
     run_url: str = Field(default="https://run.beamlit.com")
-    mcp_hub_url: str = Field(default="https://mcp-hub-server.beamlit.workers.com")
     registry_url: str = Field(default="https://us.registry.beamlit.com")
     log_level: str = Field(default="INFO")
     enable_opentelemetry: bool = Field(default=False)
@@ -72,10 +67,9 @@ class Settings(BaseSettings):
         if os.getenv('BL_ENV') == 'dev':
             self.base_url = os.getenv('BL_BASE_URL') or "https://api.beamlit.dev/v0"
             self.run_url = os.getenv('BL_RUN_URL') or "https://run.beamlit.dev"
-            self.mcp_hub_url = os.getenv('BL_MCP_HUB_URL') or "https://mcp-hub-server.beamlit.workers.dev"
             self.registry_url = os.getenv('BL_REGISTRY_URL') or "https://eu.registry.beamlit.dev"
             self.app_url = os.getenv('BL_APP_URL') or "https://app.beamlit.dev"
-            
+
     @classmethod
     def settings_customise_sources(
         cls,

{beamlit-0.0.39 → beamlit-0.0.40}/src/beamlit/functions/decorator.py

@@ -16,14 +16,12 @@ from beamlit.authentication import new_client
 from beamlit.client import AuthenticatedClient
 from beamlit.common import slugify
 from beamlit.common.settings import get_settings
-from beamlit.functions.mcp.mcp import MCPClient, MCPToolkit
 from beamlit.functions.remote.remote import RemoteToolkit
 from beamlit.models import AgentChain, Function, FunctionKit
 
 logger = getLogger(__name__)
 
 def get_functions(
-    mcp_hub:Union[list[str], None]=None,
     remote_functions:Union[list[str], None]=None,
     client:Union[AuthenticatedClient, None]=None,
     dir:Union[str, None]=None,
@@ -130,16 +128,6 @@ def get_functions(
                         except Exception as e:
                             logger.warning(f"Error processing {file_path}: {e!s}")
 
-    if mcp_hub:
-        for server in mcp_hub:
-            try:
-                mcp_client = MCPClient(client, server)
-                toolkit = MCPToolkit(client=mcp_client)
-                toolkit.initialize()
-                functions.extend(toolkit.get_tools())
-            except Exception as e:
-                logger.warn(f"Failed to initialize MCP server {server}: {e!s}")
-
     if remote_functions:
         for function in remote_functions:
             try:

{beamlit-0.0.39 → beamlit-0.0.40}/src/beamlit/functions/mcp/mcp.py

@@ -44,26 +44,22 @@ def create_dynamic_schema(name: str, schema: dict[str, t.Any]) -> type[pydantic.
 
 
 class MCPClient:
-    def __init__(self, client: AuthenticatedClient, server_name: str):
+    def __init__(self, client: AuthenticatedClient, url: str):
         self.client = client
-        self.server_name = server_name
-        self.headers = {"Api-Key": "1234567890"}
+        self.url = url
 
     def list_tools(self) -> requests.Response:
         client = self.client.get_httpx_client()
-        url = urllib.parse.urljoin(settings.mcp_hub_url, f"{self.server_name}/tools/list")
-        response = client.request("GET", url, headers=self.headers)
+        response = client.request("GET", f"{self.url}/tools/list")
         response.raise_for_status()
         return response
 
     def call_tool(self, tool_name: str, arguments: dict[str, Any] = None) -> requests.Response:
         client = self.client.get_httpx_client()
-        url = urllib.parse.urljoin(settings.mcp_hub_url, f"{self.server_name}/tools/call")
         response = client.request(
             "POST",
-            url,
+            f"{self.url}/tools/call",
             json={"name": tool_name, "arguments": arguments},
-            headers=self.headers,
         )
         response.raise_for_status()
         return response

{beamlit-0.0.39 → beamlit-0.0.40}/src/beamlit/functions/remote/remote.py

@@ -10,6 +10,7 @@ from langchain_core.tools.base import BaseTool, ToolException
 from beamlit.api.functions import get_function
 from beamlit.authentication.authentication import AuthenticatedClient
 from beamlit.common.settings import get_settings
+from beamlit.functions.mcp.mcp import MCPClient, MCPToolkit
 from beamlit.models import Function, StoreFunctionParameter
 from beamlit.run import RunClient
 
@@ -79,11 +80,9 @@ class RemoteToolkit:
     """
     Remote toolkit
     """
-
     client: AuthenticatedClient
     function: str
     _function: Function | None = None
-
     model_config = pydantic.ConfigDict(arbitrary_types_allowed=True)
 
     def initialize(self) -> None:
@@ -93,9 +92,17 @@ class RemoteToolkit:
 
     @t.override
     def get_tools(self) -> list[BaseTool]:
+        settings = get_settings()
         if self._function is None:
             raise RuntimeError("Must initialize the toolkit first")
 
+        if self._function.spec.integration_connections:
+            url = f"{settings.run_url}/{settings.workspace}/functions/{self._function.metadata.name}"
+            mcp_client = MCPClient(self.client, url)
+            mcp_toolkit = MCPToolkit(client=mcp_client)
+            mcp_toolkit.initialize()
+            return mcp_toolkit.get_tools()
+
         if self._function.spec.kit:
             return [
                 RemoteTool(