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

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

@@ -0,0 +1,73 @@
+
+from pygeai.core.base.responses import ErrorListResponse
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import KnowledgeBase
+
+
+def main():
+    manager = AILabManager()
+
+    # Step 1: Create a Knowledge Base
+    print("=== Creating Knowledge Base ===")
+    kb = KnowledgeBase(
+        name="TestKnowledgeBase2",
+        artifact_type_name=["Document"],
+    )
+    artifacts = ["artifact1", "artifact2"]
+    metadata = ["tag1", "tag2"]
+    # create_result = manager.create_knowledge_base(project_id, kb, artifacts, metadata)
+    create_result = manager.create_knowledge_base(kb)
+
+    if isinstance(create_result, ErrorListResponse):
+        print(f"Error creating knowledge base: {create_result}")
+        return
+    else:
+        print(f"Created knowledge base: {create_result.name} (ID: {create_result.id})")
+        created_kb = create_result
+
+    # Step 2: List Knowledge Bases
+    print("\n=== Listing Knowledge Bases ===")
+    list_result = manager.list_knowledge_bases(name=None, start=0, count=10)
+
+    if isinstance(list_result, ErrorListResponse):
+        print(f"Error listing knowledge bases: {list_result}")
+    else:
+        print(f"Found {len(list_result)} knowledge bases:")
+        for kb in list_result:
+            print(f"- {kb.name} (ID: {kb.id}, Artifact Type: {kb.artifact_type_name})")
+
+    # Step 3: Get Knowledge Base by Name
+    print("\n=== Getting Knowledge Base by Name ===")
+    get_result = manager.get_knowledge_base(kb_name=created_kb.name)
+
+    if isinstance(get_result, ErrorListResponse):
+        print(f"Error getting knowledge base: {get_result}")
+    else:
+        print(f"Retrieved knowledge base: {get_result.name} (ID: {get_result.id})")
+
+    # Step 4: Get Knowledge Base by ID
+    print("\n=== Getting Knowledge Base by ID ===")
+    get_result_by_id = manager.get_knowledge_base(kb_id=created_kb.id)
+
+    if isinstance(get_result_by_id, ErrorListResponse):
+        print(f"Error getting knowledge base by ID: {get_result_by_id}")
+    else:
+        print(f"Retrieved knowledge base by ID: {get_result_by_id.name} (ID: {get_result_by_id.id})")
+
+    # Step 5: Delete Knowledge Base
+    print("\n=== Deleting Knowledge Base ===")
+    delete_result = manager.delete_knowledge_base(kb_id=created_kb.id)
+
+    if isinstance(delete_result, ErrorListResponse):
+        print(f"Error deleting knowledge base: {delete_result}")
+    else:
+        print("Knowledge base deleted successfully")
+
+
+if __name__ == "__main__":
+    try:
+        main()
+    except ValueError as e:
+        print(f"ValueError: {e}")
+    except Exception as e:
+        print(f"Unexpected error: {e}")

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

@@ -0,0 +1,10 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import FilterSettings, AgenticProcessList
+
+manager = AILabManager()
+filter_settings = FilterSettings(id="30a2ab8c-7503-4220-b6a0-77f3ee40d365", start="0", count="100", allow_drafts=True)
+result = manager.list_processes(filter_settings=filter_settings)
+
+print(f"Found {len(result.processes)} processes:")
+for proc in result.processes:
+    print(f"Name: {proc.name}, ID: {proc.id}")

pygeai/tests/snippets/lab/runner_1.py

@@ -0,0 +1,212 @@
+import asyncio
+
+from pygeai.lab.runners import Runner
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import (
+    Agent, AgentData, Prompt, PromptExample, PromptOutput, LlmConfig, Sampling, Model, ModelList
+)
+from pygeai.core.models import ChatMessageList, ChatMessage, LlmSettings
+
+manager = AILabManager()
+project_id = "2ca6883f-6778-40bb-bcc1-85451fb11107"
+
+# Track created entities for rollback
+created_entities = {
+    "agent_id": None
+}
+
+
+def rollback():
+    """
+    Deletes created entities to clean up after execution or in case of errors.
+    """
+    print("\n=== Initiating Rollback ===")
+    if created_entities["agent_id"]:
+        print(f"Deleting agent {created_entities['agent_id']}...")
+        result = manager.delete_agent(project_id=project_id, agent_id=created_entities["agent_id"])
+        print(f"Rollback: {result}")
+    print("Rollback complete.")
+
+
+def create_translation_agent():
+    """
+    Creates and publishes an agent for translating text into ancient French.
+
+    :return: Agent - The created and published agent.
+    """
+    print("\n=== Agent Creation Flow ===")
+    print("Creating agent 'AncientFrenchTranslator' as draft...")
+    agent = Agent(
+        name="AncientFrenchTranslatorTest",
+        access_scope="private",
+        public_name="ancient_french_translator_test",
+        job_description="Translates modern text into ancient French",
+        avatar_image="https://example.com/ancient_french_avatar.png",
+        description="An agent for translating text into ancient French",
+        agent_data=AgentData(
+            prompt=Prompt(
+                instructions="Translate the provided text into ancient French, using vocabulary and grammar consistent with medieval French (circa 12th-14th century). Ensure the tone is formal and historically appropriate.",
+                inputs=["text"],
+                outputs=[
+                    PromptOutput(key="translated_text", description="Text translated into ancient French")
+                ],
+                examples=[
+                    PromptExample(
+                        input_data="Text: Hello, how are you today?",
+                        output='{"translated_text": "Salvete, comment estes-vous or?"}'
+                    ),
+                    PromptExample(
+                        input_data="Text: I am going to the market.",
+                        output='{"translated_text": "Je vais au marchié."}'
+                    )
+                ]
+            ),
+            llm_config=LlmConfig(
+                max_tokens=1000,
+                timeout=60,
+                sampling=Sampling(temperature=0.6, top_k=50, top_p=0.95)
+            ),
+            models=ModelList(models=[
+                Model(name="gpt-4-turbo"),  # Assuming a model capable of handling historical language
+                Model(name="mistral-7b")
+            ]),
+            resource_pools=None  # No tools or resource pools needed for this example
+        ),
+        is_draft=True,
+        revision=1,
+        status="pending"
+    )
+
+    # Create the agent
+    create_agent_result = manager.create_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    if isinstance(create_agent_result, Agent):
+        print(f"Success: Created Agent: {create_agent_result.name}, ID: {create_agent_result.id}")
+        created_entities["agent_id"] = create_agent_result.id
+    else:
+        print("Error: Agent creation failed:", create_agent_result)
+        rollback()
+        raise Exception("Agent creation failed")
+
+    # Update the agent with additional details
+    print("Updating agent with refined description...")
+    agent = create_agent_result
+    agent.description = "Specialized agent for accurate ancient French translations"
+    agent.job_description = "Translates modern text into historically accurate ancient French"
+
+    update_agent_result = manager.update_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    if isinstance(update_agent_result, Agent):
+        print(f"Success: Updated Agent: {update_agent_result.description}")
+    else:
+        print("Error: Agent update failed:", update_agent_result)
+        rollback()
+        raise Exception("Agent update failed")
+
+    # Publish the agent
+    print("Publishing agent revision '1'...")
+    publish_agent_result = manager.publish_agent_revision(project_id=project_id, agent_id=created_entities["agent_id"],
+                                                          revision="1")
+    if isinstance(publish_agent_result, Agent):
+        print(f"Success: Published Agent Revision: {publish_agent_result.name}")
+    else:
+        print("Error: Agent publish failed:", publish_agent_result)
+        rollback()
+        raise Exception("Agent publish failed")
+
+    # Retrieve the latest agent version
+    print("Retrieving latest agent version...")
+    latest_agent = manager.get_agent(project_id=project_id, agent_id=created_entities["agent_id"])
+    if isinstance(latest_agent, Agent):
+        print(f"Success: Latest Agent: {latest_agent.name}, Description: {latest_agent.description}")
+    else:
+        print("Error: Agent retrieval failed:", latest_agent.errors)
+        rollback()
+        raise Exception("Agent retrieval failed")
+
+    return latest_agent
+
+
+async def test_translation_agent(agent):
+    """
+    Tests the ancient French translation agent using the Runner class.
+
+    :param agent: Agent - The agent to test.
+    """
+    print("\n=== Testing Ancient French Translation Agent ===")
+
+    # Test case 1: Simple string input
+    print("Test 1: Translating a string input...")
+    user_input = "Good morning, my friend!"
+    try:
+        response = await Runner.run(
+            agent=agent,
+            user_input=user_input,
+            llm_settings={
+                "temperature": 0.6,
+                "max_tokens": 200,
+                "frequency_penalty": 0.1,
+                "presence_penalty": 0.2
+            }
+        )
+        print(f"Input: {user_input}")
+        print(f"Response: {response}")
+    except Exception as e:
+        print(f"Error in Test 1: {e}")
+
+    # Test case 2: ChatMessage input
+    print("\nTest 2: Translating a ChatMessage input...")
+    chat_message = ChatMessage(
+        role="user",
+        content="I am traveling to a distant land."
+    )
+    try:
+        response = await Runner.run(
+            agent=agent,
+            user_input=chat_message,
+            llm_settings=LlmSettings(
+                temperature=0.7,
+                max_tokens=300,
+                frequency_penalty=0.0,
+                presence_penalty=0.0
+            )
+        )
+        print(f"Input: {chat_message.content}")
+        print(f"Response: {response}")
+    except Exception as e:
+        print(f"Error in Test 2: {e}")
+
+    # Test case 3: ChatMessageList input
+    print("\nTest 3: Translating a ChatMessageList input...")
+    chat_message_list = ChatMessageList(messages=[
+        ChatMessage(role="system", content="Translate the following into ancient French with a formal tone."),
+        ChatMessage(role="user", content="The sun rises slowly over the hills.")
+    ])
+    try:
+        response = await Runner.run(
+            agent=agent,
+            user_input=chat_message_list
+            # Using default LLM settings
+        )
+        print(f"Input: _System: {chat_message_list.messages[0].content}\nUser: {chat_message_list.messages[1].content}")
+        print(f"Response: {response}")
+    except Exception as e:
+        print(f"Error in Test 3: {e}")
+
+
+async def main():
+    try:
+        # Create and publish the translation agent
+        agent = create_translation_agent()
+
+        # Test the agent with different input types
+        await test_translation_agent(agent)
+
+        print("\n=== Translation Agent Testing Completed Successfully ===")
+
+        rollback()
+    except Exception as e:
+        rollback()
+        print(f"\n# Critical error: {e}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())

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

@@ -0,0 +1,162 @@
+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
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import UploadFile
+from pygeai.core.models import ChatMessageList, ChatMessage, LlmSettings
+from pygeai.lab.models import Agent, AgentData, Prompt, PromptExample, PromptOutput, LlmConfig, Sampling, Model, ModelList
+
+logger = logging.getLogger(__name__)
+
+project_id = "2ca6883f-6778-40bb-bcc1-85451fb11107"
+organization_id = "4aa15b61-d3c7-4a5c-99b8-052d18a04ff2"
+
+lab_manager = AILabManager()
+file_manager = FileManager(organization_id=organization_id, project_id=project_id)
+chat_manager = ChatManager()
+
+created_entities = {
+    "agent_id": None,
+    "file_id": None
+}
+
+
+def rollback():
+    print("\n=== Initiating Rollback ===")
+    if created_entities["agent_id"]:
+        print(f"Deleting agent {created_entities['agent_id']}...")
+        result = lab_manager.delete_agent(agent_id=created_entities["agent_id"])
+        print(f"Rollback: {result}")
+    if created_entities["file_id"]:
+        print(f"Deleting file {created_entities['file_id']}...")
+        result = file_manager.delete_file(file_id=created_entities["file_id"])
+        print(f"Rollback: {result}")
+    print("Rollback complete.")
+
+
+def main():
+    # File Upload Flow
+    print("\n=== File Upload Flow ===")
+    print("Uploading file 'test.txt'...")
+    file_to_upload = UploadFile(
+        name="test.txt",
+        path="test.txt",
+        #folder="summaries"
+    )
+    upload_result = file_manager.upload_file(file=file_to_upload)
+    if isinstance(upload_result, UploadFileResponse) and upload_result.success:
+        print(f"Success: Uploaded File: test.txt, ID: {upload_result.id}")
+        created_entities["file_id"] = upload_result.id
+    else:
+        print("Error: File upload failed:", upload_result)
+        rollback()
+        exit()
+
+    # Retrieve File Metadata
+    print("Retrieving file metadata...")
+    file_data = file_manager.get_file_data(file_id=created_entities["file_id"])
+    if hasattr(file_data, "name"):
+        file_name = file_data.name
+        print(f"Success: Retrieved File Name: {file_name}")
+    else:
+        print("Error: File retrieval failed:", file_data)
+        rollback()
+        exit()
+
+    # Agent Flow
+    print("\n=== Agent Flow ===")
+    print("Creating agent 'FileSummaryAgent' as draft...")
+    agent_id = str(uuid4())
+    agent = Agent(
+        id=agent_id,
+        name="FileSummaryAgent2",
+        access_scope="private",
+        job_description="Reads and summarizes content from text files",
+        description="An agent designed to process and summarize file content",
+        agent_data=AgentData(
+            prompt=Prompt(
+                instructions="Read the provided text file by its ID and generate a concise summary.",
+                context="You are a helpful agent that can read and summarize text files.",
+                inputs=["file_id"],
+                outputs=[
+                    PromptOutput(key="summary", description="Summary of the file content")
+                ],
+                examples=[
+                    PromptExample(
+                        input_data=f"File ID: {created_entities['file_id']}",
+                        output='{"summary": "Summary of test.txt content."}'
+                    )
+                ]
+            ),
+            llm_config=LlmConfig(
+                max_tokens=2000,
+                timeout=60,
+                sampling=Sampling(temperature=0.8)
+            ),
+            models=ModelList(models=[
+                Model(name="gpt-4o-mini")
+            ])
+        ),
+        is_draft=True,
+        revision=1,
+        status="pending"
+    )
+    create_agent_result = lab_manager.create_agent(agent=agent, automatic_publish=False)
+    if isinstance(create_agent_result, Agent):
+        print(f"Success: Created Agent: {create_agent_result.name}, ID: {create_agent_result.id}")
+        created_entities["agent_id"] = create_agent_result.id
+    else:
+        print("Error: Agent creation failed:", create_agent_result)
+        rollback()
+        exit()
+
+    print("Publishing agent revision '1'...")
+    publish_agent_result = lab_manager.publish_agent_revision(agent_id=created_entities["agent_id"], revision="1")
+    if isinstance(publish_agent_result, Agent):
+        print(f"Success: Published Agent Revision: {publish_agent_result.name}")
+    else:
+        print("Error: Agent publish failed:", publish_agent_result)
+        rollback()
+        exit()
+
+    # Chat Completion Flow
+    print("\n=== Chat Completion Flow ===")
+    print(f"Asking agent about the uploaded file '{file_name}'...")
+    messages = ChatMessageList(messages=[
+        ChatMessage(
+            role="user",
+            content=f"Please summarize the content of the {{file:{file_name}}} ({file_name}) in 50 words or less."
+        )
+    ])
+    llm_settings = LlmSettings(
+        temperature=0.7,
+        max_tokens=500
+    )
+    chat_response = chat_manager.chat_completion(
+        model=f"saia:agent:{agent.name}",
+        messages=messages,
+        llm_settings=llm_settings
+    )
+    if hasattr(chat_response, "choices") and chat_response.choices:
+        print(f"Success: Chat Response: {chat_response.choices[0].message.content}")
+    else:
+        print("Error: Chat completion failed:", chat_response)
+        rollback()
+        exit()
+
+    print("\n=== Process Completed Successfully ===")
+
+
+if __name__ == "__main__":
+    try:
+        main()
+        #rollback()
+    except Exception as e:
+        rollback()
+        print(f"\n# Critical error: {e}")

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

@@ -0,0 +1,22 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import ReasoningStrategy, LocalizedDescription
+
+manager = AILabManager()
+strategy = ReasoningStrategy(
+    name="RSName3",
+    system_prompt="Let's think step by step.",
+    access_scope="private",
+    type="addendum",
+    localized_descriptions=[
+        LocalizedDescription(language="spanish", description="RSName spanish description"),
+        LocalizedDescription(language="english", description="RSName english description"),
+        LocalizedDescription(language="japanese", description="RSName japanese description")
+    ]
+)
+
+result = manager.create_reasoning_strategy(
+    strategy=strategy,
+    automatic_publish=True
+)
+
+print(f"Created: {result.name}, ID: {result.id}")

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

@@ -0,0 +1,10 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import ReasoningStrategy
+
+manager = AILabManager()
+
+result = manager.get_reasoning_strategy(
+    reasoning_strategy_id="2b757122-3e36-499d-909e-87074c3afc94"
+)
+
+print(f"Retrieved: {result.name}, ID: {result.id}")

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

@@ -0,0 +1,16 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import FilterSettings, ReasoningStrategyList
+
+manager = AILabManager()
+filter_settings = FilterSettings(
+    # name="RSName2",
+    start="0",
+    count="100",
+    allow_external=True,
+    access_scope="private"
+)
+
+result = manager.list_reasoning_strategies(filter_settings)
+print(f"Found {len(result.strategies)} strategies:")
+for strategy in result.strategies:
+    print(f"Name: {strategy.name}, ID: {strategy.id}")

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

@@ -0,0 +1,26 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import ReasoningStrategy, LocalizedDescription
+
+strategy = ReasoningStrategy(
+    id="2b757122-3e36-499d-909e-87074c3afc94",
+    name="RSName3",
+    system_prompt="Let's think step by step.",
+    access_scope="private",
+    type="addendum",
+    localized_descriptions=[
+        LocalizedDescription(language="spanish", description="RSName spanish description"),
+        LocalizedDescription(language="english", description="RSName english description"),
+        LocalizedDescription(language="japanese", description="RSName japanese description")
+    ]
+)
+
+manager = AILabManager()
+strategy.system_prompt = "Updated step-by-step thinking."
+
+result = manager.update_reasoning_strategy(
+    strategy=strategy,
+    automatic_publish=False,
+    upsert=False
+)
+
+print(f"Updated: {result.name}, Prompt: {result.system_prompt}")

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

@@ -0,0 +1,48 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Tool, ToolParameter
+
+parameters = [
+    ToolParameter(
+        key="input",
+        data_type="String",
+        description="some input that the tool needs.",
+        is_required=True
+    ),
+    ToolParameter(
+        key="some_nonsensitive_id",
+        data_type="String",
+        description="Configuration that is static, in the sense that whenever the tool is used, the value for this parameter is configured here. The llm will not know about it.",
+        is_required=True,
+        type="config",
+        from_secret=False,
+        value="b001e30b4016001f5f76b9ae9215ac40"
+    ),
+    ToolParameter(
+        key="api_token",
+        data_type="String",
+        description="Configuration that is static, but it is sensitive information . The value is stored in secret-manager",
+        is_required=True,
+        type="config",
+        value="0cd84dc7-f3f5-4a03-9288-cdfd8d72fde1"
+    )
+]
+
+tool = Tool(
+    name="sample_tool_v7",
+    description="a builtin tool that does something but really does nothing cos it does not exist.",
+    scope="builtin",
+    parameters=parameters
+)
+
+
+manager = AILabManager()
+
+
+result = manager.create_tool(
+    tool=tool,
+    automatic_publish=False
+)
+
+print(f"Created tool: {result.name}, ID: {result.id}")
+print(f"Description: {result.description}")
+print(f"Messages: {result.messages}")

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

@@ -0,0 +1,50 @@
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Tool, ToolParameter
+
+parameters = [
+    ToolParameter(
+        key="input",
+        data_type="String",
+        description="some input that the tool needs.",
+        is_required=True
+    ),
+    ToolParameter(
+        key="some_nonsensitive_id",
+        data_type="String",
+        description="Configuration that is static, in the sense that whenever the tool is used, the value for this parameter is configured here. The llm will not know about it.",
+        is_required=True,
+        type="config",
+        from_secret=False,
+        value="b001e30b4016001f5f76b9ae9215ac40"
+    ),
+    ToolParameter(
+        key="api_token",
+        data_type="String",
+        description="Configuration that is static, but it is sensitive information . The value is stored in secret-manager",
+        is_required=True,
+        type="config",
+        value="0cd84dc7-f3f5-4a03-9288-cdfd8d72fde1"
+    )
+]
+
+tool = Tool(
+    name="sample_tool_v5",
+    access_scope="private",
+    public_name="sample.tool.test#",
+    description="a builtin tool that does something but really does nothing cos it does not exist.",
+    scope="builtin",
+    parameters=parameters
+)
+
+
+manager = AILabManager()
+
+
+result = manager.create_tool(
+    tool=tool,
+    automatic_publish=False
+)
+
+print(f"Created tool: {result.name}, ID: {result.id}")
+print(f"Description: {result.description}")
+print(f"Messages: {result.messages}")

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

@@ -0,0 +1,21 @@
+from pygeai.core.base.responses import EmptyResponse
+from pygeai.lab.managers import AILabManager
+
+manager = AILabManager(api_key="your-api-key")
+
+
+result = manager.delete_tool(
+    tool_id="affd8ede-97c6-4083-b1f6-2b463ad4891e"
+)
+
+if isinstance(result, EmptyResponse):
+    print("Tool deleted successfully")
+else:
+    print("Errors:", result.errors)
+
+result = manager.delete_tool(
+    tool_name="sample tool V5"
+)
+
+print("Tool deleted successfully")
+