agenta 0.5.8__py3-none-any.whl → 0.6.0__py3-none-any.whl

agenta/cli/helper.py

@@ -57,7 +57,7 @@ def set_global_config(var_name: str, var_value: Any) -> None:
         toml.dump(global_config, config_file)
 
 
-def get_api_key() -> str:
+def get_api_key(backend_host: str) -> str:
     """
     Retrieve or request the API key for accessing the Agenta platform.
 
@@ -65,6 +65,9 @@ def get_api_key() -> str:
     If found, it prompts the user to confirm whether they'd like to use that key.
     If not found, it asks the user to input a new key.
 
+    Args:
+        backend_host (str): The URL of the backend host.
+
     Returns:
         str: The API key to be used for accessing the Agenta platform.
 
@@ -85,7 +88,7 @@ def get_api_key() -> str:
             sys.exit(0)
 
     api_key = questionary.text(
-        "(You can get your API Key here: https://cloud.agenta.ai/settings?tab=apiKeys) "
+        f"(You can get your API Key here: {backend_host}/settings?tab=apiKeys) "
         "Please provide your API key:"
     ).ask()
 

agenta/cli/main.py

@@ -8,9 +8,10 @@ import click
 import questionary
 import toml
 
+from agenta.cli import helper
 from agenta.client import client
+from agenta.cli import variant_configs
 from agenta.cli import variant_commands
-from agenta.cli import helper
 
 
 def print_version(ctx, param, value):
@@ -101,8 +102,13 @@ def init(app_name: str):
                 "Please provide the IP or URL of your remote host"
             ).ask()
         elif where_question == "On agenta cloud":
-            backend_host = "https://cloud.agenta.ai"
-            api_key = helper.get_api_key()
+            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)
             client.validate_api_key(api_key, backend_host)
 
         elif where_question is None:  # User pressed Ctrl+C
@@ -186,6 +192,7 @@ def init(app_name: str):
 
 # Add the commands to the CLI group
 cli.add_command(init)
+cli.add_command(variant_configs.config)
 cli.add_command(variant_commands.variant)
 
 if __name__ == "__main__":

agenta/cli/variant_configs.py

@@ -0,0 +1,49 @@
+import click
+from agenta.cli import helper
+
+
+@click.group()
+def config():
+    """Commands for variants configurations"""
+    pass
+
+
+def update_backend_host(backend_host: str):
+    """Check the config file and update the backend URL
+
+    Arguments:
+        app_folder -- the app folder
+        backend_host -- the backend host
+    """
+
+    click.echo(
+        click.style("\nChecking and updating global backend host...", fg="bright_black")
+    )
+    helper.set_global_config("host", backend_host)
+
+
+@config.command(
+    name="set-host",
+    context_settings=dict(
+        ignore_unknown_options=True,
+        allow_extra_args=True,
+    ),
+)
+@click.option(
+    "--backend_host", default=None, help="The URL of the backend host to use."
+)
+@click.pass_context
+def set_config_url(ctx, backend_host: str):
+    """Set the backend URL in the app configuration"""
+
+    try:
+        if not backend_host:
+            if ctx.args:
+                backend_host = ctx.args[0]
+            else:
+                click.echo(click.style("Backend host URL not specified", fg="red"))
+
+        update_backend_host(backend_host)
+        click.echo(click.style("Backend host updated successfully! 🎉\n"))
+    except Exception as ex:
+        click.echo(click.style(f"Error updating backend host: {ex}", fg="red"))

agenta/sdk/agenta_decorator.py

@@ -12,10 +12,9 @@ from typing import Any, Callable, Dict, Optional, Tuple
 import agenta
 from fastapi import Body, FastAPI, UploadFile
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.openapi.models import Server
 from fastapi.responses import JSONResponse
 
-from .context import get_contexts, save_context
+from .context import save_context
 from .router import router as router
 from .types import (
     Context,
@@ -345,7 +344,6 @@ def override_schema(openapi_schema: dict, func_name: str, endpoint: str, params:
         f"Body_{func_name}_{endpoint}_post"
     ]["properties"]
     for param_name, param_val in params.items():
-        # print(param_name, param_val)
         if isinstance(param_val, MultipleChoiceParam):
             subschema = find_in_schema(schema_to_override, param_name, "choice")
             default = str(param_val)

agenta/sdk/types.py

@@ -106,7 +106,7 @@ class Message(BaseModel):
 class MessagesInput(list):
     """Messages Input for Chat-completion.
 
-    Parameters:
+    Args:
         messages (List[Dict[str, str]]): The list of messages inputs.
         Required. Each message should be a dictionary with "role" and "content" keys.
 
@@ -116,11 +116,8 @@ class MessagesInput(list):
     """
 
     def __new__(cls, messages: List[Dict[str, str]] = None):
-        if not messages:
-            raise ValueError("Missing required parameter in MessagesInput")
-
         instance = super().__new__(cls, messages)
-        instance.messages = messages
+        instance.default = messages
         return instance
 
     @classmethod

{agenta-0.5.8.dist-info → agenta-0.6.0.dist-info}/METADATA

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

{agenta-0.5.8.dist-info → agenta-0.6.0.dist-info}/RECORD

@@ -1,8 +1,9 @@
 agenta/__init__.py,sha256=grmSigSubGpmopSRm6W3bVVOdVR6tHNfqleJPecESak,394
-agenta/cli/helper.py,sha256=Bexi77D5g9tbFND5Xh7u6KzbqKKD9jdZmiLTbXMCLgY,5928
-agenta/cli/main.py,sha256=R3i6jbn_mxTjLdJxVeYzdfYv7i1cxdqbvxUc-NDFNZY,6392
+agenta/cli/helper.py,sha256=-Xl7aDfRczrSnMVdv-tNDEq0YkxQNZxPH8mTSvmATxU,6005
+agenta/cli/main.py,sha256=AYNSMeQTJ9MVRd-5oTS1WuiWZXwQDkK_Vn1BM_F2IXQ,6660
 agenta/cli/telemetry.py,sha256=TiI4Nd1F1f1eSrs3fumDhcl8iu56aI4emyaGo5BQwMU,1165
 agenta/cli/variant_commands.py,sha256=wsY5dD2_3hX6HN2d-f1L4BtbjhvGiiw8rThxy_63A1A,15980
+agenta/cli/variant_configs.py,sha256=PLiuMKadVzs6Gi2uYaT0pZzyULNHDXaTMDWboqpwWdU,1293
 agenta/client/Readme.md,sha256=umhMce1Gq_der9pH4M_pP4NQ8rHa3MENJLM9OrdUZ50,131
 agenta/client/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agenta/client/api_models.py,sha256=zebfE2-0-SW1SvzyarzmSJMXqyiCLKrX2sHpzoX-RnU,623
@@ -17,11 +18,11 @@ agenta/docker/docker-assets/lambda_function.py,sha256=h4UZSSfqwpfsCgERv6frqwm_4J
 agenta/docker/docker-assets/main.py,sha256=BbmbFByRQ8MzL8402pJryEF34t6ba1i_JrMtWMAQDQ4,327
 agenta/docker/docker_utils.py,sha256=oK51dmJLUXcR0iEi3knDqt5BZxF3somqWHVOpTQ8rOI,5681
 agenta/sdk/__init__.py,sha256=QgjkiERHsprd59qx1bYOKD21DSQr3Bftetfgozdoe7I,470
-agenta/sdk/agenta_decorator.py,sha256=psRjROXtus2huHNOeGWvf-4Nnig5EEaJKEFCOXnEIJI,13354
+agenta/sdk/agenta_decorator.py,sha256=VmM4RAdgEc30FHvL313hQqqpzwwmjl_amT4dhBXMuI4,13259
 agenta/sdk/agenta_init.py,sha256=3-0WRpRSveWUcLhosTuym-YKV9gKVNrOTxGjSa6SxyI,7152
 agenta/sdk/context.py,sha256=q-PxL05-I84puunUAs9LGsffEXcYhDxhQxjuOz2vK90,901
 agenta/sdk/router.py,sha256=0sbajvn5C7t18anH6yNo7-oYxldHnYfwcbmQnIXBePw,269
-agenta/sdk/types.py,sha256=ep2ls42AxEIBcb0_NJVwDw2isNHFX_BNFBbom2ogP2w,3929
+agenta/sdk/types.py,sha256=JkNq4fkpOqlTC3PHotQHgLwQLWHqBXmctZGSr61CjEY,3820
 agenta/sdk/utils/globals.py,sha256=lpgflY8xovZJtHfJf41dbNCZGwx07YNkG9ldruv6xoI,360
 agenta/sdk/utils/preinit.py,sha256=YlJL7RLfel0R7DFp-jK7OV-z4ZIQJM0oupYlk7g8b5o,1278
 agenta/templates/compose_email/README.md,sha256=ss7vZPpI1Hg0VmYtFliwq_r5LnqbCy_S5OQDXg8UoIA,308
@@ -39,7 +40,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.5.8.dist-info/METADATA,sha256=1crHlBipy4gKc3JZ3a41vFXpYRew6kgb7v_xMju3xDI,10093
-agenta-0.5.8.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-agenta-0.5.8.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
-agenta-0.5.8.dist-info/RECORD,,
+agenta-0.6.0.dist-info/METADATA,sha256=kvVWmSlKSCMzZTzfIiNGeDsnprW8rHd-l_f96wi7pFo,10093
+agenta-0.6.0.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+agenta-0.6.0.dist-info/entry_points.txt,sha256=PDiu8_8AsL7ibU9v4iNoOKR1S7F2rdxjlEprjM9QOgo,46
+agenta-0.6.0.dist-info/RECORD,,