beamlit 0.0.34rc60__py3-none-any.whl → 0.0.34rc61__py3-none-any.whl

beamlit/deploy/deploy.py

@@ -1,8 +1,10 @@
 import ast
 import json
 import os
+import shutil
 import sys
-import uuid
+from pathlib import Path
+import yaml
 from logging import getLogger
 from typing import Literal
 
@@ -15,26 +17,17 @@ from beamlit.models import (
     Flavor,
     Function,
     FunctionSpec,
-    Runtime,
+    MetadataLabels,
 )
-
-from .format import arg_to_dict, format_agent_chain, format_parameters
+from beamlit.api.agents import get_agent
+from beamlit.authentication import new_client
+from beamlit.client import AuthenticatedClient
+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 +38,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 +94,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 +107,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 as e:
+        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 +135,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 +195,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 +238,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 +262,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/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.34rc60.dist-info → beamlit-0.0.34rc61.dist-info}/METADATA

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

{beamlit-0.0.34rc60.dist-info → beamlit-0.0.34rc61.dist-info}/RECORD

@@ -137,8 +137,8 @@ beamlit/common/settings.py,sha256=b2rvby-ufG3M0AB1ReoWFM-1EzF1LaE-gbokO9HvQDI,38
 beamlit/common/slugify.py,sha256=nR29r37IdWS2i44ZC6ZsXRgqKPYmvMGtFQ7BuIQUTlc,90
 beamlit/common/utils.py,sha256=jouz5igBvT37Xn_e94-foCHyQczVim-UzVcoIF6RWJ4,657
 beamlit/deploy/__init__.py,sha256=GS7l7Jtm2yKs7iNLKcfjYO-rAhUzggQ3xiYSf3oxLBY,91
-beamlit/deploy/deploy.py,sha256=FyDqSYpm_YKG8IapaFhI0EN5m71d7wkWLjfBuitxsOU,9884
-beamlit/deploy/format.py,sha256=PJ8kU7Y1pwiS3tqqyvFaag9LO3jju-4ua574163VPk4,1820
+beamlit/deploy/deploy.py,sha256=Uz-AvgEKT6dXpixEPpDBzqtRbv4_JzQ1joZOKYR_s8g,10260
+beamlit/deploy/format.py,sha256=U6UZEFAYLnGJJ7O2YmSdlUUFhnWNGAv6NZ-DW4KTgvI,2049
 beamlit/deploy/parser.py,sha256=Ga0poCZkoRnuTw082QnTcNGCBJncoRAnVsn8-1FsaJE,6907
 beamlit/functions/__init__.py,sha256=_RPG1Bfg54JGdIPnViAU6n9zD7E1cDNsdXi8oYGskzE,138
 beamlit/functions/decorator.py,sha256=ZtSQsPLI70WKwi2jPhA7DaqREQUINpqt9BDVugeV_sg,1714
@@ -262,6 +262,6 @@ beamlit/serve/app.py,sha256=gYQvUK_S7g0Em-idND8HrVDqgg5LlIemheSGlX2Jj8U,3638
 beamlit/serve/middlewares/__init__.py,sha256=1dVmnOmhAQWvWktqHkKSIX-YoF6fmMU8xkUQuhg_rJU,148
 beamlit/serve/middlewares/accesslog.py,sha256=Mu4T4_9OvHybjA0ApzZFpgi2C8f3X1NbUk-76v634XM,631
 beamlit/serve/middlewares/processtime.py,sha256=lDAaIasZ4bwvN-HKHvZpaD9r-yrkVNZYx4abvbjbrCg,411
-beamlit-0.0.34rc60.dist-info/METADATA,sha256=KQjOGNRhgwRUfHsM1l1e4DQzZsI_wEIMYQq__j1LxZo,2412
-beamlit-0.0.34rc60.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beamlit-0.0.34rc60.dist-info/RECORD,,
+beamlit-0.0.34rc61.dist-info/METADATA,sha256=vsh5bRkJV61Uqh2qDKLpesr9dElEppuWS9ixSTyBrhU,2412
+beamlit-0.0.34rc61.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beamlit-0.0.34rc61.dist-info/RECORD,,