pygeai 0.1.6__py3-none-any.whl → 0.6.0b15__py3-none-any.whl

pygeai/cli/commands/assistant.py

@@ -1,5 +1,4 @@
 import json
-import sys
 
 from pygeai.assistant.clients import AssistantClient
 from pygeai.cli.commands import Command, Option, ArgumentsEnum
@@ -8,14 +7,15 @@ from pygeai.cli.commands.common import get_llm_settings, get_welcome_data, get_m
     get_welcome_data_example_prompt
 from pygeai.cli.texts.help import ASSISTANT_HELP_TEXT
 from pygeai.core.common.exceptions import MissingRequirementException, WrongArgumentError
+from pygeai.core.utils.console import Console
 
 
-def show_assistant_help():
+def show_help():
     """
     Displays help text in stdout
     """
     help_text = build_help_text(assistant_commands, ASSISTANT_HELP_TEXT)
-    sys.stdout.write(help_text)
+    Console.write_stdout(help_text)
 
 
 def get_assistant_detail(option_list: list):
@@ -32,7 +32,7 @@ def get_assistant_detail(option_list: list):
 
     client = AssistantClient()
     result = client.get_assistant_data(assistant_id, detail)
-    sys.stdout.write(f"Assistant detail: \n{result}\n")
+    Console.write_stdout(f"Assistant detail: \n{result}")
 
 
 assistant_detail_options = [
@@ -121,7 +121,7 @@ def create_assistant(option_list: list):
         llm_settings=llm_settings,
         welcome_data=welcome_data
     )
-    sys.stdout.write(f"New assistant detail: \n{result}\n")
+    Console.write_stdout(f"New assistant detail: \n{result}")
 
 
 create_assistant_options = [
@@ -287,7 +287,7 @@ def update_assistant(option_list: list):
         llm_settings=llm_settings,
         welcome_data=welcome_data
     )
-    sys.stdout.write(f"Updated assistant detail: \n{result}\n")
+    Console.write_stdout(f"Updated assistant detail: \n{result}")
 
 
 update_assistant_options = [
@@ -401,7 +401,7 @@ def delete_assistant(option_list: list):
     result = client.delete_assistant(
         assistant_id=assistant_id,
     )
-    sys.stdout.write(f"Deleted assistant: \n{result}\n")
+    Console.write_stdout(f"Deleted assistant: \n{result}")
 
 
 delete_assistant_options = [
@@ -454,7 +454,7 @@ def send_chat_request(option_list: list):
         revision=revision,
         revision_name=revision_name
     )
-    sys.stdout.write(f"Chat request response: \n{result}\n")
+    Console.write_stdout(f"Chat request response: \n{result}")
 
 
 send_chat_request_options = [
@@ -503,7 +503,7 @@ def get_request_status(option_list: list):
 
     client = AssistantClient()
     result = client.get_request_status(request_id)
-    sys.stdout.write(f"Request status: \n{result}\n")
+    Console.write_stdout(f"Request status: \n{result}")
 
 
 request_status_options = [
@@ -527,7 +527,7 @@ def cancel_request(option_list: list):
 
     client = AssistantClient()
     result = client.cancel_request(request_id)
-    sys.stdout.write(f"Cancel request detail: \n{result}\n")
+    Console.write_stdout(f"Cancel request detail: \n{result}")
 
 
 cancel_request_options = [
@@ -545,7 +545,7 @@ assistant_commands = [
         "help",
         ["help", "h"],
         "Display help text",
-        show_assistant_help,
+        show_help,
         ArgumentsEnum.NOT_AVAILABLE,
         [],
         []

pygeai/cli/commands/auth.py

@@ -0,0 +1,299 @@
+from pygeai.auth.clients import AuthClient
+from pygeai.cli.commands import Command, Option, ArgumentsEnum
+from pygeai.cli.commands.builders import build_help_text
+from pygeai.cli.texts.help import AUTH_HELP_TEXT
+from pygeai.core.common.exceptions import MissingRequirementException, WrongArgumentError
+from pygeai.core.utils.console import Console
+
+
+def show_help():
+    """
+    Displays help text in stdout
+    """
+    help_text = build_help_text(auth_commands, AUTH_HELP_TEXT)
+    Console.write_stdout(help_text)
+
+
+def get_oauth2_access_token(option_list: list):
+    client_id = None
+    username = None
+    password = None
+    scope = "gam_user_data gam_user_roles"
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "client_id":
+            client_id = option_arg
+        if option_flag.name == "username":
+            username = option_arg
+        if option_flag.name == "password":
+            password = option_arg
+        if option_flag.name == "scope":
+            scope = option_arg
+
+    if not (client_id and username and password):
+        raise MissingRequirementException("Cannot obtain Oauth2 access token without client_id, username and password")
+
+    client = AuthClient()
+    result = client.get_oauth2_access_token(
+        client_id=client_id,
+        username=username,
+        password=password,
+        scope=scope
+    )
+    Console.write_stdout(f"Authorized projects detail: \n{result}")
+
+
+get_oauth2_access_token_options = [
+    Option(
+        "client_id",
+        ["--client-id", "--cid"],
+        "The client identifier provided by Globant.",
+        True
+    ),
+    Option(
+        "username",
+        ["--username", "-u"],
+        "Username for authentication.",
+        True
+    ),
+    Option(
+        "password",
+        ["--password", "-p"],
+        "Password for authentication.",
+        True
+    ),
+    Option(
+        "scope",
+        ["--scope", "-s"],
+        "Space-separated list of requested scopes. (Optional)",
+        True
+    ),
+
+]
+
+
+def get_user_profile_information(option_list: list):
+    access_token = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "access_token":
+            access_token = option_arg
+
+    client = AuthClient()
+    result = client.get_user_profile_information(access_token=access_token)
+    Console.write_stdout(f"User profile information: \n{result}")
+
+
+get_user_profile_information_options = [
+    Option(
+        "access_token",
+        ["--access-token", "--token"],
+        "Token obtained with the --get-access-token option",
+        True
+    ),
+]
+
+
+def create_project_api_token(option_list: list):
+    project_id = None
+    name = None
+    description = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "project_id":
+            project_id = option_arg
+        if option_flag.name == "name":
+            name = option_arg
+        if option_flag.name == "description":
+            description = option_arg
+
+    if not (project_id and name):
+        raise MissingRequirementException("Cannot create project API token without project-id and name")
+
+    client = AuthClient()
+    result = client.create_project_api_token(
+        project_id=project_id,
+        name=name,
+        description=description
+    )
+    Console.write_stdout(f"Project API token created: \n{result}")
+
+
+create_project_api_token_options = [
+    Option(
+        "project_id",
+        ["--project-id", "--pid"],
+        "The project identifier (required).",
+        True
+    ),
+    Option(
+        "name",
+        ["--name", "-n"],
+        "The name of the API token (required).",
+        True
+    ),
+    Option(
+        "description",
+        ["--description", "-d"],
+        "A description of the API token (optional).",
+        True
+    ),
+]
+
+
+def delete_project_api_token(option_list: list):
+    api_token_id = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "api_token_id":
+            api_token_id = option_arg
+
+    if not api_token_id:
+        raise MissingRequirementException("Cannot delete project API token without api-token-id")
+
+    client = AuthClient()
+    result = client.delete_project_api_token(api_token_id=api_token_id)
+    Console.write_stdout(f"Project API token deleted: \n{result}")
+
+
+delete_project_api_token_options = [
+    Option(
+        "api_token_id",
+        ["--api-token-id", "--tid"],
+        "The unique identifier of the API token to delete (required).",
+        True
+    ),
+]
+
+
+def update_project_api_token(option_list: list):
+    api_token_id = None
+    description = None
+    status = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "api_token_id":
+            api_token_id = option_arg
+        if option_flag.name == "description":
+            description = option_arg
+        if option_flag.name == "status":
+            status = option_arg
+
+    if not api_token_id:
+        raise MissingRequirementException("Cannot update project API token without api-token-id")
+
+    client = AuthClient()
+    result = client.update_project_api_token(
+        api_token_id=api_token_id,
+        description=description,
+        status=status
+    )
+    Console.write_stdout(f"Project API token updated: \n{result}")
+
+
+update_project_api_token_options = [
+    Option(
+        "api_token_id",
+        ["--api-token-id", "--tid"],
+        "The unique identifier of the API token to update (required).",
+        True
+    ),
+    Option(
+        "description",
+        ["--description", "-d"],
+        "A new description for the API token (optional).",
+        True
+    ),
+    Option(
+        "status",
+        ["--status"],
+        "The new status for the API token: 'active' or 'blocked' (optional).",
+        True
+    ),
+]
+
+
+def get_project_api_token(option_list: list):
+    api_token_id = None
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "api_token_id":
+            api_token_id = option_arg
+
+    if not api_token_id:
+        raise MissingRequirementException("Cannot get project API token without api-token-id")
+
+    client = AuthClient()
+    result = client.get_project_api_token(api_token_id=api_token_id)
+    Console.write_stdout(f"Project API token details: \n{result}")
+
+
+get_project_api_token_options = [
+    Option(
+        "api_token_id",
+        ["--api-token-id", "--tid"],
+        "The unique identifier of the API token (required).",
+        True
+    ),
+]
+
+
+auth_commands = [
+    Command(
+        "help",
+        ["help", "h"],
+        "Display help text",
+        show_help,
+        ArgumentsEnum.NOT_AVAILABLE,
+        [],
+        []
+    ),
+    Command(
+        "get_oauth2_access_token",
+        ["get-access-token", "gat"],
+        "Get Oauth acess token for Globant Enterprise AI instance",
+        get_oauth2_access_token,
+        ArgumentsEnum.REQUIRED,
+        [],
+        get_oauth2_access_token_options
+    ),
+    Command(
+        "get_user_profile_information",
+        ["get-user-information", "get-user-info", "gui"],
+        "Retrieve user profile information",
+        get_user_profile_information,
+        ArgumentsEnum.REQUIRED,
+        [],
+        get_user_profile_information_options
+    ),
+    Command(
+        "create_project_api_token",
+        ["create-project-api-token", "create-api-token", "cat"],
+        "Create a new API token for a project",
+        create_project_api_token,
+        ArgumentsEnum.REQUIRED,
+        [],
+        create_project_api_token_options
+    ),
+    Command(
+        "delete_project_api_token",
+        ["delete-project-api-token", "delete-api-token", "dat"],
+        "Revoke an API token",
+        delete_project_api_token,
+        ArgumentsEnum.REQUIRED,
+        [],
+        delete_project_api_token_options
+    ),
+    Command(
+        "update_project_api_token",
+        ["update-project-api-token", "update-api-token", "uat"],
+        "Update an existing API token",
+        update_project_api_token,
+        ArgumentsEnum.REQUIRED,
+        [],
+        update_project_api_token_options
+    ),
+    Command(
+        "get_project_api_token",
+        ["get-project-api-token", "get-api-token", "gat"],
+        "Get data of a specific project API token",
+        get_project_api_token,
+        ArgumentsEnum.REQUIRED,
+        [],
+        get_project_api_token_options
+    ),
+]

pygeai/cli/commands/base.py

@@ -1,15 +1,31 @@
-import sys
-
 from pygeai.cli.commands import ArgumentsEnum, Command, Option
 from pygeai.cli.commands.admin import admin_commands
+from pygeai.cli.commands.analytics import analytics_commands
 from pygeai.cli.commands.assistant import assistant_commands
+from pygeai.cli.commands.auth import auth_commands
 from pygeai.cli.commands.builders import build_help_text
 from pygeai.cli.commands.chat import chat_commands
 from pygeai.cli.commands.configuration import configure, configuration_options
+from pygeai.cli.commands.docs import docs_commands
+from pygeai.cli.commands.embeddings import embeddings_commands
+from pygeai.cli.commands.evaluation import evaluation_commands
+from pygeai.cli.commands.feedback import feedback_commands
+from pygeai.cli.commands.files import files_commands
+from pygeai.cli.commands.gam import gam_commands
+from pygeai.cli.commands.lab.spec import spec_commands
+from pygeai.cli.commands.llm import llm_commands
+from pygeai.cli.commands.migrate import migrate_commands
 from pygeai.cli.commands.organization import organization_commands
 from pygeai.cli.commands.rag import rag_commands
+from pygeai.cli.commands.rerank import rerank_commands
+from pygeai.cli.commands.lab.ai_lab import ai_lab_commands
+from pygeai.cli.commands.secrets import secrets_commands
+from pygeai.cli.commands.usage_limits import usage_limit_commands
+from pygeai.cli.commands.version import check_new_version
 from pygeai.cli.texts.help import HELP_TEXT
-from pygeai.cli import __version__ as cli_version
+from pygeai import __version__ as cli_version
+from pygeai.core.utils.console import Console
+from pygeai.health.clients import HealthClient
 
 
 def show_help():
@@ -17,18 +33,35 @@ def show_help():
     Displays help text in stdout
     """
     help_text = build_help_text(base_commands, HELP_TEXT)
-    sys.stdout.write(help_text)
+    Console.write_stdout(help_text)
 
 
 def show_version():
     """
     Displays version in stdout
     """
-    sys.stdout.write(
-        f" - Globant Enterprise AI: GEAI cli utility. Version: {cli_version}\n"
+    Console.write_stdout(
+        f" - Globant Enterprise AI: GEAI cli utility. Version: {cli_version}"
     )
 
 
+def check_for_updates():
+    """
+    Checks if there are updates available
+    """
+    package_name = 'pygeai'
+    version_status = check_new_version(package_name)
+    Console.write_stdout(f"{version_status}")
+
+
+def check_api_status():
+    """
+    Checks API status
+    """
+    api_status = HealthClient().check_api_status()
+    Console.write_stdout(f"API Status: {api_status}")
+
+
 """
 Commands that have available subcommands should have action None, so the parser knows that it shouldn't
 run any action but instead send it to process again to identify subcommand.
@@ -53,6 +86,24 @@ base_commands = [
         [],
         []
     ),
+    Command(
+        "check_updates",
+        ["check-updates", "cu"],
+        "Search for available updates",
+        check_for_updates,
+        ArgumentsEnum.NOT_AVAILABLE,
+        [],
+        []
+    ),
+    Command(
+        "check_status",
+        ["status", "s"],
+        "Check API status for Globant Enterprise AI instance",
+        check_api_status,
+        ArgumentsEnum.NOT_AVAILABLE,
+        [],
+        []
+    ),
     Command(
         "configure",
         ["configure", "config", "c"],
@@ -71,6 +122,15 @@ base_commands = [
         organization_commands,
         [],
     ),
+    Command(
+        "analytics",
+        ["analytics", "anl"],
+        "Invoke analytics endpoints to retrieve metrics and insights",
+        None,
+        ArgumentsEnum.REQUIRED,
+        analytics_commands,
+        [],
+    ),
     Command(
         "assistant",
         ["assistant", "ast"],
@@ -106,7 +166,135 @@ base_commands = [
         ArgumentsEnum.REQUIRED,
         admin_commands,
         []
-    )
+    ),
+    Command(
+        "auth",
+        ["auth"],
+        "Invoke auth endpoints for token generation",
+        None,
+        ArgumentsEnum.REQUIRED,
+        auth_commands,
+        []
+    ),
+
+    Command(
+        "llm",
+        ["llm"],
+        "Invoke llm endpoints for provider's and model retrieval",
+        None,
+        ArgumentsEnum.REQUIRED,
+        llm_commands,
+        []
+    ),
+    Command(
+        "files",
+        ["files"],
+        "Invoke files endpoints for file handling",
+        None,
+        ArgumentsEnum.REQUIRED,
+        files_commands,
+        []
+    ),
+    Command(
+        "usage_limit",
+        ["usage-limit", "ulim"],
+        "Invoke usage limit endpoints for organization and project",
+        None,
+        ArgumentsEnum.REQUIRED,
+        usage_limit_commands,
+        []
+    ),
+    Command(
+        "embeddings",
+        ["embeddings", "emb"],
+        "Invoke embeddings endpoints",
+        None,
+        ArgumentsEnum.REQUIRED,
+        embeddings_commands,
+        []
+    ),
+    Command(
+        "feedback",
+        ["feedback", "fbk"],
+        "Invoke feedback endpoints",
+        None,
+        ArgumentsEnum.REQUIRED,
+        feedback_commands,
+        []
+    ),
+    Command(
+        "rerank",
+        ["rerank", "rr"],
+        "Invoke rerank endpoints",
+        None,
+        ArgumentsEnum.REQUIRED,
+        rerank_commands,
+        []
+    ),
+    Command(
+        "evaluation",
+        ["evaluation", "eval"],
+        "Invoke evaluation endpoints",
+        None,
+        ArgumentsEnum.REQUIRED,
+        evaluation_commands,
+        []
+    ),
+    Command(
+        "gam",
+        ["gam"],
+        "Invoke GAM authentication endpoints",
+        None,
+        ArgumentsEnum.REQUIRED,
+        gam_commands,
+        []
+    ),
+    Command(
+        "secrets",
+        ["secrets", "sec"],
+        "Handle Globant Enterprise AI secrets",
+        None,
+        ArgumentsEnum.REQUIRED,
+        secrets_commands,
+        []
+    ),
+    Command(
+        "ai_lab",
+        ["ai-lab", "ail"],
+        "Invoke AI Lab endpoints",
+        None,
+        ArgumentsEnum.REQUIRED,
+        ai_lab_commands,
+        []
+    ),
+    Command(
+        "ai_lab_spec",
+        ["ai-lab-spec", "spec"],
+        "Invoke AI Lab endpoints",
+        None,
+        ArgumentsEnum.REQUIRED,
+        spec_commands,
+        []
+    ),
+    Command(
+        "migrate",
+        ["migrate", "mig"],
+        "Invoke migrate procedures",
+        None,
+        ArgumentsEnum.REQUIRED,
+        migrate_commands,
+        []
+    ),
+    #Command(
+    #    "docs",
+    #    ["docs"],
+    #    "View PyGEAI SDK documentation",
+    #    None,
+    #    ArgumentsEnum.NOT_AVAILABLE,
+    #    docs_commands,
+    #    []
+    #),
+
 ]
 
 
@@ -117,4 +305,10 @@ base_options = (
         "Set output file to save the command result",
         True
     ),
+    Option(
+        "verbose",
+        ["--verbose", "-v"],
+        "Enable verbose mode with detailed logging output",
+        False
+    ),
 )