naas-abi-core 1.0.6__py3-none-any.whl → 1.0.8__py3-none-any.whl

naas_abi_core/apps/api/api.py

@@ -16,6 +16,7 @@ from fastapi.security.oauth2 import OAuth2
 from fastapi.security.utils import get_authorization_scheme_param
 from fastapi.staticfiles import StaticFiles
 from naas_abi_core import logger
+from importlib.resources import files
 
 # Docs
 from naas_abi_core.apps.api.openapi_doc import API_LANDING_HTML, TAGS_METADATA
@@ -49,8 +50,10 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# Mount the static directory
-app.mount("/static", StaticFiles(directory="assets"), name="static")
+
+static_dir = os.path.join(os.path.dirname(str(files("naas_abi_core"))), "assets")
+app.mount("/static", StaticFiles(directory=str(static_dir)), name="static")
+
 
 
 # Custom OAuth2 class that accepts query parameter

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()
+# if __name__ == "__main__":
+main()

naas_abi_core/cli/deploy.py

@@ -0,0 +1,194 @@
+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 deploy(self):
+        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={
+                            "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")
+def naas():
+    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()

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.6.dist-info → naas_abi_core-1.0.8.dist-info}/METADATA

@@ -1,10 +1,11 @@
 Metadata-Version: 2.4
 Name: naas-abi-core
-Version: 1.0.6
+Version: 1.0.8
 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
@@ -14,6 +15,7 @@ Requires-Dist: langgraph>=0.6.6
 Requires-Dist: loguru<0.8,>=0.7.2
 Requires-Dist: pandas-stubs>=2.3.2.250926
 Requires-Dist: pandas>=2.3.3
+Requires-Dist: pillow>=12.0.0
 Requires-Dist: pip>=25.1.1
 Requires-Dist: psycopg[binary,pool]>=3.0.0
 Requires-Dist: pydantic>=2.11.5
@@ -29,11 +31,14 @@ Requires-Dist: starlette>=0.46.2
 Requires-Dist: types-tqdm>=4.67.0.20250809
 Provides-Extra: all
 Requires-Dist: boto3<2,>=1.38.19; extra == 'all'
+Requires-Dist: langchain-openai>=0.3.35; extra == 'all'
 Requires-Dist: paramiko<4.0.0,>=3.5.1; extra == 'all'
 Requires-Dist: qdrant-client>=1.14.3; extra == 'all'
 Requires-Dist: sshtunnel>=0.4.0; extra == 'all'
 Provides-Extra: aws
 Requires-Dist: boto3<2,>=1.38.19; extra == 'aws'
+Provides-Extra: openrouter
+Requires-Dist: langchain-openai>=0.3.35; extra == 'openrouter'
 Provides-Extra: qdrant
 Requires-Dist: qdrant-client>=1.14.3; extra == 'qdrant'
 Provides-Extra: ssh

{naas_abi_core-1.0.6.dist-info → naas_abi_core-1.0.8.dist-info}/RECORD

@@ -1,5 +1,5 @@
 naas_abi_core/__init__.py,sha256=kR93ywABUBo8-tAWCHMbRcCDCJZjqq4hWwxMk3Cr9fI,56
-naas_abi_core/apps/api/api.py,sha256=CMast2FEo6S7WbXkeELaM4QHEv_-uDLVkieZcXGGolo,7027
+naas_abi_core/apps/api/api.py,sha256=C11ItO_Xn0qh7ZubZUnQ4-Mts0Cd-Ui_4KEkU2cshqI,7127
 naas_abi_core/apps/api/api_test.py,sha256=7H_VgnbsGqu6oGY-apAHjJr_yjIcGHY0heaEThWBsUk,10537
 naas_abi_core/apps/api/openapi_doc.py,sha256=JbwzcmSP-Yl1rbN0QvUwOSmOFf-yKXECZ-2ptI7fWoU,11834
 naas_abi_core/apps/mcp/Dockerfile.mcp,sha256=936qaoMFGbMvTARjzkNHMgRfGVWnolpVCrz0fBJYrLA,826
@@ -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=k5L0LznXVHyZxINdoqe0IOVz1IaSXsUB9ptIP1dEnns,1275
+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=sdSRbeg3UPAujHjWLRW7zTpdIZF2GgXtOZ8bYYfe77Q,5701
 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
@@ -120,7 +122,8 @@ naas_abi_core/utils/onto2py/onto2py.py,sha256=_wn9qrrsWd8gO-BY4_jQJVfY87e9MJ2Er2
 naas_abi_core/utils/onto2py/tests/ttl2py_test.py,sha256=5OZqSxPffjJYiX9T4rT1mV0PT1Qhf6goqEYT_mAPnaI,8055
 naas_abi_core/workflow/__init__.py,sha256=hZD58mCB1PApxITqftP_xgjxL7NeLvOfI-rJENg1ENs,250
 naas_abi_core/workflow/workflow.py,sha256=ZufSS073JztVl0OQRTqNyK7FepFvv7gXlc4j5FAEZCI,1216
-naas_abi_core-1.0.6.dist-info/METADATA,sha256=UgONZUsRZVdOz-kFmeetC546I_5KbFnaM4LKH35xnx0,3692
-naas_abi_core-1.0.6.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-naas_abi_core-1.0.6.dist-info/entry_points.txt,sha256=q68PvlGw_rozZ0nl6mUg6l1l2IhxaTOKlf5K9goRDu0,99
-naas_abi_core-1.0.6.dist-info/RECORD,,
+assets/favicon.ico,sha256=nWk8wrHZiJV3DeuWrP2MqilXxCuoNWKGtMZfYmEVQLw,666
+naas_abi_core-1.0.8.dist-info/METADATA,sha256=7Ktah_ysJ6RSGn_ZK765ISBRn961DGiBrPLRoCCQCZw,3897
+naas_abi_core-1.0.8.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+naas_abi_core-1.0.8.dist-info/entry_points.txt,sha256=q68PvlGw_rozZ0nl6mUg6l1l2IhxaTOKlf5K9goRDu0,99
+naas_abi_core-1.0.8.dist-info/RECORD,,