pygeai 0.6.0b14__py3-none-any.whl → 0.7.0b1__py3-none-any.whl

pygeai/tests/organization/test_clients.py

@@ -18,7 +18,7 @@ class TestOrganizationClient(unittest.TestCase):
     def setUp(self):
         self.client = OrganizationClient()
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_assistant_list_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"assistants": [{"name": "assistant1"}, {"name": "assistant2"}]}
@@ -32,7 +32,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertEqual(result['assistants'][0]['name'], "assistant1")
         self.assertEqual(result['assistants'][1]['name'], "assistant2")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_assistant_list_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -45,7 +45,7 @@ class TestOrganizationClient(unittest.TestCase):
         mock_get.assert_called_once_with(endpoint=GET_ASSISTANT_LIST_V1, params={"detail": "full"})
         self.assertEqual(str(context.exception), "Unable to get assistant list: Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_list_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"projects": [{"name": "project1"}, {"name": "project2"}]}
@@ -59,7 +59,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertEqual(result['projects'][0]['name'], "project1")
         self.assertEqual(result['projects'][1]['name'], "project2")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_list_with_name(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"projects": [{"name": "specific_project"}]}
@@ -75,7 +75,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertEqual(len(result['projects']), 1)
         self.assertEqual(result['projects'][0]['name'], "specific_project")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_list_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -88,7 +88,7 @@ class TestOrganizationClient(unittest.TestCase):
         mock_get.assert_called_once_with(endpoint=GET_PROJECT_LIST_V1, params={"detail": "full"})
         self.assertEqual(str(context.exception), "Unable to get project list: Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_data_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"project": {"id": "123", "name": "project1"}}
@@ -100,7 +100,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertIsNotNone(result)
         self.assertEqual(result['project']['name'], "project1")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_data_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -113,7 +113,7 @@ class TestOrganizationClient(unittest.TestCase):
         mock_get.assert_called_once_with(endpoint=GET_PROJECT_V1.format(id="123"))
         self.assertEqual(str(context.exception), "Unable to get project data for ID '123': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_project_success(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.json.return_value = {"project": {"id": "123", "name": "project1"}}
@@ -132,7 +132,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertIsNotNone(result)
         self.assertEqual(result['project']['name'], "project1")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_project_with_usage_limit(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.json.return_value = {"project": {"id": "123", "name": "project1"}}
@@ -155,7 +155,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertIsNotNone(result)
         self.assertEqual(result['project']['name'], "project1")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_project_json_decode_error(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.status_code = 200
@@ -175,7 +175,7 @@ class TestOrganizationClient(unittest.TestCase):
         )
         self.assertEqual(str(context.exception), "Unable to create project with name 'project1': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.put")
+    @patch("pygeai.core.services.rest.GEAIApiService.put")
     def test_update_project_success(self, mock_put):
         mock_response = mock_put.return_value
         mock_response.json.return_value = {"project": {"id": "123", "name": "updated_project"}}
@@ -190,7 +190,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertIsNotNone(result)
         self.assertEqual(result['project']['name'], "updated_project")
 
-    @patch("pygeai.core.services.rest.ApiService.put")
+    @patch("pygeai.core.services.rest.GEAIApiService.put")
     def test_update_project_json_decode_error(self, mock_put):
         mock_response = mock_put.return_value
         mock_response.status_code = 200
@@ -206,7 +206,7 @@ class TestOrganizationClient(unittest.TestCase):
         )
         self.assertEqual(str(context.exception), "Unable to update project with ID '123': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.delete")
+    @patch("pygeai.core.services.rest.GEAIApiService.delete")
     def test_delete_project_success(self, mock_delete):
         mock_response = mock_delete.return_value
         mock_response.json.return_value = {"status": "deleted"}
@@ -218,7 +218,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertIsNotNone(result)
         self.assertEqual(result['status'], "deleted")
 
-    @patch("pygeai.core.services.rest.ApiService.delete")
+    @patch("pygeai.core.services.rest.GEAIApiService.delete")
     def test_delete_project_json_decode_error(self, mock_delete):
         mock_response = mock_delete.return_value
         mock_response.status_code = 200
@@ -231,7 +231,7 @@ class TestOrganizationClient(unittest.TestCase):
         mock_delete.assert_called_once_with(endpoint=DELETE_PROJECT_V1.format(id="123"))
         self.assertEqual(str(context.exception), "Unable to delete project with ID '123': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_tokens_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"tokens": ["token1", "token2"]}
@@ -245,7 +245,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertEqual(result['tokens'][0], "token1")
         self.assertEqual(result['tokens'][1], "token2")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_tokens_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -258,7 +258,7 @@ class TestOrganizationClient(unittest.TestCase):
         mock_get.assert_called_once_with(endpoint=GET_PROJECT_TOKENS_V1.format(id="123"))
         self.assertEqual(str(context.exception), "Unable to get tokens for project with ID '123': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_export_request_data_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"requests": [{"id": "1", "status": "pending"}]}
@@ -274,7 +274,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertEqual(len(result['requests']), 1)
         self.assertEqual(result['requests'][0]['status'], "pending")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_export_request_data_with_params(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"requests": [{"id": "1", "status": "completed"}]}
@@ -290,7 +290,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertEqual(len(result['requests']), 1)
         self.assertEqual(result['requests'][0]['status'], "completed")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_export_request_data_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -306,7 +306,7 @@ class TestOrganizationClient(unittest.TestCase):
         )
         self.assertEqual(str(context.exception), "Unable to export request data: Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_memberships_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {
@@ -332,7 +332,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertIsNotNone(result)
         self.assertEqual(result['count'], 1)
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_memberships_with_params(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"count": 0, "pages": 0, "organizationsMemberships": []}
@@ -360,7 +360,7 @@ class TestOrganizationClient(unittest.TestCase):
         )
         self.assertIsNotNone(result)
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_memberships_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -372,39 +372,46 @@ class TestOrganizationClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Unable to get memberships: Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_memberships_success(self, mock_get):
         mock_response = mock_get.return_value
-        mock_response.json.return_value = {
-        
-            "count": 1,
-            "pages": 1,
-            "projectsMemberships": [
-                {
-                    "organizationId": "org-123",
-                    "organizationName": "Test Org",
-                    "projectId": "proj-456",
-                    "projectName": "Test Project",
-                    "roles": []
-                }
-            ]
-        }
+        mock_response.json.side_effect = [
+            {"organizationId": "org-123"},
+            {
+                "count": 1,
+                "pages": 1,
+                "projectsMemberships": [
+                    {
+                        "organizationId": "org-123",
+                        "organizationName": "Test Org",
+                        "projectId": "proj-456",
+                        "projectName": "Test Project",
+                        "roles": []
+                    }
+                ]
+            }
+        ]
         mock_response.status_code = 200
 
         result = self.client.get_project_memberships()
 
-        mock_get.assert_called_once_with(
+        self.assertEqual(mock_get.call_count, 2)
+        mock_get.assert_called_with(
             endpoint=GET_PROJECT_MEMBERSHIPS_V2,
-            params={"startPage": 1, "pageSize": 20, "orderDirection": "desc"}
+            params={"startPage": 1, "pageSize": 20, "orderDirection": "desc"},
+            headers={"organization-id": "org-123"}
         )
         self.assertIsNotNone(result)
         self.assertEqual(result['count'], 1)
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_memberships_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
-        mock_response.json.side_effect = JSONDecodeError("Invalid JSON", "", 0)
+        mock_response.json.side_effect = [
+            {"organizationId": "org-123"},
+            JSONDecodeError("Invalid JSON", "", 0)
+        ]
         mock_response.text = "Invalid JSON response"
 
         with self.assertRaises(InvalidAPIResponseException) as context:
@@ -412,7 +419,7 @@ class TestOrganizationClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Unable to get project memberships: Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_roles_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {
@@ -425,11 +432,15 @@ class TestOrganizationClient(unittest.TestCase):
 
         result = self.client.get_project_roles(project_id="proj-123")
 
-        mock_get.assert_called_once_with(endpoint=GET_PROJECT_ROLES_V2, params={"projectId": "proj-123"})
+        mock_get.assert_called_once_with(
+            endpoint=GET_PROJECT_ROLES_V2,
+            params={"startPage": 1, "pageSize": 20, "orderDirection": "desc"},
+            headers={"project-id": "proj-123"}
+        )
         self.assertIsNotNone(result)
         self.assertEqual(len(result['roles']), 1)
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_roles_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -441,7 +452,7 @@ class TestOrganizationClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Unable to get project roles for project 'proj-123': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_members_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {
@@ -454,11 +465,15 @@ class TestOrganizationClient(unittest.TestCase):
 
         result = self.client.get_project_members(project_id="proj-123")
 
-        mock_get.assert_called_once_with(endpoint=GET_PROJECT_MEMBERS_V2, params={"projectId": "proj-123"})
+        mock_get.assert_called_once_with(
+            endpoint=GET_PROJECT_MEMBERS_V2,
+            params={"startPage": 1, "pageSize": 20, "orderDirection": "desc"},
+            headers={"project-id": "proj-123"}
+        )
         self.assertIsNotNone(result)
         self.assertEqual(len(result['members']), 1)
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_project_members_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -470,7 +485,7 @@ class TestOrganizationClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Unable to get project members for project 'proj-123': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_organization_members_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {
@@ -483,11 +498,14 @@ class TestOrganizationClient(unittest.TestCase):
 
         result = self.client.get_organization_members(organization_id="org-123")
 
-        mock_get.assert_called_once_with(endpoint=GET_ORGANIZATION_MEMBERS_V2, params={"organizationId": "org-123"})
+        mock_get.assert_called_once_with(
+            endpoint=GET_ORGANIZATION_MEMBERS_V2,
+            params={"organizationId": "org-123", "startPage": 1, "pageSize": 20, "orderDirection": "desc"}
+        )
         self.assertIsNotNone(result)
         self.assertEqual(len(result['members']), 1)
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_organization_members_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -499,7 +517,7 @@ class TestOrganizationClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Unable to get organization members for organization 'org-123': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_add_project_member_success(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.json.return_value = {"status": "invitation sent"}
@@ -522,7 +540,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertIsNotNone(result)
         self.assertEqual(result['status'], "invitation sent")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_add_project_member_with_role_guids(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.json.return_value = {"status": "invitation sent"}
@@ -544,7 +562,7 @@ class TestOrganizationClient(unittest.TestCase):
         )
         self.assertIsNotNone(result)
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_add_project_member_json_decode_error(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.status_code = 201
@@ -560,7 +578,7 @@ class TestOrganizationClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Unable to add project member 'newuser@example.com': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_organization_success(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.json.return_value = {"organizationId": "org-123", "organizationName": "Test Org"}
@@ -578,7 +596,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertIsNotNone(result)
         self.assertEqual(result['organizationName'], "Test Org")
 
-    @patch("pygeai.core.services.rest.ApiService.post")
+    @patch("pygeai.core.services.rest.GEAIApiService.post")
     def test_create_organization_json_decode_error(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.status_code = 201
@@ -590,7 +608,7 @@ class TestOrganizationClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Unable to create organization with name 'Test Org': Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_organization_list_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {
@@ -613,7 +631,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertEqual(result['count'], 2)
         self.assertEqual(len(result['organizations']), 2)
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_organization_list_with_filters(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"count": 1, "pages": 1, "organizations": [{"organizationId": "org-1", "organizationName": "Test"}]}
@@ -641,7 +659,7 @@ class TestOrganizationClient(unittest.TestCase):
         )
         self.assertIsNotNone(result)
 
-    @patch("pygeai.core.services.rest.ApiService.get")
+    @patch("pygeai.core.services.rest.GEAIApiService.get")
     def test_get_organization_list_json_decode_error(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.status_code = 200
@@ -653,7 +671,7 @@ class TestOrganizationClient(unittest.TestCase):
 
         self.assertEqual(str(context.exception), "Unable to get organization list: Invalid JSON response")
 
-    @patch("pygeai.core.services.rest.ApiService.delete")
+    @patch("pygeai.core.services.rest.GEAIApiService.delete")
     def test_delete_organization_success(self, mock_delete):
         mock_response = mock_delete.return_value
         mock_response.json.return_value = {"status": "deleted"}
@@ -665,7 +683,7 @@ class TestOrganizationClient(unittest.TestCase):
         self.assertIsNotNone(result)
         self.assertEqual(result['status'], "deleted")
 
-    @patch("pygeai.core.services.rest.ApiService.delete")
+    @patch("pygeai.core.services.rest.GEAIApiService.delete")
     def test_delete_organization_json_decode_error(self, mock_delete):
         mock_response = mock_delete.return_value
         mock_response.status_code = 200

pygeai/tests/proxy/test_clients.py

@@ -1,5 +1,5 @@
 import unittest
-from unittest.mock import Mock, patch, MagicMock
+from unittest.mock import Mock, patch
 import uuid
 import requests
 from pygeai.proxy.clients import (

pygeai/tests/proxy/test_integration.py

@@ -1,6 +1,5 @@
 import unittest
-from unittest.mock import Mock, patch, MagicMock, AsyncMock
-import asyncio
+from unittest.mock import Mock, patch, AsyncMock
 import uuid
 import json
 import sys
@@ -10,8 +9,6 @@ from pygeai.proxy.config import ProxySettingsManager
 from pygeai.proxy.tool import ProxiedTool
 from pygeai.proxy.clients import ProxyClient, ToolProxyData, ToolProxyJob, ToolProxyJobResult
 from pygeai.proxy.managers import ServerManager
-from pygeai.proxy.servers import A2AServer, MCPServer
-from types import SimpleNamespace
 
 
 class TestProxyIntegration(unittest.IsolatedAsyncioTestCase):

pygeai/tests/proxy/test_managers.py

@@ -1,11 +1,10 @@
 import unittest
-from unittest.mock import Mock, patch, MagicMock, AsyncMock
+from unittest.mock import Mock, patch, AsyncMock
 import asyncio
 import json
 import uuid
 from pygeai.proxy.managers import ServerManager
 from pygeai.proxy.config import ProxySettingsManager
-from pygeai.proxy.clients import ProxyClient, ToolProxyData, ToolProxyJobResult
 from pygeai.proxy.tool import ProxiedTool
 
 

pygeai/tests/proxy/test_servers.py

@@ -1,6 +1,5 @@
 import unittest
-from unittest.mock import Mock, patch, MagicMock, AsyncMock
-import asyncio
+from unittest.mock import Mock, patch, AsyncMock
 import httpx
 from pygeai.proxy.servers import ToolServer, MCPServer, A2AServer
 from pygeai.proxy.config import ProxySettingsManager

pygeai/tests/snippets/assistants/rag/delete_rag_assistant.py

@@ -1,5 +1,4 @@
 from pygeai.assistant.managers import AssistantManager
-from pygeai.organization.managers import OrganizationManager
 
 manager = AssistantManager()
 

pygeai/tests/snippets/assistants/rag/get_documents.py

@@ -1,5 +1,4 @@
 from pygeai.assistant.managers import AssistantManager
-from pygeai.organization.managers import OrganizationManager
 
 manager = AssistantManager()
 

pygeai/tests/snippets/assistants/rag/get_rag_assistant_data.py

@@ -1,5 +1,4 @@
 from pygeai.assistant.managers import AssistantManager
-from pygeai.organization.managers import OrganizationManager
 
 manager = AssistantManager()
 

pygeai/tests/snippets/chat/get_request_status.py

@@ -1,5 +1,4 @@
 from pygeai.assistant.managers import AssistantManager
-from pygeai.organization.managers import OrganizationManager
 
 manager = AssistantManager()
 

pygeai/tests/snippets/dbg/file_debugging.py

@@ -0,0 +1,72 @@
+"""
+Example of debugging a standalone Python file using geai-dbg.
+
+This demonstrates how to use debug_file() to debug any Python script.
+"""
+from pygeai.dbg import debug_file
+import tempfile
+import os
+
+
+def create_sample_script():
+    """Create a sample script to debug."""
+    code = '''
+def process_data(items):
+    """Process a list of items."""
+    result = []
+    for item in items:
+        result.append(item * 2)
+    return result
+
+
+def main():
+    """Main entry point."""
+    data = [1, 2, 3, 4, 5]
+    processed = process_data(data)
+    print(f"Processed: {processed}")
+    return processed
+
+
+if __name__ == "__main__":
+    main()
+'''
+    
+    fd, filepath = tempfile.mkstemp(suffix='.py', text=True)
+    try:
+        os.write(fd, code.encode())
+    finally:
+        os.close(fd)
+    
+    return filepath
+
+
+def main():
+    """Demonstrate file debugging."""
+    print("Creating sample script...")
+    script_path = create_sample_script()
+    
+    try:
+        print(f"Script created at: {script_path}")
+        print("\nSetting up debugger...")
+        
+        dbg = debug_file(script_path)
+        dbg.add_breakpoint(function_name='process_data')
+        
+        print("Debugger configured with breakpoint on 'process_data'")
+        print("\nTo run interactively, uncomment the line below:")
+        print("# dbg.run()")
+        print("\nCommands you can use:")
+        print("  c     - continue")
+        print("  s     - step into")
+        print("  n     - next line")
+        print("  p     - print variable (e.g., 'p items')")
+        print("  l     - list source")
+        print("  h     - help")
+        
+    finally:
+        os.unlink(script_path)
+        print(f"\nCleaned up {script_path}")
+
+
+if __name__ == "__main__":
+    main()

pygeai/tests/snippets/dbg/module_debugging.py

@@ -0,0 +1,60 @@
+"""
+Example of debugging a specific module and function using geai-dbg.
+
+This demonstrates how to use debug_module() to debug imported modules.
+"""
+
+
+def example_simple_module():
+    """Debug a simple built-in module."""
+    print("Example: Debugging os.path.exists function")
+    print("\nThis would set up a debugger for the os.path.exists function:")
+    print("  dbg = debug_module('os.path', 'exists')")
+    print("  dbg.run()")
+    print("\nNote: This is for demonstration. Built-in functions may not be traceable.")
+
+
+def example_pygeai_module():
+    """Debug a PyGEAI module."""
+    print("\nExample: Debugging PyGEAI modules")
+    print("\nYou can debug any PyGEAI module function:")
+    print("  from pygeai.dbg import debug_module")
+    print("  ")
+    print("  # Debug the chat module")
+    print("  dbg = debug_module('pygeai.chat', 'send_message')")
+    print("  dbg.add_breakpoint(module='pygeai.core.llm', function_name='get_completion')")
+    print("  dbg.run()")
+
+
+def example_cli_debugging():
+    """Debug the geai CLI programmatically."""
+    print("\nExample: Debugging geai CLI programmatically")
+    print("\nInstead of using 'geai-dbg', you can set it up in code:")
+    print("  from pygeai.dbg import debug_module")
+    print("  ")
+    print("  # Set up debugging with custom breakpoints")
+    print("  dbg = debug_module('pygeai.cli.geai', 'main')")
+    print("  dbg.add_breakpoint(module='pygeai.cli.commands', function_name='execute')")
+    print("  dbg.run()")
+
+
+def main():
+    """Run all examples."""
+    print("=" * 70)
+    print("Module Debugging Examples")
+    print("=" * 70)
+    
+    example_simple_module()
+    example_pygeai_module()
+    example_cli_debugging()
+    
+    print("\n" + "=" * 70)
+    print("To use module debugging from the command line:")
+    print("=" * 70)
+    print("  geai-dbg -m pygeai.cli.geai:main")
+    print("  geai-dbg -m pygeai.chat:send_message -b process_response")
+    print("=" * 70)
+
+
+if __name__ == "__main__":
+    main()

pygeai/tests/snippets/embeddings/cohere_example.py

@@ -21,7 +21,7 @@ configuration = EmbeddingConfiguration(
 embeddings = manager.generate_embeddings(configuration)
 
 print(f"Model: {embeddings.model}")
-print(f"Input type: search_document")
+print("Input type: search_document")
 print(f"Documents embedded: {len(embeddings.data)}")
 print(f"Usage - Prompt tokens: {embeddings.usage.prompt_tokens}")
 print(f"Usage - Total cost: ${embeddings.usage.total_cost} {embeddings.usage.currency}")
@@ -37,5 +37,5 @@ query_configuration = EmbeddingConfiguration(
 )
 
 query_embeddings = manager.generate_embeddings(query_configuration)
-print(f"\nQuery embedded with input_type: search_query")
+print("\nQuery embedded with input_type: search_query")
 print(f"Usage - Total cost: ${query_embeddings.usage.total_cost} {query_embeddings.usage.currency}")

pygeai/tests/snippets/embeddings/openai_base64_example.py

@@ -21,7 +21,7 @@ configuration = EmbeddingConfiguration(
 embeddings = manager.generate_embeddings(configuration)
 
 print(f"Model: {embeddings.model}")
-print(f"Encoding format: base64")
+print("Encoding format: base64")
 print(f"Embedding (base64 encoded, first 100 chars): {embeddings.data[0].embedding[:100]}")
 print(f"Type of embedding: {type(embeddings.data[0].embedding)}")
 print(f"Usage - Total cost: ${embeddings.usage.total_cost} {embeddings.usage.currency}")