pygeai 0.3.2__py3-none-any.whl → 0.4.0__py3-none-any.whl

pygeai/tests/integration/lab/tools/test_update_tool.py

@@ -0,0 +1,268 @@
+from unittest import TestCase
+import uuid
+from pygeai.core.common.exceptions import APIError
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Tool, ToolParameter
+
+
+class TestAILabUpdateToolIntegration(TestCase):  
+    def setUp(self):
+        """
+        Set up the test environment.
+        """
+        self.ai_lab_manager = AILabManager(alias="beta")
+        self.tool_to_update = self.__load_tool()
+
+    
+    def __load_tool(self):
+        self.random_str = str(uuid.uuid4())
+        return Tool(
+            id="c77e1f2e-0322-4dd0-b6ec-aff217f1cb32",
+            name=f"sdk_project_updated_tool_{self.random_str}",
+            description=f"Tool updated for sdk testing purposes {self.random_str}",
+            scope="builtin",
+            openApi="https://raw.usercontent.com//openapi.json",
+            openApiJson={"openapi": "3.0.0","info": {"title": f"Simple API overview {self.random_str}","version": "3.0.0"}},
+            accessScope="private",
+            reportEvents="None",
+            parameters=[{
+                "key": "param", 
+                "description": f"param description {self.random_str}",
+                "type":"app", 
+                "value": f"value {self.random_str}",
+                "data_type": "String",
+                "isRequired": False
+            }]
+        )
+    
+
+    def __update_tool(self, tool: Tool = None, automatic_publish: bool = False, upsert: bool = False):
+        """
+        Helper method to update a tool.
+        """
+        return self.ai_lab_manager.update_tool(
+            tool = self.tool_to_update if tool is None else tool,
+            automatic_publish=automatic_publish,
+            upsert=upsert
+        )
+
+
+    def test_update_tool_all_fields_success(self):
+        updated_tool = self.__update_tool()
+        self.assertEqual(updated_tool.name, self.tool_to_update.name)
+        self.assertEqual(updated_tool.description, self.tool_to_update.description)
+        self.assertEqual(updated_tool.open_api_json, self.tool_to_update.open_api_json)
+        self.assertEqual(updated_tool.parameters[0].description, self.tool_to_update.parameters[0].description)
+        self.assertEqual(updated_tool.parameters[0].value, self.tool_to_update.parameters[0].value)
+        self.assertTrue(updated_tool.is_draft, "Expected tool to be in draft state after update")
+
+
+    def test_update_tool_invalid_name(self):
+        test_params = [ True, False ]
+
+        for auto_publish in test_params:
+            with self.subTest(input=auto_publish):
+                tool = self.__load_tool()
+                tool2 = self.__load_tool()
+
+                with self.assertRaises(APIError) as exception:
+                    tool.name = f"{tool.name}:invalid"
+                    self.__update_tool(tool=tool, automatic_publish=auto_publish)
+                self.assertIn(
+                    "Invalid character in name (: is not allowed).",
+                    str(exception.exception),                    
+                    f"Expected an error about invalid character (:) in tool name with autopublish {'enabled' if auto_publish else 'disabled'}"
+                )
+
+                with self.assertRaises(APIError) as exception:
+                    tool2.name = f"{tool2.name}/invalid"
+                    self.__update_tool(tool=tool2, automatic_publish=auto_publish)
+                self.assertIn(
+                    "Invalid character in name (/ is not allowed).",
+                    str(exception.exception),                    
+                    f"Expected an error about invalid character (/) in tool name with autopublish {'enabled' if auto_publish else 'disabled'}"
+                )
+
+
+    def test_update_tool_duplicated_name(self):
+        test_params = [ True, False ]
+
+        for auto_publish in test_params:
+
+            with self.subTest(input=auto_publish):
+                self.tool_to_update.name = "sdk_project_gemini_tool"
+                with self.assertRaises(APIError) as exception:
+                    self.__update_tool(automatic_publish=auto_publish)
+                self.assertIn(
+                    "Tool already exists",
+                    str(exception.exception),                    
+                    f"Expected an error about duplicated tool name with autopublish {'enabled' if auto_publish else 'disabled'}"
+                )
+
+
+    def test_update_tool_no_name(self):
+        test_params = [ True, False ]
+
+        for auto_publish in test_params:
+
+            with self.subTest(input=auto_publish):
+                self.tool_to_update.name = ""
+                with self.assertRaises(APIError) as exception:
+                    self.__update_tool(automatic_publish=auto_publish)
+                self.assertIn(
+                    "Tool name cannot be empty",
+                    str(exception.exception),                    
+                    f"Expected an error when tool name is not provided with autopublish {'enabled' if auto_publish else 'disabled'}"
+                )
+
+
+    def test_update_tool_invalid_id(self):
+        test_params = [ True, False ]
+
+        for auto_publish in test_params:
+            with self.subTest(input=auto_publish):
+                invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
+                self.tool_to_update.id = invalid_id
+                with self.assertRaises(APIError) as exception:
+                    self.__update_tool(automatic_publish=auto_publish)
+                self.assertIn(
+                    f"Tool not found [IdOrName= {invalid_id}",
+                    str(exception.exception),                    
+                    f"Expected an error when tool id is invalid and autopublish is {'enabled' if auto_publish else 'disabled'}"
+                )
+
+
+    def test_update_tool_scope(self):
+        for scope in ["builtin", "external", "api"]:
+            tool = self.__load_tool()
+            tool.scope = scope
+            result = self.__update_tool(tool=tool)
+            self.assertEqual(result.scope, scope)
+
+        
+    def test_update_tool_invalid_scope(self):
+        test_params = [ True, False ]
+        self.tool_to_update.scope = "project"
+        for auto_publish in test_params:
+            with self.subTest(input=auto_publish):                
+                with self.assertRaises(ValueError) as exception:
+                    self.__update_tool(automatic_publish=auto_publish)
+                self.assertIn(
+                    f"Scope must be one of builtin, external, api, proxied",
+                    str(exception.exception),                    
+                    f"Expected an error when tool scope is invalid and autopublish {'enabled' if auto_publish else 'disabled'}"
+                )
+
+
+    def test_update_tool_access_scope(self):
+        for access_scope in ["public", "private"]:
+            tool = self.__load_tool()
+            tool.access_scope = access_scope
+
+            if access_scope == "public":
+                tool.public_name = f"com.sdk.testing.{self.random_str}"
+
+            updated_tool = self.__update_tool(tool=tool)
+            self.assertEqual(updated_tool.access_scope, access_scope)
+
+
+    def test_update_tool_invalid_public_name(self):
+        test_params = [ True, False ]
+        self.tool_to_update.access_scope = "public"
+        self.tool_to_update.public_name = "invalid#name"
+        for auto_publish in test_params:
+            with self.subTest(input=auto_publish):                
+                with self.assertRaises(APIError) as exception:
+                    self.__update_tool(automatic_publish=auto_publish)
+                self.assertIn(
+                    "Invalid public name, it can only contain lowercase letters, numbers, periods (.), dashes (-), and underscores (_). Please remove any other characters.",
+                    str(exception.exception),                    
+                    f"Expected error when invalid public name is sent and autopublish is {'enabled' if auto_publish else 'disabled'}"
+                )
+
+    def test_update_tool_duplicate_public_name(self):
+        test_params = [ True, False ]
+        self.tool_to_update.access_scope = "public"
+        self.tool_to_update.public_name = "test.sdk.beta.tool"
+        for auto_publish in test_params:
+            with self.subTest(input=auto_publish):   
+                with self.assertRaises(Exception) as exc:
+                    self.__update_tool(automatic_publish=auto_publish)
+                self.assertIn(
+                    "Tool already exists",
+                    str(exc.exception),
+                    f"Expected error when public name is duplicated and autopublish is {'enabled' if auto_publish else 'disabled'}"
+                )
+
+
+    def test_update_tool_no_public_name(self):
+        test_params = [ True, False ]
+        self.tool_to_update.access_scope = "public"
+        self.tool_to_update.public_name = ""
+
+        for auto_publish in test_params:
+            with self.subTest(input=auto_publish):  
+                with self.assertRaises(Exception) as exc:
+                    self.__update_tool(automatic_publish=auto_publish)
+                self.assertIn(
+                    "Tool publicName is required for tools with accessScope=public.",
+                    str(exc.exception),
+                    f"Expected error when public name is not provided and autopublish is {'enabled' if auto_publish else 'disabled'}"
+                )
+   
+
+    def test_update_tool_no_open_api_nor_json(self):
+        test_params = [ True, False ]
+        self.tool_to_update.scope = "api"
+        self.tool_to_update.open_api = ""
+        self.tool_to_update.open_api_json = None
+
+        for auto_publish in test_params:
+            with self.subTest(input=auto_publish):  
+                with self.assertRaises(APIError) as exc:
+                    self.__update_tool(automatic_publish=auto_publish)
+                self.assertIn(
+                    "Either openApi (URL with the OpenAPI definition) or the openApiJson (with the conent) are required for api-tools.",
+                    str(exc.exception),
+                    f"Expected error when no openApi or openApiJson are provided and autopublish is {'enabled' if auto_publish else 'disabled'}"
+                )
+
+    
+    def test_update_tool_parameters(self):
+        self.tool_to_update.parameters = [
+            ToolParameter(key="paramA", data_type="String", description="descA", is_required=True),
+            ToolParameter(key="paramB", data_type="String", description="descB", is_required=False)
+        ]
+        result = self.__update_tool()
+        self.assertEqual(len(result.parameters), 2)
+        self.assertEqual(result.parameters[0].key, "paramA")
+        self.assertEqual(result.parameters[1].key, "paramB")
+
+
+    def test_update_tool_automatic_publish(self):
+        result = self.__update_tool(automatic_publish=True)
+        self.assertFalse(result.is_draft, "Expected tool to be published after update with automatic_publish=True")
+
+   
+    def test_update_tool_upsert(self):
+        print(self.tool_to_update)
+        self.tool_to_update.id = str(uuid.uuid4())
+        self.tool_to_update.name = str(uuid.uuid4())
+        result = self.__update_tool(upsert=True)
+
+        if isinstance(result, Tool):
+            self.ai_lab_manager.delete_tool(result.id)
+
+        self.assertEqual(result.name,  self.tool_to_update.name)
+
+
+    def test_update_tool_upsert_false_not_exists(self):
+        new_id = str(uuid.uuid4())
+        self.tool_to_update.id = new_id
+        with self.assertRaises(Exception) as exc:
+            self.__update_tool(upsert=False)
+        self.assertIn(f"Tool not found [IdOrName= {new_id}]", str(exc.exception))
+
+    
+    
+    

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

@@ -10,5 +10,4 @@ result = manager.create_sharing_link(
 
 
 print(f"Sharing link created for agent ID: {result.agent_id}")
-print(f"API Token: {result.api_token}")
-print(f"Shared Link: {result.shared_link}")
+print(f"Shared Link: {result.shared_link}")

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

@@ -28,7 +28,7 @@ parameters = [
 ]
 
 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

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-0.3.2.dist-info → pygeai-0.4.0.dist-info}/METADATA

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

{pygeai-0.3.2.dist-info → pygeai-0.4.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-pygeai/__init__.py,sha256=ZhSarc8h0U0CMhQDbY9Uw3JbqrPXfXdtXYRERhLPUzY,502
+pygeai/__init__.py,sha256=robCZ_NkO88faXr08sW9mAJ-eBXAcAUU24mINsALRVs,502
 pygeai/admin/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/admin/clients.py,sha256=2wuXSmTyg-gCbempDFCeI1yCKeOlKDBZsrFyFWxcwBg,6698
 pygeai/admin/endpoints.py,sha256=Osi8UIBhrEzKlTLF2a-q2boDUl0XMR3lQ8sDrz72TL0,747
@@ -25,10 +25,10 @@ pygeai/chat/iris.py,sha256=-9pDHQpWhR_PvbZ6rD8eMPFk46PI9sCdPQ9aAyvSexs,413
 pygeai/chat/managers.py,sha256=f0BGfu9EF0G8rUyARslZi0pyDTL2yQadav0taCljI_I,3114
 pygeai/chat/session.py,sha256=k7Y6rr9x7CfAGDI-Vt3c6eGLQX57YZ74lEVJGzwwdzw,1193
 pygeai/chat/settings.py,sha256=-B2fEemZLifdsf7_7xNmWuFZYzL-yRqefivBmv3w8j8,124
-pygeai/chat/ui.py,sha256=_t4ehG-n2-nxlc4esL09ORjmjTHhLWybUiEwGstIo-g,34620
-pygeai/cli/__init__.py,sha256=7prYyn8lejShxA0xLR3cm0k34Wfo7WHsSJGwBU-bEB0,117
+pygeai/chat/ui.py,sha256=-xvjCzBwWlvyq-C0kN2YPczl4Q0alyJamXULOlGjKRA,34595
+pygeai/cli/__init__.py,sha256=9fVRZ6_hmlv9adqGukFuS_s5Yb3jSyF8vv50-d4mbQo,117
 pygeai/cli/__main__.py,sha256=2RkQaX48mS2keTpv3-9rxk5dw35PL_deqxcKUUNhp6E,154
-pygeai/cli/geai.py,sha256=W-zhw4NSJxQ5KDOjksLREjySKWs962IGV6y3YW1tDFY,4247
+pygeai/cli/geai.py,sha256=Smqxqc3XIy8RMuFVIsS9qNuYtqkNWba_CJJrhe7kW0E,4451
 pygeai/cli/geai_proxy.py,sha256=BSoeh32fhATxbsAA_B92HKDBiLgfejEQ0XwXfeOk49g,13356
 pygeai/cli/install_man.py,sha256=DjZ3k05sKSzpLFqsU4LHz1b37NHLVdOvS6oZihdBSis,3794
 pygeai/cli/parsers.py,sha256=8kKvUbg33K4VbE7ryHwq3Uwwp42BK6ZOAKycYjEj1Io,2525
@@ -37,7 +37,7 @@ pygeai/cli/commands/admin.py,sha256=LDxUrq9qGAswT4HbaN_c_ovVKbgGct7ffjXA8zVYjWY,
 pygeai/cli/commands/assistant.py,sha256=fQ_El6_BmFDpFjm_gPxzWk7bOzhimhiTwG8K0UpcxDo,18711
 pygeai/cli/commands/base.py,sha256=kkJEObpT2xSiObWAliJfGV73JS3LjLTMq7FEd4SpxkE,6930
 pygeai/cli/commands/builders.py,sha256=xXk1F4phSQxHN3NiQltl_KEZdCwwJiKLmVqQsft2OC4,1130
-pygeai/cli/commands/chat.py,sha256=vQ-NyMHdsbTy1Rc2obT3IB-OIFn0ngFmPG76ZqB4_pY,25304
+pygeai/cli/commands/chat.py,sha256=Y2e3NjDXYjutdxbpGUDmewZfKnR_QHb69Jz4L-2GHY4,25170
 pygeai/cli/commands/common.py,sha256=zL1cWKMTqzqMsHBDFfVzbZe0i2l0hgJNpzjSRgvloY8,16568
 pygeai/cli/commands/configuration.py,sha256=J1Y8AH1xYWvcY4bzqKvF-_ggEDJ22Dv4iEcCBi2pq_8,4140
 pygeai/cli/commands/embeddings.py,sha256=om_RR3CHgfmHcTN43F6TzP71n-BS8Lf589wyBMVZJ3g,4220
@@ -57,11 +57,13 @@ pygeai/cli/commands/validators.py,sha256=lNtYSJvKFrEeg_h0oYURMuoQbNG5r6QdjzDL-aT
 pygeai/cli/commands/version.py,sha256=vyJcnxwL_TfpOQI0yE2a1ZyA3VRAE7ssh9APNBXpmqk,1701
 pygeai/cli/commands/flows/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/cli/commands/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/cli/commands/lab/ai_lab.py,sha256=oiCYN_1mLXwbOQ7rh-MXVp9xUcRFTy6b0hNHcS4ytg8,137440
+pygeai/cli/commands/lab/ai_lab.py,sha256=KJMRY_xA-B_XBM2VjgniYt5UeAbcvUCZhzb1jXUUni8,129235
 pygeai/cli/commands/lab/common.py,sha256=YBenPCVgK01Xaxgj1429bp_Ri1SN4beBxZk3dCLp7X0,6590
+pygeai/cli/commands/lab/options.py,sha256=T13Vi97zochr0cU4yjyvvwWRPENILFDYpvqpU4uEBis,165
 pygeai/cli/commands/lab/spec.py,sha256=EjNdEnljKpYPQyT4d4ViAPrM1g7oIitv6ddnWVkWXPk,8301
+pygeai/cli/commands/lab/utils.py,sha256=uxhgHvCRrV6WYRxR2qd3nED_hhXcxJ1tAU8MlzoshEg,456
 pygeai/cli/texts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/cli/texts/help.py,sha256=IgifNjO22kbRB5bsuckFrzRvg7Q14y96IfxRnlJAUcw,15516
+pygeai/cli/texts/help.py,sha256=0idAj91GTubRRXgN1t8_SRiF1Ou89ssHC5vKwONDxMM,16074
 pygeai/core/__init__.py,sha256=bbNktFp7t2dOBIvWto-uGVBW8acaKIe8EKcfuLV-HmA,189
 pygeai/core/handlers.py,sha256=la1QcQbLwfiNd-xDQ3jtWRHmeEm6E93Rfx5c-5bkLmw,934
 pygeai/core/models.py,sha256=uIwrmlB6yuulScUcYIBSH3Sxm5YyzaicV7Dz2bYLi2I,24229
@@ -74,9 +76,9 @@ pygeai/core/base/models.py,sha256=_h62nnMhJXr1BLNoaldT4d9oqCTSistfF3D2LQ3bvlg,38
 pygeai/core/base/responses.py,sha256=k-mrzNO_AtEsGTUJnyRT76FJ7gfYxQ_SAhB8MBNqPZI,763
 pygeai/core/base/session.py,sha256=WVb4MmptwdgK7paHOSvfEle_HPXRRXO8CHgi0qbgtOg,2990
 pygeai/core/common/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/core/common/config.py,sha256=zrlNF-0AE54qZ-XpR0RgYzM_7v8LtS0jUpM7UXrbSqQ,4475
+pygeai/core/common/config.py,sha256=uEPqTTonya8IBX0KSRI927cjUJ39JvYExnkqep-5U6o,4395
 pygeai/core/common/decorators.py,sha256=X7Tv5XBmsuS7oZHSmI95eX8UkuukKoiOiNRl5w9lgR4,1227
-pygeai/core/common/exceptions.py,sha256=9h_el66mS838l-FINY3gO8CrSQuFhXyljHtdrn5AOnM,1085
+pygeai/core/common/exceptions.py,sha256=-eF4V0B-27zfp0aHMlZWqWRIty6P7TCOrzMRW87ZnlE,1251
 pygeai/core/embeddings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/core/embeddings/clients.py,sha256=0r-BX4ptivIBNrsOAMgw0q5nNLrIU7UxJ3SD6MkfXX4,3543
 pygeai/core/embeddings/endpoints.py,sha256=b__cuKQjribog9PSUeDzwrQ0vBO4WyYahLhLjDiUpL0,98
@@ -142,27 +144,28 @@ pygeai/health/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/health/clients.py,sha256=U2eb1tkXt1Rf_KdV0ZFQS2k4wGnJTXHHd9-Er0eWQuw,1011
 pygeai/health/endpoints.py,sha256=UAzMcqSXZtMj4r8M8B7a_a5LT6X_jMFNsCTvcsjNTYA,71
 pygeai/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pygeai/lab/clients.py,sha256=5JebyNpKCVwAQpeOAkJPIHRf_6hoKA3uUm9xewm01UQ,816
 pygeai/lab/constants.py,sha256=ddgDnXP4GD0woi-FUJaJXzaWS3H6zmDN0B-v8utM95Q,170
-pygeai/lab/managers.py,sha256=9wV6SipzsIwFP9SXsKqZ0X5x6KbUuo6iCxPZF4zNGj4,72714
+pygeai/lab/managers.py,sha256=EGU2NTpZyWtPo-VBX_wSrNOjLNPBIt7cxha9lIbuXgk,70725
 pygeai/lab/models.py,sha256=1m41gSqpXZVO9AcPVxzlsC-TgxZcCsgGUbpN5zoDMjU,71451
 pygeai/lab/runners.py,sha256=-uaCPHpFyiKtVOxlEjPjAc9h-onSdGAcYJ5IAZPqlb0,4147
 pygeai/lab/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/lab/agents/clients.py,sha256=2SYNjyD9LZaOUxmTcrwzqcHwW7K1wAmYwen_u0G-RfU,22659
+pygeai/lab/agents/clients.py,sha256=VIbKSDxSNH1P-CX6hF7LUcgtvrE3jvO_gu4KCtyzyls,21213
 pygeai/lab/agents/endpoints.py,sha256=RpWbFwqgX_GCVn29DYM46PVon5qjD7C1SmzvcjEKMzI,696
 pygeai/lab/agents/mappers.py,sha256=K6rxsO2Nq6GglmCUmyDKUNmzTG8HRbCelap6qaVKXQw,10583
 pygeai/lab/processes/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/lab/processes/clients.py,sha256=C1YYJE7c5qJSZUIDKzo5kK-LOZP6jbdrG01aE8zoEYg,51403
+pygeai/lab/processes/clients.py,sha256=4OUf36Dhyd4hJZomeLcq9XDpLiCINwmfu6tHy69TkDE,49417
 pygeai/lab/processes/endpoints.py,sha256=nFIEcNP22xe4j6URI6KcwTh7h-xgYjYYuHT6PDPiO3I,2100
 pygeai/lab/processes/mappers.py,sha256=YOWcVKdcJmLMAq-f3qevzqQ8L_hjb0_jVXBdCHutpzk,15815
 pygeai/lab/spec/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/lab/spec/loader.py,sha256=Dq9MhLqFwF4RPdBBaqKPGqt43-PrNlsHpe-NXe4S0qQ,709
 pygeai/lab/spec/parsers.py,sha256=oG7tY-GylweRxpvtCl3p53t0IoTX3UZFiB77x__3Qp8,646
 pygeai/lab/strategies/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/lab/strategies/clients.py,sha256=UZV21W-F7CHmi53oDWDloEQ4vf9DW8G_DNpB3nWDfeg,10563
+pygeai/lab/strategies/clients.py,sha256=_a1yc8kwd50Yv4g1jqfa0gRnMiROR7Dn0gx3xqFUjVE,10316
 pygeai/lab/strategies/endpoints.py,sha256=LgEvUgeeN-X6VMl-tpl9_N12GRppLPScQmiMRk7Ri28,541
 pygeai/lab/strategies/mappers.py,sha256=6C_jubAVXMKLGQy5NUD0OX7SlrU2mLe2QsgzeJ1-WKw,2437
 pygeai/lab/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/lab/tools/clients.py,sha256=TfgociWyvzao3B6g2VfjAb6FmXp1YqmJRG2TT7RwOic,27441
+pygeai/lab/tools/clients.py,sha256=3yESr0NQoO7qKi9s7VodxaV_4qjBlJ6cT_fBmEhyT48,28151
 pygeai/lab/tools/endpoints.py,sha256=HiGoMs4OVeCgH7EAERTtifFPl53NryA1Awh7D6AO8bA,699
 pygeai/lab/tools/mappers.py,sha256=bYi5k36h0k4mCvOnV-r8YOHKz0U9P0mH21GNs20w2eM,4998
 pygeai/man/__init__.py,sha256=gqGI92vUPt6RPweoWX3mTUYPWNDlm6aGUjQOnYXqthk,53
@@ -268,15 +271,29 @@ pygeai/tests/gam/test_clients.py,sha256=vNz-4ux0cubztTY-_fEPWEoMCt5VAmZLecd0V-sE
 pygeai/tests/health/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/tests/health/test_clients.py,sha256=kfakkZHFMfo2IAN-PzmtMGmgR4iNiN1RpRopI--0qHI,1525
 pygeai/tests/integration/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pygeai/tests/integration/assistants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pygeai/tests/integration/assistants/rag/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pygeai/tests/integration/assistants/rag/test_create_rag.py,sha256=yTHlfTUi7DeZRzo4T25sqAAoS3mV5apN-Elf1DX8aoM,2121
+pygeai/tests/integration/chat/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pygeai/tests/integration/chat/test_generate_image.py,sha256=byCQQK6dIy68yPAhAa66bh7N0Xz5WnKSClx1vaIIzZA,5431
 pygeai/tests/integration/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/tests/integration/lab/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/tests/integration/lab/agents/test_agents_list.py,sha256=F2KUCdeiaBC3dn8ARNWqSz_kJcRyA0HC1nquhamN35Q,4187
-pygeai/tests/integration/lab/agents/test_create_agent.py,sha256=J3_QYCQM02LVdYZN3Y4nFmqn43_BgbV6ZELtLhZnjl0,14475
+pygeai/tests/integration/lab/agents/test_create_agent.py,sha256=JLCnrij_uS4wTvu06Ng3ODfLTikFQwmPep4HG0ebUKI,14104
 pygeai/tests/integration/lab/agents/test_create_sharing_link.py,sha256=y-e8Q_TfuLz7XXMRERSKA_-OQJUMBIsJcK0lQ0Oh858,2467
 pygeai/tests/integration/lab/agents/test_delete_agent.py,sha256=sb3RfoZJdzQvcVdNcXY2C2FO3yY1ZNiAZ_6Ay6f331E,2524
 pygeai/tests/integration/lab/agents/test_get_agent.py,sha256=oW1F6SENvhL9jZC021Rj-f_Xek2DSTx3SsZBr3YT6Hk,3666
-pygeai/tests/integration/lab/agents/test_publish_agent_revision.py,sha256=4vpuAVBenLCyWvaKTA2PQVLn_e-abGDscngUzZm3dMs,5284
-pygeai/tests/integration/lab/agents/test_update_agent.py,sha256=vme9TJij46U55oOvUYGdKRyzZn-UKzXXYGJH0-zCeRs,11959
+pygeai/tests/integration/lab/agents/test_publish_agent_revision.py,sha256=PGlYn8y2L2FUfSG3NGDPHl3ZyIiohhir1spYqO6J3xY,5182
+pygeai/tests/integration/lab/agents/test_update_agent.py,sha256=X3SpA1CPr4ZZQazsCSv8Z9rDcHJTih1c8AFKdwW80xg,11624
+pygeai/tests/integration/lab/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+pygeai/tests/integration/lab/tools/test_create_tool.py,sha256=0Yzlu9ATnsxPVPa4vO4j-j5Hc6VyHQ2KoSaf0dfmqpk,12351
+pygeai/tests/integration/lab/tools/test_delete_tool.py,sha256=wy979nZh8ERd-k3jhJTjHqG4wxWE4sx-r4yn2nBc7Aw,2913
+pygeai/tests/integration/lab/tools/test_get_parameter.py,sha256=r5TIaY_ABdyrtGfQwvs6F76dxEoy5HnvOSDwYpZo4ow,3610
+pygeai/tests/integration/lab/tools/test_get_tool.py,sha256=3fVDQlklmvOUgYDp0ATv5RqRmApgD4Qw_YGqjBOaOOo,3437
+pygeai/tests/integration/lab/tools/test_list_tools.py,sha256=KMLWXUmk_hKChKo4t5xEZtuIS1UmhAH6VdEq15KCrbY,3744
+pygeai/tests/integration/lab/tools/test_publish_tool_revision.py,sha256=AoCdHF8f7jUwAEL3PUvZMEE67m8OCcn_hqY67thJplU,4589
+pygeai/tests/integration/lab/tools/test_set_parameter.py,sha256=NqLVMlOF4QiTcc2HTdVaJuWCrEvb5QkOX_xScyjyWGU,4025
+pygeai/tests/integration/lab/tools/test_update_tool.py,sha256=tZKZDoYS6RuTdRLgAmENq5tRvK-MohJGCt-s83QoBCA,11683
 pygeai/tests/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/tests/lab/test_managers.py,sha256=AsOAvyCkRpbskEy214aV2TwrqilWH6bxOiTWDOb1twQ,29778
 pygeai/tests/lab/test_mappers.py,sha256=2cLSggf168XWFpeZeBR7uJ-8C32TKb7qA91i_9fr_b0,11409
@@ -368,9 +385,11 @@ pygeai/tests/snippets/lab/runner_1.py,sha256=QD92MvC22wpWj6YyrSgpp46EcL0ciac2x1z
 pygeai/tests/snippets/lab/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/tests/snippets/lab/agents/create_agent.py,sha256=EVyfQzDST9A9KaM0ToZJKlJ7yCbfhpSVkVK_MaUVQPw,1875
 pygeai/tests/snippets/lab/agents/create_agent_2.py,sha256=jd7HKhle_c0S0vI80AejOyLaNqBWkILlRF_znzyCGcQ,1879
+pygeai/tests/snippets/lab/agents/create_agent_edge_case.py,sha256=8dA9giNdsHjFZsIWTlBFk8e1QS3YtbZxklsVu0ZrDrk,1877
+pygeai/tests/snippets/lab/agents/create_agent_without_instructions.py,sha256=jd7HKhle_c0S0vI80AejOyLaNqBWkILlRF_znzyCGcQ,1879
 pygeai/tests/snippets/lab/agents/delete_agent.py,sha256=GfDX667_V3tZMz3vjsbrxoFZggzpwjZYH_PVO2Qjw5s,269
 pygeai/tests/snippets/lab/agents/get_agent.py,sha256=bcqloJHwmNsFjEfri6QIRaTuHzwLtfEqIQPIC5pdkWQ,516
-pygeai/tests/snippets/lab/agents/get_sharing_link.py,sha256=2mYPwMgFFbqzmWZEBrFYSjdZscVyjJYCs4A3L0x8Sr8,355
+pygeai/tests/snippets/lab/agents/get_sharing_link.py,sha256=FSXuzbX8Folh1h_6liHeZyA8T21_Y-2Ws9B6-uXatDU,316
 pygeai/tests/snippets/lab/agents/list_agents.py,sha256=gY5RBFUozhkk1uJO6opjkLmKmYpF9Ws-B7Oibhs99qQ,457
 pygeai/tests/snippets/lab/agents/publish_agent_revision.py,sha256=ATbFnnSLYRmFTC4GJJUyBwxKosoTWx661Q9FpuktREM,344
 pygeai/tests/snippets/lab/agents/update_agent.py,sha256=6qCnqkowCCygEl3tv_nvdlwXG8k0-LEloKYQUdIfkrs,1943
@@ -393,7 +412,8 @@ pygeai/tests/snippets/lab/strategies/get_reasoning_strategy.py,sha256=OxxXIRWH_c
 pygeai/tests/snippets/lab/strategies/list_reasoning_strategies.py,sha256=4pqsW16m-oiv_UUVLgZrkeWZESIfDq2nvcfNC0ZpAGo,485
 pygeai/tests/snippets/lab/strategies/update_reasoning_strategy.py,sha256=OIoHNkdnXbC9GacPgXUG1jKlVizVtWfRI-E8_3xF5b0,889
 pygeai/tests/snippets/lab/tools/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/tests/snippets/lab/tools/create_tool.py,sha256=dreldR8QIu9Q9tjOG2hkD6b8KZa_VrwSfSnCrzPgQLA,1400
+pygeai/tests/snippets/lab/tools/create_tool.py,sha256=X8qoL7TOk9Gm7jyk5MvhB5V-TBvog3TETCLNL90Z3HA,1400
+pygeai/tests/snippets/lab/tools/create_tool_edge_case.py,sha256=8w4mvoRTMTyc70yYm2bgV2dr_Rh5QpPJR8VoqX-eY-s,1465
 pygeai/tests/snippets/lab/tools/delete_tool.py,sha256=pnIkYvdP7X7Gx79AMK5MSVliIXdHSpyVwRhH3kgi7ys,452
 pygeai/tests/snippets/lab/tools/get_parameter.py,sha256=dXdlHhoWxzZIYdsvHKnLLT5Vff2Tip46XCoOo-B8Gf0,490
 pygeai/tests/snippets/lab/tools/get_tool.py,sha256=-fkKAE6nflwtLY_Lf6itZJyx_9aanFp-TSDHUzub1AM,477
@@ -482,9 +502,9 @@ pygeai/vendor/a2a/utils/helpers.py,sha256=6Tbd8SVfXvdNEk6WYmLOjrAxkzFf1aIg8dkFfB
 pygeai/vendor/a2a/utils/message.py,sha256=gc_EKO69CJ4HkR76IFgsy-kENJz1dn7CfSgWJWvt-gs,2197
 pygeai/vendor/a2a/utils/task.py,sha256=BYRA_L1HpoUGJAVlyHML0lCM9Awhf2Ovjj7oPFXKbh0,1647
 pygeai/vendor/a2a/utils/telemetry.py,sha256=VvSp1Ztqaobkmq9-3sNhhPEilJS32-JTSfKzegkj6FU,10861
-pygeai-0.3.2.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
-pygeai-0.3.2.dist-info/METADATA,sha256=bhejl4MTXN0F315fIxAdpmpaqy_aIx50RpKsP89Zgjc,6938
-pygeai-0.3.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygeai-0.3.2.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
-pygeai-0.3.2.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
-pygeai-0.3.2.dist-info/RECORD,,
+pygeai-0.4.0.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
+pygeai-0.4.0.dist-info/METADATA,sha256=1mO96mgfRX1G1mJivH1rOUJP3B6maoRX3Ckb3IK_3ig,6938
+pygeai-0.4.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygeai-0.4.0.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
+pygeai-0.4.0.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
+pygeai-0.4.0.dist-info/RECORD,,