pygeai 0.6.0b7__py3-none-any.whl → 0.6.0b10__py3-none-any.whl

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

@@ -0,0 +1,17 @@
+from pygeai.evaluation.result.clients import EvaluationResultClient
+
+client = EvaluationResultClient()
+
+evaluation_plan_id = "your-evaluation-plan-id"
+
+results = client.list_evaluation_results(evaluation_plan_id=evaluation_plan_id)
+
+print(f"Found {len(results) if isinstance(results, list) else 'N/A'} evaluation results")
+
+for result in results if isinstance(results, list) else []:
+    print(f"\nResult ID: {result.get('evaluationResultId')}")
+    print(f"  Status: {result.get('evaluationResultStatus')}")
+    print(f"  Cost: ${result.get('evaluationResultCost')}")
+    print(f"  Duration: {result.get('evaluationResultDuration')}ms")
+    print(f"  Start: {result.get('evaluationResultStartDate')}")
+    print(f"  End: {result.get('evaluationResultEndDate')}")

pygeai/tests/snippets/migrate/__init__.py

@@ -0,0 +1,45 @@
+"""
+GEAI Migration Examples
+
+This package contains comprehensive examples for migrating GEAI resources
+between projects and instances using both CLI and Python SDK.
+
+## Quick Import
+
+Import specific modules:
+    from pygeai.tests.snippets.migrate import agent_migration
+    from pygeai.tests.snippets.migrate import project_migration
+
+Import all examples:
+    from pygeai.tests.snippets.migrate import *
+
+## Available Modules
+
+- project_migration:     Create projects and migrate usage limits (1 example)
+- agent_migration:       Migrate agents between projects (3 examples)
+- tool_migration:        Migrate tools between projects (2 examples)
+- process_migration:     Migrate agentic processes (2 examples)
+- assistant_migration:   Migrate RAG assistants (2 examples)
+- orchestrator_examples: Complex multi-step workflows (3 examples)
+
+## Documentation
+
+See EXAMPLES_INDEX.md for a quick reference of all available examples.
+See README.md for comprehensive usage guide and patterns.
+
+## Example Usage
+
+    from pygeai.tests.snippets.migrate.agent_migration import example_migrate_all_agents
+    
+    result = example_migrate_all_agents()
+    print(result)
+"""
+
+__all__ = [
+    'project_migration',
+    'agent_migration',
+    'tool_migration',
+    'process_migration',
+    'assistant_migration',
+    'orchestrator_examples'
+]

pygeai/tests/snippets/migrate/agent_migration.py

@@ -0,0 +1,110 @@
+"""
+Agent Migration Examples
+
+Demonstrates how to migrate agents between projects.
+Uses project-scope API keys.
+"""
+
+from pygeai.migration.strategies import AgentMigrationStrategy
+
+
+def example_migrate_specific_agents():
+    """
+    Migrate specific agents by ID to an existing project.
+    Uses project-scope API keys.
+    """
+    agent_ids = [
+        "9d8e7f6g-5h4i-3j2k-1l0m-9n8o7p6q5r4s",
+        "1a2b3c4d-5e6f-7g8h-9i0j-1k2l3m4n5o6p"
+    ]
+    
+    strategy = AgentMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+        to_instance="https://api.prod.example.ai",
+        agent_ids=agent_ids
+    )
+    
+    result = strategy.migrate()
+    
+    print(f"Migrated {len(result['migrated_agents'])} agents")
+    for agent in result['migrated_agents']:
+        print(f"  - {agent['name']} (ID: {agent['id']})")
+    
+    return result
+
+
+def example_migrate_all_agents():
+    """
+    Automatically discover and migrate all agents from source project.
+    """
+    strategy = AgentMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+        to_instance="https://api.prod.example.ai",
+        agent_ids=None  # None means "all agents"
+    )
+    
+    result = strategy.migrate()
+    print(f"Migrated all {len(result['migrated_agents'])} agents")
+    
+    return result
+
+
+def example_custom_filtered_migration():
+    """
+    Create a custom migration that only migrates agents with specific tags
+    or matching certain criteria.
+    """
+    from pygeai.lab.managers import AILabManager
+    
+    # Initialize managers for source and destination
+    source_lab = AILabManager(
+        api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        instance="https://api.test.example.ai"
+    )
+    
+    # Get all agents and filter
+    all_agents = source_lab.get_agent_list()
+    
+    # Custom filter: only migrate agents with "production" in their name
+    filtered_agent_ids = [
+        agent.id for agent in all_agents.agents 
+        if "production" in agent.name.lower()
+    ]
+    
+    print(f"Found {len(filtered_agent_ids)} agents matching filter")
+    
+    # Now migrate only the filtered agents
+    strategy = AgentMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+        agent_ids=filtered_agent_ids
+    )
+    
+    result = strategy.migrate()
+    print(f"Migrated {len(result['migrated_agents'])} filtered agents")
+    
+    return result
+
+
+if __name__ == "__main__":
+    print("Agent Migration Examples")
+    print("=" * 80)
+    print("\nThese examples demonstrate various agent migration patterns.")
+    print("Uncomment to run specific examples:\n")
+    
+    # Uncomment to run specific examples:
+    # example_migrate_specific_agents()
+    # example_migrate_all_agents()
+    # example_custom_filtered_migration()

pygeai/tests/snippets/migrate/assistant_migration.py

@@ -0,0 +1,64 @@
+"""
+RAG Assistant Migration Examples
+
+Demonstrates how to migrate RAG assistants between projects.
+Uses project-scope API keys.
+"""
+
+from pygeai.migration.strategies import RAGAssistantMigrationStrategy
+
+
+def example_migrate_all_assistants():
+    """
+    Migrate all RAG assistants from source project.
+    """
+    strategy = RAGAssistantMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+        to_instance="https://api.prod.example.ai",
+        assistant_names=None  # None for all assistants
+    )
+    
+    result = strategy.migrate()
+    print(f"Migrated {len(result['migrated_assistants'])} assistants")
+    
+    return result
+
+
+def example_migrate_specific_assistants():
+    """
+    Migrate specific RAG assistants by name.
+    """
+    assistant_names = [
+        "Customer Support Assistant",
+        "Product Documentation Assistant"
+    ]
+    
+    strategy = RAGAssistantMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+        to_instance="https://api.prod.example.ai",
+        assistant_names=assistant_names
+    )
+    
+    result = strategy.migrate()
+    print(f"Migrated {len(result['migrated_assistants'])} specific assistants")
+    
+    return result
+
+
+if __name__ == "__main__":
+    print("RAG Assistant Migration Examples")
+    print("=" * 80)
+    print("\nThese examples demonstrate RAG assistant migration patterns.")
+    print("Uncomment to run specific examples:\n")
+    
+    # Uncomment to run specific examples:
+    # example_migrate_all_assistants()
+    # example_migrate_specific_assistants()

pygeai/tests/snippets/migrate/orchestrator_examples.py

@@ -0,0 +1,179 @@
+"""
+Migration Orchestrator Examples
+
+Demonstrates how to orchestrate complex migrations with multiple strategies
+and handle error scenarios.
+"""
+
+from pygeai.migration.strategies import (
+    ProjectMigrationStrategy,
+    AgentMigrationStrategy,
+    ToolMigrationStrategy,
+    RAGAssistantMigrationStrategy
+)
+from pygeai.migration.tools import MigrationOrchestrator
+
+
+def example_orchestrated_migration():
+    """
+    Use the orchestrator to execute multiple migration strategies in sequence.
+    This is useful for complex migrations with multiple resource types.
+    """
+    # First, create the project
+    project_strategy = ProjectMigrationStrategy(
+        from_api_key="source_org_key_abc123def456ghi789jkl012mno345pqr678",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_org_key_stu901vwx234yz567abc890def123ghi456jkl789",
+        to_project_name="Orchestrated Migration Project",
+        admin_email="admin@example.com"
+    )
+    
+    # Create the orchestrator
+    orchestrator = MigrationOrchestrator()
+    
+    # Add the project creation strategy
+    orchestrator.add_strategy(project_strategy)
+    
+    # After project creation, we'll need the new project ID
+    # The orchestrator will handle this automatically
+    
+    # Add agent migration strategy
+    agent_strategy = AgentMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id=None,  # Will be filled by orchestrator after project creation
+        agent_ids=None  # Migrate all agents
+    )
+    orchestrator.add_strategy(agent_strategy)
+    
+    # Add tool migration strategy
+    tool_strategy = ToolMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id=None,  # Will be filled by orchestrator
+        tool_ids=None  # Migrate all tools
+    )
+    orchestrator.add_strategy(tool_strategy)
+    
+    # Execute all strategies in order
+    results = orchestrator.execute()
+    
+    print(f"Migration completed:")
+    print(f"  Total strategies: {results['total']}")
+    print(f"  Successful: {results['completed']}")
+    print(f"  Failed: {results['failed']}")
+    
+    return results
+
+
+def example_migration_with_error_handling():
+    """
+    Demonstrate proper error handling during migrations.
+    """
+    from pygeai.core.common.exceptions import MigrationException
+    
+    try:
+        strategy = AgentMigrationStrategy(
+            from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+            from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+            from_instance="https://api.test.example.ai",
+            to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+            to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+            agent_ids=None
+        )
+        
+        result = strategy.migrate()
+        
+        print(f"✓ Migration successful: {result['summary']}")
+        
+    except MigrationException as e:
+        print(f"✗ Migration failed: {e}")
+        print(f"  Details: {e.details}")
+        
+    except Exception as e:
+        print(f"✗ Unexpected error: {e}")
+
+
+def example_full_migration_workflow():
+    """
+    Complete workflow: create project, migrate all resources, verify.
+    """
+    # Step 1: Create new project with usage limits
+    print("Step 1: Creating new project...")
+    project_strategy = ProjectMigrationStrategy(
+        from_api_key="source_org_key_abc123def456ghi789jkl012mno345pqr678",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_org_key_stu901vwx234yz567abc890def123ghi456jkl789",
+        to_project_name="Complete Migration",
+        admin_email="admin@example.com"
+    )
+    
+    project_result = project_strategy.migrate()
+    new_project_id = project_result['project_id']
+    print(f"✓ Project created: {new_project_id}")
+    
+    # Step 2: Migrate agents
+    print("\nStep 2: Migrating agents...")
+    agent_strategy = AgentMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id=new_project_id,
+        agent_ids=None
+    )
+    agent_result = agent_strategy.migrate()
+    print(f"✓ Migrated {len(agent_result['migrated_agents'])} agents")
+    
+    # Step 3: Migrate tools
+    print("\nStep 3: Migrating tools...")
+    tool_strategy = ToolMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id=new_project_id,
+        tool_ids=None
+    )
+    tool_result = tool_strategy.migrate()
+    print(f"✓ Migrated {len(tool_result['migrated_tools'])} tools")
+    
+    # Step 4: Migrate RAG assistants
+    print("\nStep 4: Migrating RAG assistants...")
+    assistant_strategy = RAGAssistantMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id=new_project_id,
+        assistant_names=None
+    )
+    assistant_result = assistant_strategy.migrate()
+    print(f"✓ Migrated {len(assistant_result['migrated_assistants'])} assistants")
+    
+    print("\n✓ Complete migration finished successfully!")
+    
+    return {
+        'project': project_result,
+        'agents': agent_result,
+        'tools': tool_result,
+        'assistants': assistant_result
+    }
+
+
+if __name__ == "__main__":
+    print("Migration Orchestrator Examples")
+    print("=" * 80)
+    print("\nThese examples demonstrate complex migration workflows.")
+    print("Uncomment to run specific examples:\n")
+    
+    # Uncomment to run specific examples:
+    # example_orchestrated_migration()
+    # example_migration_with_error_handling()
+    # example_full_migration_workflow()

pygeai/tests/snippets/migrate/process_migration.py

@@ -0,0 +1,64 @@
+"""
+Agentic Process Migration Examples
+
+Demonstrates how to migrate agentic processes between projects.
+Uses project-scope API keys.
+"""
+
+from pygeai.migration.strategies import AgenticProcessMigrationStrategy
+
+
+def example_migrate_all_processes():
+    """
+    Migrate all agentic processes from source project.
+    """
+    strategy = AgenticProcessMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+        to_instance="https://api.prod.example.ai",
+        process_ids=None  # None for all processes
+    )
+    
+    result = strategy.migrate()
+    print(f"Migrated {len(result['migrated_processes'])} processes")
+    
+    return result
+
+
+def example_migrate_specific_processes():
+    """
+    Migrate specific agentic processes by ID.
+    """
+    process_ids = [
+        "4b5c6d7e-8f9g-0h1i-2j3k-4l5m6n7o8p9q",
+        "5c6d7e8f-9g0h-1i2j-3k4l-5m6n7o8p9q0r"
+    ]
+    
+    strategy = AgenticProcessMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+        to_instance="https://api.prod.example.ai",
+        process_ids=process_ids
+    )
+    
+    result = strategy.migrate()
+    print(f"Migrated {len(result['migrated_processes'])} specific processes")
+    
+    return result
+
+
+if __name__ == "__main__":
+    print("Agentic Process Migration Examples")
+    print("=" * 80)
+    print("\nThese examples demonstrate agentic process migration patterns.")
+    print("Uncomment to run specific examples:\n")
+    
+    # Uncomment to run specific examples:
+    # example_migrate_all_processes()
+    # example_migrate_specific_processes()

pygeai/tests/snippets/migrate/project_migration.py

@@ -0,0 +1,42 @@
+"""
+Project Migration Examples
+
+Demonstrates how to create new projects and migrate usage limits.
+Requires organization-scope API keys.
+"""
+
+from pygeai.migration.strategies import ProjectMigrationStrategy
+
+
+def example_create_project_with_usage_limits():
+    """
+    Create a new project and migrate usage limits from an existing project.
+    Requires organization-scope API keys.
+    """
+    strategy = ProjectMigrationStrategy(
+        from_api_key="source_org_key_abc123def456ghi789jkl012mno345pqr678",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_org_key_stu901vwx234yz567abc890def123ghi456jkl789",
+        to_project_name="New Migrated Project",
+        admin_email="admin@example.com",
+        to_instance="https://api.prod.example.ai"  # Optional: different instance
+    )
+    
+    # Execute the migration
+    result = strategy.migrate()
+    
+    print(f"Project created with ID: {result['project_id']}")
+    print(f"Migration result: {result}")
+    
+    return result
+
+
+if __name__ == "__main__":
+    print("Project Migration Example")
+    print("=" * 80)
+    print("\nThis example demonstrates creating a new project with usage limits.")
+    print("Uncomment to run:\n")
+    
+    # Uncomment to run:
+    # example_create_project_with_usage_limits()

pygeai/tests/snippets/migrate/tool_migration.py

@@ -0,0 +1,64 @@
+"""
+Tool Migration Examples
+
+Demonstrates how to migrate tools between projects.
+Uses project-scope API keys.
+"""
+
+from pygeai.migration.strategies import ToolMigrationStrategy
+
+
+def example_migrate_specific_tools():
+    """
+    Migrate specific tools by ID.
+    """
+    tool_ids = [
+        "2a3b4c5d-6e7f-8g9h-0i1j-2k3l4m5n6o7p",
+        "3b4c5d6e-7f8g-9h0i-1j2k-3l4m5n6o7p8q"
+    ]
+    
+    strategy = ToolMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+        to_instance="https://api.prod.example.ai",
+        tool_ids=tool_ids
+    )
+    
+    result = strategy.migrate()
+    print(f"Migrated {len(result['migrated_tools'])} tools")
+    
+    return result
+
+
+def example_migrate_all_tools():
+    """
+    Migrate all tools from source project.
+    """
+    strategy = ToolMigrationStrategy(
+        from_api_key="source_project_key_123abc456def789ghi012jkl345mno678pqr901",
+        from_project_id="7x8y9z0a-1b2c-3d4e-5f6g-7h8i9j0k1l2m",
+        from_instance="https://api.test.example.ai",
+        to_api_key="dest_project_key_789stu012vwx345yz678abc901def234ghi567",
+        to_project_id="3c4d5e6f-7g8h-9i0j-1k2l-3m4n5o6p7q8r",
+        to_instance="https://api.prod.example.ai",
+        tool_ids=None  # None means "all tools"
+    )
+    
+    result = strategy.migrate()
+    print(f"Migrated all {len(result['migrated_tools'])} tools")
+    
+    return result
+
+
+if __name__ == "__main__":
+    print("Tool Migration Examples")
+    print("=" * 80)
+    print("\nThese examples demonstrate tool migration patterns.")
+    print("Uncomment to run specific examples:\n")
+    
+    # Uncomment to run specific examples:
+    # example_migrate_specific_tools()
+    # example_migrate_all_tools()

pygeai/tests/snippets/organization/create_project.py

@@ -1,7 +1,7 @@
 from pygeai.organization.managers import OrganizationManager
 from pygeai.core.models import UsageLimit, Project
 
-manager = OrganizationManager()
+manager = OrganizationManager(alias="sdkorg")
 
 usage_limit = UsageLimit(
     subscription_type="Monthly",
@@ -12,7 +12,7 @@ usage_limit = UsageLimit(
 )
 
 project = Project(
-    name="New AI Project",
+    name="New TEST Project 2",
     description="An AI project focused on natural language processing",
     email="alejandro.trinidad@globant.com",
     usage_limit=usage_limit

{pygeai-0.6.0b7.dist-info → pygeai-0.6.0b10.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeai
-Version: 0.6.0b7
+Version: 0.6.0b10
 Summary: Software Development Kit to interact with Globant Enterprise AI.
 Author-email: Globant <geai-sdk@globant.com>
 License-Expression: MIT