beamlit 0.0.21__py3-none-any.whl → 0.0.22__py3-none-any.whl

beamlit/agents/chat.py

@@ -37,12 +37,13 @@ def get_chat_model(agent_model: AgentDeployment):
     headers = get_authentication_headers(settings)
     headers["X-Beamlit-Environment"] = agent_model.environment
 
-    jwt = headers.pop("X-Beamlit-Authorization").replace("Bearer ", "")
+    jwt = headers.get("X-Beamlit-Authorization", "").replace("Bearer ", "")
     params = {"environment": agent_model.environment}
     chat_classes = {
         "openai": {
             "func": get_openai_chat_model,
             "kwargs": {
+                "http_async_client": client.get_async_httpx_client(),
                 "http_client": client.get_httpx_client(),
             },
         },

beamlit/agents/decorator.py

@@ -1,9 +1,11 @@
 # Import necessary modules
 import ast
+import asyncio
 import importlib
 import os
 from logging import getLogger
 
+from langchain_core.tools import Tool
 from langgraph.checkpoint.memory import MemorySaver
 from langgraph.prebuilt import create_react_agent
 
@@ -35,7 +37,7 @@ def get_functions(dir="src/functions", from_decorator="function"):
                         # Look for function definitions with decorators
                         for node in ast.walk(tree):
                             if (
-                                not isinstance(node, ast.FunctionDef)
+                                (not isinstance(node, ast.FunctionDef) and not isinstance(node, ast.AsyncFunctionDef))
                                 or len(node.decorator_list) == 0
                             ):
                                 continue
@@ -73,7 +75,10 @@ def get_functions(dir="src/functions", from_decorator="function"):
                                 # Get the decorated function
                                 if not is_kit and hasattr(module, func_name):
                                     func = getattr(module, func_name)
-                                    functions.append(func)
+                                    if asyncio.iscoroutinefunction(func):
+                                        functions.append(Tool(name=func.__name__, description=func.__doc__, func=func, coroutine=func))
+                                    else:
+                                        functions.append(Tool(name=func.__name__, description=func.__doc__, func=func))
                     except Exception as e:
                         logger.warning(f"Error processing {file_path}: {e!s}")
     return functions

beamlit/authentication/authentication.py

@@ -51,7 +51,6 @@ def new_client():
     else:
         settings = get_settings()
         credentials = load_credentials_from_settings(settings)
-
         client_config = RunClientWithCredentials(
             credentials=credentials,
             workspace=settings.workspace,

beamlit/authentication/credentials.py

@@ -43,6 +43,28 @@ class Config:
         if self.context is None:
             self.context = ContextConfig()
 
+    def to_json(self) -> dict:
+        return {
+            "workspaces": [
+                {
+                    "name": ws.name,
+                    "credentials": {
+                        "api_key": ws.credentials.api_key,
+                        "access_token": ws.credentials.access_token,
+                        "refresh_token": ws.credentials.refresh_token,
+                        "expires_in": ws.credentials.expires_in,
+                        "device_code": ws.credentials.device_code,
+                        "client_credentials": ws.credentials.client_credentials,
+                    },
+                }
+                for ws in self.workspaces
+            ],
+            "context": {
+                "workspace": self.context.workspace,
+                "environment": self.context.environment,
+            },
+        }
+
 
 def load_config() -> Config:
     config = Config()
@@ -68,23 +90,6 @@ def load_config() -> Config:
 
 
 def save_config(config: Config):
-    data = {
-        "workspaces": [
-            {
-                "name": ws.name,
-                "credentials": {
-                    "access_token": ws.credentials.access_token,
-                    "api_key": ws.credentials.api_key,
-                },
-            }
-            for ws in config.workspaces
-        ],
-        "context": {
-            "workspace": config.context.workspace,
-            "environment": config.context.environment,
-        },
-    }
-
     home_dir = Path.home()
     if not home_dir:
         raise RuntimeError("Could not determine home directory")
@@ -93,9 +98,8 @@ def save_config(config: Config):
     config_file = config_dir / "config.yaml"
 
     config_dir.mkdir(mode=0o700, parents=True, exist_ok=True)
-
     with open(config_file, "w", encoding="utf-8") as f:
-        yaml.dump(data, f)
+        yaml.dump(config.to_json(), f)
 
 
 def list_workspaces() -> List[str]:
@@ -126,7 +130,7 @@ def load_credentials(workspace_name: str) -> Credentials:
 def load_credentials_from_settings(settings: Settings) -> Credentials:
     return Credentials(
         api_key=settings.authentication.api_key,
-        client_credentials=settings.authentication.client_credentials,
+        client_credentials=settings.authentication.client.credentials,
     )
 
 

beamlit/client.py

@@ -248,7 +248,6 @@ class AuthenticatedClient:
     def get_async_httpx_client(self) -> httpx.AsyncClient:
         """Get the underlying httpx.AsyncClient, constructing a new one if not previously set"""
         if self._async_client is None:
-            self._headers[self.auth_header_name] = f"{self.prefix} {self.token}" if self.prefix else self.token
             self._async_client = httpx.AsyncClient(
                 base_url=self._base_url,
                 cookies=self._cookies,

beamlit/common/settings.py

@@ -39,10 +39,14 @@ class SettingsAgent(BaseSettings):
     module: str = Field(default="main.main")
 
 
+class SettingsAuthenticationClient(BaseSettings):
+    credentials: Union[None, str] = None
+
+
 class SettingsAuthentication(BaseSettings):
     api_key: Union[None, str] = None
     jwt: Union[None, str] = None
-    client_credentials: Union[None, str] = None
+    client: SettingsAuthenticationClient = SettingsAuthenticationClient()
 
 
 class SettingsServer(BaseSettings):
@@ -56,10 +60,11 @@ class Settings(BaseSettings):
         yaml_file="beamlit.yaml",
         env_prefix="bl_",
         env_nested_delimiter="_",
+        extra="ignore",
     )
 
     workspace: str
-    environment: str
+    environment: str = Field(default="production")
     type: str = Field(default="agent")
     name: str = Field(default="beamlit-agent")
     base_url: str = Field(default="https://api.beamlit.dev/v0")

beamlit/functions/decorator.py

@@ -3,7 +3,7 @@
 from collections.abc import Callable
 from logging import getLogger
 
-from langchain_core.tools import create_schema_from_function, tool
+from langchain_core.tools import create_schema_from_function
 
 from beamlit.authentication import new_client
 from beamlit.common.settings import get_settings
@@ -58,7 +58,7 @@ def kit(bl_kit: FunctionKit = None, **kwargs: dict) -> Callable:
     def wrapper(func: Callable) -> Callable:
         if bl_kit and not func.__doc__ and bl_kit.description:
             func.__doc__ = bl_kit.description
-        return tool(func, **kwargs)
+        return func
 
     return wrapper
 
@@ -83,8 +83,7 @@ def function(
                     func,
                     parse_docstring=func.__doc__,
                 )
-            return tool(remote_func, **kwargs)
-        return tool(func, **kwargs)
+            return remote_func
+        return func
 
     return wrapper
-    return wrapper

beamlit/serve/app.py

@@ -16,7 +16,6 @@ from .middlewares import AccessLogMiddleware, AddProcessTimeHeader
 sys.path.insert(0, os.getcwd())
 sys.path.insert(0, os.path.join(os.getcwd(), "src"))
 
-
 def import_module():
     settings = get_settings()
     main_module = importlib.import_module(".".join(settings.server.module.split(".")[0:-1]))

{beamlit-0.0.21.dist-info → beamlit-0.0.22.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: beamlit
-Version: 0.0.21
+Version: 0.0.22
 Summary: Add your description here
 Author-email: cploujoux <ch.ploujoux@gmail.com>
 Requires-Python: >=3.12

{beamlit-0.0.21.dist-info → beamlit-0.0.22.dist-info}/RECORD

@@ -1,12 +1,12 @@
 beamlit/__init__.py,sha256=545gFC-wLLwUktWcOAjUWe_Glha40tBetRTOYSfHnbI,164
-beamlit/client.py,sha256=vwvjAkUKHRySnA2tOVzXI8xtm9s1k2sEklCRE4j1Vc8,12543
+beamlit/client.py,sha256=OdRbs5VVHF32HUd5RMcnkhe8YxamnAmgGlxO6pm1Xac,12431
 beamlit/errors.py,sha256=gO8GBmKqmSNgAg-E5oT-oOyxztvp7V_6XG7OUTT15q0,546
 beamlit/py.typed,sha256=8ZJUsxZiuOy1oJeVhsTWQhTG_6pTVHVXk5hJL79ebTk,25
 beamlit/run.py,sha256=y61iDBaR0917ihj5q-cJ_r3BFW1Rn5K_kDAISw5O6aU,1339
 beamlit/types.py,sha256=E1hhDh_zXfsSQ0NCt9-uw90_Mr5iIlsdfnfvxv5HarU,1005
 beamlit/agents/__init__.py,sha256=nf1iwQwGtCG6nDqyVhxfWoqR6dv6X3bvSpCeqkTCFaM,101
-beamlit/agents/chat.py,sha256=8KsUvIB-eaUApfKclT76_4HQu3VBa9nifMqmq_AAvcM,2630
-beamlit/agents/decorator.py,sha256=naeOsvfto74TZTcc15Ro98Hl8Bbe6Uhflof_SdyUkwY,6054
+beamlit/agents/chat.py,sha256=aI7pObyywRyg3dBpubzHAUWTbTk1nwtxvpY7iIP1RLY,2704
+beamlit/agents/decorator.py,sha256=sN-JsLaPwiL-zzH5eZak76y3ObluIy8NFC3xpHFcDWc,6469
 beamlit/api/__init__.py,sha256=zTSiG_ujSjAqWPyc435YXaX9XTlpMjiJWBbV-f-YtdA,45
 beamlit/api/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 beamlit/api/agents/create_agent.py,sha256=HFExosu02JZqZz7I6U6WjN81TERz6p2i8CzQCyiRYXo,4112
@@ -130,18 +130,18 @@ 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=jnz1FMRauI5qAInqeeDER8aCONx4O8ZPZGedvi3Ap_o,659
-beamlit/authentication/authentication.py,sha256=om26AteY2cCV9ctqbOCynX6PgS8YO-aCreNOFSnnWKc,3121
+beamlit/authentication/authentication.py,sha256=tZu8GoVueKDuq1RLXMvtHcV95XLikmQ19PCxLWBn2Ek,3120
 beamlit/authentication/clientcredentials.py,sha256=6kbfTjwUkXUArJX8XZLe9ZzbEicQc19tSXBvsTpiXMk,3954
-beamlit/authentication/credentials.py,sha256=_Bjj49jGeo-JTvO2GPS1yXxh5vS1NmX0haxTfg0KiEk,4965
+beamlit/authentication/credentials.py,sha256=DlfiF_FfOosPVsRoa39JSR3XoICmhBqTlHXc6b4zWtE,5349
 beamlit/authentication/device_mode.py,sha256=oQVBCDsq-pdeXF31WSTAAEdaX6eACV7SYcOSyf3ea_Q,3728
 beamlit/common/__init__.py,sha256=yDoMJDKj-xjTGl7U1YI59KpWxiOV65HSiUulgO8xdTA,277
 beamlit/common/generate.py,sha256=VJ_MiRDulXdQdnlKdM4_Bod6CO6DOGlFiosGXOLuLGs,7227
 beamlit/common/logger.py,sha256=ayabnsoHS8ncXm8EpBS01FkvSe0XRcaNdQjKVuPI5z4,1025
 beamlit/common/secrets.py,sha256=sid81bOe3LflkMKDHwBsBs9nIju8bp5-v9qU9gkyNMc,212
-beamlit/common/settings.py,sha256=Ylfw1L4a3TdYlUXwoYLYbS7qZT1zi2Hm-uLzhSZ1jcI,5222
+beamlit/common/settings.py,sha256=6iLf_7SZ8LV72ZU88ACfac0EcgtEf79rox1jOY7RmQE,5395
 beamlit/common/utils.py,sha256=jouz5igBvT37Xn_e94-foCHyQczVim-UzVcoIF6RWJ4,657
 beamlit/functions/__init__.py,sha256=_RPG1Bfg54JGdIPnViAU6n9zD7E1cDNsdXi8oYGskzE,138
-beamlit/functions/decorator.py,sha256=-2newMBztweIgFuh0ABKOdxCfUzWaRxf0ym-YAgggJI,3168
+beamlit/functions/decorator.py,sha256=qThFujZ3RSSQ2g9kSjrVtPs9SVbgp_jB3w5XaQOvFvk,3095
 beamlit/functions/github/__init__.py,sha256=gYnUkeegukOfbymdabuuJkScvH-_ZJygX05BoqkPn0o,49
 beamlit/functions/github/github.py,sha256=FajzLCNkpXcwfgnC0l9rOGT2eSPLCz8-qrMzK9N_ZNc,598
 beamlit/functions/github/kit/__init__.py,sha256=jBwPqZv6C23_utukohxqXZwrlicNlI7PYPUj0Den7Cw,136
@@ -292,10 +292,10 @@ beamlit/models/websocket_channel.py,sha256=tyNtsVR0cOwd6BK--ehBCH8bIjxtyPhiAkrxY
 beamlit/models/workspace.py,sha256=s7wS6ibswosB0FdUb3ry3BnlLa325axBdYPLI3ipe0Q,3986
 beamlit/models/workspace_labels.py,sha256=WbnUY6eCTkUNdY7hhhSF-KQCl8fWFfkCf7hzCTiNp4A,1246
 beamlit/models/workspace_user.py,sha256=70CcifQWYbeWG7TDui4pblTzUe5sVK0AS19vNCzKE8g,3423
-beamlit/serve/app.py,sha256=-CvUU7pqFDigndHpA01IR4_mxXaT8c0VsnVjMfIw6K8,2399
+beamlit/serve/app.py,sha256=_0ZesKcczd1sYm8vs3ulbXO1M1boO_5DhFf3jSmjM4g,2398
 beamlit/serve/middlewares/__init__.py,sha256=1dVmnOmhAQWvWktqHkKSIX-YoF6fmMU8xkUQuhg_rJU,148
 beamlit/serve/middlewares/accesslog.py,sha256=wM52-hcwtO-_hdM1pnsEJzerzJf1MzEyN5m85BdDccE,609
 beamlit/serve/middlewares/processtime.py,sha256=lDAaIasZ4bwvN-HKHvZpaD9r-yrkVNZYx4abvbjbrCg,411
-beamlit-0.0.21.dist-info/METADATA,sha256=EsyU4e1T5fySulg09x5vy9XZ9BNPVfrYZX8dHY5l-b0,2023
-beamlit-0.0.21.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-beamlit-0.0.21.dist-info/RECORD,,
+beamlit-0.0.22.dist-info/METADATA,sha256=5rjMq-UV1Mj14WmjhAfGw4jmzuACjWv2ykSDg3mKJiI,2023
+beamlit-0.0.22.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+beamlit-0.0.22.dist-info/RECORD,,