beamlit 0.0.60rc126__py3-none-any.whl → 0.0.60rc128__py3-none-any.whl

beamlit/authentication/clientcredentials.py

@@ -96,52 +96,7 @@ class ClientCredentials(Auth):
         Raises:
             Exception: If token refresh fails.
         """
-        err = self.do_refresh()
-        if err:
-            return err
-
+        self.get_token()
         request.headers["X-Beamlit-Authorization"] = f"Bearer {self.credentials.access_token}"
         request.headers["X-Beamlit-Workspace"] = self.workspace_name
-        yield request
-
-    def do_refresh(self) -> Optional[Exception]:
-        """
-        Performs the token refresh using the refresh token.
-
-        Returns:
-            Optional[Exception]: An exception if refreshing fails, otherwise None.
-        """
-        if not self.credentials.refresh_token:
-            return Exception("No refresh token to refresh")
-
-        url = f"{self.base_url}/oauth/token"
-        refresh_data = {
-            "grant_type": "refresh_token",
-            "refresh_token": self.credentials.refresh_token,
-            "device_code": self.credentials.device_code,
-            "client_id": "beamlit",
-        }
-
-        try:
-            response = post(url, json=refresh_data, headers={"Content-Type": "application/json"})
-            response.raise_for_status()
-            finalize_response = DeviceLoginFinalizeResponse(**response.json())
-
-            if not finalize_response.refresh_token:
-                finalize_response.refresh_token = self.credentials.refresh_token
-
-            from .credentials import Credentials, save_credentials
-
-            creds = Credentials(
-                access_token=finalize_response.access_token,
-                refresh_token=finalize_response.refresh_token,
-                expires_in=finalize_response.expires_in,
-                device_code=self.credentials.device_code,
-            )
-
-            self.credentials = creds
-            save_credentials(self.workspace_name, creds)
-            return None
-
-        except Exception as e:
-            return Exception(f"Failed to refresh token: {str(e)}")
+        yield request

beamlit/functions/local/local.py

@@ -42,7 +42,7 @@ class LocalToolKit:
                 raise RuntimeError(f"Failed to initialize local function: {e}")
 
     async def get_tools(self) -> list[BaseTool]:
-        mcp_client = MCPClient(self.client, self._function.spec["configurations"]["url"])
-        mcp_toolkit = MCPToolkit(client=mcp_client)
+        mcp_client = MCPClient(self.client, self._function.spec["configurations"]["url"], None)
+        mcp_toolkit = MCPToolkit(client=mcp_client, url=self._function.spec["configurations"]["url"])
         await mcp_toolkit.initialize()
         return await mcp_toolkit.get_tools()

{beamlit-0.0.60rc126.dist-info → beamlit-0.0.60rc128.dist-info}/METADATA

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

{beamlit-0.0.60rc126.dist-info → beamlit-0.0.60rc128.dist-info}/RECORD

@@ -115,7 +115,7 @@ beamlit/api/workspaces/update_workspace_user_role.py,sha256=Yn9iuJ4tKtauzBiJyU4-
 beamlit/authentication/__init__.py,sha256=wiXqRbc7E-ulrH_ueA9duOGFvXeo7-RvhSD1XbFogMo,1020
 beamlit/authentication/apikey.py,sha256=Q3Kz4Zbc_Lr6lYwiwiLo6AhggKcdqsd6xJjDi-nERiI,1557
 beamlit/authentication/authentication.py,sha256=OgqS6YciBnh0MomeEaL2YuZt1fHPOfQUkRhsIu6tcvk,5812
-beamlit/authentication/clientcredentials.py,sha256=oKopOql7nugarWwcuJ_2mGaPODuIsixnxyX2I-yRTGM,5191
+beamlit/authentication/clientcredentials.py,sha256=4D918PA8hZWr3_KiT5iEcF8d7E3kKNDb0UVx3S2iTe0,3585
 beamlit/authentication/credentials.py,sha256=YNMCRpNa2Cs7lN647Sz4-8LxfZtJPhSyyxCVwNzTAxo,8741
 beamlit/authentication/device_mode.py,sha256=gBJqJpWQXEG8gZbWXhlvQyk8FfHH95-aqURUnudil-4,6526
 beamlit/common/__init__.py,sha256=saX5X3hRCJ9erSlXuSkZ2VGgquvpgdcofAU_9sM4bCE,354
@@ -133,7 +133,7 @@ beamlit/deploy/parser.py,sha256=gjRUhOVtfKnc1UNc_FhXsEfj9zrMNuq8W93pNsJBpo0,7586
 beamlit/functions/__init__.py,sha256=Mnoqpa1dm7TXwjodBbF_40JyD78aXsOYWmqjDSnA1lU,317
 beamlit/functions/common.py,sha256=v4nmLP9Wotpb5e6hV30XbItgZjr5lwXcF0EOotxRngI,10127
 beamlit/functions/decorator.py,sha256=iQbLwUo0K83DFJ3ub8O5jKtkbSINnku6GZcKJ9h7-5E,2292
-beamlit/functions/local/local.py,sha256=ZKjqSC9UxLASRkrb_WCxDDEzlZR3I0s9HX2HvGciONw,1841
+beamlit/functions/local/local.py,sha256=qkd-RGObyjZD0KIwaze_TnTuiJ3lyluCcM97nbyPNBk,1897
 beamlit/functions/mcp/client.py,sha256=enLo0dzWMBHJEQf6as3UWM8tN3CjUN1YO3UPn67DLac,4072
 beamlit/functions/mcp/mcp.py,sha256=voZYIbsHipSCEKXmnU_3uYKa_cfT1M2mv5t1pEwUkkE,6040
 beamlit/functions/mcp/utils.py,sha256=V7bah6cymdtjJ_LJUrNcHDeApDHA6uXvaGVeFJGKj2U,1850
@@ -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.60rc126.dist-info/METADATA,sha256=sY4PdMGwk2yyafS7uFrczuSxDFKJssubyg_XRm3exII,3547
-beamlit-0.0.60rc126.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.60rc126.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.60rc126.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.60rc126.dist-info/RECORD,,
+beamlit-0.0.60rc128.dist-info/METADATA,sha256=B3MFxSnd094ye9cY7XsjIi7NU_k-jrm9fwgRYPPGL7E,3547
+beamlit-0.0.60rc128.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.60rc128.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
+beamlit-0.0.60rc128.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.60rc128.dist-info/RECORD,,