beamlit 0.0.34rc60__tar.gz → 0.0.34rc62__tar.gz

{beamlit-0.0.34rc60 → beamlit-0.0.34rc62}/PKG-INFO

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

{beamlit-0.0.34rc60 → beamlit-0.0.34rc62}/pyproject.toml

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

{beamlit-0.0.34rc60 → beamlit-0.0.34rc62}/src/beamlit/agents/chat.py

@@ -1,4 +1,7 @@
 from logging import getLogger
+from typing import Tuple
+
+from langchain_core.language_models import BaseChatModel
 
 from beamlit.authentication import get_authentication_headers, new_client
 from beamlit.common.settings import get_settings
@@ -39,7 +42,7 @@ def get_cohere_chat_model(**kwargs):
 
     return ChatCohere(**kwargs)
 
-def get_chat_model(name: str, agent_model: Model):
+def get_chat_model(name: str, agent_model: Model) -> Tuple[BaseChatModel, str, str]:
     settings = get_settings()
     client = new_client()
 

beamlit-0.0.34rc62/src/beamlit/agents/decorator.py

@@ -0,0 +1,108 @@
+# Import necessary modules
+import functools
+from logging import getLogger
+
+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
+from beamlit.errors import UnexpectedStatus
+from beamlit.functions import get_functions
+from beamlit.models import Agent, AgentMetadata, AgentSpec
+
+from .chat import get_chat_model
+
+
+def agent(
+    agent: Agent | dict = None,
+    override_model=None,
+    override_agent=None,
+    mcp_hub=None,
+    remote_functions=None,
+):
+    logger = getLogger(__name__)
+    try:
+        if agent is not None and not isinstance(agent, dict):
+            raise Exception(
+                'agent must be a dictionary, example: @agent(agent={"metadata": {"name": "my_agent"}})'
+            )
+
+        client = new_client()
+        chat_model = override_model or None
+        settings = init()
+
+        def wrapper(func):
+            @functools.wraps(func)
+            def wrapped(*args, **kwargs):
+                return func(
+                    settings.agent.agent,
+                    settings.agent.chat_model,
+                    settings.agent.functions,
+                    *args,
+                    **kwargs,
+                )
+
+            return wrapped
+
+        if agent is not None:
+            metadata = AgentMetadata(**agent.get("metadata", {}))
+            spec = AgentSpec(**agent.get("spec", {}))
+            agent = Agent(metadata=metadata, spec=spec)
+            if agent.spec.model and chat_model is None:
+                try:
+                    response = get_model.sync_detailed(
+                        agent.spec.model, environment=settings.environment, client=client
+                    )
+                    settings.agent.model = response.parsed
+                except UnexpectedStatus as e:
+                    if e.status_code == 404 and settings.environment != "production":
+                        try:
+                            response = get_model.sync_detailed(
+                                agent.spec.model, environment="production", client=client
+                            )
+                            settings.agent.model = response.parsed
+                        except UnexpectedStatus as e:
+                            if e.status_code == 404:
+                                raise ValueError(f"Model {agent.spec.model} not found")
+                    else:
+                        raise e
+                except Exception as e:
+                    raise e
+
+                if settings.agent.model:
+                    chat_model, provider, model = get_chat_model(agent.spec.model, settings.agent.model)
+                    settings.agent.chat_model = chat_model
+                    logger.info(f"Chat model configured, using: {provider}:{model}")
+
+        functions = get_functions(
+            client=client,
+            dir=settings.agent.functions_directory,
+            mcp_hub=mcp_hub,
+            remote_functions=remote_functions,
+            chain=agent.spec.agent_chain,
+            remote_functions_empty=not remote_functions,
+        )
+        settings.agent.functions = functions
+        
+        if override_agent is None and len(functions) == 0:
+            raise ValueError(
+                "You must define at least one function, you can define this function in directory "
+                f'"{settings.agent.functions_directory}". Here is a sample function you can use:\n\n'
+                "from beamlit.functions import function\n\n"
+                "@function()\n"
+                "def hello_world(query: str):\n"
+                "    return 'Hello, world!'\n"
+            )
+
+        if override_agent is None and chat_model is not None:
+            memory = MemorySaver()
+            _agent = create_react_agent(chat_model, functions, checkpointer=memory)
+            settings.agent.agent = _agent
+        else:
+            settings.agent.agent = override_agent
+        return wrapper
+    except Exception as e:
+        logger.error(f"Error in agent decorator: {e!s} at line {e.__traceback__.tb_lineno}")
+        raise e

{beamlit-0.0.34rc60 → beamlit-0.0.34rc62}/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.34rc60 → beamlit-0.0.34rc62}/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.34rc60 → beamlit-0.0.34rc62}/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.34rc60 → beamlit-0.0.34rc62}/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.34rc60 → beamlit-0.0.34rc62}/src/beamlit/deploy/deploy.py

@@ -1,11 +1,17 @@
 import ast
 import json
 import os
+import shutil
 import sys
-import uuid
 from logging import getLogger
+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 (
@@ -15,26 +21,15 @@ from beamlit.models import (
     Flavor,
     Function,
     FunctionSpec,
-    Runtime,
+    MetadataLabels,
 )
 
-from .format import arg_to_dict, format_agent_chain, format_parameters
+from .format import arg_to_dict
 from .parser import Resource, get_description, get_parameters, get_resources
 
 sys.path.insert(0, os.getcwd())
 sys.path.insert(0, os.path.join(os.getcwd(), "src"))
 
-random_id = str(uuid.uuid4())[: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}:{random_id}"
-    return image
-
-
 def set_default_values(resource: Resource, deployment: Agent | Function):
     settings = get_settings()
     deployment.metadata.workspace = settings.workspace
@@ -45,10 +40,6 @@ def set_default_values(resource: Resource, deployment: Agent | Function):
         deployment.metadata.display_name = deployment.metadata.name
     if not deployment.spec.description:
         deployment.spec.description = get_description(None, resource)
-    if not deployment.spec.runtime:
-        deployment.spec.runtime = Runtime()
-    if not deployment.spec.runtime.image:
-        deployment.spec.runtime.image = get_runtime_image(resource.type, deployment.metadata.name)
     return deployment
 
 def get_beamlit_deployment_from_resource(
@@ -105,7 +96,7 @@ def get_flavors(flavors: list[Flavor]) -> str:
     return json.dumps([flavor.to_dict() for flavor in flavors])
 
 def get_agent_yaml(
-    agent: Agent, functions: list[tuple[Resource, Function]], settings: Settings
+    agent: Agent, functions: list[tuple[Resource, Function]], settings: Settings, client: AuthenticatedClient
 ) -> str:
     """
     Generates YAML configuration for an agent deployment.
@@ -118,30 +109,24 @@ def get_agent_yaml(
     Returns:
         str: YAML configuration string
     """
+    try:
+        agent_response = get_agent.sync(agent.metadata.name, client=client)
+        agent.spec.repository = agent_response.spec.repository
+    except Exception:
+        pass
+    agent.spec.functions = [slugify(function.metadata.name) for (_, function) in functions]
+    agent.metadata.labels = agent.metadata.labels and MetadataLabels.from_dict(agent.metadata.labels) or MetadataLabels()
+    agent.metadata.labels["x-beamlit-auto-generated"] = "true"
+    agent_yaml = yaml.dump(agent.to_dict())
     template = f"""
 apiVersion: beamlit.com/v1alpha1
 kind: Agent
-metadata:
-  name: {slugify(agent.metadata.name)}
-  displayName: {agent.metadata.display_name or agent.metadata.name}
-  environment: {settings.environment}
-  workspace: {settings.workspace}
-  labels:
-    x-beamlit-auto-generated: "true"
-spec:
-  enabled: true
-  policies: [{", ".join(agent.spec.policies or [])}]
-  functions: [{", ".join([f"{slugify(function.metadata.name)}" for (_, function) in functions])}]
-  agentChain: {format_agent_chain(agent.spec.agent_chain)}
-  model: {agent.spec.model}
+{agent_yaml}
 """
-    if agent.spec.description:
-        template += f"""    description: |
-      {agent.spec.description}"""
     return template
 
 
-def get_function_yaml(function: Function, settings: Settings) -> str:
+def get_function_yaml(function: Function, settings: Settings, client: AuthenticatedClient) -> str:
     """
     Generates YAML configuration for a function deployment.
 
@@ -152,21 +137,13 @@ def get_function_yaml(function: Function, settings: Settings) -> str:
     Returns:
         str: YAML configuration string
     """
+    function.metadata.labels = function.metadata.labels and MetadataLabels.from_dict(function.metadata.labels) or MetadataLabels()
+    function.metadata.labels["x-beamlit-auto-generated"] = "true"
+    function_yaml = yaml.dump(function.to_dict())
     return f"""
 apiVersion: beamlit.com/v1alpha1
 kind: Function
-metadata:
-  name: {slugify(function.metadata.name)}
-  displayName: {function.metadata.display_name or function.metadata.name}
-  environment: {settings.environment}
-  labels:
-    x-beamlit-auto-generated: "true"  
-spec:
-  enabled: true
-  policies: [{", ".join(function.spec.policies or [])}]
-  description: |
-    {function.spec.description}
-  parameters: {format_parameters(function.spec.parameters)}
+{function_yaml}
 """
 
 
@@ -220,6 +197,35 @@ ENV PATH="/beamlit/.venv/bin:$PATH"
 ENTRYPOINT [{cmd_str}]
 """
 
+def clean_auto_generated(
+    directory: str,
+    type: Literal["agent", "function"],
+    deployments: list[tuple[Resource, Agent | Function]]
+):
+    """
+    Cleans up auto-generated deployments of a specific type.
+
+    Args:
+        directory (str): Base directory containing deployments
+        type (str): Type of deployment ("agent" or "function")
+        deployments (list[tuple[Resource, Agent | Function]]): List of deployment resources and configurations
+    """
+    
+    deploy_dir = Path(directory) / f"{type}s"
+    deploy_names = [d.metadata.name for (_, d) in deployments]
+
+    if deploy_dir.exists():
+        for item_dir in deploy_dir.iterdir():
+            if item_dir.is_dir() and item_dir.name not in deploy_names:
+                yaml_file = item_dir / f"{type}.yaml"
+                if yaml_file.exists():
+                    with open(yaml_file) as f:
+                        try:
+                            content = yaml.safe_load(f)
+                            if content.get("metadata", {}).get("labels", {}).get("x-beamlit-auto-generated") == "true":
+                                shutil.rmtree(item_dir)
+                        except yaml.YAMLError:
+                            continue
 
 def generate_beamlit_deployment(directory: str):
     """
@@ -234,6 +240,7 @@ def generate_beamlit_deployment(directory: str):
         - Directory structure for agents and functions
     """
     settings = init()
+    client = new_client()
     logger = getLogger(__name__)
     logger.info(f"Importing server module: {settings.server.module}")
     functions: list[tuple[Resource, Function]] = []
@@ -257,28 +264,23 @@ def generate_beamlit_deployment(directory: str):
         agent_dir = os.path.join(agents_dir, agent.metadata.name)
         os.makedirs(agent_dir, exist_ok=True)
         with open(os.path.join(agent_dir, "agent.yaml"), "w") as f:
-            content = get_agent_yaml(agent, functions, settings)
+            content = get_agent_yaml(agent, functions, settings, client)
             f.write(content)
         # write dockerfile for build
         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, "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, "function.yaml"), "w") as f:
-            content = get_function_yaml(function, settings)
+            content = get_function_yaml(function, settings, client)
             f.write(content)
         # write dockerfile for build
         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, "destination.txt"), "w") as f:
-            content = function.spec.runtime.image
-            f.write(content)
+
+    clean_auto_generated(directory, "agent", agents)
+    clean_auto_generated(directory, "function", functions)

{beamlit-0.0.34rc60 → beamlit-0.0.34rc62}/src/beamlit/deploy/format.py

@@ -48,6 +48,16 @@ def format_parameters(parameters: list[StoreFunctionParameter]) -> str:
 
     return "\n".join(formatted)
 
+def format_dict(obj: dict) -> str:
+    if not obj:
+        return "null"
+    ret = ""
+    for k, v in obj.items():
+        if not v:
+            ret += f"{k}: null\n"
+        else:
+            ret += f"{k}: {v}\n"
+    return ret
 
 def format_agent_chain(agentChain: list[AgentChain]) -> str:
     """

beamlit-0.0.34rc62/src/beamlit/functions/__init__.py

@@ -0,0 +1,5 @@
+"""Functions package providing function decorators and utilities."""
+
+from .decorator import function, get_functions, kit
+
+__all__ = ["function", "kit", "get_functions"]