beamlit 0.0.59rc125__tar.gz → 0.0.60__tar.gz

{beamlit-0.0.59rc125 → beamlit-0.0.60}/PKG-INFO

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

{beamlit-0.0.59rc125 → beamlit-0.0.60}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "beamlit"
-version = "0.0.59.pre-125"
+version = "0.0.60"
 description = "Add your description here"
 readme = "README.md"
 authors = [

{beamlit-0.0.59rc125 → beamlit-0.0.60}/src/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-0.0.59rc125 → beamlit-0.0.60}/src/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"], sse=self._function.spec["configurations"]["sse"])
-        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.59rc125 → beamlit-0.0.60}/src/integrationtest/agent.py

@@ -13,17 +13,16 @@ settings = init()
         },
         "spec": {
             "description": "A chat agent using Beamlit to handle your tasks.",
-            "model": "gpt-4o-mini",
+            "model": "sandbox-openai",
         },
     },
-    remote_functions=[],
-    # local_functions=[
-    #     {
-    #         "name": "github-search",
-    #         "description": "A tool that searches GitHub for a given query",
-    #         "url": "http://0.0.0.0:8000",
-    #     }
-    # ],
+    local_functions=[
+        {
+            "name": "brave-search",
+            "description": "A tool that searches Brave for a given query",
+            "url": "http://0.0.0.0:1400",
+        }
+    ],
 )
 async def main(
     input, agent, functions