pygeai 0.2.7b37__py3-none-any.whl → 0.2.7b39__py3-none-any.whl

pygeai/core/base/mappers.py

@@ -351,13 +351,22 @@ class ModelMapper:
     @classmethod
     def map_to_item(cls, data: dict) -> RequestItem:
         return RequestItem(
+            api_token=data.get('apiToken'),
             assistant=data.get('assistant'),
+            cost=data.get('cost'),
+            elapsed_time_ms=data.get('elapsedTimeMs'),
+            end_timestamp=data.get('endTimestamp'),
+            feedback=data.get('feedback'),
             intent=data.get('intent'),
-            timestamp=data.get('timestamp'),
+            module=data.get('module'),
             prompt=data.get('prompt'),
             output=data.get('output'),
             input_text=data.get('inputText'),
-            status=data.get('status')
+            rag_sources_consulted=data.get('ragSourcesConsulted'),
+            session_id=data.get('sessionId'),
+            start_timestamp=data.get('startTimestamp'),
+            status=data.get('status'),
+            timestamp=data.get('timestamp')
         )
 
     @classmethod

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/core/models.py

@@ -1,6 +1,6 @@
 from datetime import datetime
 
-from pydantic import Field, field_validator
+from pydantic import Field, field_validator, model_validator
 from typing import Optional, Literal, Any, List, Union, Iterator
 
 from pygeai.core import CustomBaseModel
@@ -247,23 +247,32 @@ class Project(CustomBaseModel):
 
 class RequestItem(CustomBaseModel):
     """
-    {
-      "assistant": "string",
-      "intent": "string",
-      "timestamp": "string",
-      "prompt": "string",
-      "output": "string",
-      "inputText": "string",
-      "status": "string"
-    }
+    Represents a request item with metadata about an API interaction.
+    Follows a JSON schema with camelCase aliases for external compatibility.
     """
+    api_token: str = Field(..., alias="apiToken")
     assistant: str = Field(..., alias="assistant")
+    cost: float = Field(..., alias="cost")
+    elapsed_time_ms: int = Field(..., alias="elapsedTimeMs")
+    end_timestamp: datetime = Field(..., alias="endTimestamp")
+    feedback: Optional[str] = Field(None, alias="feedback")
     intent: Optional[str] = Field(None, alias="intent")
-    timestamp: str = Field(..., alias="timestamp")
+    module: str = Field(..., alias="module")
     prompt: Optional[str] = Field(None, alias="prompt")
     output: Optional[str] = Field(None, alias="output")
     input_text: Optional[str] = Field(None, alias="inputText")
+    rag_sources_consulted: Optional[str] = Field(None, alias="ragSourcesConsulted")
+    session_id: str = Field(..., alias="sessionId")
+    start_timestamp: datetime = Field(..., alias="startTimestamp")
     status: str = Field(..., alias="status")
+    timestamp: datetime = Field(..., alias="timestamp")
+
+    @model_validator(mode="after")
+    def validate_status(self):
+        valid_statuses = {"succeeded", "failed", "pending"}
+        if self.status not in valid_statuses:
+            raise ValueError(f"Status must be one of {valid_statuses}")
+        return self
 
     def to_dict(self):
         return self.model_dump(by_alias=True, exclude_none=True)

pygeai/lab/managers.py

@@ -22,12 +22,12 @@ from pygeai.lab.tools.mappers import ToolMapper
 
 class AILabManager:
 
-    def __init__(self, api_key: str = None, base_url: str = None, alias: str = "default"):
+    def __init__(self, api_key: str = None, base_url: str = None, alias: str = "default", project_id: str = None):
         self.__agent_client = AgentClient(api_key=api_key, base_url=base_url, alias=alias)
         self.__tool_client = ToolClient(api_key=api_key, base_url=base_url, alias=alias)
         self.__reasoning_strategy_client = ReasoningStrategyClient(api_key=api_key, base_url=base_url, alias=alias)
         self.__process_client = AgenticProcessClient(api_key=api_key, base_url=base_url, alias=alias)
-        self.project_id = self.__get_project_id()
+        self.project_id = self.__get_project_id() if not project_id else project_id
 
     def __get_project_id(self):
         response = None

pygeai/organization/responses.py

@@ -22,3 +22,26 @@ class ProjectTokensResponse(BaseModel):
 
 class ProjectItemListResponse(BaseModel):
     items: list[RequestItem]
+
+    def to_list(self):
+        return [item.to_dict() for item in self.items] if self.items else []
+
+    def __getitem__(self, index: int) -> RequestItem:
+        if self.items is None:
+            raise IndexError("ProjectItemListResponse is empty")
+        return self.items[index]
+
+    def __len__(self) -> int:
+        return len(self.items) if self.items else 0
+
+    def __iter__(self):
+        """Make ProjectItemListResponse iterable over its items."""
+        if self.items is None:
+            return iter([])
+        return iter(self.items)
+
+    def append(self, item: RequestItem) -> None:
+        """Append an Agent instance to the items list."""
+        if self.items is None:
+            self.items = []
+        self.items.append(item)

pygeai/tests/lab/test_managers.py

@@ -13,9 +13,9 @@ class TestAILabManager(unittest.TestCase):
     """
     python -m unittest pygeai.tests.lab.test_managers.TestAILabManager
     """
+
     def setUp(self):
-        self.manager = AILabManager(api_key="test_key", base_url="test_url", alias="test_alias")
-        self.project_id = "test_project"
+        self.manager = AILabManager(api_key="test_key", base_url="test_url", alias="test_alias", project_id="test_project")
         self.agent_id = "test_agent"
         self.tool_id = "test_tool"
         self.process_id = "test_process"
@@ -31,7 +31,7 @@ class TestAILabManager(unittest.TestCase):
         mock_list_agents.return_value = {"data": "test_data"}
         mock_map_to_agent_list.return_value = AgentList(agents=[])
 
-        result = self.manager.get_agent_list(self.project_id, self.filter_settings)
+        result = self.manager.get_agent_list(self.filter_settings)
 
         mock_list_agents.assert_called_once()
         mock_map_to_agent_list.assert_called_once_with({"data": "test_data"})
@@ -42,7 +42,7 @@ class TestAILabManager(unittest.TestCase):
         mock_list_agents.return_value = {"error": {"id": 1, "message": "API error"}}
 
         with self.assertRaises(APIError):
-            self.manager.get_agent_list(self.project_id, self.filter_settings)
+            self.manager.get_agent_list(self.filter_settings)
 
     @patch('pygeai.lab.agents.clients.AgentClient.create_agent')
     @patch('pygeai.lab.agents.mappers.AgentMapper.map_to_agent')
@@ -51,7 +51,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_agent.return_value = Agent(id=self.agent_id, name="test_agent")
         agent = Agent(name="test_agent")
 
-        result = self.manager.create_agent(self.project_id, agent, automatic_publish=False)
+        result = self.manager.create_agent(agent, automatic_publish=False)
 
         mock_create_agent.assert_called_once()
         mock_map_to_agent.assert_called_once_with({"data": "test_data"})
@@ -64,7 +64,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_agent.return_value = Agent(id=self.agent_id, name="test_agent")
         agent = Agent(id=self.agent_id, name="test_agent")
 
-        result = self.manager.update_agent(self.project_id, agent, automatic_publish=False, upsert=False)
+        result = self.manager.update_agent(agent, automatic_publish=False, upsert=False)
 
         mock_update_agent.assert_called_once()
         mock_map_to_agent.assert_called_once_with({"data": "test_data"})
@@ -76,7 +76,7 @@ class TestAILabManager(unittest.TestCase):
         mock_get_agent.return_value = {"data": "test_data"}
         mock_map_to_agent.return_value = Agent(id=self.agent_id, name="test_agent")
 
-        result = self.manager.get_agent(self.project_id, self.agent_id, self.filter_settings)
+        result = self.manager.get_agent(self.agent_id, self.filter_settings)
 
         mock_get_agent.assert_called_once()
         mock_map_to_agent.assert_called_once_with({"data": "test_data"})
@@ -88,7 +88,7 @@ class TestAILabManager(unittest.TestCase):
         mock_create_sharing_link.return_value = {"data": "test_data"}
         mock_map_to_sharing_link.return_value = SharingLink(agent_id="test_agent", api_token="test_token", shared_link="test_link")
 
-        result = self.manager.create_sharing_link(self.project_id, self.agent_id)
+        result = self.manager.create_sharing_link(self.agent_id)
 
         mock_create_sharing_link.assert_called_once()
         mock_map_to_sharing_link.assert_called_once_with({"data": "test_data"})
@@ -100,7 +100,7 @@ class TestAILabManager(unittest.TestCase):
         mock_publish_agent_revision.return_value = {"data": "test_data"}
         mock_map_to_agent.return_value = Agent(id=self.agent_id, name="test_agent")
 
-        result = self.manager.publish_agent_revision(self.project_id, self.agent_id, self.revision)
+        result = self.manager.publish_agent_revision(self.agent_id, self.revision)
 
         mock_publish_agent_revision.assert_called_once()
         mock_map_to_agent.assert_called_once_with({"data": "test_data"})
@@ -112,7 +112,7 @@ class TestAILabManager(unittest.TestCase):
         mock_delete_agent.return_value = {"data": "test_data"}
         mock_map_to_empty_response.return_value = {"status": "success"}
 
-        result = self.manager.delete_agent(self.project_id, self.agent_id)
+        result = self.manager.delete_agent(self.agent_id)
 
         mock_delete_agent.assert_called_once()
         mock_map_to_empty_response.assert_called_once()
@@ -125,7 +125,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_tool.return_value = Tool(id=self.tool_id, name="test_tool", description="test_desc")
         tool = Tool(name="test_tool", description="test_desc")
 
-        result = self.manager.create_tool(self.project_id, tool, automatic_publish=False)
+        result = self.manager.create_tool(tool, automatic_publish=False)
 
         mock_create_tool.assert_called_once()
         mock_map_to_tool.assert_called_once_with({"data": "test_data"})
@@ -138,7 +138,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_tool.return_value = Tool(id=self.tool_id, name="test_tool", description="test_desc")
         tool = Tool(id=self.tool_id, name="test_tool", description="test_desc")
 
-        result = self.manager.update_tool(self.project_id, tool, automatic_publish=False, upsert=False)
+        result = self.manager.update_tool(tool, automatic_publish=False, upsert=False)
 
         mock_update_tool.assert_called_once()
         mock_map_to_tool.assert_called_once_with({"data": "test_data"})
@@ -150,7 +150,7 @@ class TestAILabManager(unittest.TestCase):
         mock_get_tool.return_value = {"data": "test_data"}
         mock_map_to_tool.return_value = Tool(id=self.tool_id, name="test_tool", description="test_desc")
 
-        result = self.manager.get_tool(self.project_id, self.tool_id, self.filter_settings)
+        result = self.manager.get_tool(self.tool_id, self.filter_settings)
 
         mock_get_tool.assert_called_once()
         mock_map_to_tool.assert_called_once_with({"data": "test_data"})
@@ -162,7 +162,7 @@ class TestAILabManager(unittest.TestCase):
         mock_delete_tool.return_value = {"data": "test_data"}
         mock_map_to_empty_response.return_value = {"status": "success"}
 
-        result = self.manager.delete_tool(self.project_id, tool_id=self.tool_id)
+        result = self.manager.delete_tool(tool_id=self.tool_id)
 
         mock_delete_tool.assert_called_once()
         mock_map_to_empty_response.assert_called_once()
@@ -174,7 +174,7 @@ class TestAILabManager(unittest.TestCase):
         mock_list_tools.return_value = {"data": "test_data"}
         mock_map_to_tool_list.return_value = ToolList(tools=[])
 
-        result = self.manager.list_tools(self.project_id, self.filter_settings)
+        result = self.manager.list_tools(self.filter_settings)
 
         mock_list_tools.assert_called_once()
         mock_map_to_tool_list.assert_called_once_with({"data": "test_data"})
@@ -186,7 +186,7 @@ class TestAILabManager(unittest.TestCase):
         mock_publish_tool_revision.return_value = {"data": "test_data"}
         mock_map_to_tool.return_value = Tool(id=self.tool_id, name="test_tool", description="test_desc")
 
-        result = self.manager.publish_tool_revision(self.project_id, self.tool_id, self.revision)
+        result = self.manager.publish_tool_revision(self.tool_id, self.revision)
 
         mock_publish_tool_revision.assert_called_once()
         mock_map_to_tool.assert_called_once_with({"data": "test_data"})
@@ -198,7 +198,7 @@ class TestAILabManager(unittest.TestCase):
         mock_get_parameter.return_value = {"data": "test_data"}
         mock_map_to_parameter_list.return_value = [ToolParameter(key="test_key", data_type="String", description="test_desc", is_required=True)]
 
-        result = self.manager.get_parameter(self.project_id, tool_id=self.tool_id, filter_settings=self.filter_settings)
+        result = self.manager.get_parameter(tool_id=self.tool_id, filter_settings=self.filter_settings)
 
         mock_get_parameter.assert_called_once()
         mock_map_to_parameter_list.assert_called_once_with({"data": "test_data"})
@@ -211,7 +211,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_tool.return_value = Tool(id=self.tool_id, name="test_tool", description="test_desc")
         parameters = [ToolParameter(key="test_key", data_type="String", description="test_desc", is_required=True)]
 
-        result = self.manager.set_parameter(self.project_id, tool_id=self.tool_id, parameters=parameters)
+        result = self.manager.set_parameter(tool_id=self.tool_id, parameters=parameters)
 
         mock_set_parameter.assert_called_once()
         mock_map_to_tool.assert_called_once_with({"data": "test_data"})
@@ -236,7 +236,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_strategy.return_value = ReasoningStrategy(id="test_strategy", name="test_strategy", access_scope="private", type="addendum")
         strategy = ReasoningStrategy(name="test_strategy", access_scope="private", type="addendum", localized_descriptions=[])
 
-        result = self.manager.create_reasoning_strategy(self.project_id, strategy, automatic_publish=False)
+        result = self.manager.create_reasoning_strategy(strategy, automatic_publish=False)
 
         mock_create_strategy.assert_called_once()
         mock_map_to_strategy.assert_called_once_with({"data": "test_data"})
@@ -249,7 +249,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_strategy.return_value = ReasoningStrategy(id="test_strategy", name="test_strategy", access_scope="private", type="addendum")
         strategy = ReasoningStrategy(id="test_strategy", name="test_strategy", access_scope="private", type="addendum", localized_descriptions=[])
 
-        result = self.manager.update_reasoning_strategy(self.project_id, strategy, automatic_publish=False, upsert=False)
+        result = self.manager.update_reasoning_strategy(strategy, automatic_publish=False, upsert=False)
 
         mock_update_strategy.assert_called_once()
         mock_map_to_strategy.assert_called_once_with({"data": "test_data"})
@@ -261,7 +261,7 @@ class TestAILabManager(unittest.TestCase):
         mock_get_strategy.return_value = {"data": "test_data"}
         mock_map_to_strategy.return_value = ReasoningStrategy(id="test_strategy", name="test_strategy", access_scope="private", type="addendum")
 
-        result = self.manager.get_reasoning_strategy(self.project_id, reasoning_strategy_id="test_strategy")
+        result = self.manager.get_reasoning_strategy(reasoning_strategy_id="test_strategy")
 
         mock_get_strategy.assert_called_once()
         mock_map_to_strategy.assert_called_once_with({"data": "test_data"})
@@ -274,7 +274,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_process.return_value = AgenticProcess(id=self.process_id, name="test_process")
         process = AgenticProcess(name="test_process", key="test_key")
 
-        result = self.manager.create_process(self.project_id, process, automatic_publish=False)
+        result = self.manager.create_process(process, automatic_publish=False)
 
         mock_create_process.assert_called_once()
         mock_map_to_process.assert_called_once_with({"data": "test_data"})
@@ -287,7 +287,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_process.return_value = AgenticProcess(id=self.process_id, name="test_process")
         process = AgenticProcess(id=self.process_id, name="test_process", key="test_key")
 
-        result = self.manager.update_process(self.project_id, process, automatic_publish=False, upsert=False)
+        result = self.manager.update_process(process, automatic_publish=False, upsert=False)
 
         mock_update_process.assert_called_once()
         mock_map_to_process.assert_called_once_with({"data": "test_data"})
@@ -299,7 +299,7 @@ class TestAILabManager(unittest.TestCase):
         mock_get_process.return_value = {"data": "test_data"}
         mock_map_to_process.return_value = AgenticProcess(id=self.process_id, name="test_process")
 
-        result = self.manager.get_process(self.project_id, process_id=self.process_id, filter_settings=self.filter_settings)
+        result = self.manager.get_process(process_id=self.process_id, filter_settings=self.filter_settings)
 
         mock_get_process.assert_called_once()
         mock_map_to_process.assert_called_once_with({"data": "test_data"})
@@ -311,7 +311,7 @@ class TestAILabManager(unittest.TestCase):
         mock_list_processes.return_value = {"data": "test_data"}
         mock_map_to_process_list.return_value = AgenticProcessList(processes=[])
 
-        result = self.manager.list_processes(self.project_id, self.filter_settings)
+        result = self.manager.list_processes(self.filter_settings)
 
         mock_list_processes.assert_called_once()
         mock_map_to_process_list.assert_called_once_with({"data": "test_data"})
@@ -323,7 +323,7 @@ class TestAILabManager(unittest.TestCase):
         mock_list_instances.return_value = {"data": "test_data"}
         mock_map_to_instance_list.return_value = ProcessInstanceList(instances=[])
 
-        result = self.manager.list_process_instances(self.project_id, self.process_id, self.filter_settings)
+        result = self.manager.list_process_instances(self.process_id, self.filter_settings)
 
         mock_list_instances.assert_called_once()
         mock_map_to_instance_list.assert_called_once_with({"data": "test_data"})
@@ -335,7 +335,7 @@ class TestAILabManager(unittest.TestCase):
         mock_delete_process.return_value = {"data": "test_data"}
         mock_map_to_empty_response.return_value = {"status": "success"}
 
-        result = self.manager.delete_process(self.project_id, process_id=self.process_id)
+        result = self.manager.delete_process(process_id=self.process_id)
 
         mock_delete_process.assert_called_once()
         mock_map_to_empty_response.assert_called_once()
@@ -347,7 +347,7 @@ class TestAILabManager(unittest.TestCase):
         mock_publish_process_revision.return_value = {"data": "test_data"}
         mock_map_to_process.return_value = AgenticProcess(id=self.process_id, name="test_process")
 
-        result = self.manager.publish_process_revision(self.project_id, process_id=self.process_id, revision=self.revision)
+        result = self.manager.publish_process_revision(process_id=self.process_id, revision=self.revision)
 
         mock_publish_process_revision.assert_called_once()
         mock_map_to_process.assert_called_once_with({"data": "test_data"})
@@ -360,7 +360,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_task.return_value = Task(id=self.task_id, name="test_task")
         task = Task(name="test_task")
 
-        result = self.manager.create_task(self.project_id, task, automatic_publish=False)
+        result = self.manager.create_task(task, automatic_publish=False)
 
         mock_create_task.assert_called_once()
         mock_map_to_task.assert_called_once_with({"data": "test_data"})
@@ -372,7 +372,7 @@ class TestAILabManager(unittest.TestCase):
         mock_get_task.return_value = {"data": "test_data"}
         mock_map_to_task.return_value = Task(id=self.task_id, name="test_task")
 
-        result = self.manager.get_task(self.project_id, task_id=self.task_id)
+        result = self.manager.get_task(task_id=self.task_id)
 
         mock_get_task.assert_called_once()
         mock_map_to_task.assert_called_once_with({"data": "test_data"})
@@ -384,7 +384,7 @@ class TestAILabManager(unittest.TestCase):
         mock_list_tasks.return_value = {"data": "test_data"}
         mock_map_to_task_list.return_value = TaskList(tasks=[])
 
-        result = self.manager.list_tasks(self.project_id, self.filter_settings)
+        result = self.manager.list_tasks(self.filter_settings)
 
         mock_list_tasks.assert_called_once()
         mock_map_to_task_list.assert_called_once_with({"data": "test_data"})
@@ -397,7 +397,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_task.return_value = Task(id=self.task_id, name="test_task")
         task = Task(id=self.task_id, name="test_task")
 
-        result = self.manager.update_task(self.project_id, task, automatic_publish=False, upsert=False)
+        result = self.manager.update_task(task, automatic_publish=False, upsert=False)
 
         mock_update_task.assert_called_once()
         mock_map_to_task.assert_called_once_with({"data": "test_data"})
@@ -409,7 +409,7 @@ class TestAILabManager(unittest.TestCase):
         mock_delete_task.return_value = {"data": "test_data"}
         mock_map_to_empty_response.return_value = {"status": "success"}
 
-        result = self.manager.delete_task(self.project_id, task_id=self.task_id)
+        result = self.manager.delete_task(task_id=self.task_id)
 
         mock_delete_task.assert_called_once()
         mock_map_to_empty_response.assert_called_once()
@@ -421,7 +421,7 @@ class TestAILabManager(unittest.TestCase):
         mock_publish_task_revision.return_value = {"data": "test_data"}
         mock_map_to_task.return_value = Task(id=self.task_id, name="test_task")
 
-        result = self.manager.publish_task_revision(self.project_id, task_id=self.task_id, revision=self.revision)
+        result = self.manager.publish_task_revision(task_id=self.task_id, revision=self.revision)
 
         mock_publish_task_revision.assert_called_once()
         mock_map_to_task.assert_called_once_with({"data": "test_data"})
@@ -433,7 +433,7 @@ class TestAILabManager(unittest.TestCase):
         mock_start_instance.return_value = {"data": "test_data"}
         mock_map_to_instance.return_value = ProcessInstance(id=self.instance_id, subject="test_subject", process=AgenticProcess(name="test_process"))
 
-        result = self.manager.start_instance(self.project_id, "test_process", subject="test_subject", variables=VariableList())
+        result = self.manager.start_instance("test_process", subject="test_subject", variables=VariableList())
 
         mock_start_instance.assert_called_once()
         mock_map_to_instance.assert_called_once_with({"data": "test_data"})
@@ -445,7 +445,7 @@ class TestAILabManager(unittest.TestCase):
         mock_abort_instance.return_value = {"data": "test_data"}
         mock_map_to_empty_response.return_value = {"status": "success"}
 
-        result = self.manager.abort_instance(self.project_id, self.instance_id)
+        result = self.manager.abort_instance(self.instance_id)
 
         mock_abort_instance.assert_called_once()
         mock_map_to_empty_response.assert_called_once()
@@ -457,7 +457,7 @@ class TestAILabManager(unittest.TestCase):
         mock_get_instance.return_value = {"data": "test_data"}
         mock_map_to_instance.return_value = ProcessInstance(id=self.instance_id, subject="test_subject", process=AgenticProcess(name="test_process"))
 
-        result = self.manager.get_instance(self.project_id, self.instance_id)
+        result = self.manager.get_instance(self.instance_id)
 
         mock_get_instance.assert_called_once()
         mock_map_to_instance.assert_called_once_with({"data": "test_data"})
@@ -467,7 +467,7 @@ class TestAILabManager(unittest.TestCase):
     def test_get_instance_history_success(self, mock_get_instance_history):
         mock_get_instance_history.return_value = {"history": "test_history"}
 
-        result = self.manager.get_instance_history(self.project_id, self.instance_id)
+        result = self.manager.get_instance_history(self.instance_id)
 
         mock_get_instance_history.assert_called_once()
         self.assertIsInstance(result, dict)
@@ -476,7 +476,7 @@ class TestAILabManager(unittest.TestCase):
     def test_get_thread_information_success(self, mock_get_thread_information):
         mock_get_thread_information.return_value = {"info": "test_info"}
 
-        result = self.manager.get_thread_information(self.project_id, "test_thread")
+        result = self.manager.get_thread_information("test_thread")
 
         mock_get_thread_information.assert_called_once()
         self.assertIsInstance(result, dict)
@@ -487,7 +487,7 @@ class TestAILabManager(unittest.TestCase):
         mock_send_user_signal.return_value = {"data": "test_data"}
         mock_map_to_empty_response.return_value = {"status": "success"}
 
-        result = self.manager.send_user_signal(self.project_id, self.instance_id, "test_signal")
+        result = self.manager.send_user_signal(self.instance_id, "test_signal")
 
         mock_send_user_signal.assert_called_once()
         mock_map_to_empty_response.assert_called_once()
@@ -500,7 +500,7 @@ class TestAILabManager(unittest.TestCase):
         mock_map_to_kb.return_value = KnowledgeBase(id=self.kb_id, name="test_kb")
         kb = KnowledgeBase(name="test_kb")
 
-        result = self.manager.create_knowledge_base(self.project_id, kb)
+        result = self.manager.create_knowledge_base(kb)
 
         mock_create_kb.assert_called_once()
         mock_map_to_kb.assert_called_once_with({"data": "test_data"})
@@ -512,7 +512,7 @@ class TestAILabManager(unittest.TestCase):
         mock_list_kbs.return_value = {"data": "test_data"}
         mock_map_to_kb_list.return_value = KnowledgeBaseList(knowledge_bases=[])
 
-        result = self.manager.list_knowledge_bases(self.project_id, start=0, count=10)
+        result = self.manager.list_knowledge_bases(start=0, count=10)
 
         mock_list_kbs.assert_called_once()
         mock_map_to_kb_list.assert_called_once_with({"data": "test_data"})
@@ -524,7 +524,7 @@ class TestAILabManager(unittest.TestCase):
         mock_get_kb.return_value = {"data": "test_data"}
         mock_map_to_kb.return_value = KnowledgeBase(id=self.kb_id, name="test_kb")
 
-        result = self.manager.get_knowledge_base(self.project_id, kb_id=self.kb_id)
+        result = self.manager.get_knowledge_base(kb_id=self.kb_id)
 
         mock_get_kb.assert_called_once()
         mock_map_to_kb.assert_called_once_with({"data": "test_data"})
@@ -536,7 +536,7 @@ class TestAILabManager(unittest.TestCase):
         mock_delete_kb.return_value = {"data": "test_data"}
         mock_map_to_empty_response.return_value = {"status": "success"}
 
-        result = self.manager.delete_knowledge_base(self.project_id, kb_id=self.kb_id)
+        result = self.manager.delete_knowledge_base(kb_id=self.kb_id)
 
         mock_delete_kb.assert_called_once()
         mock_map_to_empty_response.assert_called_once()
@@ -548,7 +548,7 @@ class TestAILabManager(unittest.TestCase):
         mock_list_jobs.return_value = {"data": "test_data"}
         mock_map_to_job_list.return_value = JobList(jobs=[])
 
-        result = self.manager.list_jobs(self.project_id, self.filter_settings)
+        result = self.manager.list_jobs(self.filter_settings)
 
         mock_list_jobs.assert_called_once()
         mock_map_to_job_list.assert_called_once_with({"data": "test_data"})

pygeai/tests/snippets/files/delete_file.py

@@ -1,12 +1,9 @@
-from pygeai.core.models import Organization, Project
 from pygeai.core.files.managers import FileManager
 from pygeai.core.files.models import File
 
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
 file = File(id="7fe2393d-8f86-4020-a51f-bc14ab957e1e")
 
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
+file_manager = FileManager()
 
 response = file_manager.delete_file(file_id=file.id)
 print(response)

pygeai/tests/snippets/files/get_file_content.py

@@ -1,12 +1,10 @@
-from pygeai.core.models import Organization, Project
 from pygeai.core.files.managers import FileManager
 from pygeai.core.files.models import File
 
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
+
 file = File(id="7fe2393d-8f86-4020-a51f-bc14ab957e1e")
 
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
+file_manager = FileManager()
 
 response = file_manager.get_file_content(file_id=file.id)
 print(response)

pygeai/tests/snippets/files/get_file_data.py

@@ -1,12 +1,9 @@
-from pygeai.core.models import Organization, Project
 from pygeai.core.files.managers import FileManager
 from pygeai.core.files.models import File
 
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
 file = File(id="9984b837-fe88-4014-ad14-91e1596c8ead")
 
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
+file_manager = FileManager()
 
 response = file_manager.get_file_data(file_id=file.id)
 print(response)

pygeai/tests/snippets/files/get_file_list.py

@@ -1,11 +1,6 @@
-from pygeai.core.models import Organization, Project
 from pygeai.core.files.managers import FileManager
 
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
-
-
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
+file_manager = FileManager()
 
 response = file_manager.get_file_list()
 print(response)

pygeai/tests/snippets/files/upload_file.py

@@ -1,17 +1,13 @@
-from pygeai.core.models import Organization, Project
 from pygeai.core.files.managers import FileManager
 from pygeai.core.files.models import UploadFile
 
-organization = Organization(id="4aa15b61-d3c7-4a5c-99b8-052d18a04ff2")
-project = Project(id="1956c032-3c66-4435-acb8-6a06e52f819f")
-
 file = UploadFile(
     path="test.txt",
     name="TestyFile",
     folder="TestyTestTemp"
 )
 
-file_manager = FileManager(organization_id=organization.id, project_id=project.id)
+file_manager = FileManager()
 
 response = file_manager.upload_file(file)
 print(response)

{pygeai-0.2.7b37.dist-info → pygeai-0.2.7b39.dist-info}/METADATA

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

{pygeai-0.2.7b37.dist-info → pygeai-0.2.7b39.dist-info}/RECORD

@@ -64,12 +64,12 @@ pygeai/cli/texts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,
 pygeai/cli/texts/help.py,sha256=IgifNjO22kbRB5bsuckFrzRvg7Q14y96IfxRnlJAUcw,15516
 pygeai/core/__init__.py,sha256=bbNktFp7t2dOBIvWto-uGVBW8acaKIe8EKcfuLV-HmA,189
 pygeai/core/handlers.py,sha256=QGFZESd-s3eBxe_Zd5G5n-HMgPIZLYba914WLefWleQ,596
-pygeai/core/models.py,sha256=Js1xjDa3F6kJ10QNEMyYL3bUOMPqfuiHi85Jy3ikFoI,23472
+pygeai/core/models.py,sha256=uIwrmlB6yuulScUcYIBSH3Sxm5YyzaicV7Dz2bYLi2I,24229
 pygeai/core/responses.py,sha256=wxlikhw1UAAB6Mib97xjq2eCFyZPWoosPwn9UhpKi7Y,2825
 pygeai/core/singleton.py,sha256=-U5kywQOBvbTtkWCczLZD_aoHBjDLRj07Q-UqQJpww0,242
 pygeai/core/base/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/core/base/clients.py,sha256=umtT6WoecCivXAxFajF-lMI5EbTNLKJBthuIHHixdNY,1608
-pygeai/core/base/mappers.py,sha256=yrurvsnjWnnhS1HyI1eS9-eQdbKirGyTo5r4SIhRQZo,17356
+pygeai/core/base/mappers.py,sha256=u5_UkRPPQ9vUQQANgmOHBbaaxtK77RNTTI2KoQafEDw,17793
 pygeai/core/base/models.py,sha256=_h62nnMhJXr1BLNoaldT4d9oqCTSistfF3D2LQ3bvlg,380
 pygeai/core/base/responses.py,sha256=k-mrzNO_AtEsGTUJnyRT76FJ7gfYxQ_SAhB8MBNqPZI,763
 pygeai/core/base/session.py,sha256=WVb4MmptwdgK7paHOSvfEle_HPXRRXO8CHgi0qbgtOg,2990
@@ -91,7 +91,7 @@ pygeai/core/feedback/models.py,sha256=VdeVVWTQ8qc4TEPbL2XbYI4-UP2m2eh1pkBptw_do1
 pygeai/core/files/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/core/files/clients.py,sha256=gjmm5H_BOEK_QRNb55s4CBGMA_IRV-xvkwf9YCjdEAY,6564
 pygeai/core/files/endpoints.py,sha256=hAUC28hYVcHyEyEfoLaLae76TpuVSLexPVjLJYjSWYQ,337
-pygeai/core/files/managers.py,sha256=rzkJr9pOJYV4_y_3SRhd7gAZ4n3ziH_4TLN9gwIs_EY,6269
+pygeai/core/files/managers.py,sha256=Hzq-XMEJnnNrh3pG5PViQqSt1WrbO-z5wHXsoGrSQxA,7158
 pygeai/core/files/mappers.py,sha256=8PXXsQJZEH45yLISn_xsOZmcRUEe_F6ELkUeR1lzc-M,1462
 pygeai/core/files/models.py,sha256=QOLV5kUrHOvhJXzkoFqNQX4qmVPLy6KKBk6u7oE5ttU,438
 pygeai/core/files/responses.py,sha256=fnNLK-stnXGnB9d38szWrZMu7JLTtXghAoRZnFTFJCE,356
@@ -143,7 +143,7 @@ pygeai/health/clients.py,sha256=U2eb1tkXt1Rf_KdV0ZFQS2k4wGnJTXHHd9-Er0eWQuw,1011
 pygeai/health/endpoints.py,sha256=UAzMcqSXZtMj4r8M8B7a_a5LT6X_jMFNsCTvcsjNTYA,71
 pygeai/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/lab/constants.py,sha256=ddgDnXP4GD0woi-FUJaJXzaWS3H6zmDN0B-v8utM95Q,170
-pygeai/lab/managers.py,sha256=TAMXALORbbBmxBE0JSo90ek8IFwaafkw047gxPxCcOo,72434
+pygeai/lab/managers.py,sha256=AF2CDT24NOc0xuCo52YrLNkNWElmAHfoqQBYns_axgk,72492
 pygeai/lab/models.py,sha256=7VMQBwL57hal0TIUG-dcsBJO6kRgdeCVq3GeRQGbnD0,70733
 pygeai/lab/runners.py,sha256=-uaCPHpFyiKtVOxlEjPjAc9h-onSdGAcYJ5IAZPqlb0,4147
 pygeai/lab/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -177,7 +177,7 @@ pygeai/organization/clients.py,sha256=O9RMeFm8gm5e6OoJl2y3JTS_MUZEp2Twc8AOpsUTj4
 pygeai/organization/endpoints.py,sha256=LOjwCMjx7wWBIimCzWhNt7cL1WP2IoU9Jtg1tKzRElo,794
 pygeai/organization/managers.py,sha256=KVPgmKIUGv71HkqOiW4LvUqWnuRpCu5OcxkewKbpa68,11726
 pygeai/organization/mappers.py,sha256=02-xTSE-Tmlhugh2vDIJDpixKI_4UdljcSgmwJIfJp0,2564
-pygeai/organization/responses.py,sha256=d9qJgVU1jcqMIAltXS9P8HY6ib6Qf6jH5qYEweymh0I,469
+pygeai/organization/responses.py,sha256=VetphuyPHeXlB4uNn97kwI4x4XT6-0Eede-eQ_eMXtw,1220
 pygeai/organization/limits/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/organization/limits/clients.py,sha256=g-2Fov4ArS_g0xDRaaJ2hFn-4JLW1tdbI1PE21dA-vc,19943
 pygeai/organization/limits/endpoints.py,sha256=mtca6U6l47jbbbmGc8KoXCDMnyNHtaPo8dxWVuHuUE0,2578
@@ -278,7 +278,7 @@ pygeai/tests/integration/lab/agents/test_get_agent.py,sha256=IM61b1jJG0M1fa1iffc
 pygeai/tests/integration/lab/agents/test_publish_agent_revision.py,sha256=djo-cFd4hfxmZBcto3zgxCfDqxw9a83dvvwx8jfYVyo,5916
 pygeai/tests/integration/lab/agents/test_update_agent.py,sha256=kARVT5BsKiRZXpCLhY-23Njv0q1KNLuGxxSFycCrEJo,12079
 pygeai/tests/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/tests/lab/test_managers.py,sha256=wzOkrhb1iG6Apoedtm8ppaM5g6L5DYYIL5DBQhkGDs8,30522
+pygeai/tests/lab/test_managers.py,sha256=AsOAvyCkRpbskEy214aV2TwrqilWH6bxOiTWDOb1twQ,29778
 pygeai/tests/lab/test_mappers.py,sha256=2cLSggf168XWFpeZeBR7uJ-8C32TKb7qA91i_9fr_b0,11409
 pygeai/tests/lab/test_models.py,sha256=MyOPqPb18w0yQj8tEf1zLhY_qBsRy_Cm5kYArB0I8es,54325
 pygeai/tests/lab/agents/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -343,11 +343,11 @@ pygeai/tests/snippets/embeddings/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRk
 pygeai/tests/snippets/embeddings/generate_embeddings.py,sha256=IyIlO6UDGMjT-E9uK91NdYsuWI4fCRGwqlaFr0iUzSc,785
 pygeai/tests/snippets/evaluation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/tests/snippets/files/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-pygeai/tests/snippets/files/delete_file.py,sha256=7GvC_aKPUExTt7-AApe9LnTRxo3CriWYn6vLZlk4Q24,488
-pygeai/tests/snippets/files/get_file_content.py,sha256=o1o81A3_3UrHFjBtNUyVYUJgPmRKgmylR5tfNWEhRpU,493
-pygeai/tests/snippets/files/get_file_data.py,sha256=Tnki6VpnRAMXAFQEEsRYXTqWhT2AfH-Gpyu3c5rXG00,490
-pygeai/tests/snippets/files/get_file_list.py,sha256=nlPsdEFTAYtIMx7N4CnrWhkkzt-vdzGw4ZmfxkU2ITY,379
-pygeai/tests/snippets/files/upload_file.py,sha256=SOgnkFg6FRsBfQSspHdx7oRgcth2b7-H05krs9Sb8H4,520
+pygeai/tests/snippets/files/delete_file.py,sha256=ORpPkrL9FllDonCHfoqf6LJUGyPHBoTdm1zpopzqbqU,249
+pygeai/tests/snippets/files/get_file_content.py,sha256=LM8DEmtE4MebEptcR_SUrbMfKuR9zfKfxsY9JspLVvc,255
+pygeai/tests/snippets/files/get_file_data.py,sha256=DGxbH155IcquE0t2uOifJRI6o5JQ8-DT5zQJ9MubIG4,251
+pygeai/tests/snippets/files/get_file_list.py,sha256=NItZBEqzHsS8HQEpAI2yoI5URhZB0F5eAys5Okbled8,138
+pygeai/tests/snippets/files/upload_file.py,sha256=In6XIvIFSbctrqOD8r_WMmGlXYGju4syNrBAZULjFow,280
 pygeai/tests/snippets/gam/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 pygeai/tests/snippets/gam/gam_access_token.py,sha256=168cZwF3IcAZ3neCdvh1jRxi1acpIDXHuyBZnuWwHQY,3122
 pygeai/tests/snippets/lab/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -475,9 +475,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.2.7b37.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
-pygeai-0.2.7b37.dist-info/METADATA,sha256=ZdC5hBxCKcoLr_ECX0EtEmaksswHEo587kiWToSgUzs,6883
-pygeai-0.2.7b37.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-pygeai-0.2.7b37.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
-pygeai-0.2.7b37.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
-pygeai-0.2.7b37.dist-info/RECORD,,
+pygeai-0.2.7b39.dist-info/licenses/LICENSE,sha256=eHfqo7-AWS8cMq0cg03lq7owsLeCmZA-xS5L0kuHnl8,1474
+pygeai-0.2.7b39.dist-info/METADATA,sha256=Rs88NXHwjCB6i5OIyNq7FChpuoFUxR3TFIdn3X8DNEY,6883
+pygeai-0.2.7b39.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+pygeai-0.2.7b39.dist-info/entry_points.txt,sha256=OAmwuXVCQBTCE3HeVegVd37hbhCcp9TPahvdrCuMYWw,178
+pygeai-0.2.7b39.dist-info/top_level.txt,sha256=bJFwp2tURmCfB94yXDF7ylvdSJXFDDJsyUOb-7PJgwc,7
+pygeai-0.2.7b39.dist-info/RECORD,,