pygeai 0.1.51b3__py3-none-any.whl → 0.6.0b15__py3-none-any.whl

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

@@ -1,4 +1,3 @@
-from typing import Union
 
 from pygeai.core.base.responses import ErrorListResponse
 from pygeai.lab.managers import AILabManager
@@ -6,7 +5,6 @@ from pygeai.lab.models import KnowledgeBase
 
 
 def main():
-    project_id = "2ca6883f-6778-40bb-bcc1-85451fb11107"
     manager = AILabManager()
 
     # Step 1: Create a Knowledge Base
@@ -18,7 +16,7 @@ def main():
     artifacts = ["artifact1", "artifact2"]
     metadata = ["tag1", "tag2"]
     # create_result = manager.create_knowledge_base(project_id, kb, artifacts, metadata)
-    create_result = manager.create_knowledge_base(project_id, kb)
+    create_result = manager.create_knowledge_base(kb)
 
     if isinstance(create_result, ErrorListResponse):
         print(f"Error creating knowledge base: {create_result}")
@@ -29,7 +27,7 @@ def main():
 
     # Step 2: List Knowledge Bases
     print("\n=== Listing Knowledge Bases ===")
-    list_result = manager.list_knowledge_bases(project_id, name=None, start=0, count=10)
+    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}")
@@ -40,7 +38,7 @@ def main():
 
     # Step 3: Get Knowledge Base by Name
     print("\n=== Getting Knowledge Base by Name ===")
-    get_result = manager.get_knowledge_base(project_id, kb_name=created_kb.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}")
@@ -49,7 +47,7 @@ def main():
 
     # Step 4: Get Knowledge Base by ID
     print("\n=== Getting Knowledge Base by ID ===")
-    get_result_by_id = manager.get_knowledge_base(project_id, kb_id=created_kb.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}")
@@ -58,7 +56,7 @@ def main():
 
     # Step 5: Delete Knowledge Base
     print("\n=== Deleting Knowledge Base ===")
-    delete_result = manager.delete_knowledge_base(project_id, kb_id=created_kb.id)
+    delete_result = manager.delete_knowledge_base(kb_id=created_kb.id)
 
     if isinstance(delete_result, ErrorListResponse):
         print(f"Error deleting knowledge base: {delete_result}")

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

@@ -3,10 +3,8 @@ 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(project_id="2ca6883f-6778-40bb-bcc1-85451fb11107", filter_settings=filter_settings)
-if isinstance(result, AgenticProcessList):
-    print(f"Found {len(result.processes)} processes:")
-    for proc in result.processes:
-        print(f"Name: {proc.name}, ID: {proc.id}")
-else:
-    print("Errors:", result.errors)
+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

@@ -1,6 +1,6 @@
 import asyncio
 
-from pygeai.lab import Runner
+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

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

@@ -31,7 +31,7 @@ def rollback():
     print("\n=== Initiating Rollback ===")
     if created_entities["agent_id"]:
         print(f"Deleting agent {created_entities['agent_id']}...")
-        result = lab_manager.delete_agent(project_id=project_id, agent_id=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']}...")
@@ -107,7 +107,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_agent_result = lab_manager.create_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    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
@@ -117,7 +117,7 @@ def main():
         exit()
 
     print("Publishing agent revision '1'...")
-    publish_agent_result = lab_manager.publish_agent_revision(project_id=project_id, agent_id=created_entities["agent_id"], 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:

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

@@ -15,11 +15,8 @@ strategy = ReasoningStrategy(
 )
 
 result = manager.create_reasoning_strategy(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     strategy=strategy,
     automatic_publish=True
 )
-if isinstance(result, ReasoningStrategy):
-    print(f"Created: {result.name}, ID: {result.id}")
-else:
-    print("Errors:", result.errors)
+
+print(f"Created: {result.name}, ID: {result.id}")

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

@@ -4,10 +4,7 @@ from pygeai.lab.models import ReasoningStrategy
 manager = AILabManager()
 
 result = manager.get_reasoning_strategy(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     reasoning_strategy_id="2b757122-3e36-499d-909e-87074c3afc94"
 )
-if isinstance(result, ReasoningStrategy):
-    print(f"Retrieved: {result.name}, ID: {result.id}")
-else:
-    print("Errors:", result.errors)
+
+print(f"Retrieved: {result.name}, ID: {result.id}")

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

@@ -11,9 +11,6 @@ filter_settings = FilterSettings(
 )
 
 result = manager.list_reasoning_strategies(filter_settings)
-if isinstance(result, ReasoningStrategyList):
-    print(f"Found {len(result.strategies)} strategies:")
-    for strategy in result.strategies:
-        print(f"Name: {strategy.name}, ID: {strategy.id}")
-else:
-    print("Errors:", result.errors)
+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

@@ -18,12 +18,9 @@ manager = AILabManager()
 strategy.system_prompt = "Updated step-by-step thinking."
 
 result = manager.update_reasoning_strategy(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     strategy=strategy,
     automatic_publish=False,
     upsert=False
 )
-if isinstance(result, ReasoningStrategy):
-    print(f"Updated: {result.name}, Prompt: {result.system_prompt}")
-else:
-    print("Errors:", result.errors)
+
+print(f"Updated: {result.name}, Prompt: {result.system_prompt}")

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

@@ -23,13 +23,12 @@ parameters = [
         description="Configuration that is static, but it is sensitive information . The value is stored in secret-manager",
         is_required=True,
         type="config",
-        from_secret=True,
         value="0cd84dc7-f3f5-4a03-9288-cdfd8d72fde1"
     )
 ]
 
 tool = Tool(
-    name="sample tool V5",
+    name="sample_tool_v7",
     description="a builtin tool that does something but really does nothing cos it does not exist.",
     scope="builtin",
     parameters=parameters
@@ -40,15 +39,10 @@ manager = AILabManager()
 
 
 result = manager.create_tool(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     tool=tool,
-    open_api_json=None,
     automatic_publish=False
 )
 
-if isinstance(result, Tool):
-    print(f"Created tool: {result.name}, ID: {result.id}")
-    print(f"Description: {result.description}")
-    print(f"Messages: {result.messages}")
-else:
-    print("Errors:", result.errors)
+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

@@ -5,7 +5,6 @@ manager = AILabManager(api_key="your-api-key")
 
 
 result = manager.delete_tool(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     tool_id="affd8ede-97c6-4083-b1f6-2b463ad4891e"
 )
 
@@ -15,11 +14,8 @@ else:
     print("Errors:", result.errors)
 
 result = manager.delete_tool(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     tool_name="sample tool V5"
 )
 
-if isinstance(result, EmptyResponse):
-    print("Tool deleted successfully")
-else:
-    print("Errors:", result.errors)
+print("Tool deleted successfully")
+

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

@@ -10,14 +10,12 @@ filter_settings = FilterSettings(
 )
 
 result = manager.get_parameter(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     tool_id="affd8ede-97c6-4083-b1f6-2b463ad4891e",
     filter_settings=filter_settings
 )
 
-if isinstance(result, list):
-    print(f"Retrieved {len(result)} parameters:")
-    for param in result:
-        print(f"Key: {param.key}, Type: {param.data_type}, Required: {param.is_required}")
-else:
-    print("Errors:", result.errors)
+
+print(f"Retrieved {len(result)} parameters:")
+for param in result:
+    print(f"Key: {param.key}, Type: {param.data_type}, Required: {param.is_required}")
+

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

@@ -11,14 +11,12 @@ filter_settings = FilterSettings(
 
 
 result = manager.get_tool(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     tool_id="affd8ede-97c6-4083-b1f6-2b463ad4891e",
     filter_settings=filter_settings
 )
 
-if isinstance(result, Tool):
-    print(f"Retrieved tool: {result.name}, ID: {result.id}")
-    print(f"Description: {result.description}")
-    print(f"Messages: {result.messages}")
-else:
-    print("Errors:", result.errors)
+
+print(f"Retrieved tool: {result.name}, ID: {result.id}")
+print(f"Description: {result.description}")
+print(f"Messages: {result.messages}")
+

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

@@ -14,14 +14,10 @@ filter_settings = FilterSettings(
 
 
 result = manager.list_tools(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     filter_settings=filter_settings
 )
 
+print(f"Found {len(result)} tools:")
+for tool in result:
+    print(f"Tool: {tool.name}, ID: {tool.id}, Scope: {tool.scope}")
 
-if isinstance(result, list):
-    print(f"Found {len(result)} tools:")
-    for tool in result:
-        print(f"Tool: {tool.name}, ID: {tool.id}, Scope: {tool.scope}")
-else:
-    print("Errors:", result.errors)

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

@@ -4,12 +4,10 @@ from pygeai.lab.models import Tool
 manager = AILabManager()
 
 result = manager.publish_tool_revision(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     tool_id="affd8ede-97c6-4083-b1f6-2b463ad4891e",
     revision="1"
 )
 
-if isinstance(result, Tool):
-    print(f"Published tool: {result.name}, ID: {result.id}, Revision: {result.revision}")
-else:
-    print("Errors:", result.errors)
+
+print(f"Published tool: {result.name}, ID: {result.id}, Revision: {result.revision}")
+

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

@@ -23,16 +23,11 @@ parameters = [
 ]
 
 result = manager.set_parameter(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     tool_id="affd8ede-97c6-4083-b1f6-2b463ad4891e",
     parameters=parameters
 )
 
-if isinstance(result, Tool):
-    print(f"Updated tool: {result.name}, ID: {result.id}")
-    for param in result.parameters:
-        print(f"Parameter: {param.key}, Value: {param.value}")
-elif isinstance(result, ErrorListResponse):
-    print("Errors:", result.errors)
-else:
-    print(f"Result: {result}")
+
+print(f"Updated tool: {result.name}, ID: {result.id}")
+for param in result.parameters:
+    print(f"Parameter: {param.key}, Value: {param.value}")

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

@@ -41,16 +41,12 @@ manager = AILabManager()
 
 
 result = manager.update_tool(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     tool=tool,
-    open_api_json=None,
     automatic_publish=False,
     upsert=False
 )
 
-if isinstance(result, Tool):
-    print(f"Updated tool: {result.name}, ID: {result.id}")
-    print(f"Description: {result.description}")
-    print(f"Messages: {result.messages}")
-else:
-    print("Errors:", result.errors)
+
+print(f"Updated tool: {result.name}, ID: {result.id}")
+print(f"Description: {result.description}")
+print(f"Messages: {result.messages}")