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

beamlit/agents/decorator.py

@@ -10,14 +10,15 @@ import inspect
 from logging import getLogger
 from typing import Callable
 
+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 Settings, init
 from beamlit.errors import UnexpectedStatus
 from beamlit.functions import get_functions
 from beamlit.models import Agent, AgentSpec, Metadata
-from langgraph.checkpoint.memory import MemorySaver
-from langgraph.prebuilt import create_react_agent
 
 from .chat import get_chat_model_full
 from .voice.openai import OpenAIVoiceReactAgent

beamlit/functions/common.py

@@ -21,6 +21,9 @@ import traceback
 from logging import getLogger
 from typing import Union
 
+from langchain_core.tools import StructuredTool
+from langchain_core.tools.base import create_schema_from_function
+
 from beamlit.authentication import new_client
 from beamlit.client import AuthenticatedClient
 from beamlit.common import slugify
@@ -28,8 +31,6 @@ from beamlit.common.settings import get_settings
 from beamlit.functions.local.local import LocalToolKit
 from beamlit.functions.remote.remote import RemoteToolkit
 from beamlit.models import AgentChain
-from langchain_core.tools import StructuredTool
-from langchain_core.tools.base import create_schema_from_function
 
 logger = getLogger(__name__)
 
@@ -152,8 +153,8 @@ async def get_functions(
                                         func = getattr(module, func_name)
                                         if settings.remote:
                                             toolkit = RemoteToolkit(client, slugify(func.__name__))
-                                            toolkit.initialize()
-                                            functions.extend(toolkit.get_tools())
+                                            await toolkit.initialize()
+                                            functions.extend(await toolkit.get_tools())
                                         else:
                                             if asyncio.iscoroutinefunction(func):
                                                 functions.append(

beamlit/functions/decorator.py

@@ -4,9 +4,10 @@ import functools
 from collections.abc import Callable
 from logging import getLogger
 
-from beamlit.models import Function, FunctionKit
 from fastapi import Request
 
+from beamlit.models import Function, FunctionKit
+
 logger = getLogger(__name__)
 
 def kit(bl_kit: FunctionKit = None, **kwargs: dict) -> Callable:

beamlit/functions/local/local.py

@@ -1,10 +1,11 @@
 from dataclasses import dataclass
 
 import pydantic
+from langchain_core.tools.base import BaseTool
+
 from beamlit.authentication.authentication import AuthenticatedClient
 from beamlit.functions.mcp.mcp import MCPClient, MCPToolkit
 from beamlit.models import Function
-from langchain_core.tools.base import BaseTool
 
 
 @dataclass

beamlit/functions/mcp/client.py

@@ -4,13 +4,11 @@ from typing import Any
 from urllib.parse import urljoin, urlparse
 
 import anyio
-import httpx
+import mcp.types as types
 from anyio.abc import TaskStatus
 from anyio.streams.memory import MemoryObjectReceiveStream, MemoryObjectSendStream
-from websockets.client import connect as ws_connect
 from websockets.client import WebSocketClientProtocol
-
-import mcp.types as types
+from websockets.client import connect as ws_connect
 
 logger = logging.getLogger(__name__)
 

beamlit/functions/mcp/mcp.py

@@ -12,13 +12,14 @@ 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
 

beamlit/functions/remote/remote.py

@@ -11,6 +11,8 @@ from typing import Callable
 
 import pydantic
 import typing_extensions as t
+from langchain_core.tools.base import BaseTool, ToolException
+
 from beamlit.api.functions import get_function, list_functions
 from beamlit.authentication.authentication import AuthenticatedClient
 from beamlit.common.settings import get_settings
@@ -18,7 +20,6 @@ from beamlit.errors import UnexpectedStatus
 from beamlit.functions.mcp.mcp import MCPClient, MCPToolkit
 from beamlit.models import Function, StoreFunctionParameter
 from beamlit.run import RunClient
-from langchain_core.tools.base import BaseTool, ToolException
 
 
 def create_dynamic_schema(name: str, parameters: list[StoreFunctionParameter]) -> type[pydantic.BaseModel]:

beamlit/run.py

@@ -5,6 +5,7 @@ import urllib.parse
 from typing import Any
 
 import requests
+
 from beamlit.client import AuthenticatedClient
 from beamlit.common import HTTPError, get_settings
 

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

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

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

@@ -2,12 +2,12 @@ beamlit/__init__.py,sha256=545gFC-wLLwUktWcOAjUWe_Glha40tBetRTOYSfHnbI,164
 beamlit/client.py,sha256=PnR6ybZk5dLIJPnDKAf2epHOeQC_7yL0fG4muvphHjA,12695
 beamlit/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
 beamlit/py.typed,sha256=8ZJUsxZiuOy1oJeVhsTWQhTG_6pTVHVXk5hJL79ebTk,25
-beamlit/run.py,sha256=jR_DFOCqwNCQcaZg8Cf0kUuugC6InsxtmC8UoZrbpdw,4447
+beamlit/run.py,sha256=YMI8iTPB1M9gd_1FG958tw-Prf9EwIcca4jK4igi7MM,4448
 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=ufuydptucLNe_Jyr7lQO1WfQ5pe0I5YKh-y0smwWABM,8301
-beamlit/agents/decorator.py,sha256=pKA9NIaNoToCsjMkvBpOvfQUsb0B20XwdskTonz-isI,8310
+beamlit/agents/decorator.py,sha256=yRxzrLOMhXW0S4ThRD7cm6UqUarw6gdB_ZP2jnZanvM,8311
 beamlit/agents/thread.py,sha256=PVP9Gey8fMZHA-Cs8WbfmdSlD7g-n4HKuk1sTVf8yOQ,1087
 beamlit/agents/voice/openai.py,sha256=-RDBwl16i4TbUhFo5-77Ci3zmI3Y8U2yf2MmvXR2haQ,9479
 beamlit/agents/voice/utils.py,sha256=tQidyM40Ewuy12wKqpvJLvfJgneQ0sZf50dqnerPGHg,836
@@ -131,13 +131,13 @@ beamlit/deploy/deploy.py,sha256=qpNXHvLj712Uo34MPT8z0AkPNsG03lEPe08p1XpGQhc,1116
 beamlit/deploy/format.py,sha256=W3ESUHyFv-iZDjVnHOf9YFDDXZSXYIFFbwCoL1GInE0,1162
 beamlit/deploy/parser.py,sha256=gjRUhOVtfKnc1UNc_FhXsEfj9zrMNuq8W93pNsJBpo0,7586
 beamlit/functions/__init__.py,sha256=Mnoqpa1dm7TXwjodBbF_40JyD78aXsOYWmqjDSnA1lU,317
-beamlit/functions/common.py,sha256=uPmFx8odqHVas8kSfc41B9g_raWtAAzQBNyuRjtDDb0,10114
-beamlit/functions/decorator.py,sha256=P17iKgDPe5VBEgwLPep1yVTD67C8YtRXuoe_kNEfBFg,2291
-beamlit/functions/local/local.py,sha256=GQlp-dlEd465qMwJ-oSkVtLdDGtIYEE4PsVoAzbJJyg,1890
-beamlit/functions/mcp/client.py,sha256=SwGINxSx9a3_LlDAiYuf8NNw8KZf8B5PIOls_wjJlkU,4086
-beamlit/functions/mcp/mcp.py,sha256=T9dUc1PvYO1FXADLu0rZY4FIjUFFQAOuOkiZghLd8eo,5949
+beamlit/functions/common.py,sha256=v4nmLP9Wotpb5e6hV30XbItgZjr5lwXcF0EOotxRngI,10127
+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/utils.py,sha256=V7bah6cymdtjJ_LJUrNcHDeApDHA6uXvaGVeFJGKj2U,1850
-beamlit/functions/remote/remote.py,sha256=O7zq315jHkcvNbrd4AZfsxn4PVdqGvrecKxBR-EwV9U,6727
+beamlit/functions/remote/remote.py,sha256=bkDUFiZI8YokqX8Fo76AnKLZF9PcjcDBr37hhxLevs8,6728
 beamlit/models/__init__.py,sha256=042wT7sG_YUmJJAb9edrui-DesMxLjOGVvnEtqc92CY,8916
 beamlit/models/acl.py,sha256=tH67gsl_BMaviSbTaaIkO1g9cWZgJ6VgAnYVjQSzGZY,3952
 beamlit/models/agent.py,sha256=DPZZOrDyOAXlh7uvPmhK1jgz8L4UWmaF_SH-OnAdjlw,4162
@@ -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.57rc112.dist-info/METADATA,sha256=KHt8c7Zlqs1EnvWmSVAEYgO3hasTEWMbR8AnnSmUIw0,3547
-beamlit-0.0.57rc112.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.57rc112.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.57rc112.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.57rc112.dist-info/RECORD,,
+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,,