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

pygeai/tests/snippets/files/upload_file.py

@@ -1,17 +1,13 @@
-from pygeai.core.models import Organization, Project
 from pygeai.core.files.managers import FileManager
 from pygeai.core.files.models import UploadFile
 
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
-
 file = UploadFile(
     path="test.txt",
     name="TestyFile",
     folder="TestyTestTemp"
 )
 
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
+file_manager = FileManager()
 
 response = file_manager.upload_file(file)
 print(response)

pygeai/tests/snippets/gam/gam_access_token.py

@@ -0,0 +1,87 @@
+"""
+Sample script demonstrating the usage of GAMClient to interact with GAM OAuth 2.0 endpoints.
+
+This script shows how to:
+1. Obtain an access token using username and password.
+2. Fetch user information using the access token.
+3. Refresh an access token using a refresh token.
+"""
+
+import argparse
+import getpass
+from pygeai.gam.clients import GAMClient
+
+
+def main():
+    """
+    Main function to demonstrate the usage of GAMClient for GAM OAuth 2.0 authentication.
+    """
+    parser = argparse.ArgumentParser(description="GAM OAuth 2.0 Authentication Script")
+    parser.add_argument("--username", "-u", required=True, help="Username for authentication")
+    parser.add_argument("--client-id", "--cid", required=True, help="Client ID for the application")
+    parser.add_argument("--client-secret", "--cs", required=True, help="Client secret for the application")
+    args = parser.parse_args()
+
+    password = getpass.getpass("Enter password for {}: ".format(args.username))
+
+    try:
+        gam_client = GAMClient()
+    except Exception as e:
+        print(f"Error initializing GAMClient: {e}")
+        return
+
+    # Step 1: Get Access Token
+    print("=== Step 1: Obtaining Access Token ===")
+    try:
+        access_token_response = gam_client.get_access_token(
+            client_id=args.client_id,
+            client_secret=args.client_secret,
+            username=args.username,
+            password=password,
+            scope="gam_user_data+gam_user_roles",
+            authentication_type_name="local",
+            initial_properties={"Id": "Company", "Value": "GeneXus"},
+            repository="",
+            request_token_type="OAuth"
+        )
+        print("Access Token Response:")
+        print(access_token_response)
+
+        access_token = access_token_response.get("access_token", "")
+        refresh_token = access_token_response.get("refresh_token", "")
+        if not access_token:
+            print("Error: No access token received.")
+            return
+    except Exception as e:
+        print(f"Error obtaining access token: {e}")
+        return
+
+    # Step 2: Get User Information
+    print("\n=== Step 2: Fetching User Information ===")
+    try:
+        user_info_response = gam_client.get_user_info(access_token=access_token)
+        print("User Info Response:")
+        print(user_info_response)
+    except Exception as e:
+        print(f"Error fetching user information: {e}")
+        return
+
+    # Step 3: Refresh Access Token (if refresh_token is available)
+    if refresh_token:
+        print("\n=== Step 3: Refreshing Access Token ===")
+        try:
+            refresh_token_response = gam_client.refresh_access_token(
+                client_id=args.client_id,
+                client_secret=args.client_secret,
+                refresh_token=refresh_token
+            )
+            print("Refresh Token Response:")
+            print(refresh_token_response)
+        except Exception as e:
+            print(f"Error refreshing access token: {e}")
+    else:
+        print("\n=== Step 3: Skipping Refresh Token (No refresh token available) ===")
+
+
+if __name__ == "__main__":
+    main()

pygeai/tests/snippets/lab/agentic_flow_example_1.py

@@ -24,23 +24,23 @@ def rollback():
     print("\n=== Initiating Rollback ===")
     if created_entities["instance_id"]:
         print(f"Deleting instance {created_entities['instance_id']}...")
-        result = manager.abort_instance(project_id=project_id, instance_id=created_entities["instance_id"])
+        result = manager.abort_instance(instance_id=created_entities["instance_id"])
         print(f"Rollback: {result}")
     if created_entities["process_id"]:
         print(f"Deleting process {created_entities['process_id']}...")
-        result = manager.delete_process(project_id=project_id, process_id=created_entities["process_id"])
+        result = manager.delete_process(process_id=created_entities["process_id"])
         print(f"Rollback: {result}")
     if created_entities["task_id"]:
         print(f"Deleting task {created_entities['task_id']}...")
-        result = manager.delete_task(project_id=project_id, task_id=created_entities["task_id"])
+        result = manager.delete_task(task_id=created_entities["task_id"])
         print(f"Rollback: {result}")
     if created_entities["tool_id"]:
         print(f"Deleting tool {created_entities['tool_id']}...")
-        result = manager.delete_tool(project_id=project_id, tool_id=created_entities["tool_id"])
+        result = manager.delete_tool(tool_id=created_entities["tool_id"])
         print(f"Rollback: {result}")
     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"])
+        result = manager.delete_agent(agent_id=created_entities["agent_id"])
         print(f"Rollback: {result}")
     print("Rollback complete.")
 
@@ -67,13 +67,14 @@ def main():
                 timeout=30,
                 sampling=Sampling(temperature=0.7, top_k=40, top_p=0.9)
             ),
+            strategy_name="Dynamic Prompting",
             models=ModelList(models=[
                 Model(name="gpt-4"),
                 Model(name="gpt-3.5-turbo")
             ])
         )
     )
-    create_agent_result = manager.create_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    create_agent_result = 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
@@ -84,7 +85,7 @@ def main():
 
     print("Updating agent description...")
     agent.description = "Updated agent for testing workflows with enhanced capabilities"
-    update_agent_result = manager.update_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    update_agent_result = manager.update_agent(agent=agent, automatic_publish=False)
     if isinstance(update_agent_result, Agent):
         print(f"Success: Updated Agent: {update_agent_result.description}")
     else:
@@ -93,7 +94,7 @@ def main():
         exit()
 
     print("Publishing agent revision '1'...")
-    publish_agent_result = manager.publish_agent_revision(project_id=project_id, agent_id=created_entities["agent_id"], revision="1")
+    publish_agent_result = 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:
@@ -102,7 +103,7 @@ def main():
         exit()
 
     print("Retrieving latest agent version...")
-    latest_agent = manager.get_agent(project_id=project_id, agent_id=created_entities["agent_id"])
+    latest_agent = manager.get_agent(agent_id=created_entities["agent_id"])
     if isinstance(latest_agent, Agent):
         print(f"Success: Latest Agent: {latest_agent.name}, Description: {latest_agent.description}")
     else:
@@ -121,7 +122,7 @@ def main():
             ToolParameter(key="max_length", data_type="Integer", description="Maximum length of output", is_required=False, value="100")
         ]
     )
-    create_tool_result = manager.create_tool(project_id=project_id, tool=tool, automatic_publish=False)
+    create_tool_result = manager.create_tool(tool=tool, automatic_publish=False)
     if isinstance(create_tool_result, Tool):
         print(f"Success: Created Tool: {create_tool_result.name}, ID: {create_tool_result.id}")
         created_entities["tool_id"] = create_tool_result.id
@@ -132,7 +133,7 @@ def main():
 
     print("Updating tool description...")
     tool.description = "Updated tool for enhanced text processing"
-    update_tool_result = manager.update_tool(project_id=project_id, tool=tool, automatic_publish=False)
+    update_tool_result = manager.update_tool(tool=tool, automatic_publish=False)
     if isinstance(update_tool_result, Tool):
         print(f"Success: Updated Tool: {update_tool_result.description}")
     else:
@@ -141,7 +142,7 @@ def main():
         exit()
 
     print("Publishing tool revision '1'...")
-    publish_tool_result = manager.publish_tool_revision(project_id=project_id, tool_id=created_entities["tool_id"], revision="1")
+    publish_tool_result = manager.publish_tool_revision(tool_id=created_entities["tool_id"], revision="1")
     if isinstance(publish_tool_result, Tool):
         print(f"Success: Published Tool Revision: {publish_tool_result.name}")
     else:
@@ -150,7 +151,7 @@ def main():
         exit()
 
     print("Retrieving latest tool version...")
-    latest_tool = manager.get_tool(project_id=project_id, tool_id=created_entities["tool_id"])
+    latest_tool = manager.get_tool(tool_id=created_entities["tool_id"])
     if isinstance(latest_tool, Tool):
         print(f"Success: Latest Tool: {latest_tool.name}, Description: {latest_tool.description}")
     else:
@@ -166,7 +167,7 @@ def main():
         description="Processes text data using an agent",
         title_template="Text Processing Task #{{id}}"
     )
-    create_task_result = manager.create_task(project_id=project_id, task=task, automatic_publish=False)
+    create_task_result = manager.create_task(task=task, automatic_publish=False)
     if isinstance(create_task_result, Task):
         print(f"Success: Created Task: {create_task_result.name}, ID: {create_task_result.id}")
         created_entities["task_id"] = create_task_result.id
@@ -180,7 +181,7 @@ def main():
 
     print("Updating task description...")
     task.description = "Updated task for text processing with agent collaboration"
-    update_task_result = manager.update_task(project_id=project_id, task=task, automatic_publish=False)
+    update_task_result = manager.update_task(task=task, automatic_publish=False)
     if isinstance(update_task_result, Task):
         print(f"Success: Updated Task: {update_task_result.description}")
     else:
@@ -189,7 +190,7 @@ def main():
         exit()
 
     print("Publishing task revision '1'...")
-    publish_task_result = manager.publish_task_revision(project_id=project_id, task_id=created_entities["task_id"], revision="1")
+    publish_task_result = manager.publish_task_revision(task_id=created_entities["task_id"], revision="1")
     if isinstance(publish_task_result, Task):
         print(f"Success: Published Task Revision: {publish_task_result.name}")
     else:
@@ -198,7 +199,7 @@ def main():
         exit()
 
     print("Retrieving latest task version...")
-    latest_task = manager.get_task(project_id=project_id, task_id=created_entities["task_id"])
+    latest_task = manager.get_task(task_id=created_entities["task_id"])
     if isinstance(latest_task, Task):
         print(f"Success: Latest Task: {latest_task.name}, Description: {latest_task.description}")
     else:
@@ -238,7 +239,7 @@ def main():
             SequenceFlow(key="flow4", source_key="analysis_complete", target_key="end")
         ]
     )
-    create_process_result = manager.create_process(project_id=project_id, process=process, automatic_publish=False)
+    create_process_result = manager.create_process(process=process, automatic_publish=False)
     if isinstance(create_process_result, AgenticProcess):
         print(f"Success: Created Process: {create_process_result.name}, ID: {create_process_result.id}")
         created_entities["process_id"] = create_process_result.id
@@ -249,7 +250,7 @@ def main():
 
     print("Updating process description...")
     process.description = "Updated process for advanced text analysis"
-    update_process_result = manager.update_process(project_id=project_id, process=process, automatic_publish=False)
+    update_process_result = manager.update_process(process=process, automatic_publish=False)
     if isinstance(update_process_result, AgenticProcess):
         print(f"Success: Updated Process: {update_process_result.description}")
     else:
@@ -258,7 +259,7 @@ def main():
         exit()
 
     print("Publishing process revision '1'...")
-    publish_process_result = manager.publish_process_revision(project_id=project_id, process_id=created_entities["process_id"], revision="1")
+    publish_process_result = manager.publish_process_revision(process_id=created_entities["process_id"], revision="1")
     if isinstance(publish_process_result, AgenticProcess):
         print(f"Success: Published Process Revision: {publish_process_result.name}")
     else:
@@ -267,7 +268,7 @@ def main():
         exit()
 
     print("Retrieving latest process version...")
-    latest_process = manager.get_process(project_id=project_id, process_id=created_entities["process_id"])
+    latest_process = manager.get_process(process_id=created_entities["process_id"])
     if isinstance(latest_process, AgenticProcess):
         print(f"Success: Latest Process: {latest_process.name}, Description: {latest_process.description}")
     else:
@@ -296,7 +297,7 @@ def main():
         exit()
 
     print("Retrieving instance information...")
-    instance_info = manager.get_instance(project_id=project_id, instance_id=created_entities["instance_id"])
+    instance_info = manager.get_instance(instance_id=created_entities["instance_id"])
     if isinstance(instance_info, ProcessInstance):
         print(f"Success: Instance Info: Process: {instance_info.process.name}, Subject: {instance_info.subject}")
     else:
@@ -305,7 +306,7 @@ def main():
         exit()
 
     print("Retrieving instance history...")
-    history = manager.get_instance_history(project_id=project_id, instance_id=created_entities["instance_id"])
+    history = manager.get_instance_history(instance_id=created_entities["instance_id"])
     if isinstance(history, dict):
         print(f"Success: Instance History: {history}")
     else:
@@ -320,5 +321,6 @@ if __name__ == "__main__":
     try:
         main()
     except Exception as e:
+        breakpoint()
         rollback()
         print(f"\n# Critical error: {e}")

pygeai/tests/snippets/lab/agentic_flow_example_4.py

@@ -22,23 +22,23 @@ def rollback():
     print("\n=== Initiating Rollback ===")
     if created_entities["instance_id"]:
         print(f"Deleting instance {created_entities['instance_id']}...")
-        result = manager.abort_instance(project_id=project_id, instance_id=created_entities["instance_id"])
+        result = manager.abort_instance(instance_id=created_entities["instance_id"])
         print(f"Rollback: {result}")
     if created_entities["process_id"]:
         print(f"Deleting process {created_entities['process_id']}...")
-        result = manager.delete_process(project_id=project_id, process_id=created_entities["process_id"])
+        result = manager.delete_process(process_id=created_entities["process_id"])
         print(f"Rollback: {result}")
     if created_entities["task_id"]:
         print(f"Deleting task {created_entities['task_id']}...")
-        result = manager.delete_task(project_id=project_id, task_id=created_entities["task_id"])
+        result = manager.delete_task(task_id=created_entities["task_id"])
         print(f"Rollback: {result}")
     if created_entities["tool_id"]:
         print(f"Deleting tool {created_entities['tool_id']}...")
-        result = manager.delete_tool(project_id=project_id, tool_id=created_entities["tool_id"])
+        result = manager.delete_tool(tool_id=created_entities["tool_id"])
         print(f"Rollback: {result}")
     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"])
+        result = manager.delete_agent(agent_id=created_entities["agent_id"])
         print(f"Rollback: {result}")
     print("Rollback complete.")
 
@@ -82,7 +82,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_tool_result = manager.create_tool(project_id=project_id, tool=tool, automatic_publish=False)
+    create_tool_result = manager.create_tool(tool=tool, automatic_publish=False)
     if isinstance(create_tool_result, Tool):
         print(f"Success: Created Tool: {create_tool_result.name}, ID: {create_tool_result.id}")
         created_entities["tool_id"] = create_tool_result.id
@@ -107,7 +107,7 @@ def main():
             value="English"
         )
     )
-    update_tool_result = manager.update_tool(project_id=project_id, tool=tool, automatic_publish=False)
+    update_tool_result = manager.update_tool(tool=tool, automatic_publish=False)
     if isinstance(update_tool_result, Tool):
         print(f"Success: Updated Tool: {update_tool_result.description}")
     else:
@@ -116,7 +116,7 @@ def main():
         exit()
 
     print("Publishing tool revision '1'...")
-    publish_tool_result = manager.publish_tool_revision(project_id=project_id, tool_id=created_entities["tool_id"], revision="1")
+    publish_tool_result = manager.publish_tool_revision(tool_id=created_entities["tool_id"], revision="1")
     if isinstance(publish_tool_result, Tool):
         print(f"Success: Published Tool Revision: {publish_tool_result.name}")
     else:
@@ -125,7 +125,7 @@ def main():
         exit()
 
     print("Retrieving latest tool version...")
-    latest_tool = manager.get_tool(project_id=project_id, tool_id=created_entities["tool_id"])
+    latest_tool = manager.get_tool(tool_id=created_entities["tool_id"])
     if isinstance(latest_tool, Tool):
         print(f"Success: Latest Tool: {latest_tool.name}, Description: {latest_tool.description}")
     else:
@@ -184,7 +184,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_agent_result = manager.create_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    create_agent_result = 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
@@ -200,7 +200,7 @@ def main():
     agent.description = "Improved agent for text summarization tasks"
     agent.job_description = "Summarizes text and extracts key insights"
 
-    update_agent_result = manager.update_agent(project_id=project_id, agent=agent, automatic_publish=False)
+    update_agent_result = manager.update_agent(agent=agent, automatic_publish=False)
     if isinstance(update_agent_result, Agent):
         print(f"Success: Updated Agent: {update_agent_result.description}")
     else:
@@ -209,7 +209,7 @@ def main():
         exit()
 
     print("Publishing agent revision '1'...")
-    publish_agent_result = manager.publish_agent_revision(project_id=project_id, agent_id=created_entities["agent_id"], revision="1")
+    publish_agent_result = 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:
@@ -218,7 +218,7 @@ def main():
         exit()
 
     print("Retrieving latest agent version...")
-    latest_agent = manager.get_agent(project_id=project_id, agent_id=created_entities["agent_id"])
+    latest_agent = manager.get_agent(agent_id=created_entities["agent_id"])
     if isinstance(latest_agent, Agent):
         print(f"Success: Latest Agent: {latest_agent.name}, Description: {latest_agent.description}")
     else:
@@ -259,7 +259,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_task_result = manager.create_task(project_id=project_id, task=task, automatic_publish=False)
+    create_task_result = manager.create_task(task=task, automatic_publish=False)
     if isinstance(create_task_result, Task):
         print(f"Success: Created Task: {create_task_result.name}, ID: {create_task_result.id}")
         created_entities["task_id"] = create_task_result.id
@@ -275,7 +275,7 @@ def main():
     task.description = "Enhanced task for text summarization and analysis"
     task.prompt_data.instructions = "Summarize text and highlight key insights."
 
-    update_task_result = manager.update_task(project_id=project_id, task=task, automatic_publish=False)
+    update_task_result = manager.update_task(task=task, automatic_publish=False)
     if isinstance(update_task_result, Task):
         print(f"Success: Updated Task: {update_task_result.description}")
     else:
@@ -284,7 +284,7 @@ def main():
         exit()
 
     print("Publishing task revision '1'...")
-    publish_task_result = manager.publish_task_revision(project_id=project_id, task_id=created_entities["task_id"], revision="1")
+    publish_task_result = manager.publish_task_revision(task_id=created_entities["task_id"], revision="1")
     if isinstance(publish_task_result, Task):
         print(f"Success: Published Task Revision: {publish_task_result.name}")
     else:
@@ -293,7 +293,7 @@ def main():
         exit()
 
     print("Retrieving latest task version...")
-    latest_task = manager.get_task(project_id=project_id, task_id=created_entities["task_id"])
+    latest_task = manager.get_task(task_id=created_entities["task_id"])
     if isinstance(latest_task, Task):
         print(f"Success: Latest Task: {latest_task.name}, Description: {latest_task.description}")
     else:
@@ -358,7 +358,7 @@ def main():
         revision=1,
         status="pending"
     )
-    create_process_result = manager.create_process(project_id=project_id, process=process, automatic_publish=False)
+    create_process_result = manager.create_process(process=process, automatic_publish=False)
     if isinstance(create_process_result, AgenticProcess):
         print(f"Success: Created Process: {create_process_result.name}, ID: {create_process_result.id}")
         created_entities["process_id"] = create_process_result.id
@@ -373,7 +373,7 @@ def main():
     process.description = "Optimized process for text summarization workflows"
     process.agentic_activities[0].name = "Advanced Text Summarization"
 
-    update_process_result = manager.update_process(project_id=project_id, process=process, automatic_publish=False)
+    update_process_result = manager.update_process(process=process, automatic_publish=False)
     if isinstance(update_process_result, AgenticProcess):
         print(f"Success: Updated Process: {update_process_result.description}")
     else:
@@ -382,7 +382,7 @@ def main():
         exit()
 
     print("Publishing process revision '1'...")
-    publish_process_result = manager.publish_process_revision(project_id=project_id, process_id=created_entities["process_id"], revision="1")
+    publish_process_result = manager.publish_process_revision(process_id=created_entities["process_id"], revision="1")
     if isinstance(publish_process_result, AgenticProcess):
         print(f"Success: Published Process Revision: {publish_process_result.name}")
     else:
@@ -391,7 +391,7 @@ def main():
         exit()
 
     print("Retrieving latest process version...")
-    latest_process = manager.get_process(project_id=project_id, process_id=created_entities["process_id"])
+    latest_process = manager.get_process(process_id=created_entities["process_id"])
     if isinstance(latest_process, AgenticProcess):
         print(f"Success: Latest Process: {latest_process.name}, Description: {latest_process.description}")
     else:
@@ -417,7 +417,7 @@ def main():
         exit()
 
     print("Retrieving instance information...")
-    instance_info = manager.get_instance(project_id=project_id, instance_id=created_entities["instance_id"])
+    instance_info = manager.get_instance(instance_id=created_entities["instance_id"])
     if isinstance(instance_info, ProcessInstance):
         print(f"Success: Instance Info: Process: {instance_info.process.name}, Subject: {instance_info.subject}")
     else:
@@ -426,7 +426,7 @@ def main():
         exit()
 
     print("Retrieving instance history...")
-    history = manager.get_instance_history(project_id=project_id, instance_id=created_entities["instance_id"])
+    history = manager.get_instance_history(instance_id=created_entities["instance_id"])
     if isinstance(history, dict):
         print(f"Success: Instance History: {history}")
     else:

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

@@ -3,9 +3,9 @@ from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model, Sampli
 
 agent = Agent(
     status="active",
-    name="Public Translator V24",
-    access_scope="public",
-    public_name="com.genexus.geai.public_translator_24",
+    name="Private Translator V6",
+    access_scope="private",
+    # public_name="com.genexus.geai.public_translator_24",
     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.",
@@ -26,6 +26,7 @@ agent = Agent(
                 PromptExample(input_data="esto es una prueba pincheguey [keep-slang]", output='{"translated_text":"this is a test pal","summary":"prueba"}')
             ]
         ),
+        strategy_name="Dynamic Prompting",
         llm_config=LlmConfig(
             max_tokens=5000,
             timeout=0,
@@ -38,14 +39,10 @@ agent = Agent(
 
 manager = AILabManager()
 result = manager.create_agent(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     agent=agent,
     automatic_publish=False
 )
 
 
-if isinstance(result, Agent):
-    print(f"Agent: {agent.to_dict()}")
-else:
-    print("Errors:", result.errors)
+print(f"Agent: {agent.to_dict()}")
 

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

@@ -39,14 +39,10 @@ agent = Agent(
 
 manager = AILabManager()
 result = manager.create_agent(
-    project_id="2ca6883f-6778-40bb-bcc1-85451fb11107",
     agent=agent,
     automatic_publish=False
 )
 
 
-if isinstance(result, Agent):
-    print(f"Agent: {agent.to_dict()}")
-else:
-    print("Errors:", result.errors)
+print(f"Agent: {agent.to_dict()}")
 

pygeai/tests/snippets/lab/agents/create_agent_edge_case.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#",
+    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/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}")