pygeai 0.1.51b3__py3-none-any.whl → 0.6.0b15__py3-none-any.whl

pygeai/cli/commands/evaluation.py

@@ -1,5 +1,4 @@
 import json
-import sys
 
 from pygeai.cli.commands import Command, Option, ArgumentsEnum
 from pygeai.cli.commands.builders import build_help_text
@@ -7,6 +6,7 @@ from pygeai.cli.commands.common import get_boolean_value
 from pygeai.cli.commands.validators import validate_row_structure, validate_system_metric
 from pygeai.cli.texts.help import EVALUATION_HELP_TEXT
 from pygeai.core.common.exceptions import MissingRequirementException, WrongArgumentError
+from pygeai.core.utils.console import Console
 from pygeai.evaluation.dataset.clients import EvaluationDatasetClient
 from pygeai.evaluation.plan.clients import EvaluationPlanClient
 from pygeai.evaluation.result.clients import EvaluationResultClient
@@ -17,7 +17,7 @@ def show_help():
     Displays help text in stdout
     """
     help_text = build_help_text(evaluation_commands, EVALUATION_HELP_TEXT)
-    sys.stdout.write(help_text)
+    Console.write_stdout(help_text)
 
 
 """
@@ -28,7 +28,7 @@ def show_help():
 def list_datasets():
     client = EvaluationDatasetClient()
     result = client.list_datasets()
-    sys.stdout.write(f"Feedback detail: \n{result}\n")
+    Console.write_stdout(f"Feedback detail: \n{result}")
 
 
 def create_dataset(option_list: list):
@@ -91,7 +91,7 @@ def create_dataset(option_list: list):
             rows=rows
         )
 
-    sys.stdout.write(f"New dataset detail: \n{result}\n")
+    Console.write_stdout(f"New dataset detail: \n{result}")
 
 
 create_dataset_options = [
@@ -149,7 +149,7 @@ def get_dataset(option_list: list):
     result = client.get_dataset(
         dataset_id=dataset_id
     )
-    sys.stdout.write(f"Dataset detail: \n{result}\n")
+    Console.write_stdout(f"Dataset detail: \n{result}")
 
 
 get_dataset_options = [
@@ -217,7 +217,7 @@ def update_dataset(option_list: list):
         dataset_active=dataset_active,
         rows=rows,
     )
-    sys.stdout.write(f"Updated dataset detail: \n{result}\n")
+    Console.write_stdout(f"Updated dataset detail: \n{result}")
 
 
 update_dataset_options = [
@@ -274,7 +274,7 @@ def delete_dataset(option_list: list):
     result = client.delete_dataset(
         dataset_id=dataset_id
     )
-    sys.stdout.write(f"Deleted dataset detail: \n{result}\n")
+    Console.write_stdout(f"Deleted dataset detail: \n{result}")
 
 
 delete_dataset_options = [
@@ -323,7 +323,7 @@ def create_dataset_row(option_list: list):
         dataset_id=dataset_id,
         row=row
     )
-    sys.stdout.write(f"Deleted dataset detail: \n{result}\n")
+    Console.write_stdout(f"Deleted dataset detail: \n{result}")
 
 
 create_dataset_row_options = [
@@ -356,7 +356,7 @@ def list_dataset_rows(option_list: list):
     result = client.list_dataset_rows(
         dataset_id=dataset_id
     )
-    sys.stdout.write(f"Dataset rows: \n{result}\n")
+    Console.write_stdout(f"Dataset rows: \n{result}")
 
 
 list_dataset_rows_options = [
@@ -387,7 +387,7 @@ def get_dataset_row(option_list: list):
         dataset_id=dataset_id,
         dataset_row_id=row_id
     )
-    sys.stdout.write(f"Row detail: \n{result}\n")
+    Console.write_stdout(f"Row detail: \n{result}")
 
 
 get_dataset_row_options = [
@@ -446,7 +446,7 @@ def update_dataset_row(option_list: list):
         dataset_row_id=row_id,
         row=row
     )
-    sys.stdout.write(f"Row detail: \n{result}\n")
+    Console.write_stdout(f"Row detail: \n{result}")
 
 
 update_dataset_row_options = [
@@ -489,7 +489,7 @@ def delete_dataset_row(option_list: list):
         dataset_id=dataset_id,
         dataset_row_id=row_id
     )
-    sys.stdout.write(f"Deleted row detail: \n{result}\n")
+    Console.write_stdout(f"Deleted row detail: \n{result}")
 
 
 delete_dataset_row_options = [
@@ -538,7 +538,7 @@ def create_dataset_row_expected_source(option_list: list):
         expected_source_value=value,
         expected_source_extension=extension,
     )
-    sys.stdout.write(f"Extended source detail: \n{result}\n")
+    Console.write_stdout(f"Extended source detail: \n{result}")
 
 
 create_dataset_row_expected_source_options = [
@@ -593,7 +593,7 @@ def list_dataset_row_expected_sources(option_list: list):
         dataset_id=dataset_id,
         dataset_row_id=row_id
     )
-    sys.stdout.write(f"Expected sources: \n{result}\n")
+    Console.write_stdout(f"Expected sources: \n{result}")
 
 
 list_dataset_row_expected_sources_options = [
@@ -634,7 +634,7 @@ def get_dataset_row_expected_source(option_list: list):
         dataset_row_id=row_id,
         expected_source_id=expected_source_id
     )
-    sys.stdout.write(f"Expected source detail: \n{result}\n")
+    Console.write_stdout(f"Expected source detail: \n{result}")
 
 
 get_dataset_row_expected_source_options = [
@@ -694,7 +694,7 @@ def update_dataset_row_expected_source(option_list: list):
         expected_source_extension=extension,
 
     )
-    sys.stdout.write(f"Updated expected source detail: \n{result}\n")
+    Console.write_stdout(f"Updated expected source detail: \n{result}")
 
 
 update_dataset_row_expected_source_options = [
@@ -759,7 +759,7 @@ def delete_dataset_row_expected_source(option_list: list):
         dataset_row_id=row_id,
         expected_source_id=expected_source_id
     )
-    sys.stdout.write(f"Deleted expected source detail: \n{result}\n")
+    Console.write_stdout(f"Deleted expected source detail: \n{result}")
 
 
 delete_dataset_row_expected_source_options = [
@@ -818,7 +818,7 @@ def create_dataset_row_filter_variable(option_list: list):
         filter_variable_value=value,
         filter_variable_operator=operator,
     )
-    sys.stdout.write(f"Filter variable detail: \n{result}\n")
+    Console.write_stdout(f"Filter variable detail: \n{result}")
 
 
 create_dataset_row_filter_variable_options = [
@@ -879,7 +879,7 @@ def list_dataset_row_filter_variables(option_list: list):
         dataset_id=dataset_id,
         dataset_row_id=row_id
     )
-    sys.stdout.write(f"Filter variables: \n{result}\n")
+    Console.write_stdout(f"Filter variables: \n{result}")
 
 
 list_dataset_row_filter_variables_options = [
@@ -920,7 +920,7 @@ def get_dataset_row_filter_variable(option_list: list):
         dataset_row_id=row_id,
         filter_variable_id=filter_variable_id
     )
-    sys.stdout.write(f"Filter variable detail: \n{result}\n")
+    Console.write_stdout(f"Filter variable detail: \n{result}")
 
 
 get_dataset_row_filter_variable_options = [
@@ -983,7 +983,7 @@ def update_dataset_row_filter_variable(option_list: list):
         filter_variable_value=value,
         filter_variable_operator=operator,
     )
-    sys.stdout.write(f"Updated filter variable detail: \n{result}\n")
+    Console.write_stdout(f"Updated filter variable detail: \n{result}")
 
 
 update_dataset_row_filter_variable_options = [
@@ -1054,7 +1054,7 @@ def delete_dataset_row_filter_variable(option_list: list):
         dataset_row_id=row_id,
         filter_variable_id=filter_variable_id
     )
-    sys.stdout.write(f"Deleted filter variable detail: \n{result}\n")
+    Console.write_stdout(f"Deleted filter variable detail: \n{result}")
 
 
 delete_dataset_row_filter_variable_options = [
@@ -1097,7 +1097,7 @@ def update_dataset_rows_file(option_list: list):
         dataset_id=dataset_id,
         file_path=dataset_rows_file
     )
-    sys.stdout.write(f"Dataset rows file detail: \n{result}\n")
+    Console.write_stdout(f"Dataset rows file detail: \n{result}")
 
 
 update_dataset_rows_file_options = [
@@ -1124,7 +1124,7 @@ update_dataset_rows_file_options = [
 def list_evaluation_plans():
     client = EvaluationPlanClient()
     result = client.list_evaluation_plans()
-    sys.stdout.write(f"Evaluation plans: \n{result}\n")
+    Console.write_stdout(f"Evaluation plans: \n{result}")
 
 
 def create_evaluation_plan(option_list: list):
@@ -1184,7 +1184,7 @@ def create_evaluation_plan(option_list: list):
         dataset_id=dataset_id,
         system_metrics=system_metrics,
     )
-    sys.stdout.write(f"New evaluation plan detail: \n{result}\n")
+    Console.write_stdout(f"New evaluation plan detail: \n{result}")
 
 
 create_evaluation_plan_options = [
@@ -1254,7 +1254,7 @@ def get_evaluation_plan(option_list: list):
     result = client.get_evaluation_plan(
         evaluation_plan_id=evaluation_plan_id
     )
-    sys.stdout.write(f"Evaluation plan detail: \n{result}\n")
+    Console.write_stdout(f"Evaluation plan detail: \n{result}")
 
 
 get_evaluation_plan_options = [
@@ -1327,7 +1327,7 @@ def update_evaluation_plan(option_list: list):
         dataset_id=dataset_id,
         system_metrics=system_metrics,
     )
-    sys.stdout.write(f"Updated evaluation plan: \n{result}\n")
+    Console.write_stdout(f"Updated evaluation plan: \n{result}")
 
 
 update_evaluation_plan_options = [
@@ -1403,7 +1403,7 @@ def delete_evaluation_plan(option_list: list):
     result = client.delete_evaluation_plan(
         evaluation_plan_id=evaluation_plan_id
     )
-    sys.stdout.write(f"Deleted evaluation plan: \n{result}\n")
+    Console.write_stdout(f"Deleted evaluation plan: \n{result}")
 
 
 delete_evaluation_plan_options = [
@@ -1430,7 +1430,7 @@ def list_evaluation_plan_system_metrics(option_list: list):
     result = client.list_evaluation_plan_system_metrics(
         evaluation_plan_id=evaluation_plan_id
     )
-    sys.stdout.write(f"Evaluation plan's system metrics: \n{result}\n")
+    Console.write_stdout(f"Evaluation plan's system metrics: \n{result}")
 
 
 list_evaluation_plan_system_metrics_options = [
@@ -1466,7 +1466,7 @@ def add_evaluation_plan_system_metric(option_list: list):
         system_metric_weight=system_metric_weight
 
     )
-    sys.stdout.write(f"Evaluation plan's system metrics: \n{result}\n")
+    Console.write_stdout(f"Evaluation plan's system metrics: \n{result}")
 
 
 add_evaluation_plan_system_metric_options = [
@@ -1509,7 +1509,7 @@ def get_evaluation_plan_system_metric(option_list: list):
         evaluation_plan_id=evaluation_plan_id,
         system_metric_id=system_metric_id
     )
-    sys.stdout.write(f"Evaluation plan's system metric: \n{result}\n")
+    Console.write_stdout(f"Evaluation plan's system metric: \n{result}")
 
 
 get_evaluation_plan_system_metric_options = [
@@ -1551,7 +1551,7 @@ def update_evaluation_plan_system_metric(option_list: list):
         system_metric_weight=system_metric_weight
 
     )
-    sys.stdout.write(f"Evaluation plan's system metrics: \n{result}\n")
+    Console.write_stdout(f"Evaluation plan's system metrics: \n{result}")
 
 
 update_evaluation_plan_system_metric_options = [
@@ -1594,7 +1594,7 @@ def delete_evaluation_plan_system_metric(option_list: list):
         evaluation_plan_id=evaluation_plan_id,
         system_metric_id=system_metric_id
     )
-    sys.stdout.write(f"Evaluation plan's system metric: \n{result}\n")
+    Console.write_stdout(f"Evaluation plan's system metric: \n{result}")
 
 
 delete_evaluation_plan_system_metric_options = [
@@ -1616,7 +1616,7 @@ delete_evaluation_plan_system_metric_options = [
 def list_system_metrics():
     client = EvaluationPlanClient()
     result = client.list_system_metrics()
-    sys.stdout.write(f"Available system metrics: \n{result}\n")
+    Console.write_stdout(f"Available system metrics: \n{result}")
 
 
 def get_system_metric(option_list: list):
@@ -1633,7 +1633,7 @@ def get_system_metric(option_list: list):
     result = client.get_system_metric(
         system_metric_id=system_metric_id
     )
-    sys.stdout.write(f"System metric: \n{result}\n")
+    Console.write_stdout(f"System metric: \n{result}")
 
 
 get_system_metric_options = [
@@ -1660,7 +1660,7 @@ def execute_evaluation_plan(option_list: list):
     result = client.execute_evaluation_plan(
         evaluation_plan_id=evaluation_plan_id,
     )
-    sys.stdout.write(f"Evaluation plan execution: \n{result}\n")
+    Console.write_stdout(f"Evaluation plan execution: \n{result}")
 
 
 execute_evaluation_plan_options = [
@@ -1691,7 +1691,7 @@ def list_evaluation_results(option_list: list):
     result = client.list_evaluation_results(
         evaluation_plan_id=evaluation_plan_id,
     )
-    sys.stdout.write(f"Evaluation results: \n{result}\n")
+    Console.write_stdout(f"Evaluation results: \n{result}")
 
 
 list_evaluation_results_options = [
@@ -1718,7 +1718,7 @@ def get_evaluation_result(option_list: list):
     result = client.get_evaluation_result(
         evaluation_result_id=evaluation_result_id,
     )
-    sys.stdout.write(f"Evaluation result: \n{result}\n")
+    Console.write_stdout(f"Evaluation result: \n{result}")
 
 
 get_evaluation_result_options = [

pygeai/cli/commands/feedback.py

@@ -1,10 +1,9 @@
-import sys
-
 from pygeai.cli.commands import Command, Option, ArgumentsEnum
 from pygeai.cli.commands.builders import build_help_text
 from pygeai.cli.texts.help import FEEDBACK_HELP_TEXT
 from pygeai.core.common.exceptions import MissingRequirementException, WrongArgumentError
 from pygeai.core.feedback.clients import FeedbackClient
+from pygeai.core.utils.console import Console
 
 
 def show_help():
@@ -12,7 +11,7 @@ def show_help():
     Displays help text in stdout
     """
     help_text = build_help_text(feedback_commands, FEEDBACK_HELP_TEXT)
-    sys.stdout.write(help_text)
+    Console.write_stdout(help_text)
 
 
 def send_feedback(option_list: list):
@@ -41,7 +40,7 @@ def send_feedback(option_list: list):
         answer_score=answer_score,
         comments=comments
     )
-    sys.stdout.write(f"Feedback detail: \n{result}\n")
+    Console.write_stdout(f"Feedback detail: \n{result}")
 
 
 send_feedback_options = [

pygeai/cli/commands/files.py

@@ -1,10 +1,9 @@
-import sys
-
 from pygeai.cli.commands import Command, ArgumentsEnum, Option
 from pygeai.cli.commands.builders import build_help_text
 from pygeai.cli.texts.help import FILES_HELP_TEXT
 from pygeai.core.common.exceptions import MissingRequirementException
 from pygeai.core.files.clients import FileClient
+from pygeai.core.utils.console import Console
 
 
 def show_help():
@@ -12,7 +11,7 @@ def show_help():
     Displays help text in stdout
     """
     help_text = build_help_text(files_commands, FILES_HELP_TEXT)
-    sys.stdout.write(help_text)
+    Console.write_stdout(help_text)
 
 
 def upload_file(option_list: list):
@@ -45,7 +44,7 @@ def upload_file(option_list: list):
         file_name=file_name,
         folder=folder
     )
-    sys.stdout.write(f"Uploaded file: \n{result}\n")
+    Console.write_stdout(f"Uploaded file: \n{result}")
 
 
 upload_file_options = [
@@ -103,7 +102,7 @@ def get_file(option_list: list):
         project=project,
         file_id=file_id,
     )
-    sys.stdout.write(f"File: \n{result}\n")
+    Console.write_stdout(f"File: \n{result}")
 
 
 get_file_options = [
@@ -149,7 +148,7 @@ def delete_file(option_list: list):
         organization=organization,
         project=project
     )
-    sys.stdout.write(f"Deleted file: \n{result}\n")
+    Console.write_stdout(f"Deleted file: \n{result}")
 
 
 delete_file_options = [
@@ -195,7 +194,7 @@ def get_file_content(option_list: list):
         organization=organization,
         project=project
     )
-    sys.stdout.write(f"File content: \n{result}\n")
+    Console.write_stdout(f"File content: \n{result}")
 
 
 get_file_content_options = [
@@ -237,7 +236,7 @@ def get_file_list(option_list: list):
         organization=organization,
         project=project
     )
-    sys.stdout.write(f"Files list: \n{result}\n")
+    Console.write_stdout(f"Files list: \n{result}")
 
 
 get_file_list_options = [

pygeai/cli/commands/gam.py

@@ -1,11 +1,8 @@
-import json
-import sys
-
 from pygeai.cli.commands import Command, Option, ArgumentsEnum
 from pygeai.cli.commands.builders import build_help_text
-from pygeai.cli.commands.common import get_messages, get_boolean_value, get_penalty_float_value
 from pygeai.cli.texts.help import GAM_HELP_TEXT
 from pygeai.core.common.exceptions import MissingRequirementException, WrongArgumentError
+from pygeai.core.utils.console import Console
 from pygeai.gam.clients import GAMClient
 
 
@@ -14,7 +11,74 @@ def show_help():
     Displays help text in stdout
     """
     help_text = build_help_text(gam_commands, GAM_HELP_TEXT)
-    sys.stdout.write(help_text)
+    Console.write_stdout(help_text)
+
+
+def generate_signin_url(option_list: list):
+    client_id = None
+    redirect_uri = None
+    scope = "gam_user_data"
+    state = None
+    response_type = "code"
+
+    for option_flag, option_arg in option_list:
+        if option_flag.name == "client_id":
+            client_id = option_arg
+        if option_flag.name == "redirect_uri":
+            redirect_uri = option_arg
+        if option_flag.name == "scope":
+            scope = option_arg
+        if option_flag.name == "state":
+            state = option_arg
+        if option_flag.name == "response_type":
+            response_type = option_arg
+
+    if not all([client_id, redirect_uri, state]):
+        raise MissingRequirementException("client_id, redirect_uri, and state are required for generating signin URL")
+
+    client = GAMClient()
+    result = client.generate_signing_url(
+        client_id=client_id,
+        redirect_uri=redirect_uri,
+        scope=scope,
+        state=state,
+        response_type=response_type
+    )
+    Console.write_stdout(f"GAM Signin URL: \n{result}")
+
+
+generate_signin_url_options = [
+    Option(
+        "client_id",
+        ["--client-id", "--cid"],
+        "Application Client ID.",
+        True
+    ),
+    Option(
+        "redirect_uri",
+        ["--redirect-uri", "--ru"],
+        "Callback URL configured in the application (must match the one in GAM).",
+        True
+    ),
+    Option(
+        "scope",
+        ["--scope", "-s"],
+        'Scope of the user account to access. Default: "gam_user_data"',
+        True
+    ),
+    Option(
+        "state",
+        ["--state", "--st"],
+        "Random string to store the status before the request.",
+        True
+    ),
+    Option(
+        "response_type",
+        ["--response-type", "--rt"],
+        'Response type for the signin request. Default: "code"',
+        True
+    )
+]
 
 
 def get_access_token(option_list: list):
@@ -67,7 +131,7 @@ def get_access_token(option_list: list):
         repository=repository,
         request_token_type=request_token_type,
     )
-    sys.stdout.write(f"GAM Access Token: \n{result}\n")
+    Console.write_stdout(f"GAM Access Token: \n{result}")
 
 
 get_access_token_options = [
@@ -98,7 +162,9 @@ get_access_token_options = [
     Option(
         "scope",
         ["--scope", "-s"],
-        'Scope of the user account you want to access. gam_user_data+gam_user_roles. Default: "gam_user_data"',
+        'Scope of the user account you want to access. gam_user_data+gam_user_roles. '
+        'Valid scopes: gam_user_data, gam_user_additional_data, gam_user_roles, session_initial_prop, '
+        'session_application_data, fullcontrol. Default: "gam_user_data"',
         True
     ),
     Option(
@@ -148,7 +214,7 @@ def get_user_info(option_list: list):
     result = client.get_user_info(
         access_token=access_token,
     )
-    sys.stdout.write(f"GAM User info: \n{result}\n")
+    Console.write_stdout(f"GAM User info: \n{result}")
 
 
 get_user_info_options = [
@@ -187,7 +253,7 @@ def refresh_access_token(option_list: list):
         grant_type=grant_type,
         refresh_token=refresh_token,
     )
-    sys.stdout.write(f"GAM Access Token: \n{result}\n")
+    Console.write_stdout(f"GAM Access Token: \n{result}")
 
 
 refresh_access_token_options = [
@@ -221,7 +287,7 @@ refresh_access_token_options = [
 def get_authentication_types():
     client = GAMClient()
     result = client.get_authentication_types()
-    sys.stdout.write(f"GAM Authentication Types: \n{result}\n")
+    Console.write_stdout(f"GAM Authentication Types: \n{result}")
 
 
 gam_commands = [
@@ -261,6 +327,15 @@ gam_commands = [
         [],
         refresh_access_token_options
     ),
+    Command(
+        "generate_signin_url",
+        ["generate-signin-url", "gsu"],
+        "Generate signin URL for browser-based OAuth flow",
+        generate_signin_url,
+        ArgumentsEnum.REQUIRED,
+        [],
+        generate_signin_url_options
+    )
     # Command(
     #     "get_authentication_types",
     #     ["get-authentication-types", "get-auth-types"],