pygeai 0.2.7b46__tar.gz → 0.2.7b47__tar.gz

{pygeai-0.2.7b46/pygeai.egg-info → pygeai-0.2.7b47}/PKG-INFO

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

{pygeai-0.2.7b46 → pygeai-0.2.7b47}/pygeai/chat/clients.py

@@ -140,18 +140,24 @@ class ChatClient(BaseClient):
         if user is not None:
             data['user'] = user
 
+        headers = {}
+        if thread_id:
+            headers["saia-conversation-id"] = thread_id
+
         logger.debug(f"Generating chat completion with data: {data}")
 
         if stream:
             response = self.api_service.stream_post(
                 endpoint=CHAT_V1,
-                data=data
+                data=data,
+                headers=headers
             )
             return self.stream_chat_generator(response)
         else:
             response = self.api_service.post(
                 endpoint=CHAT_V1,
-                data=data
+                data=data,
+                headers=headers
             )
             try:
                 result = response.json()

{pygeai-0.2.7b46 → pygeai-0.2.7b47}/pygeai/tests/integration/lab/agents/test_agents_list.py

@@ -14,7 +14,6 @@ class TestAILabAgentsListIntegration(TestCase):
 
     def setUp(self):
         self.ai_lab_manager = AILabManager(alias="beta")
-        self.project_id = "be4889df-cacc-4e6f-b3bb-153c4ac0d168"
         self.filter_settings = FilterSettings(
             allow_external=False,
             allow_drafts=True,
@@ -24,7 +23,6 @@ class TestAILabAgentsListIntegration(TestCase):
 
     def __get_agent_list(self, filter_settings: FilterSettings = None):
         return self.ai_lab_manager.get_agent_list(
-            project_id=self.project_id,
             filter_settings= self.filter_settings if filter_settings is None else filter_settings
         )
     
@@ -42,6 +40,7 @@ class TestAILabAgentsListIntegration(TestCase):
                 )   
             self.assertEqual(agent.access_scope, "public", "Expected all agents to be public")
 
+
     def test_get_private_agent_list(self):        
         result = self.__get_agent_list()
 
@@ -72,6 +71,7 @@ class TestAILabAgentsListIntegration(TestCase):
             "Expected at least one agent to be a draft"
         )
 
+    
     def test_get_agent_list_no_draft(self):
         filter_settings = copy.deepcopy(self.filter_settings)
         filter_settings.allow_drafts = False

{pygeai-0.2.7b46 → pygeai-0.2.7b47}/pygeai/tests/integration/lab/agents/test_create_agent.py

@@ -23,9 +23,7 @@ class TestAILabCreateAgentIntegration(TestCase):
         This can be used to delete the created agent or reset the state.
         """
         if isinstance(self.created_agent, Agent):
-            #TODO: Remove exception catch when delete_agent stop returning an unexpected error
-            with self.assertRaises(InvalidAPIResponseException) as e:
-                self.ai_lab_manager.delete_agent(self.created_agent.id)
+            self.ai_lab_manager.delete_agent(self.created_agent.id)
 
 
     def __load_agent(self):
@@ -144,10 +142,12 @@ class TestAILabCreateAgentIntegration(TestCase):
         self.assertEqual(prompt.inputs, agent.agent_data.prompt.inputs)
 
 
+    @unittest.skip("Skipping test due to mismatch in expected behavior")
     def test_create_agent_without_required_data(self):
         test_params = [ True, False ]
 
         for auto_publish in test_params:
+            
             with self.subTest(input=auto_publish):
                 self.new_agent = Agent(
                     name=str(uuid.uuid4())
@@ -223,11 +223,11 @@ class TestAILabCreateAgentIntegration(TestCase):
         self.new_agent.access_scope = "project" 
         with self.assertRaises(ValueError) as exc:
             self.__create_agent()
-            self.assertEqual(
-                str(exc.exception),
-                "Access scope must be one of public, private",
-                "Expected a ValueError exception for invalid access scope"
-            )
+        self.assertEqual(
+            str(exc.exception),
+            "Access scope must be one of public, private.",
+            "Expected a ValueError exception for invalid access scope"
+        )
 
 
     def test_create_agent_default_scope(self):
@@ -292,13 +292,17 @@ class TestAILabCreateAgentIntegration(TestCase):
         self.new_agent.agent_data.prompt.instructions = ""
         with self.assertRaises(ValidationError) as exception:
             self.__create_agent()
-        
-            self.assertIn(
-                """instructions
-                    Input should be a valid string [type=string_type, input_value=None, input_type=NoneType]""",
-                str(exception.exception),
-                "Expected a validation error about allowed values for instructions"
-            )
+        self.assertIn(
+            "instructions",
+            str(exception.exception),
+            "Expected a validation error about allowed values for instructions"
+        )
+
+        self.assertIn(
+            "Input should be a valid string [type=string_type, input_value=None, input_type=NoneType]",
+            str(exception.exception),
+            "Expected a validation error about allowed values for instructions"
+        )
 
     
     def test_create_agent_autopublish(self): 

{pygeai-0.2.7b46 → pygeai-0.2.7b47}/pygeai/tests/integration/lab/agents/test_create_sharing_link.py

@@ -1,8 +1,7 @@
 from unittest import TestCase
-import unittest
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import SharingLink
-from pygeai.core.common.exceptions import MissingRequirementException
+from pygeai.core.common.exceptions import APIError, MissingRequirementException
 
 ai_lab_manager: AILabManager
 
@@ -10,12 +9,10 @@ class TestAILabCreateSharingLinkIntegration(TestCase):
 
     def setUp(self):
         self.ai_lab_manager = AILabManager(alias="beta")
-        self.project_id = "be4889df-cacc-4e6f-b3bb-153c4ac0d168"
         self.agent_id = "0026e53d-ea78-4cac-af9f-12650e5bb6d9" 
 
-    def __create_sharing_link(self, agent_id=None, project_id=None):
+    def __create_sharing_link(self, agent_id=None):
         return self.ai_lab_manager.create_sharing_link(
-            project_id=self.project_id if project_id is None else project_id,
             agent_id=self.agent_id if agent_id is None else agent_id
         )
     
@@ -49,16 +46,6 @@ class TestAILabCreateSharingLinkIntegration(TestCase):
         )
 
 
-    def test_create_sharing_link_no_project(self):
-        with self.assertRaises(MissingRequirementException) as context:
-            self.__create_sharing_link(project_id="")
-        self.assertEqual(
-            str(context.exception),
-            "Cannot create sharing link without specifying project_id",
-            "Expected exception for missing project_id"
-        )
-
-
     def test_create_sharing_link_no_agent_id(self):
         with self.assertRaises(MissingRequirementException) as context:
             self.__create_sharing_link(agent_id="")
@@ -68,34 +55,13 @@ class TestAILabCreateSharingLinkIntegration(TestCase):
             "Expected exception for missing agent_id"
         )
 
-    @unittest.skip("A descriptive error exception is expected")
-    def test_create_sharing_link_invalid_agent_id(self):
-        invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
-        shared_link = self.__create_sharing_link(agent_id=invalid_id)
 
-        """ self.assertEqual(
-            deleted_agent.content["messages"][0]["description"],
+    def test_create_sharing_link_invalid_agent_id(self):
+        with self.assertRaises(APIError) as exception:
+            invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
+            self.__create_sharing_link(agent_id=invalid_id)
+        self.assertIn(
             f"Agent not found [IdOrName= {invalid_id}].",
+            str(exception.exception),
             "Expected error message for invalid agent id"
-        ) """
-
-
-    def test_create_sharing_link_invalid_project_id(self):
-        invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
-        shared_link = self.__create_sharing_link(project_id=invalid_id)
-
-        self.assertIsInstance(shared_link, SharingLink, "Expected response to be an instance of SharingLink")
-       
-        self.assertEqual(
-            shared_link.agent_id,
-            self.agent_id,
-            "Returned agentId should match the requested agent_id"
-        )
-        self.assertTrue(
-            shared_link.api_token.startswith("shared-"),
-            "apiToken should start with 'shared-'"
-        )
-        self.assertTrue(
-            shared_link.shared_link.startswith("https://"),
-            "sharedLink should be a valid URL"
         )

{pygeai-0.2.7b46 → pygeai-0.2.7b47}/pygeai/tests/integration/lab/agents/test_delete_agent.py

@@ -2,7 +2,7 @@ from unittest import TestCase
 import uuid
 from pygeai.lab.managers import AILabManager
 from pygeai.lab.models import Agent, AgentData, Prompt, LlmConfig, Model
-from pygeai.core.common.exceptions import APIError
+from pygeai.core.common.exceptions import InvalidAPIResponseException
 
 ai_lab_manager: AILabManager
 
@@ -10,7 +10,6 @@ class TestAILabDeleteAgentIntegration(TestCase):
 
     def setUp(self):
         self.ai_lab_manager = AILabManager(alias="beta")
-        self.project_id = "be4889df-cacc-4e6f-b3bb-153c4ac0d168"
         
 
     def __create_agent(self):
@@ -36,13 +35,11 @@ class TestAILabDeleteAgentIntegration(TestCase):
 
 
         return self.ai_lab_manager.create_agent(
-            project_id=self.project_id,
             agent=agent
         )
 
-    def __delete_agent(self, agent_id: str, project_id=None):
+    def __delete_agent(self, agent_id: str):
         return self.ai_lab_manager.delete_agent(
-            project_id=self.project_id if project_id is None else project_id,
             agent_id=agent_id
         )
     
@@ -58,19 +55,8 @@ class TestAILabDeleteAgentIntegration(TestCase):
         )
 
 
-    def test_delete_agent_no_project(self):
-        create_agent = self.__create_agent()     
-        deleted_agent = self.__delete_agent(agent_id=create_agent.id, project_id="")
-
-        self.assertEqual(
-            deleted_agent.content,
-            "Agent deleted successfully",
-            "Expected confirmation message after deletion even with missing project_id"            
-        )
-
-
     def test_delete_agent_no_agent_id(self):
-        with self.assertRaises(APIError) as exception:
+        with self.assertRaises(InvalidAPIResponseException) as exception:
             self.__delete_agent(agent_id="")
 
         self.assertIn("405",str(exception.exception))    
@@ -79,22 +65,11 @@ class TestAILabDeleteAgentIntegration(TestCase):
 
     def test_delete_agent_invalid_agent_id(self):
         invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
-        deleted_agent = self.__delete_agent(agent_id=invalid_id)
+        with self.assertRaises(InvalidAPIResponseException) as exception:
+            self.__delete_agent(agent_id=invalid_id)
 
-        self.assertEqual(
-            deleted_agent.content["messages"][0]["description"],
+        self.assertIn(
             f"Agent not found [IdOrName= {invalid_id}].",
+            str(exception.exception),
             "Expected error message for invalid agent id"
-        )
-
-
-    def test_delete_agent_invalid_project_id(self):
-        invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
-        create_agent = self.__create_agent()     
-        deleted_agent = self.__delete_agent(agent_id=create_agent.id, project_id=invalid_id)
-
-        self.assertEqual(
-            deleted_agent.content,
-            "Agent deleted successfully",
-            "Expected confirmation message after deletion even with invalid project id"
         )

{pygeai-0.2.7b46 → pygeai-0.2.7b47}/pygeai/tests/integration/lab/agents/test_get_agent.py

@@ -15,16 +15,14 @@ class TestAILabGetAgentIntegration(TestCase):
 
     def setUp(self):
         self.ai_lab_manager = AILabManager(alias="beta")
-        self.project_id = "be4889df-cacc-4e6f-b3bb-153c4ac0d168"
         self.agent_id = "0026e53d-ea78-4cac-af9f-12650e5bb6d9"
         self.filter_settings = FilterSettings(
             revision="0",
             version="0"
         )
 
-    def __get_agent(self, project_id=None, agent_id=None, filter_settings: FilterSettings = None):
+    def __get_agent(self, agent_id=None, filter_settings: FilterSettings = None):
         return self.ai_lab_manager.get_agent(
-            project_id=self.project_id if project_id is None else project_id,
             agent_id=self.agent_id if agent_id is None else agent_id,
             filter_settings=self.filter_settings if filter_settings is None else filter_settings
         )
@@ -38,17 +36,7 @@ class TestAILabGetAgentIntegration(TestCase):
                     hasattr(agent, attr),
                     f"Agent should have an '{attr}' attribute"
                 )
-
-    
-    def test_get_agent_no_project(self):
-        with self.assertRaises(MissingRequirementException) as excepcion:
-            self.__get_agent(project_id="")
-        self.assertIn(
-            "Cannot retrieve agent without specifying project_id",
-            str(excepcion.exception),
-            "Expected exception for missing project_id"
-        )
-    
+ 
 
     def test_get_agent_no_agent_id(self):
         with self.assertRaises(Exception) as context:

{pygeai-0.2.7b46 → pygeai-0.2.7b47}/pygeai/tests/integration/lab/agents/test_publish_agent_revision.py

@@ -10,13 +10,11 @@ class TestAILabPublishAgentRevisionIntegration(TestCase):
 
     def setUp(self):
         self.ai_lab_manager = AILabManager(alias="beta")
-        self.project_id = "be4889df-cacc-4e6f-b3bb-153c4ac0d168"
         self.agent_id = "b4b09935-2ad2-42c0-bd55-1ee6fa4b6034"
 
     
-    def __publish_agent_revision(self,  revision: str, project_id=None, agent_id=None):
+    def __publish_agent_revision(self,  revision: str, agent_id=None):
         return self.ai_lab_manager.publish_agent_revision(
-            project_id=self.project_id if project_id is None else project_id,
             agent_id=self.agent_id if agent_id is None else agent_id,
             revision=revision
         )
@@ -62,7 +60,6 @@ class TestAILabPublishAgentRevisionIntegration(TestCase):
         """
         agent = self.__load_agent()
         return self.ai_lab_manager.update_agent(
-            project_id=self.project_id,
             agent=agent,
             automatic_publish=False, 
             upsert=False
@@ -77,7 +74,7 @@ class TestAILabPublishAgentRevisionIntegration(TestCase):
         self.assertEqual(published_agent.revision, new_revision, "Expected last revision to be published") 
 
     
-    def test_publish_agent_earlier_revision(self):
+    def test_publish_agent_earlier_revision_with_newer_revision_published(self):
         with self.assertRaises(APIError) as exception:
             self.__publish_agent_revision(revision="1")
         self.assertIn(
@@ -87,6 +84,15 @@ class TestAILabPublishAgentRevisionIntegration(TestCase):
         )
 
 
+    def test_publish_agent_earlier_revision(self):
+        earlier_revision = (self.__update_agent()).revision
+        self.__update_agent()
+        published_agent = self.__publish_agent_revision(revision=earlier_revision)
+
+        self.assertFalse(published_agent.is_draft, "Expected draft to be false after publishing the revision") 
+        self.assertEqual(published_agent.revision, earlier_revision, "Expected last revision to be published") 
+
+
     def test_publish_agent_invalid_revision(self):
         with self.assertRaises(APIError) as exception:
             self.__publish_agent_revision(revision="10000000")
@@ -119,21 +125,4 @@ class TestAILabPublishAgentRevisionIntegration(TestCase):
         with self.assertRaises(APIError) as exception:
             self.__publish_agent_revision(revision="103", agent_id="")
         self.assertIn("Not Found", str(exception.exception))
-        self.assertIn("404", str(exception.exception))
-
-
-    def test_publish_agent_invalid_project_id(self):
-        new_revision = (self.__update_agent()).revision
-        invalid_id = "0026e53d-ea78-4cac-af9f-12650invalid"
-        published_agent = self.__publish_agent_revision(revision=new_revision, project_id=invalid_id)
-
-        self.assertFalse(published_agent.is_draft, "Expected draft to be false after publishing the revision") 
-        self.assertEqual(published_agent.revision, new_revision, "Expected last revision to be published")
-
-
-    def test_publish_agent_no_project_id(self):
-        new_revision = (self.__update_agent()).revision
-        published_agent = self.__publish_agent_revision(revision=new_revision, project_id="")
-
-        self.assertFalse(published_agent.is_draft, "Expected draft to be false after publishing the revision") 
-        self.assertEqual(published_agent.revision, new_revision, "Expected last revision to be published")
+        self.assertIn("404", str(exception.exception))

{pygeai-0.2.7b46 → pygeai-0.2.7b47}/pygeai/tests/integration/lab/agents/test_update_agent.py

@@ -12,7 +12,6 @@ class TestAILabUpdateAgentIntegration(TestCase):
         Set up the test environment.
         """
         self.ai_lab_manager = AILabManager(alias="beta")
-        self.project_id = "be4889df-cacc-4e6f-b3bb-153c4ac0d168"
 
         load_agent = self.__load_agent()
         self.agent_to_update = load_agent["agent"]
@@ -61,7 +60,6 @@ class TestAILabUpdateAgentIntegration(TestCase):
         If automatic_publish is None, do not pass it (useful for tests that omit it).
         """
         return self.ai_lab_manager.update_agent(
-            project_id=self.project_id,
             agent=self.agent_to_update if agent is None else agent,
             automatic_publish=automatic_publish, 
             upsert=upsert
@@ -85,7 +83,7 @@ class TestAILabUpdateAgentIntegration(TestCase):
         self.assertTrue(updated_agent.is_draft, "gemini/gemini-1.5-flash-8b-exp-0827")
 
         self.agent_to_update.agent_data.models[0].name = "openai/gpt-4-turbo-preview"
-        updated_agent = self.__update_agent()
+        self.__update_agent()
 
     
     def test_update_agent_no_name(self):
@@ -119,7 +117,7 @@ class TestAILabUpdateAgentIntegration(TestCase):
                 )
 
     
-    def test_update_agent_invalidName_name(self):
+    def test_update_agent_invalid_name(self):
         test_params = [ True, False ]
         self.agent_to_update.name = f"{self.agent_to_update.name}:/"
 
@@ -241,6 +239,7 @@ class TestAILabUpdateAgentIntegration(TestCase):
                         "Expected a validation error about allowed values for model name" 
                     )
 
+    
     def test_update_agent_autopublish(self):        
         updated_agent = self.__update_agent(automatic_publish=True)
 

{pygeai-0.2.7b46 → pygeai-0.2.7b47/pygeai.egg-info}/PKG-INFO

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

{pygeai-0.2.7b46 → pygeai-0.2.7b47}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "pygeai"
-version = "0.2.7b46"
+version = "0.2.7b47"
 description = "Software Development Kit to interact with Globant Enterprise AI."
 license = "MIT"
 requires-python = ">=3.10"