agenta 0.10.0__py3-none-any.whl → 0.10.2__py3-none-any.whl

agenta/cli/main.py

@@ -79,7 +79,9 @@ def cli():
 
 @click.command()
 @click.option("--app_name", default="")
-def init(app_name: str):
+@click.option("--backend_host", default="")
+def init(app_name: str, backend_host: str):
+    init_option = "Blank App" if backend_host != "" and app_name != "" else ""
     """Initialize a new Agenta app with the template files."""
     if not app_name:
         while True:
@@ -95,28 +97,38 @@ def init(app_name: str):
                     )
 
     try:
-        where_question = questionary.select(
-            "Where are you running agenta?",
-            choices=["On agenta cloud", "On my local machine", "On a remote machine"],
-        ).ask()
-
-        if where_question == "On my local machine":
-            backend_host = "http://localhost"
-        elif where_question == "On a remote machine":
-            backend_host = questionary.text(
-                "Please provide the IP or URL of your remote host"
+        backend_hosts = {
+            "https://cloud.agenta.ai": "On agenta cloud",
+            "http://localhost": "On my local machine",
+        }
+        where_question = backend_hosts.get(backend_host, "On a remote machine")
+        if not backend_host:
+            where_question = questionary.select(
+                "Where are you running agenta?",
+                choices=[
+                    "On agenta cloud",
+                    "On my local machine",
+                    "On a remote machine",
+                ],
             ).ask()
-        elif where_question == "On agenta cloud":
-            global_backend_host = helper.get_global_config("host")
-            if global_backend_host:
-                backend_host = global_backend_host
-            else:
-                backend_host = "https://cloud.agenta.ai"
 
-            api_key = helper.get_api_key(backend_host)
+            if where_question == "On my local machine":
+                backend_host = "http://localhost"
+            elif where_question == "On a remote machine":
+                backend_host = questionary.text(
+                    "Please provide the IP or URL of your remote host"
+                ).ask()
+            elif where_question == "On agenta cloud":
+                global_backend_host = helper.get_global_config("host")
+                if global_backend_host:
+                    backend_host = global_backend_host
+                else:
+                    backend_host = "https://cloud.agenta.ai"
+
+                api_key = helper.get_api_key(backend_host)
 
-        elif where_question is None:  # User pressed Ctrl+C
-            sys.exit(0)
+            elif where_question is None:  # User pressed Ctrl+C
+                sys.exit(0)
         backend_host = (
             backend_host
             if backend_host.startswith("http://") or backend_host.startswith("https://")
@@ -160,41 +172,42 @@ def init(app_name: str):
             toml.dump(config, config_file)
 
         # Ask for init option
-        init_option = questionary.select(
-            "How do you want to initialize your app?",
-            choices=["Blank App", "Start from template"],
-        ).ask()
-
-        # If the user selected the second option, show a list of available templates
-        if init_option == "Start from template":
-            current_dir = Path.cwd()
-            template_dir = Path(__file__).parent.parent / "templates"
-            templates = [
-                folder.name for folder in template_dir.iterdir() if folder.is_dir()
-            ]
-            template_desc = [
-                toml.load((template_dir / name / "template.toml"))["short_desc"]
-                for name in templates
-            ]
-
-            # Show the templates to the user
-            template = questionary.select(
-                "Which template do you want to use?",
-                choices=[
-                    questionary.Choice(
-                        title=f"{template} - {template_desc}", value=template
-                    )
-                    for template, template_desc in zip(templates, template_desc)
-                ],
+        if not init_option:
+            init_option = questionary.select(
+                "How do you want to initialize your app?",
+                choices=["Blank App", "Start from template"],
             ).ask()
 
-            # Copy the template files to the current directory
-            chosen_template_dir = template_dir / template
-            for file in chosen_template_dir.glob("*"):
-                if file.name != "template.toml" and not file.is_dir():
-                    shutil.copy(file, current_dir / file.name)
-        elif init_option is None:  # User pressed Ctrl+C
-            sys.exit(0)
+            # If the user selected the second option, show a list of available templates
+            if init_option == "Start from template":
+                current_dir = Path.cwd()
+                template_dir = Path(__file__).parent.parent / "templates"
+                templates = [
+                    folder.name for folder in template_dir.iterdir() if folder.is_dir()
+                ]
+                template_desc = [
+                    toml.load((template_dir / name / "template.toml"))["short_desc"]
+                    for name in templates
+                ]
+
+                # Show the templates to the user
+                template = questionary.select(
+                    "Which template do you want to use?",
+                    choices=[
+                        questionary.Choice(
+                            title=f"{template} - {template_desc}", value=template
+                        )
+                        for template, template_desc in zip(templates, template_desc)
+                    ],
+                ).ask()
+
+                # Copy the template files to the current directory
+                chosen_template_dir = template_dir / template
+                for file in chosen_template_dir.glob("*"):
+                    if file.name != "template.toml" and not file.is_dir():
+                        shutil.copy(file, current_dir / file.name)
+            elif init_option is None:  # User pressed Ctrl+C
+                sys.exit(0)
 
         # Create a .gitignore file and add some default environment folder names to it
         gitignore_content = (

agenta/sdk/agenta_init.py

@@ -75,16 +75,21 @@ class AgentaSingleton:
                 )
             else:
                 try:
-                    get_app_id = client.list_apps(app_name=app_name)
-                    app_id = get_app_id.app_id
+                    apps = client.list_apps(app_name=app_name)
+                    if len(apps) == 0:
+                        raise APIRequestError(f"App with name {app_name} not found")
 
+                    app_id = apps[0].app_id
                     if not app_id:
                         raise APIRequestError(
                             f"App with name {app_name} does not exist on the server."
                         )
 
-                    get_base_id = client.list_bases(app_id=app_id, base_name=base_name)
-                    base_id = get_base_id.base_id
+                    bases = client.list_bases(app_id=app_id, base_name=base_name)
+                    if len(bases) == 0:
+                        raise APIRequestError(f"No base was found for the app {app_id}")
+
+                    base_id = bases[0].base_id
                 except Exception as ex:
                     raise APIRequestError(
                         f"Failed to get base id and/or app_id from the server with error: {ex}"
@@ -171,7 +176,7 @@ class Config:
                     + str(ex)
                 )
         try:
-            self.set(**config["parameters"])
+            self.set(**config.parameters)
         except Exception as ex:
             logger.warning("Failed to set the configuration with error: " + str(ex))
 

{agenta-0.10.0.dist-info → agenta-0.10.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.10.0
+Version: 0.10.2
 Summary: The SDK for agenta is an open-source LLMOps platform.
 Home-page: https://agenta.ai
 Keywords: LLMOps,LLM,evaluation,prompt engineering

{agenta-0.10.0.dist-info → agenta-0.10.2.dist-info}/RECORD

@@ -1,7 +1,7 @@
 agenta/__init__.py,sha256=Sg_4PW8MM-r9p96FeqTdu1h7fcQzN4nF7KPuQD7lgIc,493
 agenta/cli/evaluation_commands.py,sha256=fs6492tprPId9p8eGO02Xy-NCBm2RZNJLZWcUxugwd8,474
 agenta/cli/helper.py,sha256=X_0GtjugLQiM_f53_l3AF3uPS8if96ejTh2nThYqLuE,6199
-agenta/cli/main.py,sha256=3XuIlEQLG9X3NFW3beKPqSe-zTNdBGdT9jh8e0dU5PU,7599
+agenta/cli/main.py,sha256=sYkNSqSJRTQOi6u-JulHiwC4ObqzygP13Dk1tcwEktk,8310
 agenta/cli/telemetry.py,sha256=GaFFRsE_NtrcSSJ10r2jhgFs5Sk8gf2C09Ox3gOr3eU,1317
 agenta/cli/variant_commands.py,sha256=wJCfTp1uNQwSadg4rraxZF5NUVfpfg_0RAK76qD5xDo,16858
 agenta/cli/variant_configs.py,sha256=PLiuMKadVzs6Gi2uYaT0pZzyULNHDXaTMDWboqpwWdU,1293
@@ -86,7 +86,7 @@ agenta/docker/docker-assets/main.py,sha256=BbmbFByRQ8MzL8402pJryEF34t6ba1i_JrMtW
 agenta/docker/docker_utils.py,sha256=gtwD0XZ_1Vk-rzNUYctlnRjZPpn01mX7tYvaNK7o7OQ,3542
 agenta/sdk/__init__.py,sha256=TUTEndClnM2SKeQsC7Swv_K3uWkBZTd7jYKvp2-tQeY,565
 agenta/sdk/agenta_decorator.py,sha256=B0tQCBsbSUFfdo1Ibadd0MPQCN7xdGtBcTurmsclL6U,15429
-agenta/sdk/agenta_init.py,sha256=NdXNUlfabALJINJlZr5oXFt4XwMeAomp83YImdTZ_KU,7792
+agenta/sdk/agenta_init.py,sha256=67THCATEW_oC8juCzdS-AouirksuXouyrRPyuUkLFxw,8025
 agenta/sdk/context.py,sha256=q-PxL05-I84puunUAs9LGsffEXcYhDxhQxjuOz2vK90,901
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
 agenta/sdk/types.py,sha256=GiUiXDkbM2pMx1-mldzQT8uDUcm9sRqVUytThEkbjj4,4658
@@ -108,7 +108,7 @@ agenta/templates/simple_prompt/app.py,sha256=kODgF6lhzsaJPdgL5b21bUki6jkvqjWZzWR
 agenta/templates/simple_prompt/env.example,sha256=g9AE5bYcGPpxawXMJ96gh8oenEPCHTabsiOnfQo3c5k,70
 agenta/templates/simple_prompt/requirements.txt,sha256=ywRglRy7pPkw8bljmMEJJ4aOOQKrt9FGKULZ-DGkoBU,23
 agenta/templates/simple_prompt/template.toml,sha256=DQBtRrF4GU8LBEXOZ-GGuINXMQDKGTEG5y37tnvIUIE,60
-agenta-0.10.0.dist-info/METADATA,sha256=oAcCBLqZqUXCKD1y1MTwDxHuyXwz_cw8hTAfEupgg3M,27296
-agenta-0.10.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-agenta-0.10.0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.10.0.dist-info/RECORD,,
+agenta-0.10.2.dist-info/METADATA,sha256=EOO3EgKztl8qTAHATQwW4IjsyoDxREGmc50ygS0vWc8,27296
+agenta-0.10.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+agenta-0.10.2.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.10.2.dist-info/RECORD,,