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

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/tests/snippets/organization/get_memberships.py

@@ -0,0 +1,12 @@
+from pygeai.organization.managers import OrganizationManager
+
+manager = OrganizationManager(alias="sdkorg")
+
+response = manager.get_memberships(
+    email="alejandro.trinidad@globant.com",
+    start_page=1,
+    page_size=10,
+    #order_key="organizationName",
+    #order_direction="asc"
+)
+print(f"response: {response}")

pygeai/tests/snippets/organization/get_organization_members.py

@@ -0,0 +1,6 @@
+from pygeai.organization.managers import OrganizationManager
+
+manager = OrganizationManager()
+
+response = manager.get_organization_members(organization_id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
+print(f"response: {response}")

pygeai/tests/snippets/organization/get_project_members.py

@@ -0,0 +1,6 @@
+from pygeai.organization.managers import OrganizationManager
+
+manager = OrganizationManager()
+
+response = manager.get_project_members(project_id="1956c032-3c66-4435-acb8-6a06e52f819f")
+print(f"response: {response}")

pygeai/tests/snippets/organization/get_project_memberships.py

@@ -0,0 +1,12 @@
+from pygeai.organization.managers import OrganizationManager
+
+manager = OrganizationManager()
+
+response = manager.get_project_memberships(
+    email="user@example.com",
+    start_page=1,
+    page_size=10,
+    order_key="projectName",
+    order_direction="desc"
+)
+print(f"response: {response}")

pygeai/tests/snippets/organization/get_project_roles.py

@@ -0,0 +1,6 @@
+from pygeai.organization.managers import OrganizationManager
+
+manager = OrganizationManager(alias="sdkorg")
+
+response = manager.get_project_roles(project_id="2ca6883f-6778-40bb-bcc1-85451fb11107")
+print(f"response: {response}")

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

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