pygeai 0.1.6__py3-none-any.whl → 0.6.0b15__py3-none-any.whl

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

@@ -0,0 +1,48 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Sampling, PromptExample, PromptOutput
+
+agent = Agent(
+    id="f64ba214-152b-4dd4-be0d-2920da415f5d",
+    status="active",
+    name="Private Translator V25",
+    access_scope="private",
+    public_name="com.genexus.geai.private_translator_25",
+    job_description="Translates",
+    avatar_image="https://www.shareicon.net/data/128x128/2016/11/09/851442_logo_512x512.png",
+    description="Agent that translates from any language to english.",
+    is_draft=False,
+    is_readonly=False,
+    revision=1,
+    version=None,
+    agent_data=AgentData(
+        prompt=Prompt(
+            instructions="the user will provide a text, you must return the same text translated to english",
+            inputs=["text", "avoid slang indicator"],
+            outputs=[
+                PromptOutput(key="translated_text", description="translated text, with slang or not depending on the indication. in plain text."),
+                PromptOutput(key="summary", description="a summary in the original language of the text to be translated, also in plain text.")
+            ],
+            examples=[
+                PromptExample(input_data="opitiiiis mundo [no-slang]", output='{"translated_text":"hello world","summary":"saludo"}'),
+                PromptExample(input_data="esto es una prueba pincheguey [keep-slang]", output='{"translated_text":"this is a test pal","summary":"prueba"}')
+            ]
+        ),
+        llm_config=LlmConfig(
+            max_tokens=5000,
+            timeout=0,
+            sampling=Sampling(temperature=0.5, top_k=0, top_p=0)
+        ),
+        models=[Model(name="gpt-4-turbo-preview")]
+    )
+)
+
+
+manager = AILabManager()
+result = manager.create_agent(
+    agent=agent,
+    automatic_publish=False
+)
+
+
+print(f"Agent: {agent.to_dict()}")
+

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

@@ -0,0 +1,12 @@
+from pygeai.core.base.responses import EmptyResponse
+from pygeai.lab.managers import AILabManager
+
+manager = AILabManager()
+
+agent_id = "3c06e604-26a9-485c-b84e-8eba3ff9a218"
+
+result = manager.delete_agent(
+    agent_id=agent_id
+)
+
+print(f"Agent deleted successfully")

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

@@ -0,0 +1,24 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import FilterSettings, Agent
+
+
+manager = AILabManager()
+
+
+result = manager.get_agent(
+    agent_id="9716a0a1-5eab-4cc9-a611-fa2c3237c511"
+)
+
+print(f"Retrieved agent: {result.to_dict()}")
+
+filter_settings = FilterSettings(
+    revision="0",
+    version="0",
+    allow_drafts=False
+)
+result = manager.get_agent(
+    agent_id="9716a0a1-5eab-4cc9-a611-fa2c3237c511",
+    filter_settings=filter_settings
+)
+
+print(f"Retrieved agent: {result.to_dict()}")

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/get_sharing_link.py

@@ -0,0 +1,13 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import SharingLink
+
+manager = AILabManager()
+
+
+result = manager.create_sharing_link(
+    agent_id="9716a0a1-5eab-4cc9-a611-fa2c3237c511"
+)
+
+
+print(f"Sharing link created for agent ID: {result.agent_id}")
+print(f"Shared Link: {result.shared_link}")

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

@@ -0,0 +1,18 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import FilterSettings, AgentList
+
+ai_lab_manager = AILabManager()
+
+project_id = "2ca6883f-6778-40bb-bcc1-85451fb11107"
+filter_settings = FilterSettings(
+    allow_external=False,
+    allow_drafts=True,
+    access_scope="private"
+)
+
+result = ai_lab_manager.get_agent_list(
+    filter_settings=filter_settings
+)
+
+for agent in result.agents:
+    print(f"Agent: {agent.name}, ID: {agent.id}")

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

@@ -0,0 +1,12 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Agent
+
+manager = AILabManager()
+
+result = manager.publish_agent_revision(
+    agent_id="9716a0a1-5eab-4cc9-a611-fa2c3237c511",
+    revision="6"
+)
+
+print(f"Published agent: {result.name}, ID: {result.id}")
+print(f"Revision: {result.revision}, Draft: {result.is_draft}")

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

@@ -0,0 +1,50 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Sampling, PromptExample, PromptOutput, ModelList
+
+updated_agent = Agent(
+    id="9716a0a1-5eab-4cc9-a611-fa2c3237c511",
+    status="active",
+    name="Public Translator V21 Updated",
+    access_scope="public",
+    public_name="com.genexus.geai.public_translator_21",
+    job_description="Translates updated",
+    avatar_image="https://www.shareicon.net/data/128x128/2016/11/09/851442_logo_512x512.png",
+    description="Updated agent that translates from any language to English.",
+    is_draft=True,
+    is_readonly=False,
+    revision=1,
+    version=None,
+    agent_data=AgentData(
+        prompt=Prompt(
+            instructions="the user will provide a text, you must return the same text translated to English with updates",
+            inputs=["text", "avoid slang indicator"],
+            outputs=[
+                PromptOutput(key="translated_text", description="translated text, updated version"),
+                PromptOutput(key="summary", description="summary in original language, updated")
+            ],
+            examples=[
+                PromptExample(input_data="opitiiiis mundo [no-slang]", output='{"translated_text":"hello world","summary":"saludo"}'),
+                PromptExample(input_data="esto es una prueba [keep-slang]", output='{"translated_text":"this is a test","summary":"prueba"}')
+            ]
+        ),
+        llm_config=LlmConfig(
+            max_tokens=6000,
+            timeout=0,
+            sampling=Sampling(temperature=0.6, top_k=0, top_p=0.0)
+        ),
+        models=ModelList(models=[
+            Model(name="gpt-4-turbo-preview"),
+            Model(name="xlm-roberta-large")
+        ])
+    )
+)
+
+manager = AILabManager()
+
+result = manager.update_agent(
+    agent=updated_agent,
+    automatic_publish=False,
+    upsert=False
+)
+
+print(f"Updated agent: {result.to_dict()}")

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/lab/assistant_to_agent.py

@@ -0,0 +1,191 @@
+from typing import Union
+from uuid import uuid4
+from pydantic import ValidationError
+import logging
+from pygeai.core.common.exceptions import MissingRequirementException
+from pygeai.core.models import Assistant
+from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Sampling, Model, ModelList
+from pygeai.core.base.responses import ErrorListResponse
+from pygeai.assistant.managers import AssistantManager
+from pygeai.lab.managers import AILabManager
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+
+def transform_assistant_to_agent(assistant: Assistant) -> Agent:
+    """
+    Transform an Assistant object into an Agent object, using Assistant.name as Agent.name.
+    """
+    try:
+        if not assistant.name:
+            raise ValueError("Assistant.name cannot be None or empty")
+
+        # Map status
+        status_map = {1: "active", 2: "inactive"}
+        agent_status = status_map.get(assistant.status, "active")
+
+        # Map LLM settings to LlmConfig
+        llm_config = None
+        if assistant.llm_settings:
+            sampling = Sampling(
+                temperature=assistant.llm_settings.temperature or 0.7,  # Default from Assistant example
+                top_k=50,  # Default, as Agent example uses top_k=0 which may not be suitable
+                top_p=assistant.llm_settings.top_p or 1.0
+            )
+            llm_config = LlmConfig(
+                max_tokens=assistant.llm_settings.max_tokens or 1000,  # Default from Assistant example
+                timeout=60,  # Default, as Agent example uses 0 which may not be suitable
+                sampling=sampling
+            )
+
+        # Map revisions to models
+        models = []
+        if assistant.revisions:
+            for revision in assistant.revisions:
+                model_prompt = {"instructions": revision.prompt} if revision.prompt else None
+                model = Model(
+                    name=revision.model_name or f"model_{revision.revision_id or uuid4().hex}",
+                    llm_config=llm_config,
+                    prompt=model_prompt
+                )
+                models.append(model)
+        else:
+            # Use llm_settings.model_name if no revisions, aligning with Assistant example
+            if assistant.llm_settings and assistant.llm_settings.model_name:
+                models.append(Model(
+                    name=str(assistant.llm_settings.model_name),  # e.g., "gpt-4"
+                    llm_config=llm_config,
+                    prompt=None
+                ))
+        models = ModelList(models=models) if models else ModelList(models=[])
+
+        # Create Prompt
+        prompt = Prompt(
+            instructions=assistant.prompt or "Default instructions for the agent",
+            inputs=[],  # Assistant doesn't provide inputs, unlike Agent example
+            outputs=[],  # Assistant doesn't provide outputs, unlike Agent example
+            context=None,
+            examples=None
+        )
+
+        # Create AgentData
+        agent_data = AgentData(
+            prompt=prompt,
+            llm_config=llm_config or LlmConfig(
+                max_tokens=1000,  # Default from Assistant example
+                timeout=60,
+                sampling=Sampling(temperature=0.7, top_k=50, top_p=1.0)  # Defaults from Assistant example
+            ),
+            models=models,
+            resource_pools=None
+        )
+
+        # Use welcome_data.description for job_description if available, else fallback to description
+        job_description = (
+            assistant.welcome_data.description
+            if assistant.welcome_data and assistant.welcome_data.description
+            else assistant.description
+        )
+
+        # Create Agent
+        agent = Agent(
+            id=assistant.id or str(uuid4()),
+            status=agent_status,
+            name=assistant.name,  # Use Assistant.name directly
+            access_scope="private",  # Default from Agent example
+            public_name=None,  # Agent example uses a specific format, but Assistant doesn't provide this
+            avatar_image=None,  # Agent example has a URL, but Assistant doesn't provide this
+            description=assistant.description,
+            job_description=job_description,
+            is_draft=True,  # Default, unlike Agent example's False
+            is_readonly=False,
+            revision=assistant.revisions[0].revision_id if assistant.revisions else None,
+            version=None,
+            agent_data=agent_data
+        )
+
+        return agent
+
+    except ValidationError as e:
+        logger.error(f"Validation error while transforming assistant {assistant.id}: {e}")
+        raise
+    except ValueError as e:
+        logger.error(f"Value error while transforming assistant {assistant.id}: {e}")
+        raise
+    except Exception as e:
+        logger.error(f"Unexpected error while transforming assistant {assistant.id}: {e}")
+        raise
+
+def fetch_transform_create_agent(
+    assistant_id: str,
+    project_id: str,
+    detail: str = "full",
+    automatic_publish: bool = False
+) -> Union[Agent, ErrorListResponse]:
+    """
+    Fetch an Assistant by ID, transform it to an Agent, and create it in the specified project.
+
+    Args:
+        assistant_id (str): The ID of the Assistant to fetch.
+        project_id (str): The ID of the project where the Agent will be created.
+        detail (str, optional): The detail level for fetching Assistant ("summary" or "full"). Defaults to "full".
+        automatic_publish (bool, optional): Whether to automatically publish the created Agent. Defaults to False.
+
+    Returns:
+        Union[Agent, ErrorListResponse]: The created Agent or an error response.
+    """
+    try:
+        # Initialize managers
+        assistant_manager = AssistantManager()
+        ailab_manager = AILabManager()
+
+        # Fetch Assistant
+        logger.info(f"Fetching Assistant with ID: {assistant_id}")
+        assistant = assistant_manager.get_assistant_data(assistant_id=assistant_id, detail=detail)
+        if isinstance(assistant, ErrorListResponse):
+            logger.error(f"Failed to fetch Assistant: {assistant}")
+            return assistant
+
+        # Transform to Agent
+        logger.info(f"Transforming Assistant {assistant_id} to Agent")
+        agent = transform_assistant_to_agent(assistant)
+
+        # Create Agent
+        logger.info(f"Creating Agent in project {project_id}")
+        result = ailab_manager.create_agent(
+            project_id=project_id,
+            agent=agent,
+            automatic_publish=automatic_publish
+        )
+
+        logger.info(f"Successfully created Agent with ID: {result.id}, Name: {result.name}")
+        return result
+
+    except MissingRequirementException as e:
+        logger.error(f"Missing requirement: {e}")
+    except ValidationError as e:
+        logger.error(f"Validation error: {e}")
+    except ValueError as e:
+        logger.error(f"Value error: {e}")
+    except Exception as e:
+        logger.error(f"Unexpected error: {e}")
+
+
+if __name__ == "__main__":
+    try:
+        ASSISTANT_ID = "3806a85f-cb06-4724-8dba-da19b0a702eb"
+        PROJECT_ID = "1956c032-3c66-4435-acb8-6a06e52f819f"
+
+        result = fetch_transform_create_agent(
+            assistant_id=ASSISTANT_ID,
+            project_id=PROJECT_ID,
+            detail="full",
+            automatic_publish=True
+        )
+        if isinstance(result, Agent):
+            print(f"Created Agent: ID={result.id}, Name={result.name}")
+        else:
+            print(f"Error: {result}")
+    except Exception as e:
+        print(f"Error in main: {e}")