agno 1.8.1__py3-none-any.whl → 2.0.0a1__py3-none-any.whl

agno/cli/operator.py

@@ -1,357 +0,0 @@
-from pathlib import Path
-from typing import List, Optional
-
-from typer import launch as typer_launch
-
-from agno.cli.config import AgnoCliConfig
-from agno.cli.console import print_heading, print_info
-from agno.cli.settings import AGNO_CLI_CONFIG_DIR, agno_cli_settings
-from agno.infra.resources import InfraResources
-from agno.utils.log import logger
-
-
-def delete_agno_config() -> None:
-    from agno.utils.filesystem import delete_from_fs
-
-    logger.debug("Removing existing Agno configuration")
-    delete_from_fs(AGNO_CLI_CONFIG_DIR)
-
-
-def authenticate_user() -> None:
-    """Authenticate the user using credentials from agno.com
-    Steps:
-    1. Authenticate the user by opening the agno sign-in url.
-        Once authenticated, agno.com will post an auth token to a
-        mini http server running on the auth_server_port.
-    2. Using the auth_token, authenticate the user with the api.
-    3. After the user is authenticated update the AgnoCliConfig.
-    4. Save the auth_token locally for future use.
-    """
-    from agno.api.schemas.user import UserSchema
-    from agno.api.user import authenticate_and_get_user
-    from agno.cli.auth_server import (
-        get_auth_token_from_web_flow,
-        get_port_for_auth_server,
-    )
-    from agno.cli.credentials import save_auth_token
-
-    print_heading("Authenticating with agno.com")
-
-    auth_server_port = get_port_for_auth_server()
-    redirect_uri = "http%3A%2F%2Flocalhost%3A{}%2Fauth".format(auth_server_port)
-    auth_url = "{}?source=cli&action=signin&redirection_supported=true&redirecturi={}".format(
-        agno_cli_settings.cli_auth_url, redirect_uri
-    )
-    print_info("\nYour browser will be opened to visit:\n{}".format(auth_url))
-    typer_launch(auth_url)
-    print_info("\nWaiting for a response from the browser...\n")
-
-    auth_token = get_auth_token_from_web_flow(auth_server_port)
-    if auth_token is None:
-        logger.error("Could not authenticate, please set AGNO_API_KEY or try again")
-        return
-
-    agno_config: Optional[AgnoCliConfig] = AgnoCliConfig.from_saved_config()
-    existing_user: Optional[UserSchema] = agno_config.user if agno_config is not None else None
-    # Authenticate the user and claim any workspaces from anon user
-    try:
-        user: Optional[UserSchema] = authenticate_and_get_user(auth_token=auth_token, existing_user=existing_user)
-    except Exception as e:
-        logger.exception(e)
-        logger.error("Could not authenticate, please set AGNO_API_KEY or try again")
-        return
-
-    # Save the auth token if user is authenticated
-    if user is not None:
-        save_auth_token(auth_token)
-    else:
-        logger.error("Could not authenticate, please set AGNO_API_KEY or try again")
-        return
-
-    if agno_config is None:
-        agno_config = AgnoCliConfig(user)
-        agno_config.save_config()
-    else:
-        agno_config.user = user
-
-    print_info("Welcome {}".format(user.email))
-
-
-def initialize_agno(reset: bool = False, login: bool = False) -> Optional[AgnoCliConfig]:
-    """Initialize Agno on the users machine.
-
-    Steps:
-    1. Check if AGNO_CLI_CONFIG_DIR exists, if not, create it. If reset == True, recreate AGNO_CLI_CONFIG_DIR.
-    2. Authenticates the user if login == True.
-    3. If AgnoCliConfig exists and auth is valid, returns AgnoCliConfig.
-    """
-    from agno.api.user import create_anon_user
-    from agno.utils.filesystem import delete_from_fs
-
-    print_heading("Welcome to Agno!")
-    if reset:
-        delete_agno_config()
-
-    logger.debug("Initializing Agno")
-
-    # Check if ~/.config/ag exists, if it is not a dir - delete it and create the directory
-    if AGNO_CLI_CONFIG_DIR.exists():
-        logger.debug(f"{AGNO_CLI_CONFIG_DIR} exists")
-        if not AGNO_CLI_CONFIG_DIR.is_dir():
-            try:
-                delete_from_fs(AGNO_CLI_CONFIG_DIR)
-            except Exception as e:
-                logger.exception(e)
-                raise Exception(f"Something went wrong, please delete {AGNO_CLI_CONFIG_DIR} and run again")
-            AGNO_CLI_CONFIG_DIR.mkdir(parents=True, exist_ok=True)
-    else:
-        AGNO_CLI_CONFIG_DIR.mkdir(parents=True)
-        logger.debug(f"Created {AGNO_CLI_CONFIG_DIR}")
-
-    # Confirm AGNO_CLI_CONFIG_DIR exists otherwise we should return
-    if AGNO_CLI_CONFIG_DIR.exists():
-        logger.debug(f"Agno config location: {AGNO_CLI_CONFIG_DIR}")
-    else:
-        raise Exception("Something went wrong, please try again")
-
-    agno_config: Optional[AgnoCliConfig] = AgnoCliConfig.from_saved_config()
-    if agno_config is None:
-        logger.debug("Creating new AgnoCliConfig")
-        agno_config = AgnoCliConfig()
-        agno_config.save_config()
-
-    # Authenticate user
-    if login:
-        print_info("")
-        authenticate_user()
-    else:
-        anon_user = create_anon_user()
-        if anon_user is not None and agno_config is not None:
-            agno_config.user = anon_user
-
-    logger.debug("Agno initialized")
-    return agno_config
-
-
-def start_resources(
-    agno_config: AgnoCliConfig,
-    resources_file_path: Path,
-    target_env: Optional[str] = None,
-    target_infra: Optional[str] = None,
-    target_group: Optional[str] = None,
-    target_name: Optional[str] = None,
-    target_type: Optional[str] = None,
-    dry_run: Optional[bool] = False,
-    auto_confirm: Optional[bool] = False,
-    force: Optional[bool] = None,
-    pull: Optional[bool] = False,
-) -> None:
-    print_heading(f"Starting resources in: {resources_file_path}")
-    logger.debug(f"\ttarget_env   : {target_env}")
-    logger.debug(f"\ttarget_infra : {target_infra}")
-    logger.debug(f"\ttarget_name  : {target_name}")
-    logger.debug(f"\ttarget_type  : {target_type}")
-    logger.debug(f"\ttarget_group : {target_group}")
-    logger.debug(f"\tdry_run      : {dry_run}")
-    logger.debug(f"\tauto_confirm : {auto_confirm}")
-    logger.debug(f"\tforce        : {force}")
-    logger.debug(f"\tpull         : {pull}")
-
-    from agno.workspace.config import WorkspaceConfig
-
-    if not resources_file_path.exists():
-        logger.error(f"File does not exist: {resources_file_path}")
-        return
-
-    # Get resources to deploy
-    resource_groups_to_create: List[InfraResources] = WorkspaceConfig.get_resources_from_file(
-        resource_file=resources_file_path,
-        env=target_env,
-        infra=target_infra,
-        order="create",
-    )
-
-    # Track number of resource groups created
-    num_rgs_created = 0
-    num_rgs_to_create = len(resource_groups_to_create)
-    # Track number of resources created
-    num_resources_created = 0
-    num_resources_to_create = 0
-
-    if num_rgs_to_create == 0:
-        print_info("No resources to create")
-        return
-
-    logger.debug(f"Deploying {num_rgs_to_create} resource groups")
-    for rg in resource_groups_to_create:
-        _num_resources_created, _num_resources_to_create = rg.create_resources(
-            group_filter=target_group,
-            name_filter=target_name,
-            type_filter=target_type,
-            dry_run=dry_run,
-            auto_confirm=auto_confirm,
-            force=force,
-            pull=pull,
-        )
-        if _num_resources_created > 0:
-            num_rgs_created += 1
-        num_resources_created += _num_resources_created
-        num_resources_to_create += _num_resources_to_create
-        logger.debug(f"Deployed {num_resources_created} resources in {num_rgs_created} resource groups")
-
-    if dry_run:
-        return
-
-    if num_resources_created == 0:
-        return
-
-    print_heading(f"\n--**-- ResourceGroups deployed: {num_rgs_created}/{num_rgs_to_create}\n")
-    if num_resources_created != num_resources_to_create:
-        logger.error("Some resources failed to create, please check logs")
-
-
-def stop_resources(
-    agno_config: AgnoCliConfig,
-    resources_file_path: Path,
-    target_env: Optional[str] = None,
-    target_infra: Optional[str] = None,
-    target_group: Optional[str] = None,
-    target_name: Optional[str] = None,
-    target_type: Optional[str] = None,
-    dry_run: Optional[bool] = False,
-    auto_confirm: Optional[bool] = False,
-    force: Optional[bool] = None,
-) -> None:
-    print_heading(f"Stopping resources in: {resources_file_path}")
-    logger.debug(f"\ttarget_env   : {target_env}")
-    logger.debug(f"\ttarget_infra : {target_infra}")
-    logger.debug(f"\ttarget_name  : {target_name}")
-    logger.debug(f"\ttarget_type  : {target_type}")
-    logger.debug(f"\ttarget_group : {target_group}")
-    logger.debug(f"\tdry_run      : {dry_run}")
-    logger.debug(f"\tauto_confirm : {auto_confirm}")
-    logger.debug(f"\tforce        : {force}")
-
-    from agno.workspace.config import WorkspaceConfig
-
-    if not resources_file_path.exists():
-        logger.error(f"File does not exist: {resources_file_path}")
-        return
-
-    # Get resource groups to shutdown
-    resource_groups_to_shutdown: List[InfraResources] = WorkspaceConfig.get_resources_from_file(
-        resource_file=resources_file_path,
-        env=target_env,
-        infra=target_infra,
-        order="create",
-    )
-
-    # Track number of resource groups deleted
-    num_rgs_shutdown = 0
-    num_rgs_to_shutdown = len(resource_groups_to_shutdown)
-    # Track number of resources created
-    num_resources_shutdown = 0
-    num_resources_to_shutdown = 0
-
-    if num_rgs_to_shutdown == 0:
-        print_info("No resources to delete")
-        return
-
-    logger.debug(f"Deleting {num_rgs_to_shutdown} resource groups")
-    for rg in resource_groups_to_shutdown:
-        _num_resources_shutdown, _num_resources_to_shutdown = rg.delete_resources(
-            group_filter=target_group,
-            name_filter=target_name,
-            type_filter=target_type,
-            dry_run=dry_run,
-            auto_confirm=auto_confirm,
-            force=force,
-        )
-        if _num_resources_shutdown > 0:
-            num_rgs_shutdown += 1
-        num_resources_shutdown += _num_resources_shutdown
-        num_resources_to_shutdown += _num_resources_to_shutdown
-        logger.debug(f"Deleted {num_resources_shutdown} resources in {num_rgs_shutdown} resource groups")
-
-    if dry_run:
-        return
-
-    if num_resources_shutdown == 0:
-        return
-
-    print_heading(f"\n--**-- ResourceGroups deleted: {num_rgs_shutdown}/{num_rgs_to_shutdown}\n")
-    if num_resources_shutdown != num_resources_to_shutdown:
-        logger.error("Some resources failed to delete, please check logs")
-
-
-def patch_resources(
-    agno_config: AgnoCliConfig,
-    resources_file_path: Path,
-    target_env: Optional[str] = None,
-    target_infra: Optional[str] = None,
-    target_group: Optional[str] = None,
-    target_name: Optional[str] = None,
-    target_type: Optional[str] = None,
-    dry_run: Optional[bool] = False,
-    auto_confirm: Optional[bool] = False,
-    force: Optional[bool] = None,
-) -> None:
-    print_heading(f"Updating resources in: {resources_file_path}")
-    logger.debug(f"\ttarget_env   : {target_env}")
-    logger.debug(f"\ttarget_infra : {target_infra}")
-    logger.debug(f"\ttarget_name  : {target_name}")
-    logger.debug(f"\ttarget_type  : {target_type}")
-    logger.debug(f"\ttarget_group : {target_group}")
-    logger.debug(f"\tdry_run      : {dry_run}")
-    logger.debug(f"\tauto_confirm : {auto_confirm}")
-    logger.debug(f"\tforce        : {force}")
-
-    from agno.workspace.config import WorkspaceConfig
-
-    if not resources_file_path.exists():
-        logger.error(f"File does not exist: {resources_file_path}")
-        return
-
-    # Get resource groups to update
-    resource_groups_to_patch: List[InfraResources] = WorkspaceConfig.get_resources_from_file(
-        resource_file=resources_file_path,
-        env=target_env,
-        infra=target_infra,
-        order="create",
-    )
-
-    num_rgs_patched = 0
-    num_rgs_to_patch = len(resource_groups_to_patch)
-    # Track number of resources updated
-    num_resources_patched = 0
-    num_resources_to_patch = 0
-
-    if num_rgs_to_patch == 0:
-        print_info("No resources to patch")
-        return
-
-    logger.debug(f"Patching {num_rgs_to_patch} resource groups")
-    for rg in resource_groups_to_patch:
-        _num_resources_patched, _num_resources_to_patch = rg.update_resources(
-            group_filter=target_group,
-            name_filter=target_name,
-            type_filter=target_type,
-            dry_run=dry_run,
-            auto_confirm=auto_confirm,
-            force=force,
-        )
-        if _num_resources_patched > 0:
-            num_rgs_patched += 1
-        num_resources_patched += _num_resources_patched
-        num_resources_to_patch += _num_resources_to_patch
-        logger.debug(f"Patched {num_resources_patched} resources in {num_rgs_patched} resource groups")
-
-    if dry_run:
-        return
-
-    if num_resources_patched == 0:
-        return
-
-    print_heading(f"\n--**-- ResourceGroups patched: {num_rgs_patched}/{num_rgs_to_patch}\n")
-    if num_resources_patched != num_resources_to_patch:
-        logger.error("Some resources failed to patch, please check logs")

agno/cli/settings.py

@@ -1,96 +0,0 @@
-from __future__ import annotations
-
-from importlib import metadata
-from pathlib import Path
-
-from pydantic import Field, field_validator
-from pydantic_core.core_schema import ValidationInfo
-from pydantic_settings import BaseSettings, SettingsConfigDict
-
-from agno.utils.log import logger
-
-AGNO_CLI_CONFIG_DIR: Path = Path.home().resolve().joinpath(".config").joinpath("ag")
-
-
-class AgnoCliSettings(BaseSettings):
-    app_name: str = "agno"
-    app_version: str = metadata.version("agno")
-
-    tmp_token_path: Path = AGNO_CLI_CONFIG_DIR.joinpath("tmp_token")
-    config_file_path: Path = AGNO_CLI_CONFIG_DIR.joinpath("config.json")
-    credentials_path: Path = AGNO_CLI_CONFIG_DIR.joinpath("credentials.json")
-    ai_conversations_path: Path = AGNO_CLI_CONFIG_DIR.joinpath("ai_conversations.json")
-    auth_token_cookie: str = "__agno_session"
-    auth_token_header: str = "X-AGNO-AUTH-TOKEN"
-
-    api_runtime: str = "prd"
-    api_enabled: bool = True
-    alpha_features: bool = False
-    api_url: str = Field("https://api.agno.com", validate_default=True)
-    cli_auth_url: str = Field("https://app.agno.com", validate_default=True)
-    signin_url: str = Field("https://app.agno.com/login", validate_default=True)
-    playground_url: str = Field("https://app.agno.com/playground", validate_default=True)
-
-    model_config = SettingsConfigDict(env_prefix="AGNO_")
-
-    @field_validator("api_runtime", mode="before")
-    def validate_runtime_env(cls, v):
-        """Validate api_runtime."""
-
-        valid_api_runtimes = ["dev", "stg", "prd"]
-        if v.lower() not in valid_api_runtimes:
-            raise ValueError(f"Invalid api_runtime: {v}")
-
-        return v.lower()
-
-    @field_validator("cli_auth_url", mode="before")
-    def update_cli_auth_url(cls, v, info: ValidationInfo):
-        api_runtime = info.data["api_runtime"]
-        if api_runtime == "dev":
-            return "http://localhost:3000/cli-auth"
-        elif api_runtime == "stg":
-            return "https://app-stg.agno.com/cli-auth"
-        else:
-            return "https://app.agno.com/cli-auth"
-
-    @field_validator("signin_url", mode="before")
-    def update_signin_url(cls, v, info: ValidationInfo):
-        api_runtime = info.data["api_runtime"]
-        if api_runtime == "dev":
-            return "http://localhost:3000/login"
-        elif api_runtime == "stg":
-            return "https://app-stg.agno.com/login"
-        else:
-            return "https://app.agno.com/login"
-
-    @field_validator("playground_url", mode="before")
-    def update_playground_url(cls, v, info: ValidationInfo):
-        api_runtime = info.data["api_runtime"]
-        if api_runtime == "dev":
-            return "http://localhost:3000/playground"
-        elif api_runtime == "stg":
-            return "https://app-stg.agno.com/playground"
-        else:
-            return "https://app.agno.com/playground"
-
-    @field_validator("api_url", mode="before")
-    def update_api_url(cls, v, info: ValidationInfo):
-        api_runtime = info.data["api_runtime"]
-        if api_runtime == "dev":
-            from os import getenv
-
-            if getenv("AGNO_RUNTIME") == "docker":
-                return "http://host.docker.internal:7070"
-            return "http://localhost:7070"
-        elif api_runtime == "stg":
-            return "https://api-stg.agno.com"
-        else:
-            return "https://api.agno.com"
-
-    def gate_alpha_feature(self):
-        if not self.alpha_features:
-            logger.error("This is an Alpha feature not for general use.\nPlease message the Agno team for access.")
-            exit(1)
-
-
-agno_cli_settings = AgnoCliSettings()