beamlit 0.0.57rc120__py3-none-any.whl → 0.0.58__py3-none-any.whl

beamlit/agents/chat.py

@@ -1,11 +1,12 @@
 from logging import getLogger
 from typing import Tuple, Union
 
+from langchain_core.language_models import BaseChatModel
+
 from beamlit.api.models import get_model
 from beamlit.authentication import get_authentication_headers, new_client
 from beamlit.common.settings import get_settings
 from beamlit.models import Model
-from langchain_core.language_models import BaseChatModel
 
 from .voice.openai import OpenAIVoiceReactAgent
 

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/agents/thread.py

@@ -2,7 +2,6 @@
 
 Defines threading capabilities for agents.
 """
-import jwt
 from fastapi import Request
 
 

beamlit/authentication/authentication.py

@@ -7,21 +7,18 @@ It also includes utilities for creating authenticated clients and managing authe
 from dataclasses import dataclass
 from typing import Dict, Generator
 
-from httpx import Auth, Request, Response
-
 from beamlit.common.settings import Settings, get_settings
+from httpx import Auth, Request, Response
 
 from ..client import AuthenticatedClient
 from .apikey import ApiKeyProvider
 from .clientcredentials import ClientCredentials
-from .credentials import (
-    Credentials,
-    current_context,
-    load_credentials,
-    load_credentials_from_settings,
-)
+from .credentials import (Credentials, current_context, load_credentials,
+                          load_credentials_from_settings)
 from .device_mode import BearerToken
 
+global provider_singleton
+provider_singleton = None
 
 class PublicProvider(Auth):
     """
@@ -144,6 +141,11 @@ def get_authentication_headers(settings: Settings) -> Dict[str, str]:
     Returns:
         Dict[str, str]: A dictionary of authentication headers.
     """
+    global provider_singleton
+
+    if provider_singleton:
+        return provider_singleton.get_headers()
+
     context = current_context()
     if context.workspace and not settings.authentication.client.credentials:
         credentials = load_credentials(context.workspace)
@@ -165,4 +167,5 @@ def get_authentication_headers(settings: Settings) -> Dict[str, str]:
 
     if provider is None:
         return None
+    provider_singleton = provider
     return provider.get_headers()

beamlit/authentication/clientcredentials.py

@@ -4,16 +4,13 @@ authentication for Beamlit. It manages token refreshing and authentication flows
 client credentials and refresh tokens.
 """
 
-import base64
-import json
 from dataclasses import dataclass
 from datetime import datetime, timedelta
 from typing import Generator, Optional
 
 import requests
-from httpx import Auth, Request, Response, post
-
 from beamlit.common.settings import get_settings
+from httpx import Auth, Request, Response, post
 
 
 @dataclass
@@ -39,6 +36,7 @@ class ClientCredentials(Auth):
             base_url (str): The base URL for authentication.
         """
         self.credentials = credentials
+        self.expires_at = None
         self.workspace_name = workspace_name
         self.base_url = base_url
 
@@ -52,16 +50,15 @@ class ClientCredentials(Auth):
         Raises:
             Exception: If token refresh fails.
         """
-        err = self.refresh_if_needed()
+        err = self.get_token()
         if err:
             raise err
-
         return {
             "X-Beamlit-Authorization": f"Bearer {self.credentials.access_token}",
             "X-Beamlit-Workspace": self.workspace_name,
         }
 
-    def refresh_if_needed(self) -> Optional[Exception]:
+    def get_token(self) -> Optional[Exception]:
         """
         Checks if the access token needs to be refreshed and performs the refresh if necessary.
 
@@ -69,33 +66,23 @@ class ClientCredentials(Auth):
             Optional[Exception]: An exception if refreshing fails, otherwise None.
         """
         settings = get_settings()
-        if self.credentials.client_credentials and not self.credentials.refresh_token:
+        if self.need_token():
             headers = {"Authorization": f"Basic {self.credentials.client_credentials}", "Content-Type": "application/json"}
             body = {"grant_type": "client_credentials"}
             response = requests.post(f"{settings.base_url}/oauth/token", headers=headers, json=body)
             response.raise_for_status()
-            self.credentials.access_token = response.json()["access_token"]
-            self.credentials.refresh_token = response.json()["refresh_token"]
-            self.credentials.expires_in = response.json()["expires_in"]
-
-        # Need to refresh token if expires in less than 10 minutes
-        parts = self.credentials.access_token.split(".")
-        if len(parts) != 3:
-            return Exception("Invalid JWT token format")
-        try:
-            claims_bytes = base64.urlsafe_b64decode(parts[1] + "=" * (-len(parts[1]) % 4))
-            claims = json.loads(claims_bytes)
-        except Exception as e:
-            return Exception(f"Failed to decode/parse JWT claims: {str(e)}")
-
-        exp_time = datetime.fromtimestamp(claims["exp"])
-        current_time = datetime.now()
-        # Refresh if token expires in less than 10 minutes
-        if current_time + timedelta(minutes=10) > exp_time:
-            return self.do_refresh()
-
+            creds = response.json()
+            self.credentials.access_token = creds["access_token"]
+            self.credentials.refresh_token = creds["refresh_token"]
+            self.credentials.expires_in = creds["expires_in"]
+            self.expires_at = datetime.now() + timedelta(seconds=self.credentials.expires_in)
         return None
 
+    def need_token(self):
+        if not self.expires_at:
+            return True
+        return datetime.now() > self.expires_at - timedelta(minutes=10)
+
     def auth_flow(self, request: Request) -> Generator[Request, Response, None]:
         """
         Processes the authentication flow by ensuring tokens are valid and adding necessary headers.

beamlit/deploy/deploy.py

@@ -14,13 +14,21 @@ from pathlib import Path
 from typing import Literal
 
 import yaml
+
 from beamlit.api.agents import get_agent
 from beamlit.authentication import new_client
 from beamlit.client import AuthenticatedClient
 from beamlit.common import slugify
 from beamlit.common.settings import Settings, get_settings, init
-from beamlit.models import (Agent, AgentSpec, Flavor, Function, FunctionSpec,
-                            Metadata, MetadataLabels)
+from beamlit.models import (
+    Agent,
+    AgentSpec,
+    Flavor,
+    Function,
+    FunctionSpec,
+    Metadata,
+    MetadataLabels,
+)
 
 from .format import arg_to_dict
 from .parser import Resource, get_description, get_parameters, get_resources

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-0.0.57rc120.dist-info → beamlit-0.0.58.dist-info}/METADATA

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

{beamlit-0.0.57rc120.dist-info → beamlit-0.0.58.dist-info}/RECORD

@@ -6,9 +6,9 @@ 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=_g86rpzeDS5XTHTu8N2mTUP0LjMG7rY4iZUjrdKT0ZE,8942
-beamlit/agents/decorator.py,sha256=wgdbgJ65LfmhldGWxG8PBYU_7datnVpqKRzom6Ok3GU,8337
-beamlit/agents/thread.py,sha256=QecahZCH8_U91cvCw2X6cYn9N1liVXVl3u3lFxVDdNg,798
+beamlit/agents/chat.py,sha256=elnJbS28aSRXwmIsAlS0guVK7swRpbY61yFL1QVeGbI,8943
+beamlit/agents/decorator.py,sha256=VtFtMu8rNIS5fAUr39MnUIW8EGrY_84fUnkz2859yK4,8338
+beamlit/agents/thread.py,sha256=kMLqMlWZIHNJdsQHvtlh3C99LuMsS0XrMks-vJaV9U4,787
 beamlit/agents/voice/openai.py,sha256=-RDBwl16i4TbUhFo5-77Ci3zmI3Y8U2yf2MmvXR2haQ,9479
 beamlit/agents/voice/utils.py,sha256=tQidyM40Ewuy12wKqpvJLvfJgneQ0sZf50dqnerPGHg,836
 beamlit/api/__init__.py,sha256=zTSiG_ujSjAqWPyc435YXaX9XTlpMjiJWBbV-f-YtdA,45
@@ -114,8 +114,8 @@ beamlit/api/workspaces/update_workspace.py,sha256=qa5DV2UJSUYuB_ibALb4E9ghKpT1Ha
 beamlit/api/workspaces/update_workspace_user_role.py,sha256=Yn9iuJ4tKtauzBiJyU4-wYUMS9g98X2Om8zs7UkzrY8,4917
 beamlit/authentication/__init__.py,sha256=wiXqRbc7E-ulrH_ueA9duOGFvXeo7-RvhSD1XbFogMo,1020
 beamlit/authentication/apikey.py,sha256=Q3Kz4Zbc_Lr6lYwiwiLo6AhggKcdqsd6xJjDi-nERiI,1557
-beamlit/authentication/authentication.py,sha256=u1DLlF2XKwkvQCjdBSTVp23pgC1eN4U_9EaRx4BszVM,5613
-beamlit/authentication/clientcredentials.py,sha256=_TY3op9afFh2wUiIdOc7sL8ehgz5EiVKoUYL4FFLfPk,5734
+beamlit/authentication/authentication.py,sha256=OgqS6YciBnh0MomeEaL2YuZt1fHPOfQUkRhsIu6tcvk,5812
+beamlit/authentication/clientcredentials.py,sha256=a3KNT3x3c2izS1EQXbL-bO9Xm-6T16qY7cLf97eE8u8,5198
 beamlit/authentication/credentials.py,sha256=Xb2Ajj-72FdU1l_pI8OMDRSXruCJHUQjME1NxTBmkTU,8713
 beamlit/authentication/device_mode.py,sha256=gBJqJpWQXEG8gZbWXhlvQyk8FfHH95-aqURUnudil-4,6526
 beamlit/common/__init__.py,sha256=saX5X3hRCJ9erSlXuSkZ2VGgquvpgdcofAU_9sM4bCE,354
@@ -127,7 +127,7 @@ beamlit/common/settings.py,sha256=9LshW8yVXmX9ZeNhfYH-nBnQV5nKSpi1RTZDNUOxe2s,52
 beamlit/common/slugify.py,sha256=QPJqa1VrSjRle931RN37t24rUjAbyZEg1UFAp2fLgac,679
 beamlit/common/utils.py,sha256=eG201z9gMRnhoHkaZGNtfFUbCzfg_Y59JR4ciMgidW8,1465
 beamlit/deploy/__init__.py,sha256=uRsI_-gTbbki59LlvubeTfG6wfI3o2XqZODW0QXA-Ao,292
-beamlit/deploy/deploy.py,sha256=xt6MY8J3jxMEEA9NCCZXYiy9eGQ5NxdPK0u_z3wW24w,11650
+beamlit/deploy/deploy.py,sha256=AP1FjmoqE3tdKcNKtRZLojTaJ1PR73hwrk6ORRSgRmQ,11654
 beamlit/deploy/format.py,sha256=W3ESUHyFv-iZDjVnHOf9YFDDXZSXYIFFbwCoL1GInE0,1162
 beamlit/deploy/parser.py,sha256=gjRUhOVtfKnc1UNc_FhXsEfj9zrMNuq8W93pNsJBpo0,7586
 beamlit/functions/__init__.py,sha256=Mnoqpa1dm7TXwjodBbF_40JyD78aXsOYWmqjDSnA1lU,317
@@ -135,7 +135,7 @@ beamlit/functions/common.py,sha256=v4nmLP9Wotpb5e6hV30XbItgZjr5lwXcF0EOotxRngI,1
 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=wejtA256-aMYEgUjfFQibD2-D2UJIySTVIfphPr_tuo,6039
+beamlit/functions/mcp/mcp.py,sha256=voZYIbsHipSCEKXmnU_3uYKa_cfT1M2mv5t1pEwUkkE,6040
 beamlit/functions/mcp/utils.py,sha256=V7bah6cymdtjJ_LJUrNcHDeApDHA6uXvaGVeFJGKj2U,1850
 beamlit/functions/remote/remote.py,sha256=bkDUFiZI8YokqX8Fo76AnKLZF9PcjcDBr37hhxLevs8,6728
 beamlit/models/__init__.py,sha256=042wT7sG_YUmJJAb9edrui-DesMxLjOGVvnEtqc92CY,8916
@@ -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.57rc120.dist-info/METADATA,sha256=TxuRw5IiySfgjmt-GvjfTpk06r4YpjgY0d64VqMiSfY,3547
-beamlit-0.0.57rc120.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.57rc120.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
-beamlit-0.0.57rc120.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
-beamlit-0.0.57rc120.dist-info/RECORD,,
+beamlit-0.0.58.dist-info/METADATA,sha256=1JuqbE0ek155mK2KttjVURNzVtGu6efMKMX_4aj59FU,3542
+beamlit-0.0.58.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.58.dist-info/entry_points.txt,sha256=zxhgdn7SP-Otk4rEv7LMPAAa9w4TUCLbu9TJi9-K3xg,115
+beamlit-0.0.58.dist-info/licenses/LICENSE,sha256=p5PNQvpvyDT_0aYBDgmV1fFI_vAD2aSV0wWG7VTgRis,1069
+beamlit-0.0.58.dist-info/RECORD,,