pygeai 0.6.0b11__py3-none-any.whl → 0.6.0b12__py3-none-any.whl

pygeai/tests/snippets/chat/get_response_complete_example.py

@@ -0,0 +1,67 @@
+from pygeai.chat.clients import ChatClient
+
+# Example: Comprehensive usage of get_response with all major parameters
+client = ChatClient()
+
+model = "openai/gpt-4-turbo-preview"
+input_text = "Analyze the benefits of renewable energy and suggest implementation strategies"
+
+# Define a tool for additional data retrieval
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "get_energy_stats",
+            "description": "Get statistics about renewable energy adoption",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "country": {
+                        "type": "string",
+                        "description": "Country name"
+                    },
+                    "year": {
+                        "type": "integer",
+                        "description": "Year for statistics"
+                    }
+                },
+                "required": ["country"]
+            }
+        }
+    }
+]
+
+# Comprehensive configuration
+response = client.get_response(
+    model=model,
+    input=input_text,
+    instructions="You are an energy policy expert. Provide detailed, data-driven analysis with actionable recommendations.",
+    tools=tools,
+    tool_choice="auto",  # Let the model decide when to use tools
+    parallel_tool_calls=True,
+    temperature=0.7,
+    max_output_tokens=2000,
+    top_p=0.9,
+    metadata={
+        "request_id": "energy_analysis_001",
+        "department": "sustainability",
+        "priority": "high"
+    },
+    user="analyst_john_doe",
+    reasoning={
+        "effort": "high"  # Request thorough reasoning
+    },
+    truncation="auto",
+    store=True  # Store for future reference
+)
+
+print("=== Comprehensive Response ===")
+print(response)
+
+# Extract key information if response is a dict
+if isinstance(response, dict):
+    print("\n=== Response Metadata ===")
+    if "usage" in response:
+        print(f"Tokens used: {response['usage']}")
+    if "model" in response:
+        print(f"Model used: {response['model']}")

pygeai/tests/snippets/chat/get_response_with_instructions.py

@@ -0,0 +1,19 @@
+from pygeai.chat.clients import ChatClient
+
+# Example: Using get_response with custom instructions
+client = ChatClient()
+
+model = "openai/gpt-4-turbo-preview"
+input_text = "Tell me about machine learning"
+
+# Using instructions to customize the model's behavior
+response = client.get_response(
+    model=model,
+    input=input_text,
+    instructions="You are an expert educator who explains complex topics in simple terms suitable for beginners. Always use analogies and examples.",
+    temperature=0.8,
+    max_output_tokens=1500
+)
+
+print("Response with custom instructions:")
+print(response)

pygeai/tests/snippets/chat/get_response_with_metadata.py

@@ -0,0 +1,24 @@
+from pygeai.chat.clients import ChatClient
+
+# Example: Using get_response with metadata and user identification
+client = ChatClient()
+
+model = "openai/gpt-4-turbo-preview"
+input_text = "What are the benefits of using AI in healthcare?"
+
+# Using metadata to pass additional context
+response = client.get_response(
+    model=model,
+    input=input_text,
+    metadata={
+        "request_id": "req_12345",
+        "department": "healthcare",
+        "priority": "high"
+    },
+    user="user_jane_doe",  # User identifier for tracking
+    temperature=0.7,
+    max_output_tokens=1500
+)
+
+print("Response with metadata:")
+print(response)

pygeai/tests/snippets/chat/get_response_with_parallel_tools.py

@@ -0,0 +1,58 @@
+from pygeai.chat.clients import ChatClient
+
+# Example: Using get_response with parallel tool calls
+client = ChatClient()
+
+model = "openai/gpt-4-turbo-preview"
+input_text = "What's the weather in New York and what time is it in Tokyo?"
+
+# Define tools
+tools = [
+    {
+        "type": "function",
+        "function": {
+            "name": "get_weather",
+            "description": "Get current weather for a location",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "location": {
+                        "type": "string",
+                        "description": "City name"
+                    }
+                },
+                "required": ["location"]
+            }
+        }
+    },
+    {
+        "type": "function",
+        "function": {
+            "name": "get_time",
+            "description": "Get current time for a location",
+            "parameters": {
+                "type": "object",
+                "properties": {
+                    "location": {
+                        "type": "string",
+                        "description": "City name"
+                    }
+                },
+                "required": ["location"]
+            }
+        }
+    }
+]
+
+# Enable parallel tool calls to execute both tools simultaneously
+response = client.get_response(
+    model=model,
+    input=input_text,
+    tools=tools,
+    parallel_tool_calls=True,  # Allow multiple tools to be called in parallel
+    temperature=0.7,
+    max_output_tokens=1000
+)
+
+print("Response with parallel tool calls:")
+print(response)

pygeai/tests/snippets/chat/get_response_with_reasoning.py

@@ -0,0 +1,21 @@
+from pygeai.chat.clients import ChatClient
+
+# Example: Using get_response with reasoning parameter
+client = ChatClient()
+
+model = "openai/o1-preview"
+input_text = "Explain the concept of quantum entanglement in simple terms"
+
+# Using reasoning parameter to control reasoning effort
+response = client.get_response(
+    model=model,
+    input=input_text,
+    reasoning={
+        "type": "default",  # Can be "default", "high", "medium", "low"
+        "effort": "medium"
+    },
+    max_output_tokens=2000
+)
+
+print("Response with reasoning:")
+print(response)

pygeai/tests/snippets/chat/get_response_with_store.py

@@ -0,0 +1,38 @@
+from pygeai.chat.clients import ChatClient
+
+# Example: Using get_response with store parameter to persist conversation
+client = ChatClient()
+
+model = "openai/gpt-4-turbo-preview"
+input_text = "Remember that my favorite color is blue. What did I just tell you?"
+
+# Use store parameter to enable conversation storage
+response = client.get_response(
+    model=model,
+    input=input_text,
+    store=True,  # Store this conversation for future reference
+    user="user_123",  # Associate with specific user
+    metadata={
+        "session_id": "session_abc",
+        "conversation_type": "preference_learning"
+    },
+    temperature=0.7,
+    max_output_tokens=500
+)
+
+print("Response with conversation storage:")
+print(response)
+
+# Follow-up request can reference stored context
+follow_up = client.get_response(
+    model=model,
+    input="What's my favorite color?",
+    store=True,
+    user="user_123",
+    metadata={
+        "session_id": "session_abc"
+    }
+)
+
+print("\nFollow-up response:")
+print(follow_up)

pygeai/tests/snippets/chat/get_response_with_truncation.py

@@ -0,0 +1,24 @@
+from pygeai.chat.clients import ChatClient
+
+# Example: Using get_response with truncation strategy
+client = ChatClient()
+
+model = "openai/gpt-4-turbo-preview"
+input_text = """
+This is a very long input text that might exceed the context window.
+Lorem ipsum dolor sit amet, consectetur adipiscing elit. Sed do eiusmod 
+tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam,
+quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat.
+""" * 100  # Simulate long input
+
+# Using truncation to handle long inputs
+response = client.get_response(
+    model=model,
+    input=input_text,
+    truncation="auto",  # Options: "auto", "disabled"
+    max_output_tokens=1000,
+    temperature=0.7
+)
+
+print("Response with truncation handling:")
+print(response)

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

@@ -0,0 +1,39 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Permission
+
+# Create an agent with permissions configured
+agent = Agent(
+    name="Agent_With_Permissions",
+    description="Agent demonstrating permissions configuration",
+    access_scope="private",
+    is_draft=True,
+    sharing_scope="organization",
+    permissions=Permission(
+        allow_chat_sharing=True,
+        allow_external_execution=False
+    ),
+    agent_data=AgentData(
+        prompt=Prompt(
+            instructions="You are a collaborative assistant that can be shared within the organization",
+            inputs=["user_query"]
+        ),
+        llm_config=LlmConfig(max_tokens=2000),
+        strategy_name="Dynamic Prompting",
+        models=[Model(name="gpt-4-turbo-preview")]
+    )
+)
+
+manager = AILabManager()
+result = manager.create_agent(agent=agent, automatic_publish=False)
+
+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(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(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/create_agent_with_properties.py

@@ -0,0 +1,46 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Property
+
+# Create an agent with properties
+agent = Agent(
+    name="Agent_With_Properties",
+    description="Agent demonstrating the properties field",
+    access_scope="private",
+    is_draft=True,
+    agent_data=AgentData(
+        prompt=Prompt(
+            instructions="You are a helpful assistant with custom properties",
+            inputs=["user_query"]
+        ),
+        llm_config=LlmConfig(max_tokens=2000),
+        strategy_name="Dynamic Prompting",
+        models=[Model(name="gpt-4-turbo-preview")],
+        properties=[
+            Property(
+                data_type="string",
+                key="environment",
+                value="production"
+            ),
+            Property(
+                data_type="number",
+                key="max_retries",
+                value="3"
+            ),
+            Property(
+                data_type="boolean",
+                key="enable_logging",
+                value="true"
+            )
+        ]
+    )
+)
+
+manager = AILabManager()
+result = manager.create_agent(agent=agent, automatic_publish=False)
+
+print(f"Created agent with {len(result.agent_data.properties) if result.agent_data and result.agent_data.properties else 0} properties")
+print(f"Agent ID: {result.id}")
+if result.agent_data and result.agent_data.properties:
+    print("\nProperties:")
+    for prop in result.agent_data.properties:
+        print(f"  - {prop.key} ({prop.data_type}): {prop.value}")

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

@@ -0,0 +1,62 @@
+from pygeai.lab.managers import AILabManager
+
+# Get an agent and inspect all new fields
+manager = AILabManager()
+
+# First, list agents to get an ID
+agents = manager.get_agent_list()
+if len(agents) == 0:
+    print("No agents found. Please create an agent first.")
+    exit(1)
+
+# Get the first agent
+agent_id = agents[0].id
+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 ===")
+
+# Sharing scope
+print(f"\nSharing Scope: {agent.sharing_scope}")
+
+# Permissions
+if agent.permissions:
+    print(f"\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")
+
+# Effective permissions
+if agent.effective_permissions:
+    print(f"\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")
+
+# Agent data properties
+if agent.agent_data:
+    print(f"\nAgent Data:")
+    print(f"  - Strategy Name: {agent.agent_data.strategy_name}")
+    
+    if agent.agent_data.properties:
+        print(f"  - Properties ({len(agent.agent_data.properties)}):")
+        for prop in agent.agent_data.properties:
+            print(f"    * {prop.key} ({prop.data_type}): {prop.value}")
+    else:
+        print(f"  - Properties: None")
+    
+    if agent.agent_data.models:
+        print(f"  - Models: {len(agent.agent_data.models)} configured")
+    else:
+        print(f"  - 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")
+else:
+    print(f"\nAgent Data: None")

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

@@ -0,0 +1,50 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Property
+
+# Update an agent's properties
+manager = AILabManager()
+
+# Get an existing agent
+agents = manager.get_agent_list()
+if len(agents) == 0:
+    print("No agents found. Please create an agent first.")
+    exit(1)
+
+agent = manager.get_agent(agents[0].id)
+print(f"Updating agent: {agent.name}")
+
+# Update agent data properties
+if agent.agent_data:
+    # Add or update properties
+    agent.agent_data.properties = [
+        Property(
+            data_type="string",
+            key="version",
+            value="2.0.0"
+        ),
+        Property(
+            data_type="string",
+            key="region",
+            value="us-east-1"
+        ),
+        Property(
+            data_type="boolean",
+            key="auto_scale",
+            value="true"
+        )
+    ]
+    
+    # Update the agent
+    updated_agent = manager.update_agent(agent)
+    
+    print(f"\nAgent updated successfully!")
+    print(f"Agent ID: {updated_agent.id}")
+    
+    if updated_agent.agent_data and updated_agent.agent_data.properties:
+        print(f"\nUpdated Properties ({len(updated_agent.agent_data.properties)}):")
+        for prop in updated_agent.agent_data.properties:
+            print(f"  - {prop.key} ({prop.data_type}): {prop.value}")
+    else:
+        print("\nNo properties found after update")
+else:
+    print("Agent has no agent_data, cannot update properties")

pygeai/tests/snippets/organization/add_project_member.py

@@ -0,0 +1,10 @@
+from pygeai.organization.managers import OrganizationManager
+
+manager = OrganizationManager()
+
+response = manager.add_project_member(
+    project_id="1956c032-3c66-4435-acb8-6a06e52f819f",
+    user_email="newuser@example.com",
+    roles=["Project member", "Project administrator"]
+)
+print(f"response: {response}")

pygeai/tests/snippets/organization/add_project_member_batch.py

@@ -0,0 +1,44 @@
+import csv
+from pygeai.organization.clients import OrganizationClient
+
+client = OrganizationClient()
+
+batch_file = "project_members.csv"
+
+successful = 0
+failed = 0
+errors = []
+
+with open(batch_file, 'r') as f:
+    csv_reader = csv.reader(f)
+    for line_num, row in enumerate(csv_reader, start=1):
+        if len(row) < 3:
+            error_msg = f"Line {line_num}: Invalid format - expected at least 3 columns (project_id, email, role1, ...)"
+            errors.append(error_msg)
+            failed += 1
+            continue
+
+        project_id = row[0].strip()
+        email = row[1].strip()
+        roles = [r.strip() for r in row[2:] if r.strip()]
+
+        if not (project_id and email and roles):
+            error_msg = f"Line {line_num}: Missing required fields"
+            errors.append(error_msg)
+            failed += 1
+            continue
+
+        try:
+            result = client.add_project_member(project_id, email, roles)
+            print(f"Successfully added {email} to project {project_id}")
+            successful += 1
+        except Exception as e:
+            error_msg = f"Line {line_num}: Failed to add {email} to project {project_id}: {str(e)}"
+            errors.append(error_msg)
+            failed += 1
+
+print(f"\nBatch processing complete: {successful} successful, {failed} failed")
+if errors:
+    print("\nErrors:")
+    for error in errors:
+        print(f"  - {error}")

{pygeai-0.6.0b11.dist-info → pygeai-0.6.0b12.dist-info}/METADATA

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