pygeai 0.6.0b13__py3-none-any.whl → 0.6.1__py3-none-any.whl

pygeai/tests/snippets/evaluation/dataset/complete_workflow_example.py

@@ -123,7 +123,7 @@ def main():
         dataset_type="T",
         dataset_active=True
     )
-    print(f"✓ Updated dataset description")
+    print("✓ Updated dataset description")
     
     # Update a row
     updated_row = client.update_dataset_row(
@@ -135,7 +135,7 @@ def main():
             "dataSetRowContextDocument": "DL is particularly effective for image and speech recognition."
         }
     )
-    print(f"✓ Updated row 2")
+    print("✓ Updated row 2")
     
     # Update expected source
     updated_source = client.update_expected_source(
@@ -146,7 +146,7 @@ def main():
         expected_source_value="Updated source content about Machine Learning.",
         expected_source_extension="txt"
     )
-    print(f"✓ Updated expected source")
+    print("✓ Updated expected source")
     
     # Update filter variable
     updated_filter = client.update_filter_variable(
@@ -158,7 +158,7 @@ def main():
         filter_variable_value="intermediate",
         filter_variable_operator="="
     )
-    print(f"✓ Updated filter variable")
+    print("✓ Updated filter variable")
     
     # Step 7: Cleanup
     print("\n[7/7] Cleaning up...")
@@ -166,20 +166,20 @@ def main():
     # Delete filter variables
     client.delete_filter_variable(dataset_id=dataset_id, dataset_row_id=row1_id, filter_variable_id=filter1_id)
     client.delete_filter_variable(dataset_id=dataset_id, dataset_row_id=row1_id, filter_variable_id=filter2_id)
-    print(f"✓ Deleted filter variables")
+    print("✓ Deleted filter variables")
     
     # Delete expected source
     client.delete_expected_source(dataset_id=dataset_id, dataset_row_id=row1_id, expected_source_id=source1_id)
-    print(f"✓ Deleted expected source")
+    print("✓ Deleted expected source")
     
     # Delete rows
     client.delete_dataset_row(dataset_id=dataset_id, dataset_row_id=row1_id)
     client.delete_dataset_row(dataset_id=dataset_id, dataset_row_id=row2_id)
-    print(f"✓ Deleted rows")
+    print("✓ Deleted rows")
     
     # Delete dataset
     client.delete_dataset(dataset_id=dataset_id)
-    print(f"✓ Deleted dataset")
+    print("✓ Deleted dataset")
     
     print("\n" + "="*60)
     print("Workflow completed successfully!")

pygeai/tests/snippets/evaluation/plan/complete_workflow_example.py

@@ -83,7 +83,7 @@ def main():
         system_metric_id="accuracy",
         system_metric_weight=0.8
     )
-    print(f"✓ Updated metric weight")
+    print("✓ Updated metric weight")
     
     # Get specific metric details
     metric_detail = client.get_evaluation_plan_system_metric(
@@ -99,12 +99,12 @@ def main():
         name="Updated Performance Test",
         assistant_revision="2.0"
     )
-    print(f"✓ Updated plan")
+    print("✓ Updated plan")
     
     # Execute the evaluation plan
     print("\n[EXECUTE] Running evaluation plan...")
     execution = client.execute_evaluation_plan(evaluation_plan_id=plan_id)
-    print(f"✓ Execution started")
+    print("✓ Execution started")
     print(f"  Execution ID: {execution.get('executionId')}")
     print(f"  Status: {execution.get('status')}")
     
@@ -116,11 +116,11 @@ def main():
         evaluation_plan_id=plan_id,
         system_metric_id="relevance"
     )
-    print(f"✓ Deleted added metric")
+    print("✓ Deleted added metric")
     
     # Delete the plan
     client.delete_evaluation_plan(evaluation_plan_id=plan_id)
-    print(f"✓ Deleted evaluation plan")
+    print("✓ Deleted evaluation plan")
     
     print("\n" + "="*60)
     print("Workflow completed successfully!")

pygeai/tests/snippets/evaluation/result/complete_workflow_example.py

@@ -98,7 +98,7 @@ def main():
         
         # Show sample row
         if rows:
-            print(f"\n  Sample Row:")
+            print("\n  Sample Row:")
             sample_row = rows[0]
             print(f"    Dataset Row ID: {sample_row.get('dataSetRowId')}")
             print(f"    Status: {sample_row.get('evaluationResultRowStatus')}")
@@ -108,7 +108,7 @@ def main():
             
             output = sample_row.get('evaluationResultRowOutput', '')
             if output:
-                print(f"    Output (first 200 chars):")
+                print("    Output (first 200 chars):")
                 print(f"      {output[:200]}...")
     
     # Step 4: Display aggregated analytics
@@ -122,7 +122,7 @@ def main():
         print(f"Average Cost per Evaluation: ${total_cost/len(results):.4f}")
         print(f"Average Duration per Evaluation: {total_duration/len(results):.0f}ms")
     
-    print(f"\nStatus Distribution:")
+    print("\nStatus Distribution:")
     for status, count in status_counts.items():
         percentage = (count / len(results) * 100) if results else 0
         print(f"  {status}: {count} ({percentage:.1f}%)")

pygeai/tests/snippets/lab/agentic_flow_example_1.py

@@ -5,7 +5,7 @@ from pygeai.lab.models import (
     Tool, ToolParameter,
     Task,
     AgenticProcess, KnowledgeBase, AgenticActivity, ArtifactSignal, UserSignal, Event, SequenceFlow,
-    Variable, ProcessInstance, VariableList
+    ProcessInstance
 )
 
 manager = AILabManager()

pygeai/tests/snippets/lab/agentic_flow_example_2.py

@@ -1,10 +1,9 @@
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import (
-    Agent, AgentData, Prompt, PromptExample, PromptOutput, LlmConfig, Sampling, Model, ModelList,
-    Tool, ToolParameter,
-    Task,
+    Agent, AgentData, Prompt, PromptOutput, LlmConfig, Sampling, Model, ModelList,
+    Tool, Task,
     AgenticProcess, KnowledgeBase, AgenticActivity, ArtifactSignal, UserSignal, Event, SequenceFlow,
-    Variable, ProcessInstance, VariableList
+    ProcessInstance
 )
 
 manager = AILabManager()

pygeai/tests/snippets/lab/agents/create_agent_with_permissions.py

@@ -30,10 +30,10 @@ print(f"Created agent: {result.name}")
 print(f"Agent ID: {result.id}")
 print(f"Sharing scope: {result.sharing_scope}")
 if result.permissions:
-    print(f"\nPermissions:")
+    print("\nPermissions:")
     print(f"  - Allow chat sharing: {result.permissions.allow_chat_sharing}")
     print(f"  - Allow external execution: {result.permissions.allow_external_execution}")
 if result.effective_permissions:
-    print(f"\nEffective Permissions:")
+    print("\nEffective Permissions:")
     print(f"  - Allow chat sharing: {result.effective_permissions.allow_chat_sharing}")
     print(f"  - Allow external execution: {result.effective_permissions.allow_external_execution}")

pygeai/tests/snippets/lab/agents/delete_agent.py

@@ -1,4 +1,3 @@
-from pygeai.core.base.responses import EmptyResponse
 from pygeai.lab.managers import AILabManager
 
 manager = AILabManager()
@@ -9,4 +8,4 @@ result = manager.delete_agent(
     agent_id=agent_id
 )
 
-print(f"Agent deleted successfully")
+print("Agent deleted successfully")

pygeai/tests/snippets/lab/agents/get_agent.py

@@ -1,5 +1,5 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import FilterSettings, Agent
+from pygeai.lab.models import FilterSettings
 
 
 manager = AILabManager()

pygeai/tests/snippets/lab/agents/get_agent_with_new_fields.py

@@ -16,30 +16,30 @@ agent = manager.get_agent(agent_id)
 print(f"Agent: {agent.name} (ID: {agent.id})")
 print(f"Status: {agent.status}")
 print(f"Is Draft: {agent.is_draft}")
-print(f"\n=== New Fields ===")
+print("\n=== New Fields ===")
 
 # Sharing scope
 print(f"\nSharing Scope: {agent.sharing_scope}")
 
 # Permissions
 if agent.permissions:
-    print(f"\nPermissions:")
+    print("\nPermissions:")
     print(f"  - Allow chat sharing: {agent.permissions.allow_chat_sharing}")
     print(f"  - Allow external execution: {agent.permissions.allow_external_execution}")
 else:
-    print(f"\nPermissions: None")
+    print("\nPermissions: None")
 
 # Effective permissions
 if agent.effective_permissions:
-    print(f"\nEffective Permissions:")
+    print("\nEffective Permissions:")
     print(f"  - Allow chat sharing: {agent.effective_permissions.allow_chat_sharing}")
     print(f"  - Allow external execution: {agent.effective_permissions.allow_external_execution}")
 else:
-    print(f"\nEffective Permissions: None")
+    print("\nEffective Permissions: None")
 
 # Agent data properties
 if agent.agent_data:
-    print(f"\nAgent Data:")
+    print("\nAgent Data:")
     print(f"  - Strategy Name: {agent.agent_data.strategy_name}")
     
     if agent.agent_data.properties:
@@ -47,16 +47,16 @@ if agent.agent_data:
         for prop in agent.agent_data.properties:
             print(f"    * {prop.key} ({prop.data_type}): {prop.value}")
     else:
-        print(f"  - Properties: None")
+        print("  - Properties: None")
     
     if agent.agent_data.models:
         print(f"  - Models: {len(agent.agent_data.models)} configured")
     else:
-        print(f"  - Models: None")
+        print("  - Models: None")
     
     if agent.agent_data.resource_pools:
         print(f"  - Resource Pools: {len(agent.agent_data.resource_pools)} configured")
     else:
-        print(f"  - Resource Pools: None")
+        print("  - Resource Pools: None")
 else:
-    print(f"\nAgent Data: None")
+    print("\nAgent Data: None")

pygeai/tests/snippets/lab/agents/get_sharing_link.py

@@ -1,5 +1,4 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import SharingLink
 
 manager = AILabManager()
 

pygeai/tests/snippets/lab/agents/list_agents.py

@@ -1,5 +1,5 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import FilterSettings, AgentList
+from pygeai.lab.models import FilterSettings
 
 ai_lab_manager = AILabManager()
 

pygeai/tests/snippets/lab/agents/publish_agent_revision.py

@@ -1,5 +1,4 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import Agent
 
 manager = AILabManager()
 

pygeai/tests/snippets/lab/agents/update_agent_properties.py

@@ -37,7 +37,7 @@ if agent.agent_data:
     # Update the agent
     updated_agent = manager.update_agent(agent)
     
-    print(f"\nAgent updated successfully!")
+    print("\nAgent updated successfully!")
     print(f"Agent ID: {updated_agent.id}")
     
     if updated_agent.agent_data and updated_agent.agent_data.properties:

pygeai/tests/snippets/lab/crud_ui.py

@@ -1,12 +1,10 @@
 import json
 import streamlit as st
-from typing import Optional, List, Dict, Any
+from typing import Any
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import (
-    Agent, AgentData, Prompt, PromptExample, PromptOutput, LlmConfig, Sampling, Model, ModelList,
-    Tool, ToolParameter, ResourcePool, ResourcePoolTool, ResourcePoolList,
-    Task, AgenticProcess, KnowledgeBase, AgenticActivity, ArtifactSignal, UserSignal, Event, SequenceFlow,
-    Variable, VariableList, FilterSettings
+    Agent, AgentData, Prompt, LlmConfig, Sampling, Model, ModelList,
+    Tool, ToolParameter, Task, AgenticProcess, KnowledgeBase
 )
 
 # Initialize the manager

pygeai/tests/snippets/lab/processes/kbs/get_kb.py

@@ -1,5 +1,4 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import KnowledgeBase
 
 manager = AILabManager()
 

pygeai/tests/snippets/lab/processes/kbs/list_kbs.py

@@ -1,5 +1,4 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import KnowledgeBaseList
 
 manager = AILabManager()
 

pygeai/tests/snippets/lab/processes/list_processes.py

@@ -1,5 +1,5 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import FilterSettings, AgenticProcessList
+from pygeai.lab.models import FilterSettings
 
 manager = AILabManager()
 filter_settings = FilterSettings(id="30a2ab8c-7503-4220-b6a0-77f3ee40d365", start="0", count="100", allow_drafts=True)

pygeai/tests/snippets/lab/samples/summarize_files.py

@@ -1,9 +1,6 @@
 import logging
-import os
 from uuid import uuid4
 
-from dotenv import load_dotenv
-from pydantic_settings import SettingsConfigDict
 from pygeai.chat.managers import ChatManager
 from pygeai.core.files.responses import UploadFileResponse
 from pygeai.lab.managers import AILabManager

pygeai/tests/snippets/lab/strategies/get_reasoning_strategy.py

@@ -1,5 +1,4 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import ReasoningStrategy
 
 manager = AILabManager()
 

pygeai/tests/snippets/lab/strategies/list_reasoning_strategies.py

@@ -1,5 +1,5 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import FilterSettings, ReasoningStrategyList
+from pygeai.lab.models import FilterSettings
 
 manager = AILabManager()
 filter_settings = FilterSettings(

pygeai/tests/snippets/lab/tools/get_tool.py

@@ -1,5 +1,5 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import FilterSettings, Tool
+from pygeai.lab.models import FilterSettings
 
 manager = AILabManager()
 

pygeai/tests/snippets/lab/tools/publish_tool_revision.py

@@ -1,5 +1,4 @@
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import Tool
 
 manager = AILabManager()
 

pygeai/tests/snippets/lab/tools/set_parameters.py

@@ -1,6 +1,5 @@
-from pygeai.core.base.responses import ErrorListResponse
 from pygeai.lab.managers import AILabManager
-from pygeai.lab.models import ToolParameter, Tool
+from pygeai.lab.models import ToolParameter
 
 manager = AILabManager()
 

pygeai/tests/snippets/lab/use_cases/c_code_fixer_agent_flow.py

@@ -1,9 +1,8 @@
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import (
     Agent, AgentData, Prompt, PromptExample, PromptOutput, LlmConfig, Sampling, Model, ModelList,
-    Tool, ToolParameter, ToolMessage,
-    Task,
-    AgenticProcess, KnowledgeBase, AgenticActivity, ArtifactSignal, UserSignal, Event, SequenceFlow,
+    Tool, ToolParameter, Task,
+    AgenticProcess, AgenticActivity, ArtifactSignal, Event, SequenceFlow,
     Variable, ProcessInstance, VariableList
 )
 

pygeai/tests/snippets/lab/use_cases/file_summarizer_example_2.py

@@ -84,7 +84,7 @@ def main():
                 ],
                 examples=[
                     PromptExample(
-                        input_data=f"File Name: test.txt",
+                        input_data="File Name: test.txt",
                         output='{"summary": "Summary of test.txt content."}'
                     )
                 ]

pygeai/tests/snippets/lab/use_cases/update_cli_expert.py

@@ -1,6 +1,5 @@
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Sampling, PromptExample, PromptOutput
-import uuid
 
 GEAI_CLI_HELP = """
 # PyGEAI - SDK for Globant Enterprise AI

pygeai/tests/snippets/lab/use_cases/update_lab_expert.py

@@ -1,6 +1,5 @@
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Sampling, PromptExample, PromptOutput
-import uuid
 
 GEAI_LAB_HELP = """\
 The Lab

pygeai/tests/snippets/lab/use_cases/update_web_designer.py

@@ -1,6 +1,5 @@
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Sampling, PromptExample, PromptOutput, ResourcePool, ResourcePoolTool, ResourcePoolList, ModelList
-import uuid
 import json
 
 WEB_DESIGN_GUIDE = """

pygeai/tests/snippets/lab/use_cases/update_web_reader.py

@@ -1,6 +1,5 @@
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Sampling, PromptExample, PromptOutput, ResourcePool, ResourcePoolTool, ResourcePoolList, ModelList
-import uuid
 import json
 
 WEB_READING_GUIDE = """

pygeai/tests/snippets/migrate/orchestrator_examples.py

@@ -63,7 +63,7 @@ def example_orchestrated_migration():
     # Execute all strategies in order
     results = orchestrator.execute()
     
-    print(f"Migration completed:")
+    print("Migration completed:")
     print(f"  Total strategies: {results['total']}")
     print(f"  Successful: {results['completed']}")
     print(f"  Failed: {results['failed']}")

{pygeai-0.6.0b13.dist-info → pygeai-0.6.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeai
-Version: 0.6.0b13
+Version: 0.6.1
 Summary: Software Development Kit to interact with Globant Enterprise AI.
 Author-email: Globant <geai-sdk@globant.com>
 License-Expression: MIT
@@ -48,35 +48,60 @@ By using the Python SDK to interact with Globant Enterprise AI, you agree with t
 [Terms and Conditions](https://www.globant.com/enterprise-ai/terms-of-use)
 
 ## Compatibility
-This package is compatible with the Globant Enterprise AI release from November 2025.
+This package is compatible with the Globant Enterprise AI release from January 2026.
 
 ## Configuration
 
-Before using the SDK, you need to define `GEAI_API_KEY` (`$SAIA_APITOKEN`) and `GEAI_API_BASE_URL` (`$BASE_URL`). You can achieve this in three ways:
+The SDK supports two authentication methods: **API Key** authentication and **OAuth 2.0** authentication.
+
+### API Key Authentication
+
+Before using the SDK with API Key authentication, you need to define `GEAI_API_KEY` and `GEAI_API_BASE_URL`. You can achieve this in three ways:
 
 * **Environment variables:** Set `GEAI_API_KEY` and `GEAI_API_BASE_URL` as environment variables in your operating system.
 * **Credentials file:** Create a file named credentials in the `.geai` directory within your user home directory (`$USER_HOME/.geai/credentials`) and define `GEAI_API_KEY` and `GEAI_API_BASE_URL` within this file.
 * **Client instantiation:** Specify the `api_key` and `base_url` parameters directly when creating an instance of a client class.
 
+### OAuth 2.0 Authentication
+
+For OAuth 2.0 authentication, you need to define:
+
+* `GEAI_OAUTH_ACCESS_TOKEN` - Your OAuth 2.0 access token
+* `GEAI_PROJECT_ID` - Your project ID (required with OAuth)
+* `GEAI_API_BASE_URL` - Base URL for the API
+* `GEAI_ORGANIZATION_ID` - (Optional) Your organization ID
+
+These can be configured via environment variables, credentials file, or client instantiation.
+
 **Note:** If you plan to use the [Evaluation Module](https://wiki.genexus.com/enterprise-ai/wiki?896,Evaluation), you must also define `GEAI_API_EVAL_URL`
 
 ### Credentials file
-The credentials file is organized via profiles, so one can interact with different instances of Globant Enterprise AI by just
+The credentials file is organized via profiles (aliases), so one can interact with different instances of Globant Enterprise AI by just
 referencing which profile one wishes to use. This also applies to different levels of permissions.
 
 The structure of the credentials file is as follows:
 
+**API Key Authentication:**
+
 ```
 [default]
 geai_api_key = <API_TOKEN>
 geai_api_base_url = <GEAI_BASE_URL>
-geai_api_evaul_url = <GEAI_EVALUATION_MODULE_URL>
+geai_api_eval_url = <GEAI_EVALUATION_MODULE_URL>
 
-[<another alias>]
+[production]
 geai_api_key = <API_TOKEN>
 geai_api_base_url = <GEAI_BASE_URL>
-geai_api_evaul_url = <GEAI_EVALUATION_MODULE_URL>
+```
+
+**OAuth 2.0 Authentication:**
 
+```
+[oauth-profile]
+geai_oauth_access_token = <OAUTH_ACCESS_TOKEN>
+geai_project_id = <PROJECT_ID>
+geai_api_base_url = <GEAI_BASE_URL>
+geai_organization_id = <ORGANIZATION_ID>  # Optional
 ```
 
 After setting the profiles, one can use them with the --alias option, for example: