naas-abi-core 1.0.7__py3-none-any.whl → 1.1.0__py3-none-any.whl

naas_abi_core/cli/__init__.py

@@ -7,7 +7,7 @@ from .init import init
 from .module import module
 from .new import new
 from .secret import secrets
-
+from .deploy import deploy
 # from dotenv import load_dotenv
 
 # load_dotenv()
@@ -48,6 +48,7 @@ main.add_command(agent)
 main.add_command(chat)
 main.add_command(new)
 main.add_command(init)
+main.add_command(deploy)
 
 # if __name__ == "__main__":
 main()

naas_abi_core/cli/deploy.py

@@ -0,0 +1,199 @@
+import json
+import subprocess
+from uuid import uuid4
+
+import click
+import requests
+from pydantic import BaseModel
+from rich.console import Console
+from rich.markdown import Markdown
+
+from naas_abi_core import logger
+from naas_abi_core.engine.engine_configuration.EngineConfiguration import (
+    EngineConfiguration,
+)
+
+
+@click.group("deploy")
+def deploy():
+    pass
+
+
+class Container(BaseModel):
+    name: str
+    image: str
+    port: int
+    cpu: str
+    memory: str
+    env: dict
+
+
+class Space(BaseModel):
+    name: str
+    containers: list[Container]
+
+
+class NaasAPIClient:
+    naas_api_key: str
+    base_url: str
+
+    def __init__(self, naas_api_key: str):
+        self.naas_api_key = naas_api_key
+        self.base_url = "https://api.naas.ai"
+
+    def create_registry(self, name: str):
+        response = requests.post(
+            f"{self.base_url}/registry/",
+            headers={"Authorization": f"Bearer {self.naas_api_key}"},
+            json={"name": name},
+        )
+        if response.status_code == 409:
+            return self.get_registry(name)
+        response.raise_for_status()
+        return response.json()
+
+    def get_registry(self, name: str):
+        response = requests.get(
+            f"{self.base_url}/registry/{name}",
+            headers={"Authorization": f"Bearer {self.naas_api_key}"},
+        )
+        response.raise_for_status()
+        return response.json()
+
+    def get_registry_credentials(self, name: str):
+        response = requests.get(
+            f"{self.base_url}/registry/{name}/credentials",
+            headers={"Authorization": f"Bearer {self.naas_api_key}"},
+        )
+        response.raise_for_status()
+        return response.json()
+
+    def update_space(self, space: Space) -> dict:
+        response = requests.put(
+            f"{self.base_url}/space/{space.name}",
+            headers={"Authorization": f"Bearer {self.naas_api_key}"},
+            json=space.model_dump(),
+        )
+        response.raise_for_status()
+        return response.json()
+
+    def create_space(self, space: Space) -> dict:
+        response = requests.post(
+            f"{self.base_url}/space/",
+            headers={"Authorization": f"Bearer {self.naas_api_key}"},
+            json=space.model_dump(),
+        )
+
+        if response.status_code == 409:
+            return self.update_space(space)
+
+        response.raise_for_status()
+        return response.json()
+
+    def get_space(self, name: str) -> dict:
+        response = requests.get(
+            f"{self.base_url}/space/{name}",
+            headers={"Authorization": f"Bearer {self.naas_api_key}"},
+        )
+        response.raise_for_status()
+        return response.json()
+
+
+class NaasDeployer:
+    image_name: str
+
+    naas_api_client: NaasAPIClient
+
+    def __init__(self, configuration: EngineConfiguration):
+        self.configuration = configuration
+        self.image_name = str(uuid4())
+        assert configuration.deploy is not None
+        self.naas_api_client = NaasAPIClient(configuration.deploy.naas_api_key)
+
+    def docker_build(self, image_name: str):
+        subprocess.run(
+            f"docker build -t {image_name} . --platform linux/amd64", shell=True
+        )
+
+    def env_list_to_dict(self, env: list[str]) -> dict:
+        return {env_var.split("=", 1)[0]: env_var.split("=", 1)[1] for env_var in env}
+
+    def deploy(self, env: list[str]):
+        registry = self.naas_api_client.create_registry(
+            self.configuration.deploy.space_name
+        )
+
+        uid = str(uuid4())
+
+        image_name = f"{registry['registry']['uri']}:{uid}"
+        self.docker_build(image_name)
+        credentials = self.naas_api_client.get_registry_credentials(
+            self.configuration.deploy.space_name
+        )
+        docker_login_command = f"docker login -u {credentials['credentials']['username']} -p {credentials['credentials']['password']} {registry['registry']['uri']}"
+        subprocess.run(docker_login_command, shell=True)
+        subprocess.run(f"docker push {image_name}", shell=True)
+
+        image_sha = (
+            subprocess.run(
+                "docker inspect --format='{{index .RepoDigests 0}}' "
+                + image_name
+                + " | cut -d'@' -f2",
+                shell=True,
+                capture_output=True,
+            )
+            .stdout.strip()
+            .decode("utf-8")
+        )
+
+        image_name_with_sha = f"{image_name.replace(':' + uid, '')}@{image_sha}"
+
+
+        self.naas_api_client.create_space(
+            Space(
+                name=self.configuration.deploy.space_name,
+                containers=[
+                    Container(
+                        name="api",
+                        image=image_name_with_sha,
+                        port=9879,
+                        cpu="1",
+                        memory="1Gi",
+                        env=self.env_list_to_dict(env) | {
+                            "NAAS_API_KEY": self.configuration.deploy.naas_api_key,
+                            "ENV": "prod",
+                        },
+                    )
+                ],
+            )
+        )
+
+        space = self.naas_api_client.get_space(self.configuration.deploy.space_name)
+
+        Console().print(
+            Markdown(f"""
+# Deployment successful
+
+- Space: {self.configuration.deploy.space_name}
+- Image: {image_name_with_sha}
+- URL: https://{self.configuration.deploy.space_name}.default.space.naas.ai
+
+```json
+{json.dumps(space, indent=4)}
+```
+                                 """)
+        )
+
+
+@deploy.command("naas")
+@click.option("-e", "--env", multiple=True, help="Environment variables to set (e.g. -e FOO=BAR -e BAZ=QUX)")
+def naas(env: list[str]):
+    configuration: EngineConfiguration = EngineConfiguration.load_configuration()
+
+    if configuration.deploy is None:
+        logger.error(
+            "Deploy configuration not found in the yaml configuration file. Please add a deploy section to the configuration file."
+        )
+
+    deployer = NaasDeployer(configuration)
+    deployer.deploy(env)

naas_abi_core/engine/engine_configuration/EngineConfiguration.py

@@ -20,7 +20,7 @@ from naas_abi_core.engine.engine_configuration.EngineConfiguration_VectorStoreSe
 from naas_abi_core.services.secret.Secret import Secret
 from pydantic import BaseModel, model_validator
 from typing_extensions import Literal
-
+from naas_abi_core.engine.engine_configuration.EngineConfiguration_Deploy import DeployConfiguration
 
 class ServicesConfiguration(BaseModel):
     object_storage: ObjectStorageServiceConfiguration
@@ -80,6 +80,8 @@ class GlobalConfig(BaseModel):
 
 class EngineConfiguration(BaseModel):
     api: ApiConfiguration
+    
+    deploy: DeployConfiguration | None = None
 
     services: ServicesConfiguration
 

naas_abi_core/engine/engine_configuration/EngineConfiguration_Deploy.py

@@ -0,0 +1,6 @@
+from pydantic import BaseModel
+
+class DeployConfiguration(BaseModel):
+    workspace_id: str
+    space_name: str
+    naas_api_key: str

{naas_abi_core-1.0.7.dist-info → naas_abi_core-1.1.0.dist-info}/METADATA

@@ -1,10 +1,11 @@
 Metadata-Version: 2.4
 Name: naas-abi-core
-Version: 1.0.7
+Version: 1.1.0
 Summary: Abi framework allowing you to build your AI system.
 Author-email: Maxime Jublou <maxime@naas.ai>, Florent Ravenel <florent@naas.ai>, Jeremy Ravenel <jeremy@naas.ai>
 Requires-Python: <4,>=3.10
 Requires-Dist: click<8.2,>=8.1.1
+Requires-Dist: docker>=7.1.0
 Requires-Dist: dotenv>=0.9.9
 Requires-Dist: fastapi<0.116,>=0.115.5
 Requires-Dist: fastmcp>=2.13.2

{naas_abi_core-1.0.7.dist-info → naas_abi_core-1.1.0.dist-info}/RECORD

@@ -7,10 +7,11 @@ naas_abi_core/apps/mcp/mcp_server.py,sha256=U-wDl8rswMk4nefGg2SUByw4x8CM10M59FfL
 naas_abi_core/apps/mcp/mcp_server_test.py,sha256=8jixnDyERM_HjjDTRtPDQcJ_rzA0ESAn147sl2Gk8gw,5636
 naas_abi_core/apps/terminal_agent/main.py,sha256=ucrNCGjjixyiO47Eily1yAmrwznS6KOcB1kk02yt_m8,19631
 naas_abi_core/apps/terminal_agent/terminal_style.py,sha256=YOpfvBlKU52wNyGEKbZPiRQVKxug-hI92H8ScV5L8Ew,5254
-naas_abi_core/cli/__init__.py,sha256=UHfBoHZGZ_cRDENqfAtm0gEX3tWKwUxsMezHZ3SXJqg,1273
+naas_abi_core/cli/__init__.py,sha256=P89Ihofvv_pQf2wsfYL-PjLJMl6KRj3fNNIZDB77Qy0,1324
 naas_abi_core/cli/agent.py,sha256=fMdbC7HsrOfZSf5zVRHWSmyrejI5mUdRlAT5v5YHXzk,658
 naas_abi_core/cli/chat.py,sha256=3t_TJ7vqCNs0MIIXOtlSke3nzy4rMSEJtB3P6pKItMo,856
 naas_abi_core/cli/config.py,sha256=CcdDX6HKCP32NjRhbVsCOwLUC9LmaqTm2sv8W5rOt00,1484
+naas_abi_core/cli/deploy.py,sha256=ZMf0EbwGoL4fwae3he2wCLMm6ghRWJthy1uW_tX6fC0,6018
 naas_abi_core/cli/init.py,sha256=Pcy2-hy-FvpXf3UOKMP6agWyFrCl9z-KP5ktEWltPy0,220
 naas_abi_core/cli/module.py,sha256=TBl-SpeGUcy1Rrp40Irbt34yQS00xJcNje-OijNE4Hk,717
 naas_abi_core/cli/new.py,sha256=aFhKbTHwqYkPdzrd7r8i_h9dfzXNjI03t8qVeqME8w8,262
@@ -20,7 +21,8 @@ naas_abi_core/engine/EngineProxy.py,sha256=o-D8LlP-PjQ_Yct4JWrDZw7mA7yporxb2XrJF
 naas_abi_core/engine/Engine_test.py,sha256=8eLZEnkL0IR4VAr6LF8fJ_fxZzi9s1mXCLgTVOIsR3E,161
 naas_abi_core/engine/IEngine.py,sha256=u-m-Qrvt3SP3gYKWPFPVjV8rs05D6NGnzO3XA0FInnw,2865
 naas_abi_core/engine/conftest.py,sha256=Al-SRVLrEdbTrX8sxQ3bBK4w1bbzE4GoBkzoBK-aXlg,932
-naas_abi_core/engine/engine_configuration/EngineConfiguration.py,sha256=XA8BDyuYWvjwREGHDBB1iGFe5neJKFKsaHFRnPPKJXI,5378
+naas_abi_core/engine/engine_configuration/EngineConfiguration.py,sha256=12rKgF1LaLC_YMhKEABFSdU_Fb04qGpW4tft3eVCnPI,5529
+naas_abi_core/engine/engine_configuration/EngineConfiguration_Deploy.py,sha256=NbPMxiNt9C4zfNF70fs7a8prELpd3QAHbZlvHfwcbxs,134
 naas_abi_core/engine/engine_configuration/EngineConfiguration_GenericLoader.py,sha256=KK2TQx6cNmoqFcwr9En00NKrX4ckkZl4ecv9QCUwPyc,1995
 naas_abi_core/engine/engine_configuration/EngineConfiguration_ObjectStorageService.py,sha256=cPZBs5-2dqX-piIZ7KTqiOce6O6GbnDDwjPGcfDN_U4,4736
 naas_abi_core/engine/engine_configuration/EngineConfiguration_ObjectStorageService_test.py,sha256=h1PdIbMTJWi_lG83YgpI6zg8gRo0WEWvGSE6R4uKQp4,1063
@@ -121,7 +123,7 @@ naas_abi_core/utils/onto2py/tests/ttl2py_test.py,sha256=5OZqSxPffjJYiX9T4rT1mV0P
 naas_abi_core/workflow/__init__.py,sha256=hZD58mCB1PApxITqftP_xgjxL7NeLvOfI-rJENg1ENs,250
 naas_abi_core/workflow/workflow.py,sha256=ZufSS073JztVl0OQRTqNyK7FepFvv7gXlc4j5FAEZCI,1216
 assets/favicon.ico,sha256=nWk8wrHZiJV3DeuWrP2MqilXxCuoNWKGtMZfYmEVQLw,666
-naas_abi_core-1.0.7.dist-info/METADATA,sha256=gTjTHO7PttQKEINKCxX0toiOwO14J3Nhxkxh2m224EA,3868
-naas_abi_core-1.0.7.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-naas_abi_core-1.0.7.dist-info/entry_points.txt,sha256=q68PvlGw_rozZ0nl6mUg6l1l2IhxaTOKlf5K9goRDu0,99
-naas_abi_core-1.0.7.dist-info/RECORD,,
+naas_abi_core-1.1.0.dist-info/METADATA,sha256=W0UO3c5K34ADisfJVFt0_pl2YadivJRqKH1UEAiAGvQ,3897
+naas_abi_core-1.1.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+naas_abi_core-1.1.0.dist-info/entry_points.txt,sha256=q68PvlGw_rozZ0nl6mUg6l1l2IhxaTOKlf5K9goRDu0,99
+naas_abi_core-1.1.0.dist-info/RECORD,,