pygeai 0.4.0b3__tar.gz → 0.4.0b4__tar.gz

{pygeai-0.4.0b3/pygeai.egg-info → pygeai-0.4.0b4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeai
-Version: 0.4.0b3
+Version: 0.4.0b4
 Summary: Software Development Kit to interact with Globant Enterprise AI.
 Author-email: Globant <geai-sdk@globant.com>
 License-Expression: MIT
@@ -41,11 +41,11 @@ Dynamic: license-file
 
 PyGEAI is a Software Development Kit (SDK) for interacting with [Globant Enterprise AI](https://wiki.genexus.com/enterprise-ai/wiki?8,Table+of+contents%3AEnterprise+AI). It comprises libraries, tools, code samples, and documentation to simplify your experience with the platform.
 
-## Repository
+## Terms and conditions
 
-Find the PyGEAI source code and documentation in the following GitHub repository:
+By using the Python SDK to interact with Globant Enterprise AI, you agree with the following Terms and Conditions:
 
-[GitHub repository](https://github.com/RT-GEN029-GI/pygeai)
+[Terms and Conditions](https://www.globant.com/enterprise-ai/terms-of-use)
 
 ## Compatibility
 This package is compatible with the Globant Enterprise AI release from June 2025.

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/README.md

@@ -2,11 +2,11 @@
 
 PyGEAI is a Software Development Kit (SDK) for interacting with [Globant Enterprise AI](https://wiki.genexus.com/enterprise-ai/wiki?8,Table+of+contents%3AEnterprise+AI). It comprises libraries, tools, code samples, and documentation to simplify your experience with the platform.
 
-## Repository
+## Terms and conditions
 
-Find the PyGEAI source code and documentation in the following GitHub repository:
+By using the Python SDK to interact with Globant Enterprise AI, you agree with the following Terms and Conditions:
 
-[GitHub repository](https://github.com/RT-GEN029-GI/pygeai)
+[Terms and Conditions](https://www.globant.com/enterprise-ai/terms-of-use)
 
 ## Compatibility
 This package is compatible with the Globant Enterprise AI release from June 2025.

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/pygeai/assistant/rag/models.py

@@ -227,7 +227,7 @@ class SearchOptions(CustomBaseModel):
             "ingestion": self.ingestion.to_dict() if self.ingestion else None,
             "options": self.options,
             "rerank": self.rerank,
-            "variables": self.variables.to_dict() if self.variables else None,
+            "variables": self.variables.to_list() if self.variables else None,
             "vectorStore": self.vector_store
         }
         return {k: v for k, v in result.items() if v is not None}

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/pygeai/lab/models.py

@@ -94,7 +94,7 @@ class Model(CustomBaseModel):
     :param llm_config: Optional[LlmConfig] - Overrides default agent LLM settings.
     :param prompt: Optional[dict] - A tailored prompt specific to this model.
     """
-    name: str = Field(..., alias="name")
+    name: Optional[str] = Field(None, alias="name")
     llm_config: Optional[LlmConfig] = Field(None, alias="llmConfig")
     prompt: Optional[Dict[str, Any]] = Field(None, alias="prompt")
 

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/pygeai/lab/processes/mappers.py

@@ -110,8 +110,8 @@ class AgenticProcessMapper:
         """
         return [
             Variable(
-                key=var["key"],
-                value=var["value"]
+                key=var.get("key"),
+                value=var.get("value")
             )
             for var in data
         ]

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/pygeai/lab/tools/mappers.py

@@ -37,8 +37,8 @@ class ToolMapper:
         """
         return [
             ToolMessage(
-                description=msg["description"],
-                type=msg["type"]
+                description=msg.get("description"),
+                type=msg.get("type")
             )
             for msg in messages_data
         ]
@@ -55,9 +55,9 @@ class ToolMapper:
         """
         tool_data = data.get("tool", data)
 
-        name = tool_data["name"]
-        description = tool_data["description"]
-        scope = tool_data["scope"]
+        name = tool_data.get("name")
+        description = tool_data.get("description")
+        scope = tool_data.get("scope")
         parameter_data = tool_data.get("parameters")
         parameters = cls._map_parameters(parameter_data) if parameter_data else None
 

pygeai-0.4.0b4/pygeai/tests/integration/chat/test_generate_image.py

@@ -0,0 +1,162 @@
+from unittest import TestCase
+from pygeai.chat.clients import ChatClient
+
+chat_client: ChatClient
+
+class TestChatGenerateImageIntegration(TestCase):    
+
+    def setUp(self):
+        self.chat_client = ChatClient(alias="beta")
+        self.new_image = self.__load_image()
+
+
+    def __load_image(self):
+        return {
+            "model": "openai/gpt-image-1",
+            "prompt": "generate an image of a futuristic city skyline at sunset",
+            "n": 1,
+            "quality": "high",
+            "size": "1024x1536"
+        }
+    
+
+    def __generate_image(self, image = None):
+        image = image if image is not None else self.new_image
+        return self.chat_client.generate_image(
+            model=image["model"],
+            prompt=image["prompt"],
+            n=image["n"],
+            quality=image["quality"],
+            size=image["size"],
+            aspect_ratio= image["aspect_ratio"] if "aspect_ratio" in image else None
+        )
+
+    
+    def test_generate_image(self):
+        created_image = self.__generate_image()
+        self.assertEqual(len(created_image["data"]), 1, "Expected an image to be generated")
+
+
+    def test_generate_image_invalid_model(self):
+        self.new_image["model"] = "openai/gpt-image-10",
+        created_image = self.__generate_image()
+
+        self.assertEqual(
+            created_image["error"]["code"], 400,
+            "Expected a 400 code for invalid model"
+        )
+        self.assertEqual(
+            created_image["error"]["message"],
+            'Provider \'["openai\' does not exists.', 
+            "Expected an error message when model does not exists"
+        )
+
+
+    def test_generate_image_no_model(self):
+        self.new_image["model"] = ""
+        created_image = self.__generate_image()
+
+        self.assertEqual(
+            created_image["error"]["code"], 400,
+            "Expected a 400 code for no model"
+        )
+        self.assertEqual(
+            created_image["error"]["message"],
+            "Invalid 'model' name. Must follow pattern {provider}/{modelName}", 
+            "Expected an error message when no model is provided"
+        )
+
+
+    def test_generate_image_no_prompt(self):
+        self.new_image["prompt"] = ""
+        created_image = self.__generate_image()
+
+        self.assertEqual(
+            created_image["error"]["type"],
+            "invalid_request_error",
+            "Expected a 400 code for no model"
+        )
+        self.assertEqual(
+            created_image["error"]["param"], "prompt", 
+            "Expected an error message when no model is provided"
+        )
+
+
+    def test_generate_image_specific_n(self):
+        self.new_image["n"] = 2
+
+        created_image = self.__generate_image()
+        self.assertEqual(len(created_image["data"]), 2, "Expected two images to be generated")
+
+
+    def test_generate_image_no_n(self):
+        self.new_image["n"] = None # default is 1
+
+        created_image = self.__generate_image()
+        self.assertEqual(len(created_image["data"]), 1, "Expected an image to be generated")
+
+
+    def test_generate_image_no_supported_n(self):
+        self.new_image["model"] = "openai/dall-e-3"
+        self.new_image["n"] = 5
+
+        created_image = self.__generate_image()
+        self.assertIn(
+            "Invalid 'n': integer above maximum value",
+            created_image["error"]["message"],
+            "Expected an error message when n is not supported by the model"
+        )
+
+
+    def test_generate_image_no_quality(self):
+        self.new_image["quality"] = ""
+
+        created_image = self.__generate_image()
+        self.assertIn(
+            "Invalid value: ''. Supported values are: 'low', 'medium', 'high', and 'auto'",
+            created_image["error"]["message"],
+            "Expected an error message when quality is not provided"
+        )
+    
+
+    def test_generate_image_no_supported_quality(self):
+        self.new_image["model"] = "openai/dall-e-3"
+
+        created_image = self.__generate_image()
+        self.assertIn(
+            "Invalid value: 'high'. Supported values are: 'standard' and 'hd'",
+            created_image["error"]["message"],
+            "Expected an error message when quality is not supported by the model"
+        )
+
+
+    def test_generate_image_no_size(self):
+        self.new_image["size"] = ""
+
+        created_image = self.__generate_image()
+        self.assertIn(
+            "Invalid value: ''. Supported values are: '1024x1024', '1024x1536', '1536x1024', and 'auto'",
+            created_image["error"]["message"],
+            "Expected an error message when no size is provided"
+        )
+
+
+    def test_generate_image_no_supported_size(self):
+        self.new_image["size"] = 1024
+        self.new_image["quality"] = None
+
+        created_image = self.__generate_image()
+        self.assertIn(
+            "Invalid type for 'size': expected one of '1024x1024', '1024x1536', '1536x1024', or 'auto', but got an integer instead",
+            created_image["error"]["message"],
+            "Expected an error message when no size is provided"
+        )
+    
+
+    def test_generate_image_with_aspect_ratio(self):
+        self.new_image["model"] = "vertex_ai/imagen-3.0-generate-001"
+        self.new_image["aspect_ratio"] = "4:3"
+        self.new_image["quality"] = None
+
+        created_image = self.__generate_image()
+        self.assertEqual(len(created_image["data"]), 1, "Expected an image to be generated")

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/pygeai/tests/integration/lab/agents/test_create_agent.py

@@ -287,21 +287,18 @@ class TestAILabCreateAgentIntegration(TestCase):
                 )
 
     
-    @unittest.skip("Agent is getting created regardless of the prompt instructions being empty")
     def test_create_agent_no_prompt_instructions(self):
         self.new_agent.agent_data.prompt.instructions = ""
-        with self.assertRaises(ValidationError) as exception:
-            self.__create_agent()
-        self.assertIn(
-            "instructions",
-            str(exception.exception),
-            "Expected a validation error about allowed values for instructions"
-        )
+        self.created_agent = self.__create_agent()
 
-        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"
+        self.assertTrue(
+            isinstance(self.created_agent, Agent),
+            "Expected a created agent"
+        )
+        
+        self.assertIsNone(
+            self.created_agent.agent_data.prompt.instructions,
+            "Expected the created agent to not have prompt instructions"
         )
 
     

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/pygeai/tests/integration/lab/agents/test_update_agent.py

@@ -207,28 +207,20 @@ class TestAILabUpdateAgentIntegration(TestCase):
                     f"Expected a validation error about allowed values for instructions when autopublish is {'enabled' if auto_publish else 'disabled'}"
                 )
 
-    
+    # TODO: Change validation when API behavior is fixed
     def test_update_agent_no_model(self):        
         test_params = [ True, False ]
         self.agent_to_update.agent_data.models[0].name = ""
         
         for auto_publish in test_params:
             with self.subTest(input=auto_publish):
-                if auto_publish == False:
-                    with self.assertRaises(ValidationError) as exception:
-                        self.__update_agent(automatic_publish=auto_publish)
+                
+                # If the agent is not published, the API returns a warning message for invalid model name. However, the sdk mapping is not returning it.
+                if auto_publish == False:                    
+                    updated_agent = self.__update_agent(automatic_publish=auto_publish)
                     error_msg = str(exception.exception)
                     
-                    self.assertIn(
-                        "name",
-                        error_msg,
-                        "Expected a validation error about empty model name" 
-                    )
-                    self.assertIn(
-                        "Input should be a valid string [type=string_type, input_value=None, input_type=NoneType]",
-                        error_msg,
-                        "Expected a validation error about empty model name" 
-                    )
+                    self.assertTrue(self.isInstance(updated_agent), Agent)
                 else:                 
                     with self.assertRaises(APIError) as exception:
                         self.__update_agent(automatic_publish=auto_publish)

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/pygeai/tests/integration/lab/tools/test_create_tool.py

@@ -28,7 +28,6 @@ class TestAILabCreateToolIntegration(TestCase):
 
     
     def __load_tool(self):
-        #random_str = str(uuid.uuid4())
         return Tool(
             name=str(uuid.uuid4()),
             description="Tool created for sdk testing purposes",

pygeai-0.4.0b4/pygeai/tests/integration/lab/tools/test_list_tools.py

@@ -0,0 +1,38 @@
+from unittest import TestCase
+from pygeai.lab.managers import AILabManager
+from pygeai.lab.models import ToolList, FilterSettings
+import copy
+
+ai_lab_manager: AILabManager
+
+class TestAILabListToolsIntegration(TestCase):    
+
+    def setUp(self):
+        self.ai_lab_manager = AILabManager(alias="beta")
+        self.filter_settings = FilterSettings(
+            allow_external=False,
+            allow_drafts=True,
+            access_scope="private"
+        )
+    """ list-tools or lt            List tools
+        --project-id or --pid           ID of the project
+        --id                    ID of the tool to filter by. Defaults to an empty string (no filtering).
+        --count                 Number of tools to retrieve. Defaults to '100'.
+        --access-scope          Access scope of the tools, either "public" or "private". Defaults to "public".
+        --allow-drafts          Whether to include draft tools. Defaults to 1 (True).
+        --scope                 Scope of the tools, must be 'builtin', 'external', or 'api'. Defaults to 'api'.
+        --allow-external                Whether to include external tools. Defaults to 1 (True). """
+
+    def __list_tools(self, filter_settings: FilterSettings = None):
+        filter_settings = filter_settings if filter_settings != None else self.filter_settings
+        return self.ai_lab_manager.list_tools(filter_settings=filter_settings)
+    
+    def test_private_list_tools(self):
+        result = self.__list_tools()
+        self.assertIsInstance(result, ToolList , "Expected a list of tools") 
+        print(result)
+        for tool in result.tools:
+            self.assertTrue(tool.access_scope == "private", "Expected all tools to be private")
+        
+
+

{pygeai-0.4.0b3 → pygeai-0.4.0b4/pygeai.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: pygeai
-Version: 0.4.0b3
+Version: 0.4.0b4
 Summary: Software Development Kit to interact with Globant Enterprise AI.
 Author-email: Globant <geai-sdk@globant.com>
 License-Expression: MIT
@@ -41,11 +41,11 @@ Dynamic: license-file
 
 PyGEAI is a Software Development Kit (SDK) for interacting with [Globant Enterprise AI](https://wiki.genexus.com/enterprise-ai/wiki?8,Table+of+contents%3AEnterprise+AI). It comprises libraries, tools, code samples, and documentation to simplify your experience with the platform.
 
-## Repository
+## Terms and conditions
 
-Find the PyGEAI source code and documentation in the following GitHub repository:
+By using the Python SDK to interact with Globant Enterprise AI, you agree with the following Terms and Conditions:
 
-[GitHub repository](https://github.com/RT-GEN029-GI/pygeai)
+[Terms and Conditions](https://www.globant.com/enterprise-ai/terms-of-use)
 
 ## Compatibility
 This package is compatible with the Globant Enterprise AI release from June 2025.

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/pygeai.egg-info/SOURCES.txt

@@ -279,6 +279,8 @@ pygeai/tests/gam/test_clients.py
 pygeai/tests/health/__init__.py
 pygeai/tests/health/test_clients.py
 pygeai/tests/integration/__init__.py
+pygeai/tests/integration/chat/__init__.py
+pygeai/tests/integration/chat/test_generate_image.py
 pygeai/tests/integration/lab/__init__.py
 pygeai/tests/integration/lab/agents/__init__.py
 pygeai/tests/integration/lab/agents/test_agents_list.py
@@ -292,6 +294,7 @@ pygeai/tests/integration/lab/tools/__init__.py
 pygeai/tests/integration/lab/tools/test_create_tool.py
 pygeai/tests/integration/lab/tools/test_delete_tool.py
 pygeai/tests/integration/lab/tools/test_get_tool.py
+pygeai/tests/integration/lab/tools/test_list_tools.py
 pygeai/tests/lab/__init__.py
 pygeai/tests/lab/test_managers.py
 pygeai/tests/lab/test_mappers.py

{pygeai-0.4.0b3 → pygeai-0.4.0b4}/pyproject.toml

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