beamlit 0.0.56rc105__py3-none-any.whl → 0.0.56rc107__py3-none-any.whl

beamlit/functions/mcp/mcp.py

@@ -17,7 +17,10 @@ from mcp import ClientSession
 from mcp.client.sse import sse_client
 from mcp.types import CallToolResult, ListToolsResult
 
+
+
 from beamlit.authentication.authentication import AuthenticatedClient
+from beamlit.authentication import get_authentication_headers
 from beamlit.common.settings import get_settings
 
 from .utils import create_schema_model
@@ -36,7 +39,7 @@ class MCPClient:
     async def list_sse_tools(self) -> ListToolsResult:
         # Create a new context for each SSE connection
         try:
-            async with sse_client(f"{self.url}/sse") as (read_stream, write_stream):
+            async with sse_client(f"{self.url}/sse", headers=get_authentication_headers(settings)) as (read_stream, write_stream):
                 async with ClientSession(read_stream, write_stream) as session:
                     await session.initialize()
                     response = await session.list_tools()
@@ -66,7 +69,7 @@ class MCPClient:
         arguments: dict[str, Any] = None,
     ) -> requests.Response | AsyncIterator[CallToolResult]:
         if self._sse:
-            async with sse_client(f"{self.url}/sse") as (read_stream, write_stream):
+            async with sse_client(f"{self.url}/sse", headers=get_authentication_headers(settings)) as (read_stream, write_stream):
                 async with ClientSession(read_stream, write_stream) as session:
                     await session.initialize()
                     response = await session.call_tool(tool_name, arguments or {})

beamlit/functions/remote/remote.py

@@ -119,15 +119,15 @@ class RemoteToolkit:
 
     def initialize(self) -> None:
         """Initialize the session and retrieve the remote function details."""
+        settings = get_settings()
         if self._function is None:
             try:
-                response = get_function.sync_detailed(self.function, client=self.client)
+                response = get_function.sync_detailed(self.function, client=self.client, environment=settings.environment)
                 function_name = self.function.upper().replace("-", "_")
                 if os.getenv(f"BL_FUNCTION_{function_name}_SERVICE_NAME"):
                     self._service_name = os.getenv(f"BL_FUNCTION_{function_name}_SERVICE_NAME")
                 self._function = response.parsed
             except UnexpectedStatus as e:
-                settings = get_settings()
                 functions = list_functions.sync_detailed(
                     client=self.client,
                     environment=settings.environment,

beamlit/models/agent_information_response.py

@@ -15,13 +15,11 @@ class AgentInformationResponse:
     Attributes:
         description (Union[Unset, str]): Description of the agent
         display_name (Union[Unset, str]): Display name of the agent
-        name (Union[Unset, str]): Name of the agent
         prompt (Union[Unset, str]): Prompt of the agent
     """
 
     description: Union[Unset, str] = UNSET
     display_name: Union[Unset, str] = UNSET
-    name: Union[Unset, str] = UNSET
     prompt: Union[Unset, str] = UNSET
     additional_properties: dict[str, Any] = _attrs_field(init=False, factory=dict)
 
@@ -30,8 +28,6 @@ class AgentInformationResponse:
 
         display_name = self.display_name
 
-        name = self.name
-
         prompt = self.prompt
 
         field_dict: dict[str, Any] = {}
@@ -41,8 +37,6 @@ class AgentInformationResponse:
             field_dict["description"] = description
         if display_name is not UNSET:
             field_dict["displayName"] = display_name
-        if name is not UNSET:
-            field_dict["name"] = name
         if prompt is not UNSET:
             field_dict["prompt"] = prompt
 
@@ -57,14 +51,11 @@ class AgentInformationResponse:
 
         display_name = d.pop("displayName", UNSET)
 
-        name = d.pop("name", UNSET)
-
         prompt = d.pop("prompt", UNSET)
 
         agent_information_response = cls(
             description=description,
             display_name=display_name,
-            name=name,
             prompt=prompt,
         )
 

{beamlit-0.0.56rc105.dist-info → beamlit-0.0.56rc107.dist-info}/METADATA

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

{beamlit-0.0.56rc105.dist-info → beamlit-0.0.56rc107.dist-info}/RECORD

@@ -149,9 +149,9 @@ beamlit/functions/__init__.py,sha256=Mnoqpa1dm7TXwjodBbF_40JyD78aXsOYWmqjDSnA1lU
 beamlit/functions/common.py,sha256=4SK1N0VoQiydUT4frGT9OWMRxgHgMHUcsBhicMkcMmc,9996
 beamlit/functions/decorator.py,sha256=iQbLwUo0K83DFJ3ub8O5jKtkbSINnku6GZcKJ9h7-5E,2292
 beamlit/functions/local/local.py,sha256=KjkHWBxGlG9fliXnOOwZQEop1g2o10IaLiPo4Zu_XAk,1929
-beamlit/functions/mcp/mcp.py,sha256=6pHWTPbglw1-SlqIztAigN284OyYR7ovn29u0IuF73E,5611
+beamlit/functions/mcp/mcp.py,sha256=7qftM5-EyY8DriXgVV1H-7KHoO7_Dn23QzQpMolRFk8,5767
 beamlit/functions/mcp/utils.py,sha256=V7bah6cymdtjJ_LJUrNcHDeApDHA6uXvaGVeFJGKj2U,1850
-beamlit/functions/remote/remote.py,sha256=aeOjvBB2fStuerF7UQ7dDe3dXqrnEUX_54VCgaKkJpk,6717
+beamlit/functions/remote/remote.py,sha256=BQkRHOITBndco3ewH2g0_PXMef5OCx2g21GEQlS1_L4,6743
 beamlit/models/__init__.py,sha256=Pb6m_N_UwQ7N-EgdlhaciTHuXyD-8OCwbLicC9Lee1w,10084
 beamlit/models/acl.py,sha256=tH67gsl_BMaviSbTaaIkO1g9cWZgJ6VgAnYVjQSzGZY,3952
 beamlit/models/agent.py,sha256=pkFemfg0OUAuiqebiT3PurXhjAKvgCa_YOPuyqFVE2o,4264
@@ -159,7 +159,7 @@ beamlit/models/agent_chain.py,sha256=Vz6galRoT20JRM3mAfaJ_HbOX2pS1pL_fyjYM35b0NI
 beamlit/models/agent_history.py,sha256=76ZHoaz7oq3-2ikNQj25NmzyXOzx5JIPxOlfUyljRQg,5124
 beamlit/models/agent_history_event.py,sha256=Ed_xaedwLEPNuM807Ldj5xqUBj4Eua9NqpCzpTRi0Og,3820
 beamlit/models/agent_information_request.py,sha256=l5t_9R2IoZeS-FyPnrwkbQQT7Avzy8fe4ZzFuaNJls4,1868
-beamlit/models/agent_information_response.py,sha256=I8rOeh790OP0glPSYjPZft9uaUg0RmVrfSPdar9AdFs,2587
+beamlit/models/agent_information_response.py,sha256=I-mm4d-mhd521Pkd3aZwVRv0Q9C4u0bCjglK50bnSkE,2345
 beamlit/models/agent_release.py,sha256=AXtHX_Ze7cMh2tKnRw2usRWnLf2B9u_iFPTF6WYdJLM,1928
 beamlit/models/agent_spec.py,sha256=EKYoV5NTc1AVz_8_R6-Cm2LUvR6kiJsd-aPk4iaEd5c,13221
 beamlit/models/api_key.py,sha256=oKuqDF0xe2Z2-6yJqulbzlXEQyM3W7lvQn6FXCktaaU,4278
@@ -282,8 +282,8 @@ beamlit/serve/app.py,sha256=lM59fdUtfkfAYNPWSCU9pkXIPBnhgVGvvgfoMkSVtks,4531
 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.56rc105.dist-info/METADATA,sha256=VzHgElHsaMg3BYdYPWniSMozZmhUiL9AbJDQxdzju7M,3515
-beamlit-0.0.56rc105.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.56rc105.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.56rc105.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.56rc105.dist-info/RECORD,,
+beamlit-0.0.56rc107.dist-info/METADATA,sha256=KgH3gUuxMqxqWvDfWar59ZjNAsqE1RrCRivH1PMftvs,3515
+beamlit-0.0.56rc107.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.56rc107.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
+beamlit-0.0.56rc107.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.56rc107.dist-info/RECORD,,