beamlit 0.0.56rc106__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-0.0.56rc106.dist-info → beamlit-0.0.56rc107.dist-info}/METADATA

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

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

@@ -149,7 +149,7 @@ 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=BQkRHOITBndco3ewH2g0_PXMef5OCx2g21GEQlS1_L4,6743
 beamlit/models/__init__.py,sha256=Pb6m_N_UwQ7N-EgdlhaciTHuXyD-8OCwbLicC9Lee1w,10084
@@ -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.56rc106.dist-info/METADATA,sha256=BAqHlVOBGJ3jN0GS1y9vTHHvRBZDP0ccIU0j9VPzS5o,3515
-beamlit-0.0.56rc106.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.56rc106.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.56rc106.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.56rc106.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,,