beamlit 0.0.31rc37__tar.gz → 0.0.31rc39__tar.gz

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/PKG-INFO

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

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/pyproject.toml

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

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/agents/chain.py

@@ -1,17 +1,17 @@
 import asyncio
-from dataclasses import dataclass
 import warnings
-from typing import Any, Callable
+from dataclasses import dataclass
+from typing import Callable
 
 import pydantic
 import typing_extensions as t
+from langchain_core.tools.base import BaseTool, ToolException
+
 from beamlit.api.agents import list_agents
 from beamlit.authentication.authentication import AuthenticatedClient
+from beamlit.common.settings import get_settings
 from beamlit.models import Agent, AgentChain
 from beamlit.run import RunClient
-from beamlit.common.settings import get_settings
-from langchain_core.tools.base import BaseTool, ToolException
-
 
 
 class ChainTool(BaseTool):

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/agents/decorator.py

@@ -6,6 +6,10 @@ 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
+
 from beamlit.api.models import get_model
 from beamlit.authentication import new_client
 from beamlit.common.settings import init
@@ -13,9 +17,6 @@ from beamlit.errors import UnexpectedStatus
 from beamlit.functions.mcp.mcp import MCPClient, MCPToolkit
 from beamlit.functions.remote.remote import RemoteToolkit
 from beamlit.models import Agent, AgentMetadata, AgentSpec
-from langchain_core.tools import Tool
-from langgraph.checkpoint.memory import MemorySaver
-from langgraph.prebuilt import create_react_agent
 
 from .chain import ChainToolkit
 from .chat import get_chat_model

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/api/service_accounts/create_workspace_service_account.py

@@ -6,7 +6,9 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.create_workspace_service_account_body import CreateWorkspaceServiceAccountBody
-from ...models.create_workspace_service_account_response_200 import CreateWorkspaceServiceAccountResponse200
+from ...models.create_workspace_service_account_response_200 import (
+    CreateWorkspaceServiceAccountResponse200,
+)
 from ...types import Response
 
 

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/api/service_accounts/delete_workspace_service_account.py

@@ -5,7 +5,9 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.delete_workspace_service_account_response_200 import DeleteWorkspaceServiceAccountResponse200
+from ...models.delete_workspace_service_account_response_200 import (
+    DeleteWorkspaceServiceAccountResponse200,
+)
 from ...types import Response
 
 

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/api/service_accounts/get_workspace_service_accounts.py

@@ -5,7 +5,9 @@ import httpx
 
 from ... import errors
 from ...client import AuthenticatedClient, Client
-from ...models.get_workspace_service_accounts_response_200_item import GetWorkspaceServiceAccountsResponse200Item
+from ...models.get_workspace_service_accounts_response_200_item import (
+    GetWorkspaceServiceAccountsResponse200Item,
+)
 from ...types import Response
 
 

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/api/service_accounts/update_workspace_service_account.py

@@ -6,7 +6,9 @@ import httpx
 from ... import errors
 from ...client import AuthenticatedClient, Client
 from ...models.update_workspace_service_account_body import UpdateWorkspaceServiceAccountBody
-from ...models.update_workspace_service_account_response_200 import UpdateWorkspaceServiceAccountResponse200
+from ...models.update_workspace_service_account_response_200 import (
+    UpdateWorkspaceServiceAccountResponse200,
+)
 from ...types import Response
 
 

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/authentication/authentication.py

@@ -1,15 +1,19 @@
-import os
 from dataclasses import dataclass
 from typing import Dict, Generator
 
-from beamlit.common.settings import Settings, get_settings
 from httpx import Auth, Request, Response
 
+from beamlit.common.settings import Settings, get_settings
+
 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
 
 

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/common/logger.py

@@ -27,5 +27,5 @@ def init(log_level: str):
     logging.getLogger("httpx").propagate = False
 
     handler = logging.StreamHandler()
-    handler.setFormatter(ColoredFormatter(f"%(levelname)s %(name)s - %(message)s"))
+    handler.setFormatter(ColoredFormatter("%(levelname)s %(name)s - %(message)s"))
     logging.basicConfig(level=log_level, handlers=[handler])

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/common/settings.py

@@ -2,6 +2,16 @@ import os
 from logging import getLogger
 from typing import Tuple, Type, Union
 
+from langchain_core.language_models.chat_models import BaseChatModel
+from langgraph.graph.graph import CompiledGraph
+from pydantic import Field
+from pydantic_settings import (
+    BaseSettings,
+    PydanticBaseSettingsSource,
+    SettingsConfigDict,
+    YamlConfigSettingsSource,
+)
+
 from beamlit.api.agents import get_agent
 from beamlit.api.functions import get_function
 from beamlit.api.models import get_model
@@ -9,11 +19,6 @@ from beamlit.client import AuthenticatedClient
 from beamlit.common.logger import init as init_logger
 from beamlit.models import Agent, Function, Model
 from beamlit.types import UNSET, Unset
-from langchain_core.language_models.chat_models import BaseChatModel
-from langgraph.graph.graph import CompiledGraph
-from pydantic import Field
-from pydantic_settings import (BaseSettings, PydanticBaseSettingsSource,
-                               SettingsConfigDict, YamlConfigSettingsSource)
 
 global SETTINGS
 SETTINGS = None
@@ -99,7 +104,6 @@ def init_agent(
     client: AuthenticatedClient,
     destination: str = f"{os.getcwd()}/src/beamlit_generated.py",
 ):
-    from beamlit.api.agents import get_agent_deployment
     from beamlit.common.generate import generate
 
     logger = getLogger(__name__)

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/deploy/deploy.py

@@ -2,12 +2,20 @@ import ast
 import json
 import os
 import sys
+import uuid
 from logging import getLogger
 from typing import Literal
 
 from beamlit.common.settings import Settings, get_settings, init
-from beamlit.models import (Agent, AgentChain, AgentSpec, EnvironmentMetadata,
-                            Flavor, Function, FunctionSpec, Runtime)
+from beamlit.models import (
+    Agent,
+    AgentSpec,
+    EnvironmentMetadata,
+    Flavor,
+    Function,
+    FunctionSpec,
+    Runtime,
+)
 
 from .format import arg_to_dict, format_agent_chain, format_parameters
 from .parser import Resource, get_description, get_parameters, get_resources
@@ -23,6 +31,8 @@ def get_runtime_image(type: str, name: str) -> str:
     settings = get_settings()
     registry_url = settings.registry_url.replace("https://", "").replace("http://", "")
     image = f"{registry_url}/{settings.workspace}/{type}s/{name}"
+    # Generate a random ID to ensure unique image tags
+    image = f"{image}:{str(uuid.uuid4())[:8]}"
     return image
 
 
@@ -247,29 +257,29 @@ def generate_beamlit_deployment(directory: str):
         # write deployment file
         agent_dir = os.path.join(agents_dir, agent.metadata.name)
         os.makedirs(agent_dir, exist_ok=True)
-        with open(os.path.join(agent_dir, f"agent.yaml"), "w") as f:
+        with open(os.path.join(agent_dir, "agent.yaml"), "w") as f:
             content = get_agent_yaml(agent, functions, settings)
             f.write(content)
         # write dockerfile for build
-        with open(os.path.join(agent_dir, f"Dockerfile"), "w") as f:
+        with open(os.path.join(agent_dir, "Dockerfile"), "w") as f:
             content = dockerfile("agent", resource, agent)
             f.write(content)
         # write destination docker
-        with open(os.path.join(agent_dir, f"destination.txt"), "w") as f:
+        with open(os.path.join(agent_dir, "destination.txt"), "w") as f:
             content = agent.spec.runtime.image
             f.write(content)
     for resource, function in functions:
         # write deployment file
         function_dir = os.path.join(functions_dir, function.metadata.name)
         os.makedirs(function_dir, exist_ok=True)
-        with open(os.path.join(function_dir, f"function.yaml"), "w") as f:
+        with open(os.path.join(function_dir, "function.yaml"), "w") as f:
             content = get_function_yaml(function, settings)
             f.write(content)
         # write dockerfile for build
-        with open(os.path.join(function_dir, f"Dockerfile"), "w") as f:
+        with open(os.path.join(function_dir, "Dockerfile"), "w") as f:
             content = dockerfile("function", resource, function)
             f.write(content)
         # write destination docker
-        with open(os.path.join(function_dir, f"destination.txt"), "w") as f:
+        with open(os.path.join(function_dir, "destination.txt"), "w") as f:
             content = function.spec.runtime.image
             f.write(content)

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/deploy/format.py

@@ -1,7 +1,9 @@
 
 import ast
+
 from beamlit.models import AgentChain, StoreFunctionParameter
 
+
 def arg_to_list(arg: ast.List):
     value = []
     for v in arg.elts:

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/functions/decorator.py

@@ -4,11 +4,12 @@ import json
 from collections.abc import Callable
 from logging import getLogger
 
+from langchain_core.tools import create_schema_from_function
+
 from beamlit.authentication import new_client
 from beamlit.common.settings import get_settings
 from beamlit.models import Function, FunctionKit
 from beamlit.run import RunClient
-from langchain_core.tools import create_schema_from_function
 
 logger = getLogger(__name__)
 

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/functions/mcp/mcp.py

@@ -7,11 +7,12 @@ import pydantic
 import pydantic_core
 import requests
 import typing_extensions as t
-from beamlit.authentication.authentication import AuthenticatedClient
-from beamlit.common.settings import get_settings
 from langchain_core.tools.base import BaseTool, BaseToolkit, ToolException
 from mcp import ListToolsResult
 
+from beamlit.authentication.authentication import AuthenticatedClient
+from beamlit.common.settings import get_settings
+
 settings = get_settings()
 
 def create_dynamic_schema(name: str, schema: dict[str, t.Any]) -> type[pydantic.BaseModel]:

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/functions/remote/remote.py

@@ -1,16 +1,17 @@
 import asyncio
 import warnings
 from dataclasses import dataclass
-from typing import Callable, Literal
+from typing import Callable
 
 import pydantic
 import typing_extensions as t
+from langchain_core.tools.base import BaseTool, ToolException
+
 from beamlit.api.functions import get_function
 from beamlit.authentication.authentication import AuthenticatedClient
 from beamlit.common.settings import get_settings
 from beamlit.models import Function, StoreFunctionParameter
 from beamlit.run import RunClient
-from langchain_core.tools.base import BaseTool, ToolException
 
 
 def create_dynamic_schema(name: str, parameters: list[StoreFunctionParameter]) -> type[pydantic.BaseModel]:

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/models/__init__.py

@@ -32,7 +32,9 @@ from .function_spec import FunctionSpec
 from .get_trace_ids_response_200 import GetTraceIdsResponse200
 from .get_trace_logs_response_200 import GetTraceLogsResponse200
 from .get_trace_response_200 import GetTraceResponse200
-from .get_workspace_service_accounts_response_200_item import GetWorkspaceServiceAccountsResponse200Item
+from .get_workspace_service_accounts_response_200_item import (
+    GetWorkspaceServiceAccountsResponse200Item,
+)
 from .increase_and_rate_metric import IncreaseAndRateMetric
 from .integration_config import IntegrationConfig
 from .integration_connection import IntegrationConnection
@@ -75,8 +77,12 @@ from .resource_deployment_metrics_query_per_second_per_region_per_code import (
     ResourceDeploymentMetricsQueryPerSecondPerRegionPerCode,
 )
 from .resource_environment_metrics import ResourceEnvironmentMetrics
-from .resource_environment_metrics_inference_per_region import ResourceEnvironmentMetricsInferencePerRegion
-from .resource_environment_metrics_query_per_region_per_code import ResourceEnvironmentMetricsQueryPerRegionPerCode
+from .resource_environment_metrics_inference_per_region import (
+    ResourceEnvironmentMetricsInferencePerRegion,
+)
+from .resource_environment_metrics_query_per_region_per_code import (
+    ResourceEnvironmentMetricsQueryPerRegionPerCode,
+)
 from .resource_log import ResourceLog
 from .resource_metrics import ResourceMetrics
 from .runtime import Runtime

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/models/resource_environment_metrics.py

@@ -8,7 +8,9 @@ from ..types import UNSET, Unset
 if TYPE_CHECKING:
     from ..models.metric import Metric
     from ..models.qps import QPS
-    from ..models.resource_environment_metrics_inference_per_region import ResourceEnvironmentMetricsInferencePerRegion
+    from ..models.resource_environment_metrics_inference_per_region import (
+        ResourceEnvironmentMetricsInferencePerRegion,
+    )
     from ..models.resource_environment_metrics_query_per_region_per_code import (
         ResourceEnvironmentMetricsQueryPerRegionPerCode,
     )

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/run.py

@@ -2,6 +2,7 @@ import urllib.parse
 from typing import Any
 
 import requests
+
 from beamlit.client import AuthenticatedClient
 from beamlit.common import HTTPError, get_settings
 

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/beamlit/serve/app.py

@@ -7,14 +7,18 @@ from logging import getLogger
 from uuid import uuid4
 
 from asgi_correlation_id import CorrelationIdMiddleware
-from beamlit.common import HTTPError, get_settings, init
-from beamlit.common.instrumentation import (get_metrics_exporter,
-                                            get_resource_attributes,
-                                            get_span_exporter, instrument_app)
 from fastapi import FastAPI, Request, Response
 from fastapi.responses import JSONResponse
 from traceloop.sdk import Traceloop
 
+from beamlit.common import HTTPError, get_settings, init
+from beamlit.common.instrumentation import (
+    get_metrics_exporter,
+    get_resource_attributes,
+    get_span_exporter,
+    instrument_app,
+)
+
 from .middlewares import AccessLogMiddleware, AddProcessTimeHeader
 
 sys.path.insert(0, os.getcwd())
@@ -55,6 +59,7 @@ app.add_middleware(AddProcessTimeHeader)
 app.add_middleware(AccessLogMiddleware)
 instrument_app(app)
 
+
 @app.get("/health")
 async def health():
     return {"status": "ok"}
@@ -67,7 +72,7 @@ async def root(request: Request):
     try:
         body = await request.json()
 
-        original_func = getattr(func, '__wrapped__', func)
+        original_func = getattr(func, "__wrapped__", func)
         if asyncio.iscoroutinefunction(func) or asyncio.iscoroutinefunction(original_func):
             response = await func(body)
         else:

{beamlit-0.0.31rc37 → beamlit-0.0.31rc39}/src/integrationtest/.beamlit/agents/agent-custom/agent.yaml

@@ -9,10 +9,10 @@ metadata:
 spec:
   enabled: true
   policies: []
-  functions: [helloworld_2]
+  functions: [helloworld]
   agentChain: []
   model: gpt-4o-mini
   runtime:
-    image: us.registry.beamlit.com/main/agents/agent-custom
+    image: us.registry.beamlit.com/main/agents/agent-custom:9280b44d
     description: |
       A chat agent using Beamlit to handle your tasks.

beamlit-0.0.31rc39/src/integrationtest/.beamlit/agents/agent-custom/destination.txt

@@ -0,0 +1 @@
+us.registry.beamlit.com/main/agents/agent-custom:9280b44d

beamlit-0.0.31rc39/src/integrationtest/.beamlit/functions/helloworld/Dockerfile

@@ -0,0 +1,23 @@
+
+FROM python:3.12-slim
+
+ARG UV_VERSION="latest"
+RUN apt update && apt install -y curl
+
+# Install uv.
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
+RUN curl -fsSL https://raw.githubusercontent.com/beamlit/toolkit/main/install.sh | BINDIR=/bin sh
+WORKDIR /beamlit
+
+# Install the application dependencies.
+COPY pyproject.toml /beamlit/pyproject.toml
+COPY uv.lock /beamlit/uv.lock
+RUN uv sync --no-cache
+
+COPY README.m[d] /beamlit/README.md
+COPY LICENS[E] /beamlit/LICENSE
+COPY . /beamlit/src
+
+ENV PATH="/beamlit/.venv/bin:$PATH"
+
+ENTRYPOINT ["bl","serve","--port","80","--module","functions.helloworld.helloworld"]

beamlit-0.0.31rc39/src/integrationtest/.beamlit/functions/helloworld/destination.txt

@@ -0,0 +1 @@
+us.registry.beamlit.com/main/functions/helloworld:109b61a4

beamlit-0.0.31rc39/src/integrationtest/.beamlit/functions/helloworld/function.yaml

@@ -0,0 +1,19 @@
+
+apiVersion: beamlit.com/v1alpha1
+kind: Function
+metadata:
+  name: helloworld
+  displayName: helloworld
+  environment: production
+spec:
+  enabled: true
+  policies: []
+  description: |
+    A function for saying hello to the world.
+  parameters: 
+    - name: query
+      type: string
+      required: true
+      description: Parameter query
+  runtime:
+    image: us.registry.beamlit.com/main/functions/helloworld:109b61a4

beamlit-0.0.31rc37/src/integrationtest/.beamlit/agents/agent-custom/destination.txt

@@ -1 +0,0 @@
-us.registry.beamlit.com/main/agents/agent-custom