pygeai 0.2.7b38__tar.gz → 0.2.7b40__tar.gz

{pygeai-0.2.7b38/pygeai.egg-info → pygeai-0.2.7b40}/PKG-INFO

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

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/cli/commands/configuration.py

@@ -1,5 +1,3 @@
-import sys
-
 from pygeai.cli.commands import Option
 from pygeai.core.common.config import get_settings
 from pygeai.core.utils.console import Console
@@ -34,7 +32,7 @@ def configure(option_list: list[str, str] = None):
         for option_flag, option_arg in option_list:
             if option_flag.name == "list":
                 list_alias = True
-            if option_flag.name == "alias":
+            if option_flag.name == "profile_alias":
                 alias = option_arg
             if option_flag.name == "api_key":
                 api_key = option_arg
@@ -98,8 +96,8 @@ configuration_options = (
         True
     ),
     Option(
-        "alias",
-        ["--alias", "-a"],
+        "profile_alias",
+        ["--profile-alias", "--pa"],
         "Set alias for settings section",
         True
     ),

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/core/files/managers.py

@@ -1,4 +1,5 @@
 from pygeai import logger
+from pygeai.admin.clients import AdminClient
 from pygeai.core.base.mappers import ErrorMapper, ResponseMapper
 from pygeai.core.base.responses import EmptyResponse
 from pygeai.core.files.clients import FileClient
@@ -28,8 +29,26 @@ class FileManager:
             base_url,
             alias
         )
-        self.__organization_id = organization_id
-        self.__project_id = project_id
+        self.organization_id = self.__get_organization_id() if not organization_id else organization_id
+        self.project_id = self.__get_project_id() if not project_id else project_id
+
+    def __get_organization_id(self):
+        response = None
+        try:
+            response = AdminClient().validate_api_token()
+            return response.get("organizationId")
+        except Exception as e:
+            logger.error(f"Error retrieving organization_id from GEAI. Response: {response}: {e}")
+            raise APIError(f"Error retrieving organization_id from GEAI: {e}")
+
+    def __get_project_id(self):
+        response = None
+        try:
+            response = AdminClient().validate_api_token()
+            return response.get("projectId")
+        except Exception as e:
+            logger.error(f"Error retrieving project_id from GEAI. Response: {response}: {e}")
+            raise APIError(f"Error retrieving project_id from GEAI: {e}")
 
     def upload_file(
             self,
@@ -47,8 +66,8 @@ class FileManager:
         """
         response_data = self.__client.upload_file(
             file_path=file.path,
-            organization_id=self.__organization_id,
-            project_id=self.__project_id,
+            organization_id=self.organization_id,
+            project_id=self.project_id,
             folder=file.folder,
             file_name=file.name,
         )
@@ -75,8 +94,8 @@ class FileManager:
         :raises APIError: If the API returns errors.
         """
         response_data = self.__client.get_file(
-            organization=self.__organization_id,
-            project=self.__project_id,
+            organization=self.organization_id,
+            project=self.project_id,
             file_id=file_id
         )
         if ErrorHandler.has_errors(response_data):
@@ -101,8 +120,8 @@ class FileManager:
         :raises APIError: If the API returns errors.
         """
         response_data = self.__client.delete_file(
-            organization=self.__organization_id,
-            project=self.__project_id,
+            organization=self.organization_id,
+            project=self.project_id,
             file_id=file_id
         )
         if ErrorHandler.has_errors(response_data):
@@ -128,8 +147,8 @@ class FileManager:
         :raises APIError: If the API returns errors.
         """
         response_data = self.__client.get_file_content(
-            organization=self.__organization_id,
-            project=self.__project_id,
+            organization=self.organization_id,
+            project=self.project_id,
             file_id=file_id
         )
         if isinstance(response_data, dict) and "errors" in response_data:
@@ -151,8 +170,8 @@ class FileManager:
         :raises APIError: If the API returns errors.
         """
         response_data = self.__client.get_file_list(
-            organization=self.__organization_id,
-            project=self.__project_id
+            organization=self.organization_id,
+            project=self.project_id
         )
         if ErrorHandler.has_errors(response_data):
             error = ErrorHandler.extract_error(response_data)

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/core/handlers.py

@@ -6,7 +6,16 @@ class ErrorHandler:
     @classmethod
     def has_errors(cls, response):
         error_found = False
-        if "errors" in response or "error" in response:
+        if (
+                "errors" in response or
+                "error" in response or
+                (
+                        "message" in response and
+                        isinstance(response.get("message"), list) and
+                        len(response.get("message")) > 0 and
+                        response.get("message")[0].get("type") == "error"
+                )
+        ):
             error_found = True
 
         return error_found

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/lab/agents/clients.py

@@ -73,6 +73,7 @@ class AgentClient(BaseClient):
             description: str,
             agent_data_prompt: dict,
             agent_data_llm_config: dict,
+            agent_data_strategy_name: str,
             agent_data_models: list,
             agent_data_resource_pools: Optional[List[Dict]] = None,
             automatic_publish: bool = False
@@ -89,6 +90,7 @@ class AgentClient(BaseClient):
         :param description: str - Detailed description of the agent’s purpose (optional).
         :param agent_data_prompt: dict - Prompt configuration, including 'context', 'instructions', and optional 'examples' (e.g., {'context': str, 'instructions': str, 'examples': [{'inputData': str, 'output': str}]}).
         :param agent_data_llm_config: dict - LLM configuration (e.g., {'maxTokens': int, 'timeout': int, 'sampling': {'temperature': float}}).
+        :param agent_data_strategy_name: str - Strategy name to be used.
         :param agent_data_models: list - List of models the agent can use (e.g., [{'name': 'gpt-4o', 'llmConfig': dict}]).
         :param agent_data_resource_pools: Optional[List[Dict]] - Resource pools for tools and helper agents (e.g., [{'name': str, 'tools': [{'name': str, 'revision': int}], 'agents': [{'name': str, 'revision': int}]}]).
         :param automatic_publish: bool - Automatically publish the agent after creation (default: False).
@@ -107,17 +109,23 @@ class AgentClient(BaseClient):
                 "jobDescription": job_description,
                 "avatarImage": avatar_image,
                 "description": description,
-                "agentData": {
-                    "prompt": agent_data_prompt,
-                    "llmConfig": agent_data_llm_config,
-                    "models": agent_data_models
-                }
             }
         }
-
+        if (
+                agent_data_prompt or agent_data_strategy_name or agent_data_prompt or agent_data_resource_pools or
+                agent_data_llm_config or agent_data_models
+        ):
+            data["agentDefinition"]["agentData"] = {}
         if agent_data_resource_pools is not None:
             data["agentDefinition"]["agentData"]["resourcePools"] = agent_data_resource_pools
-
+        if agent_data_prompt is not None:
+            data["agentDefinition"]["agentData"]["prompt"] = agent_data_prompt
+        if agent_data_llm_config is not None:
+            data["agentDefinition"]["agentData"]["llmConfig"] = agent_data_llm_config
+        if agent_data_strategy_name is not None:
+            data["agentDefinition"]["agentData"]["strategyName"] = agent_data_strategy_name
+        if agent_data_models is not None:
+            data["agentDefinition"]["agentData"]["models"] = agent_data_models
         logger.debug(f"Creating agent with data: {data}")
 
         endpoint = CREATE_AGENT_V2
@@ -300,13 +308,11 @@ class AgentClient(BaseClient):
             headers=headers,
             data={}
         )
-        try:
-            result = response.json()
-        except JSONDecodeError as e:
-            logger.error(f"Unable to delete agent {agent_id} from project {project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
+        if response.status_code != 204:
+            logger.error(f"Unable to delete agent {agent_id} from project {project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
             raise InvalidAPIResponseException(f"Unable to delete agent {agent_id} from project {project_id}: {response.text}")
-
-        return result
+        else:
+            return {}
 
     def update_agent(
             self,
@@ -320,6 +326,7 @@ class AgentClient(BaseClient):
             description: str,
             agent_data_prompt: dict,
             agent_data_llm_config: dict,
+            agent_data_strategy_name: dict,
             agent_data_models: list,
             agent_data_resource_pools: Optional[List[Dict]] = None,
             automatic_publish: bool = False,
@@ -338,6 +345,7 @@ class AgentClient(BaseClient):
         :param description: str - Updated purpose description (optional).
         :param agent_data_prompt: dict - Updated prompt configuration (e.g., {'context': str, 'instructions': str, 'examples': [{'inputData': str, 'output': str}]}).
         :param agent_data_llm_config: dict - Updated LLM configuration (e.g., {'maxTokens': int, 'timeout': int, 'sampling': {'temperature': float}}).
+        :param agent_data_strategy_name: str - Updated StrategyName configuration.
         :param agent_data_models: list - Updated model list (e.g., [{'name': 'gpt-4o', 'llmConfig': dict}]).
         :param agent_data_resource_pools: Optional[List[Dict]] - Updated resource pools (e.g., [{'name': str, 'tools': [{'name': str, 'revision': int}], 'agents': [{'name': str, 'revision': int}]}]).
         :param automatic_publish: bool - Automatically publish the agent after updating (default: False).
@@ -357,15 +365,23 @@ class AgentClient(BaseClient):
                 "jobDescription": job_description,
                 "avatarImage": avatar_image,
                 "description": description,
-                "agentData": {
-                    "prompt": agent_data_prompt,
-                    "llmConfig": agent_data_llm_config,
-                    "models": agent_data_models
-                }
             }
         }
+        if (
+                agent_data_prompt or agent_data_strategy_name or agent_data_prompt or agent_data_resource_pools or
+                agent_data_llm_config or agent_data_models
+        ):
+            data["agentDefinition"]["agentData"] = {}
         if agent_data_resource_pools is not None:
             data["agentDefinition"]["agentData"]["resourcePools"] = agent_data_resource_pools
+        if agent_data_prompt is not None:
+            data["agentDefinition"]["agentData"]["prompt"] = agent_data_prompt
+        if agent_data_llm_config is not None:
+            data["agentDefinition"]["agentData"]["llmConfig"] = agent_data_llm_config
+        if agent_data_strategy_name is not None:
+            data["agentDefinition"]["agentData"]["strategyName"] = agent_data_strategy_name
+        if agent_data_models is not None:
+            data["agentDefinition"]["agentData"]["models"] = agent_data_models
 
         logger.debug(f"Updating agent with ID {agent_id} with data: {data}")
 
@@ -392,4 +408,4 @@ class AgentClient(BaseClient):
             logger.error(f"Unable to update agent {agent_id} in project {project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
             raise InvalidAPIResponseException(f"Unable to update agent {agent_id} in project {project_id}: {response.text}")
 
-        return result
+        return result

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/lab/agents/mappers.py

@@ -65,10 +65,12 @@ class AgentMapper:
         prompt_data = data.get("prompt")
         llm_config_data = data.get("llmConfig")
         models_list = data.get("models")
+        strategy_name = data.get("strategyName")
         resource_pool_list = data.get("resourcePools")
         return AgentData(
             prompt=cls._map_to_prompt(prompt_data) if prompt_data else None,
             llm_config=cls._map_to_llm_config(llm_config_data) if llm_config_data else None,
+            strategy_name=strategy_name,
             models=cls._map_to_model_list(models_list) if models_list else None,
             resource_pools=cls._map_to_resource_pool_list(resource_pool_list) if resource_pool_list else None
         )

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/lab/managers.py

@@ -101,6 +101,7 @@ class AILabManager:
             avatar_image=agent.avatar_image,
             description=agent.description,
             agent_data_prompt=agent.agent_data.prompt.to_dict() if agent.agent_data is not None else None,
+            agent_data_strategy_name=agent.agent_data.strategy_name if agent.agent_data is not None else None,
             agent_data_llm_config=agent.agent_data.llm_config.to_dict() if agent.agent_data is not None else None,
             agent_data_models=agent.agent_data.models.to_dict() if agent.agent_data is not None else None,
             agent_data_resource_pools=agent.agent_data.resource_pools.to_dict() if agent.agent_data and agent.agent_data.resource_pools else None,
@@ -149,6 +150,7 @@ class AILabManager:
             description=agent.description,
             agent_data_prompt=agent.agent_data.prompt.to_dict() if agent.agent_data is not None else None,
             agent_data_llm_config=agent.agent_data.llm_config.to_dict() if agent.agent_data is not None else None,
+            agent_data_strategy_name=agent.agent_data.strategy_name if agent.agent_data is not None else None,
             agent_data_models=agent.agent_data.models.to_dict() if agent.agent_data is not None else None,
             automatic_publish=automatic_publish,
             upsert=upsert

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/lab/models.py

@@ -166,6 +166,14 @@ class Prompt(CustomBaseModel):
     context: Optional[str] = Field(None, alias="context", description="Background context for the agent")
     examples: Optional[List[PromptExample]] = Field(None, alias="examples")
 
+    @field_validator("instructions")
+    @classmethod
+    def validate_instructions(cls, value: str) -> str:
+        if not value.strip():
+            raise ValueError("instructions cannot be blank")
+
+        return value
+
     @field_validator("outputs", mode="before")
     @classmethod
     def normalize_outputs(cls, value):
@@ -380,11 +388,13 @@ class AgentData(CustomBaseModel):
 
     :param prompt: dict - Prompt instructions, inputs, outputs, and examples for the agent.
     :param llm_config: dict - Configuration parameters for the LLM (e.g., max tokens, timeout, temperature).
+    :param strategy_name: str - Strategy name used for the agent (e.g., Dynamic Prompting, Chain of Thought, Question Refinement, etc)
     :param models: ModelList - List of models available for the agent.
     :param resource_pools: Optional[List[ResourcePool]] - List of resource pools organizing tools and helper agents.
     """
     prompt: Prompt = Field(..., alias="prompt")
     llm_config: LlmConfig = Field(..., alias="llmConfig")
+    strategy_name: Optional[str] = Field("Dynamic Prompting", alias="strategyName")
     models: Optional[Union[ModelList, List[Model]]] = Field(None, alias="models")
     resource_pools: Optional[ResourcePoolList] = Field(None, alias="resourcePools", description="List of resource pools organizing tools and helper agents")
 

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/lab/processes/clients.py

@@ -399,13 +399,12 @@ class AgenticProcessClient(BaseClient):
             endpoint=endpoint,
             headers=headers
         )
-        try:
-            result = response.json()
-        except JSONDecodeError as e:
-            logger.error(f"Unable to delete process {process_id or process_name} from project {project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
-            raise InvalidAPIResponseException(f"Unable to delete process {process_id or process_name} from project {project_id}: {response.text}")
 
-        return result
+        if response.status_code != 204:
+            logger.error(f"Unable to delete process {process_id or process_name} from project {project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
+            raise InvalidAPIResponseException(f"Unable to delete process {process_id or process_name} from project {project_id}: {response.text}")
+        else:
+            return {}
 
     def publish_process_revision(
             self,
@@ -699,11 +698,12 @@ class AgenticProcessClient(BaseClient):
             logger.debug(f"Deleting task with name {task_name}")
 
         response = self.api_service.delete(endpoint=endpoint, headers=headers)
-        try:
-            return response.json()
-        except JSONDecodeError as e:
-            logger.error(f"Unable to delete task {task_id or task_name} from project {project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
+
+        if response.status_code != 204:
+            logger.error(f"Unable to delete task {task_id or task_name} from project {project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
             raise InvalidAPIResponseException(f"Unable to delete task {task_id or task_name} from project {project_id}: {response.text}")
+        else:
+            return {}
 
     def publish_task_revision(
             self,
@@ -1121,11 +1121,12 @@ class AgenticProcessClient(BaseClient):
             logger.debug(f"Deleting KB with name {kb_name}")
 
         response = self.api_service.delete(endpoint=endpoint, headers=headers)
-        try:
-            return response.json()
-        except JSONDecodeError as e:
-            logger.error(f"Unable to delete knowledge base {kb_id or kb_name} from project {project_id}: JSONDecodeError parsing error (status {response.status_code}): {e}. Response: {response.text}")
+
+        if response.status_code != 204:
+            logger.error(f"Unable to delete knowledge base {kb_id or kb_name} from project {project_id}: JSONDecodeError parsing error (status {response.status_code}). Response: {response.text}")
             raise InvalidAPIResponseException(f"Unable to delete knowledge base {kb_id or kb_name} from project {project_id}: {response.text}")
+        else:
+            return {}
 
     def list_jobs(
             self,

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/lab/tools/clients.py

@@ -256,13 +256,11 @@ class ToolClient(BaseClient):
             headers=headers
         )
 
-        try:
-            result = response.json()
-        except JSONDecodeError as e:
-            logger.error(f"Unable to delete tool {tool_id or tool_name} from project {project_id}: JSON parsing error (status {response.status_code}): {e}. Response: {response.text}")
+        if response.status_code != 204:
+            logger.error(f"Unable to delete tool {tool_id or tool_name} from project {project_id}: JSON parsing error (status {response.status_code}). Response: {response.text}")
             raise InvalidAPIResponseException(f"Unable to delete tool {tool_id or tool_name} from project {project_id}: {response.text}")
-
-        return result
+        else:
+            return {}
 
     def update_tool(
             self,

{pygeai-0.2.7b38 → pygeai-0.2.7b40}/pygeai/tests/integration/lab/agents/test_create_agent.py

@@ -13,7 +13,6 @@ class TestAILabCreateAgentIntegration(TestCase):
         Set up the test environment.
         """
         self.ai_lab_manager = AILabManager(alias="beta")
-        self.project_id = "be4889df-cacc-4e6f-b3bb-153c4ac0d168"
         self.new_agent = self.__load_agent()
         self.created_agent: Agent = None
 
@@ -26,7 +25,7 @@ class TestAILabCreateAgentIntegration(TestCase):
         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.project_id, self.created_agent.id)
+                self.ai_lab_manager.delete_agent(self.created_agent.id)
 
 
     def __load_agent(self):
@@ -66,7 +65,6 @@ class TestAILabCreateAgentIntegration(TestCase):
         If automatic_publish is None, do not pass it (useful for tests that omit it).
         """
         return self.ai_lab_manager.create_agent(
-            project_id=self.project_id,
             agent=self.new_agent if agent is None else agent,
             automatic_publish=automatic_publish
         )

pygeai-0.2.7b40/pygeai/tests/snippets/files/delete_file.py

@@ -0,0 +1,9 @@
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import File
+
+file = File(id="7fe2393d-8f86-4020-a51f-bc14ab957e1e")
+
+file_manager = FileManager()
+
+response = file_manager.delete_file(file_id=file.id)
+print(response)

pygeai-0.2.7b40/pygeai/tests/snippets/files/get_file_content.py

@@ -0,0 +1,10 @@
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import File
+
+
+file = File(id="7fe2393d-8f86-4020-a51f-bc14ab957e1e")
+
+file_manager = FileManager()
+
+response = file_manager.get_file_content(file_id=file.id)
+print(response)

pygeai-0.2.7b40/pygeai/tests/snippets/files/get_file_data.py

@@ -0,0 +1,9 @@
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import File
+
+file = File(id="9984b837-fe88-4014-ad14-91e1596c8ead")
+
+file_manager = FileManager()
+
+response = file_manager.get_file_data(file_id=file.id)
+print(response)

pygeai-0.2.7b40/pygeai/tests/snippets/files/get_file_list.py

@@ -0,0 +1,6 @@
+from pygeai.core.files.managers import FileManager
+
+file_manager = FileManager()
+
+response = file_manager.get_file_list()
+print(response)

pygeai-0.2.7b40/pygeai/tests/snippets/files/upload_file.py

@@ -0,0 +1,13 @@
+from pygeai.core.files.managers import FileManager
+from pygeai.core.files.models import UploadFile
+
+file = UploadFile(
+    path="test.txt",
+    name="TestyFile",
+    folder="TestyTestTemp"
+)
+
+file_manager = FileManager()
+
+response = file_manager.upload_file(file)
+print(response)