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

naas_abi_core/cli/deploy.py

@@ -115,7 +115,10 @@ class NaasDeployer:
             f"docker build -t {image_name} . --platform linux/amd64", shell=True
         )
 
-    def deploy(self):
+    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
         )
@@ -145,6 +148,7 @@ class NaasDeployer:
 
         image_name_with_sha = f"{image_name.replace(':' + uid, '')}@{image_sha}"
 
+
         self.naas_api_client.create_space(
             Space(
                 name=self.configuration.deploy.space_name,
@@ -155,7 +159,7 @@ class NaasDeployer:
                         port=9879,
                         cpu="1",
                         memory="1Gi",
-                        env={
+                        env=self.env_list_to_dict(env) | {
                             "NAAS_API_KEY": self.configuration.deploy.naas_api_key,
                             "ENV": "prod",
                         },
@@ -182,7 +186,8 @@ class NaasDeployer:
 
 
 @deploy.command("naas")
-def 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:
@@ -191,4 +196,4 @@ def naas():
         )
 
     deployer = NaasDeployer(configuration)
-    deployer.deploy()
+    deployer.deploy(env)

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: naas-abi-core
-Version: 1.0.8
+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

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

@@ -11,7 +11,7 @@ naas_abi_core/cli/__init__.py,sha256=P89Ihofvv_pQf2wsfYL-PjLJMl6KRj3fNNIZDB77Qy0
 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/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
@@ -123,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.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,,
+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,,