agenta 0.5.7__tar.gz → 0.6.0__tar.gz

{agenta-0.5.7 → agenta-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: agenta
-Version: 0.5.7
+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.7 → agenta-0.6.0}/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-0.5.7 → agenta-0.6.0}/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,9 +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"
+            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()
+            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
@@ -187,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-0.6.0/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-0.5.7 → agenta-0.6.0}/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-0.5.7 → agenta-0.6.0}/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.7 → agenta-0.6.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "agenta"
-version = "0.5.7"
+version = "0.6.0"
 description = "The SDK for agenta is an open-source LLMOps platform."
 readme = "README.md"
 authors = ["Mahmoud Mabrouk <mahmoud@agenta.ai>"]