pygeai 0.4.0b10__tar.gz → 0.4.0b12__tar.gz

{pygeai-0.4.0b10/pygeai.egg-info → pygeai-0.4.0b12}/PKG-INFO

@@ -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 → pygeai-0.4.0b12}/pygeai/chat/ui.py

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

{pygeai-0.4.0b10 → pygeai-0.4.0b12}/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-0.4.0b10 → pygeai-0.4.0b12}/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-0.4.0b10 → pygeai-0.4.0b12}/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-0.4.0b10 → pygeai-0.4.0b12}/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-0.4.0b10 → pygeai-0.4.0b12}/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-0.4.0b12/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-0.4.0b10 → pygeai-0.4.0b12}/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-0.4.0b10 → pygeai-0.4.0b12}/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-0.4.0b10 → pygeai-0.4.0b12}/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-0.4.0b12/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-0.4.0b12/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))