pygeai 0.4.0b10__py3-none-any.whl → 0.4.0b12__py3-none-any.whl

pygeai/chat/ui.py

@@ -1,5 +1,4 @@
 import logging
-from pathlib import Path
 
 import streamlit as st
 import argparse

pygeai/cli/commands/chat.py

@@ -362,7 +362,7 @@ def chat_with_agent(option_list: list):
         raise MissingRequirementException(f"Agent name must be specified.")
 
     project_id = get_project_id()
-    agent_data = AgentClient().get_agent(project_id=project_id, agent_id=agent_name)
+    agent_data = AgentClient(project_id=project_id).get_agent(agent_id=agent_name)
     if 'errors' in agent_data:
         raise InvalidAgentException(f"There is no agent with that name: {agent_data.get('errors')}")
 
@@ -374,62 +374,54 @@ def chat_with_agent(option_list: list):
             Console.write_stderr("Streamlit is required for GUI mode. Install it with 'pip install streamlit'.")
             sys.exit(1)
 
-        if use_gui:
-            try:
-                import streamlit
-            except ImportError:
-                logger.error("Streamlit not installed")
-                Console.write_stderr("Streamlit is required for GUI mode. Install it with 'pip install streamlit'.")
-                sys.exit(1)
+        try:
+            ui_path = resources.files("pygeai.chat").joinpath("ui.py")
+            ui_file_path = str(ui_path)
+
+            # Add the top-level project root to PYTHONPATH
+            package_root = str(Path(ui_file_path).resolve().parents[2])
+            env = os.environ.copy()
+            env["PYTHONPATH"] = package_root + os.pathsep + env.get("PYTHONPATH", "")
+
+            streamlit_cmd = [
+                sys.executable, "-m", "streamlit", "run", ui_file_path,
+                "--server.address", "127.0.0.1",
+                "--", "--agent-name", agent_name
+            ]
+
+            if platform.system() == "Linux":
+                streamlit_cmd.insert(5, "--server.headless=true")
+
+            process = subprocess.Popen(
+                streamlit_cmd,
+                stdout=subprocess.PIPE,
+                stderr=subprocess.PIPE,
+                text=True
+            )
 
-            try:
-                ui_path = resources.files("pygeai.chat").joinpath("ui.py")
-                ui_file_path = str(ui_path)
-
-                # Add the top-level project root to PYTHONPATH
-                package_root = str(Path(ui_file_path).resolve().parents[2])
-                env = os.environ.copy()
-                env["PYTHONPATH"] = package_root + os.pathsep + env.get("PYTHONPATH", "")
-
-                streamlit_cmd = [
-                    sys.executable, "-m", "streamlit", "run", ui_file_path,
-                    "--server.address", "127.0.0.1",
-                    "--", "--agent-name", agent_name
-                ]
-
-                if platform.system() == "Linux":
-                    streamlit_cmd.insert(5, "--server.headless=true")
-
-                process = subprocess.Popen(
-                    streamlit_cmd,
-                    stdout=subprocess.PIPE,
-                    stderr=subprocess.PIPE,
-                    text=True
-                )
-
-                url = "http://localhost:8501"
-                if platform.system() == "Linux":
-                    Console.write_stdout(f"Open Streamlit app at {url} (or next port like 8502 if 8501 is taken)")
-
-                try:
-                    stdout, stderr = process.communicate()
-                    if stderr:
-                        logger.error(f"Streamlit stderr:\n{stderr}")
-                        Console.write_stderr(f"Streamlit error:\n{stderr}")
-                except KeyboardInterrupt:
-                    process.terminate()
-                    Console.write_stdout("Streamlit stopped.")
-                    sys.exit(0)
+            url = "http://localhost:8501"
+            if platform.system() == "Linux":
+                Console.write_stdout(f"Open Streamlit app at {url} (or next port like 8502 if 8501 is taken)")
 
-            except FileNotFoundError:
-                logger.error("Could not locate pygeai/chat/ui.py")
-                Console.write_stderr("Streamlit UI file not found. Ensure pygeai is installed correctly.")
-                sys.exit(1)
-            except Exception as e:
-                logger.error(f"Streamlit error: {e}")
-                Console.write_stderr(
-                    f"Failed to launch Streamlit. Check port with 'lsof -i :8501' and kill any process, or try {url}.")
-                sys.exit(1)
+            try:
+                stdout, stderr = process.communicate()
+                if stderr:
+                    logger.error(f"Streamlit stderr:\n{stderr}")
+                    Console.write_stderr(f"Streamlit error:\n{stderr}")
+            except KeyboardInterrupt:
+                process.terminate()
+                Console.write_stdout("Streamlit stopped.")
+                sys.exit(0)
+
+        except FileNotFoundError:
+            logger.error("Could not locate pygeai/chat/ui.py")
+            Console.write_stderr("Streamlit UI file not found. Ensure pygeai is installed correctly.")
+            sys.exit(1)
+        except Exception as e:
+            logger.error(f"Streamlit error: {e}")
+            Console.write_stderr(
+                f"Failed to launch Streamlit. Check port with 'lsof -i :8501' and kill any process, or try {url}.")
+            sys.exit(1)
     else:
         chat_session = AgentChatSession(agent_name)
         messages = list()

pygeai/cli/commands/lab/ai_lab.py

@@ -15,7 +15,6 @@ from pygeai.lab.processes.clients import AgenticProcessClient
 from pygeai.lab.strategies.clients import ReasoningStrategyClient
 from pygeai.lab.tools.clients import ToolClient
 from pygeai.lab.constants import VALID_SCOPES
-from pygeai.tests.snippets.lab.crud_ui import project_id
 
 
 def show_help():

pygeai/core/common/config.py

@@ -114,9 +114,7 @@ def get_settings():
 
 if __name__ == "__main__":
     settings = get_settings()
-    geai_api_key = settings.get_api_key()
     geai_base_url = settings.get_base_url()
     geai_eval_url = settings.get_eval_url()
-    print(f"api_key: {geai_api_key}")
     print(f"base_url: {geai_base_url}")
     print(f"eval_url: {geai_eval_url}")

pygeai/lab/managers.py

@@ -559,7 +559,7 @@ class AILabManager:
             tool_id: Optional[str] = None,
             tool_public_name: Optional[str] = None,
             parameters: List[ToolParameter] = None
-    ) -> Tool:
+    ) -> EmptyResponse:
         """
         Sets or updates parameters for a specific tool in the specified project.
 
@@ -591,7 +591,7 @@ class AILabManager:
             logger.error(f"Error received while setting tool parameters: {error}")
             raise APIError(f"Error received while setting tool parameters: {error}")
 
-        result = ToolMapper.map_to_tool(response_data)
+        result = ResponseMapper.map_to_empty_response(response_data or "Parameter set successfully")
         return result
 
     def list_reasoning_strategies(

pygeai/lab/tools/clients.py

@@ -114,7 +114,8 @@ class ToolClient(AILabClient):
             headers=headers,
             data=data
         )
-        if response.status_code != 200:
+        
+        if response.status_code >= 300:
             logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
             raise APIResponseError(f"API returned an error: {response.text}")
         try:
@@ -168,7 +169,7 @@ class ToolClient(AILabClient):
                 "allowExternal": allow_external
             }
         )
-        if response.status_code != 200:
+        if response.status_code >= 300:
             logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
             raise APIResponseError(f"API returned an error: {response.text}")
         try:
@@ -212,7 +213,7 @@ class ToolClient(AILabClient):
                 "allowDrafts": allow_drafts,
             }
         )
-        if response.status_code != 200:
+        if response.status_code >= 300:
             logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
             raise APIResponseError(f"API returned an error: {response.text}")
         try:
@@ -255,7 +256,7 @@ class ToolClient(AILabClient):
             endpoint=endpoint,
             headers=headers
         )
-        if response.status_code != 200:
+        if response.status_code >= 300:
             logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
             raise APIResponseError(f"API returned an error: {response.text}")
 
@@ -375,7 +376,7 @@ class ToolClient(AILabClient):
             headers=headers,
             data=data
         )
-        if response.status_code != 200:
+        if response.status_code >= 300:
             logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
             raise APIResponseError(f"API returned an error: {response.text}")
         try:
@@ -463,7 +464,7 @@ class ToolClient(AILabClient):
                 "allowDrafts": allow_drafts,
             }
         )
-        if response.status_code != 200:
+        if response.status_code >= 300:
             logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
             raise APIResponseError(f"API returned an error: {response.text}")
         try:
@@ -520,16 +521,11 @@ class ToolClient(AILabClient):
             headers=headers,
             data=data
         )
-        if response.status_code != 200:
-            logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
-            raise APIResponseError(f"API returned an error: {response.text}")
-        try:
-            result = response.json()
-        except JSONDecodeError as e:
-            logger.error(f"Unable to set parameters for tool {tool_id or tool_public_name} in project {self.project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to set parameters for tool {tool_id or tool_public_name} in project {self.project_id}: {response.text}")
-
-        return result
+        if response.status_code != 204:
+            logger.error(f"Unable to set parameter for tool {tool_id or tool_public_name} from project {self.project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
+            raise InvalidAPIResponseException(f"Unable to set parameter for tool {tool_id or tool_public_name} from project {self.project_id}: {response.text}")
+        else:
+            return {}
 
     def export_tool(
             self,
@@ -559,7 +555,7 @@ class ToolClient(AILabClient):
             endpoint=endpoint,
             headers=headers,
         )
-        if response.status_code != 200:
+        if response.status_code >= 300:
             logger.error(f"Invalid status code returned from the API endpoint: {response.text}")
             raise APIResponseError(f"API returned an error: {response.text}")
         try:

pygeai/tests/integration/assistants/rag/test_create_rag.py

@@ -0,0 +1,72 @@
+from unittest import TestCase
+from pygeai.assistant.managers import AssistantManager
+from pygeai.core.models import WelcomeData, LlmSettings
+from pygeai.assistant.rag.models import (
+    Search,
+    RetrieverOptions,
+    SearchOptions,
+    ChunkOptions,
+    IndexOptions,
+    RAGAssistant,
+)
+
+assistant_manager = AssistantManager(alias="beta")
+
+class TestAssistantCreateRagIntegration(TestCase):
+
+    def __load_rag(self):
+        llm_options = LlmSettings(
+            cache=False,
+            temperature=0.1,
+            max_tokens=999,
+            model_name="gpt-3.5-turbo-16k",
+            n=1,
+            presence_penalty=0,
+            frequency_penalty=0,
+            provider="OpenAI",
+            stream=False,
+            top_p=1.0,
+            type=None,
+            verbose=True,
+        )
+
+        retriever_options = RetrieverOptions(type="vectorStore")
+
+        search_options = SearchOptions(
+            history_count=2,
+            llm=llm_options,
+            search=Search(
+                k=5,
+                return_source_documents=False,
+                score_threshold=0,
+                prompt="Use {context} and {question}",
+                template="",
+            ),
+            retriever=retriever_options,
+        )
+
+        chunk_options = ChunkOptions(chunk_size=999, chunk_overlap=0)
+
+        index_options = IndexOptions(chunks=chunk_options)
+
+        welcome_data = WelcomeData(
+            title="Test Profile Welcome Data",
+            description="Test Profile with WelcomeData",
+            features=[],
+            examples_prompt=[],
+        )
+
+        return RAGAssistant(
+            name="Testaasdfasdfsasdfdfsdfsdf123in123g12355",
+            description="Test Profile with WelcomeData",
+            search_options=search_options,
+            index_options=index_options,
+            welcome_data=welcome_data,
+        )
+
+
+    def test_create_rag_assistant(self):
+        rag_assistant = self.__load_rag()
+        response = assistant_manager.create_assistant(rag_assistant)
+        
+        self.assertIsInstance(response, RAGAssistant, "Failed to create RAG assistant")

pygeai/tests/integration/lab/agents/test_create_agent.py

@@ -33,7 +33,6 @@ class TestAILabCreateAgentIntegration(TestCase):
             access_scope="public",
             public_name=f"public_{random_str}",
             job_description="Translator",
-            avatar_image="https://www.shareicon.net/data/128x128/2016/11/09/851442_logo_512x512.png",
             description="Agent that translates from any language to english.",
             agent_data=AgentData(
                 prompt=Prompt(

pygeai/tests/integration/lab/agents/test_publish_agent_revision.py

@@ -28,7 +28,6 @@ class TestAILabPublishAgentRevisionIntegration(TestCase):
             access_scope="public",
             public_name=f"public_{random_str}",
             job_description=f"SummarizerAgent{random_str}",
-            avatar_image="https://www.shareicon.net/data/128x128/2016/11/09/851442_logo_512x512.png",
             description=f"Agent that summarized documents. {random_str}",
             agent_data=AgentData(
                 prompt=Prompt(

pygeai/tests/integration/lab/agents/test_update_agent.py

@@ -25,7 +25,6 @@ class TestAILabUpdateAgentIntegration(TestCase):
             access_scope="public",
             public_name=f"public_{random_str}",
             job_description=f"SummarizerAgent{random_str}",
-            avatar_image="https://www.shareicon.net/data/128x128/2016/11/09/851442_logo_512x512.png",
             description=f"Agent that summarized documents. {random_str}",
             agent_data=AgentData(
                 prompt=Prompt(

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

@@ -0,0 +1,98 @@
+from unittest import TestCase
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import FilterSettings, ToolParameter
+from pygeai.core.common.exceptions import APIResponseError, MissingRequirementException
+import copy
+
+ai_lab_manager: AILabManager
+
+class TestAILabGetParameterIntegration(TestCase):   
+
+    def setUp(self):
+        self.ai_lab_manager = AILabManager(alias="beta")
+        self.tool_id = "42610b61-cb47-49bf-9475-36ee652f31f3"
+        self.filter_settings = FilterSettings(
+            revision="0",
+            version="0",
+            allow_drafts=True
+        )
+
+
+    def __get_parameter(self, tool_id: str = None, tool_public_name: str = None, filter_settings: FilterSettings = None):
+        return self.ai_lab_manager.get_parameter(
+            tool_id=self.tool_id if tool_id is None else tool_id,
+            tool_public_name=tool_public_name,
+            filter_settings=self.filter_settings if filter_settings is None else filter_settings
+        )   
+    
+
+    def test_get_parameter(self):
+        tool_parameters = self.__get_parameter()
+       
+        for param in tool_parameters:
+            self.assertIsInstance(param, ToolParameter, "Expected a tool parameter")
+ 
+
+    def test_get_parameter_invalid_tool_id(self):
+        invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
+
+        with self.assertRaises(APIResponseError) as exception:
+            self.__get_parameter(tool_id=invalid_id)
+        self.assertIn(
+            f"Tool not found [IdOrName= {invalid_id}]",
+            str(exception.exception),
+            "Expected error when trying to send an invalid tool id"
+        )
+
+  
+    def test_get_parameter_no_tool_id(self):
+        with self.assertRaises(MissingRequirementException) as exception:
+            self.__get_parameter(tool_id="")
+        self.assertIn(
+            "Either tool_id or tool_public_name must be provided.",
+            str(exception.exception),
+            "Expected error when trying to send a request without tool_id"
+        )
+
+
+    def test_get_parameter_by_tool_public_name(self):
+        tool_parameters = self.__get_parameter(tool_public_name="test.sdk.beta.tool")
+
+        for param in tool_parameters:
+            self.assertIsInstance(param, ToolParameter, "Expected a tool parameter")
+
+
+    def test_get_parameter_by_invalid_public_name(self):
+        invalid_public_name = "test.sdk.beta.tool.invalid"
+
+        with self.assertRaises(APIResponseError) as exception:
+            self.__get_parameter(tool_public_name=invalid_public_name)
+        self.assertIn(
+             f"Tool not found [IdOrName= {invalid_public_name}]",
+            str(exception.exception),
+            "Expected error when trying to send a request without tool_id"
+        )
+
+
+    def test_get_parameter_by_past_revision(self):
+        filter_settings = copy.deepcopy(self.filter_settings)
+        filter_settings.revision = "3"
+
+        with self.assertRaises(APIResponseError) as exception:
+            self.__get_parameter(filter_settings=filter_settings)
+        self.assertIn(
+            "Requested revision not found [revision=3].",
+            str(exception.exception),
+            "Expected error when trying to send a request without tool_id"
+        )
+
+    
+    def test_get_parameter_by_draft_revision(self):
+        filter_settings = copy.deepcopy(self.filter_settings)
+        filter_settings.revision = "10"
+        tool_parameters = self.__get_parameter(filter_settings=filter_settings)
+
+        for param in tool_parameters:
+            self.assertIsInstance(param, ToolParameter, "Expected a tool parameter")
+
+    

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

@@ -0,0 +1,119 @@
+from unittest import TestCase
+import uuid
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import Tool
+from pygeai.core.common.exceptions import APIError
+
+ai_lab_manager: AILabManager
+
+class TestAILabPublishToolRevisionIntegration(TestCase):   
+
+    def setUp(self):
+        self.ai_lab_manager = AILabManager(alias="beta")
+        self.tool_id = "c77e1f2e-0322-4dd0-b6ec-aff217f1cb32"
+
+
+    def __publish_tool_revision(self, revision: str, tool_id=None):
+        return self.ai_lab_manager.publish_tool_revision(
+            tool_id=self.tool_id if tool_id is None else tool_id,
+            revision=revision
+        )
+
+    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.
+        """
+        tool = self.__load_tool()
+        return self.ai_lab_manager.update_tool(
+            tool = tool,
+            automatic_publish=False, 
+            upsert=False
+        )
+    
+
+    def test_publish_tool_revision(self):
+        updated_tool = self.__update_tool()
+        new_revision = updated_tool.revision
+
+        published_tool = self.__publish_tool_revision(revision=str(new_revision))
+
+        self.assertFalse(published_tool.is_draft, "Expected draft to be false after publishing the revision") 
+        self.assertEqual(published_tool.revision, new_revision, "Expected last revision to be published") 
+ 
+
+    def test_publish_tool_earlier_revision_with_newer_revision_published(self):
+        with self.assertRaises(APIError) as exception:
+            self.__publish_tool_revision(revision="1")
+        self.assertIn(
+            "There are newer published revisions.",
+            str(exception.exception),
+            "Expected error when trying to send a earlier revision"
+        )
+
+    
+    def test_publish_tool_earlier_revision(self):
+        earlier_revision = (self.__update_tool()).revision
+        #update the tool to create a newer revision
+        self.__update_tool()
+
+        published_tool = self.__publish_tool_revision(revision=earlier_revision)
+
+        self.assertFalse(published_tool.is_draft, "Expected draft to be false after publishing the revision") 
+        self.assertEqual(published_tool.revision, earlier_revision, "Expected last revision to be published") 
+
+
+    def test_publish_tool_invalid_revision(self):
+        with self.assertRaises(APIError) as exception:
+            self.__publish_tool_revision(revision="10000000")
+        self.assertIn(
+            "Invalid revision [rev=10000000]",
+            str(exception.exception),
+            "Expected error when trying to send a revision that does not exist"
+        )
+
+
+    def test_publish_tool_string_revision(self):
+        with self.assertRaises(APIError) as exception:
+            self.__publish_tool_revision(revision="revision")
+        self.assertIn("Bad Request", str(exception.exception))
+        self.assertIn("400", str(exception.exception))
+
+
+    def test_publish_tool_invalid_tool_id(self):
+        invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
+        with self.assertRaises(APIError) as exception:
+            self.__publish_tool_revision(revision="103", tool_id=invalid_id)
+        self.assertIn(
+            f"Tool not found [IdOrName= {invalid_id}].",
+            str(exception.exception),
+            "Expected error when sending and invalid agent id"
+        )
+
+    
+    def test_publish_tool_no_tool_id(self):
+        with self.assertRaises(APIError) as exception:
+            self.__publish_tool_revision(revision="103", tool_id="")
+        self.assertIn("Not Found", str(exception.exception))
+        self.assertIn("404", str(exception.exception))

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

@@ -0,0 +1,114 @@
+from unittest import TestCase
+import uuid
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import ToolParameter
+from pygeai.core.common.exceptions import MissingRequirementException, InvalidAPIResponseException
+
+ai_lab_manager: AILabManager
+
+class TestAILabSetParameterIntegration(TestCase):
+    def setUp(self):
+        self.ai_lab_manager = AILabManager(alias="beta")
+        self.tool_id = "42610b61-cb47-49bf-9475-36ee652f31f3"
+        self.tool_parameters = self.__load_parameters()
+
+
+    def __set_parameter(self, tool_id: str = None, tool_public_name: str = None, parameters: list = None):        
+        return self.ai_lab_manager.set_parameter(
+            tool_id = self.tool_id if tool_id is None else tool_id,
+            tool_public_name = tool_public_name,
+            parameters = self.tool_parameters if parameters is None else parameters,
+        )
+    
+    
+    def __load_parameters(self):
+        random_str = str(uuid.uuid4())
+        return [
+            ToolParameter(
+                key="param3",
+                data_type="String",
+                description=f"{random_str}_config",
+                is_required=True,
+                type="config",
+                value=random_str
+            )
+        ]
+        
+
+    def test_set_parameter(self):  
+        result = self.__set_parameter()
+        self.assertEqual(
+            result.content, 
+            "Parameter set successfully",
+            "Expected successful parameter update"
+        )
+
+
+    def test_set_parameter_invalid_tool_id(self):
+        invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
+
+        with self.assertRaises(InvalidAPIResponseException) as context:
+            self.__set_parameter(tool_id=invalid_id)
+
+        self.assertIn(
+            f"Unable to set parameter for tool {invalid_id}",
+            str(context.exception),
+            "Expected error for invalid tool ID"
+        )
+
+    
+    def test_set_parameter_no_tool_id(self):
+        with self.assertRaises(MissingRequirementException) as context:
+            self.__set_parameter(tool_id="")
+
+        self.assertIn(
+            "Either tool_id or tool_public_name must be provided.",
+            str(context.exception),
+            "Expected error for invalid tool ID"
+        )
+
+
+    def test_set_parameter_by_tool_public_name(self):
+        self.tool_parameters[0].key = "Param1"
+
+        result = self.__set_parameter(tool_id = "", tool_public_name="test.sdk.beta.tool")
+        self.assertEqual(
+            result.content, 
+            "Parameter set successfully",
+            "Expected successful parameter update"
+        )
+
+
+    def test_set_parameter_by_invalid_public_name(self):
+        invalid_public_name = "test.sdk.beta.tool.invalid"
+
+        with self.assertRaises(InvalidAPIResponseException) as exception:
+            self.__set_parameter(tool_id = "", tool_public_name=invalid_public_name)
+        self.assertIn(
+            f"Unable to set parameter for tool {invalid_public_name}",
+            str(exception.exception),
+            "Expected error when trying to send a request without tool_id"
+        )
+
+
+    def test_set_parameter_no_parameters(self):
+        self.tool_parameters = []
+
+        with self.assertRaises(MissingRequirementException) as exception:
+            self.__set_parameter()
+        self.assertIn(
+            "Parameters list must be provided and non-empty.",
+            str(exception.exception),
+            "Expected error when trying to send a request without parameters"
+        )
+
+
+    def test_set_parameter_not_added_parameters(self):
+        #param3 set on self.tool_parameters, does not exist in the public tool
+        with self.assertRaises(InvalidAPIResponseException) as exception:
+            self.__set_parameter(tool_id = "", tool_public_name = "test.sdk.beta.tool")
+        self.assertIn(
+            "Parameter param3 not found in this tool",
+            str(exception.exception),
+            "Expected error when trying to send a request without parameters"
+        )

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

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

{pygeai-0.4.0b10.dist-info → pygeai-0.4.0b12.dist-info}/RECORD

@@ -25,7 +25,7 @@ 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/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=Smqxqc3XIy8RMuFVIsS9qNuYtqkNWba_CJJrhe7kW0E,4451
@@ -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=F9SOJWfW577y9lF6xR_LLpvVB2jevGvLH1KG7NqOdfc,25640
+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,7 +57,7 @@ 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=Ecvp8E6tlEh56sPsmcAwYF0CjNn3zdJwggs-BmtdKqU,129292
+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
@@ -76,7 +76,7 @@ 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=-eF4V0B-27zfp0aHMlZWqWRIty6P7TCOrzMRW87ZnlE,1251
 pygeai/core/embeddings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -146,7 +146,7 @@ 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=_i9tQACTGblKkw05WMu5RLpCYlyL2PtLvNI42F11u8Y,70670
+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
@@ -165,7 +165,7 @@ pygeai/lab/strategies/clients.py,sha256=_a1yc8kwd50Yv4g1jqfa0gRnMiROR7Dn0gx3xqFU
 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=Jnc4AU_1fnEqhHOsey3AoKpNl1Pfw01qq_hDOmmDvE8,28394
+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
@@ -271,22 +271,28 @@ 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=yL3owPJP0RoHSy3lEAdH06LJXqxgt3f2l2plk75d4cc,14206
+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=1seho_GOtOHik_YJ9GPA4McSZorohhFRvuzY7UYXbxo,11726
+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
@@ -383,7 +389,7 @@ pygeai/tests/snippets/lab/agents/create_agent_edge_case.py,sha256=8dA9giNdsHjFZs
 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
@@ -406,7 +412,7 @@ 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
@@ -496,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.4.0b10.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
-pygeai-0.4.0b10.dist-info/METADATA,sha256=v2dh0SCr1-9mioQuxBd4PjRxun1SntD_QHjMJZUjO-w,6941
-pygeai-0.4.0b10.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygeai-0.4.0b10.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
-pygeai-0.4.0b10.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
-pygeai-0.4.0b10.dist-info/RECORD,,
+pygeai-0.4.0b12.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
+pygeai-0.4.0b12.dist-info/METADATA,sha256=4GKLJjLWFsGzD8vMzbE90o9HKovdZsG4_Ip_EBAOBU0,6941
+pygeai-0.4.0b12.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygeai-0.4.0b12.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
+pygeai-0.4.0b12.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
+pygeai-0.4.0b12.dist-info/RECORD,,