pygeai 0.6.0b7__py3-none-any.whl → 0.6.0b10__py3-none-any.whl

pygeai/tests/migration/test_tools.py

@@ -1,7 +1,7 @@
 import unittest
 from unittest.mock import Mock
 
-from pygeai.migration.tools import MigrationTool
+from pygeai.migration.tools import MigrationTool, MigrationPlan, MigrationOrchestrator
 from pygeai.migration.strategies import MigrationStrategy
 
 
@@ -11,25 +11,149 @@ class TestMigrationTool(unittest.TestCase):
     """
 
     def setUp(self):
-        """Set up test fixtures"""
         self.mock_strategy = Mock(spec=MigrationStrategy)
 
     def test_migration_tool_initialization(self):
-        """Test MigrationTool initialization with a strategy"""
         tool = MigrationTool(self.mock_strategy)
-        self.assertEqual(tool.strategy, self.mock_strategy)
+        self.assertEqual(tool._strategy, self.mock_strategy)
 
     def test_migration_tool_set_strategy(self):
-        """Test setting a new strategy on MigrationTool"""
         tool = MigrationTool(self.mock_strategy)
         new_strategy = Mock(spec=MigrationStrategy)
         tool.set_strategy(new_strategy)
-
-        self.assertEqual(tool.strategy, new_strategy)
+        self.assertEqual(tool._strategy, new_strategy)
 
     def test_migration_tool_run_migration(self):
-        """Test running migration with the strategy"""
         tool = MigrationTool(self.mock_strategy)
         tool.run_migration()
+        self.mock_strategy.migrate.assert_called_once()
+
+
+class TestMigrationPlan(unittest.TestCase):
+    """
+    python -m unittest pygeai.tests.migration.test_tools.TestMigrationPlan
+    """
+
+    def test_migration_plan_initialization(self):
+        strategies = [Mock(spec=MigrationStrategy), Mock(spec=MigrationStrategy)]
+        plan = MigrationPlan(strategies=strategies)
+        self.assertEqual(len(plan.strategies), 2)
+        self.assertEqual(plan.dependencies, {})
+        self.assertFalse(plan.dry_run)
+        self.assertTrue(plan.stop_on_error)
+
+    def test_migration_plan_with_dependencies(self):
+        strategies = [Mock(spec=MigrationStrategy), Mock(spec=MigrationStrategy)]
+        dependencies = {1: [0]}
+        plan = MigrationPlan(strategies=strategies, dependencies=dependencies)
+        self.assertEqual(plan.dependencies, {1: [0]})
+
+
+class TestMigrationOrchestrator(unittest.TestCase):
+    """
+    python -m unittest pygeai.tests.migration.test_tools.TestMigrationOrchestrator
+    """
+
+    def setUp(self):
+        self.strategy1 = Mock(spec=MigrationStrategy)
+        self.strategy2 = Mock(spec=MigrationStrategy)
+        self.strategy3 = Mock(spec=MigrationStrategy)
+
+    def test_orchestrator_simple_execution(self):
+        plan = MigrationPlan(strategies=[self.strategy1, self.strategy2])
+        orchestrator = MigrationOrchestrator(plan)
+        
+        result = orchestrator.execute()
+        
+        self.strategy1.migrate.assert_called_once()
+        self.strategy2.migrate.assert_called_once()
+        self.assertEqual(result['total'], 2)
+        self.assertEqual(result['completed'], 2)
+        self.assertEqual(result['failed'], 0)
+        self.assertEqual(result['success_rate'], 1.0)
+
+    def test_orchestrator_with_dependencies(self):
+        plan = MigrationPlan(
+            strategies=[self.strategy1, self.strategy2, self.strategy3],
+            dependencies={1: [0], 2: [1]}
+        )
+        orchestrator = MigrationOrchestrator(plan)
+        
+        result = orchestrator.execute()
+        
+        self.assertEqual(result['completed'], 3)
+        self.assertEqual(result['completed_indices'], [0, 1, 2])
+
+    def test_orchestrator_stop_on_error(self):
+        self.strategy1.migrate.side_effect = Exception("Migration failed")
+        plan = MigrationPlan(
+            strategies=[self.strategy1, self.strategy2],
+            stop_on_error=True
+        )
+        orchestrator = MigrationOrchestrator(plan)
+        
+        with self.assertRaises(ValueError) as context:
+            orchestrator.execute()
+        
+        self.assertIn("Migration failed at strategy", str(context.exception))
+        self.strategy1.migrate.assert_called_once()
+        self.strategy2.migrate.assert_not_called()
+
+    def test_orchestrator_continue_on_error(self):
+        self.strategy1.migrate.side_effect = Exception("Migration failed")
+        plan = MigrationPlan(
+            strategies=[self.strategy1, self.strategy2],
+            stop_on_error=False
+        )
+        orchestrator = MigrationOrchestrator(plan)
+        
+        result = orchestrator.execute()
+        
+        self.strategy1.migrate.assert_called_once()
+        self.strategy2.migrate.assert_called_once()
+        self.assertEqual(result['completed'], 1)
+        self.assertEqual(result['failed'], 1)
+        self.assertEqual(result['success_rate'], 0.5)
+
+    def test_orchestrator_dry_run(self):
+        plan = MigrationPlan(
+            strategies=[self.strategy1, self.strategy2],
+            dry_run=True
+        )
+        orchestrator = MigrationOrchestrator(plan)
+        
+        result = orchestrator.execute()
+        
+        self.strategy1.migrate.assert_not_called()
+        self.strategy2.migrate.assert_not_called()
+        self.assertTrue(result['valid'])
+        self.assertEqual(result['execution_order'], [0, 1])
+
+    def test_orchestrator_circular_dependency(self):
+        plan = MigrationPlan(
+            strategies=[self.strategy1, self.strategy2],
+            dependencies={0: [1], 1: [0]}
+        )
+        orchestrator = MigrationOrchestrator(plan)
+        
+        with self.assertRaises(ValueError) as context:
+            orchestrator.execute()
+        
+        self.assertIn("Circular dependency", str(context.exception))
+
+    def test_orchestrator_dry_run_circular_dependency(self):
+        plan = MigrationPlan(
+            strategies=[self.strategy1, self.strategy2],
+            dependencies={0: [1], 1: [0]},
+            dry_run=True
+        )
+        orchestrator = MigrationOrchestrator(plan)
+        
+        result = orchestrator.execute()
+        
+        self.assertFalse(result['valid'])
+        self.assertIn("Circular dependency", result['error'])
+
 
-        self.mock_strategy.migrate.assert_called_once()
+if __name__ == '__main__':
+    unittest.main()

pygeai/tests/organization/limits/test_clients.py

@@ -47,6 +47,7 @@ class TestUsageLimitClient(unittest.TestCase):
     def test_set_organization_usage_limit_success(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.json.return_value = self.valid_response
+        mock_response.status_code = 200
 
         result = self.client.set_organization_usage_limit(self.organization, self.valid_usage_limit)
 
@@ -79,6 +80,7 @@ class TestUsageLimitClient(unittest.TestCase):
         invalid_usage_limit["renewalStatus"] = "InvalidStatus"
         mock_response = mock_post.return_value
         mock_response.json.return_value = {"error": "Invalid renewal status"}
+        mock_response.status_code = 200
 
         result = self.client.set_organization_usage_limit(self.organization, invalid_usage_limit)
 
@@ -94,6 +96,7 @@ class TestUsageLimitClient(unittest.TestCase):
     def test_get_organization_latest_usage_limit_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = self.valid_response
+        mock_response.status_code = 200
 
         result = self.client.get_organization_latest_usage_limit(self.organization)
 
@@ -123,6 +126,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response = {"limits": [self.valid_response]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.get_all_usage_limits_from_organization(self.organization)
 
@@ -153,6 +157,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response = {"status": "deleted"}
         mock_response = mock_delete.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.delete_usage_limit_from_organization(self.organization, self.limit_id)
 
@@ -183,6 +188,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response["hardLimit"] = hard_limit
         mock_response = mock_put.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.set_organization_hard_limit(self.organization, self.limit_id, hard_limit)
 
@@ -217,6 +223,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response["softLimit"] = soft_limit
         mock_response = mock_put.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.set_organization_soft_limit(self.organization, self.limit_id, soft_limit)
 
@@ -251,6 +258,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response["renewalStatus"] = renewal_status
         mock_response = mock_put.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.set_organization_renewal_status(self.organization, self.limit_id, renewal_status)
 
@@ -282,6 +290,7 @@ class TestUsageLimitClient(unittest.TestCase):
     def test_set_project_usage_limit_success(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.json.return_value = self.valid_response
+        mock_response.status_code = 200
 
         result = self.client.set_project_usage_limit(self.organization, self.project, self.valid_usage_limit)
 
@@ -314,6 +323,7 @@ class TestUsageLimitClient(unittest.TestCase):
         invalid_usage_limit["subscriptionType"] = "InvalidType"
         mock_response = mock_post.return_value
         mock_response.json.return_value = {"error": "Invalid subscription type"}
+        mock_response.status_code = 200
 
         result = self.client.set_project_usage_limit(self.organization, self.project, invalid_usage_limit)
 
@@ -330,6 +340,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response = {"limits": [self.valid_response]}
         mock_response = mock_get.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.get_all_usage_limits_from_project(self.organization, self.project)
 
@@ -359,6 +370,7 @@ class TestUsageLimitClient(unittest.TestCase):
     def test_get_latest_usage_limit_from_project_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = self.valid_response
+        mock_response.status_code = 200
 
         result = self.client.get_latest_usage_limit_from_project(self.organization, self.project)
 
@@ -387,6 +399,7 @@ class TestUsageLimitClient(unittest.TestCase):
     def test_get_active_usage_limit_from_project_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = self.valid_response
+        mock_response.status_code = 200
 
         result = self.client.get_active_usage_limit_from_project(self.organization, self.project)
 
@@ -416,6 +429,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response = {"status": "deleted"}
         mock_response = mock_delete.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.delete_usage_limit_from_project(self.organization, self.project, self.limit_id)
 
@@ -446,6 +460,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response["hardLimit"] = hard_limit
         mock_response = mock_put.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.set_hard_limit_for_active_usage_limit_from_project(
             self.organization, self.project, self.limit_id, hard_limit
@@ -484,6 +499,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response["softLimit"] = soft_limit
         mock_response = mock_put.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.set_soft_limit_for_active_usage_limit_from_project(
             self.organization, self.project, self.limit_id, soft_limit
@@ -522,6 +538,7 @@ class TestUsageLimitClient(unittest.TestCase):
         response["renewalStatus"] = renewal_status
         mock_response = mock_put.return_value
         mock_response.json.return_value = response
+        mock_response.status_code = 200
 
         result = self.client.set_project_renewal_status(self.organization, self.project, self.limit_id, renewal_status)
 

pygeai/tests/organization/test_clients.py

@@ -21,6 +21,7 @@ class TestOrganizationClient(unittest.TestCase):
     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"}]}
+        mock_response.status_code = 200
 
         result = self.client.get_assistant_list(detail="summary")
 
@@ -47,6 +48,7 @@ class TestOrganizationClient(unittest.TestCase):
     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"}]}
+        mock_response.status_code = 200
 
         result = self.client.get_project_list(detail="summary")
 
@@ -60,6 +62,7 @@ class TestOrganizationClient(unittest.TestCase):
     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"}]}
+        mock_response.status_code = 200
 
         result = self.client.get_project_list(detail="full", name="specific_project")
 
@@ -88,6 +91,7 @@ class TestOrganizationClient(unittest.TestCase):
     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"}}
+        mock_response.status_code = 200
 
         result = self.client.get_project_data(project_id="123")
 
@@ -112,6 +116,7 @@ class TestOrganizationClient(unittest.TestCase):
     def test_create_project_success(self, mock_post):
         mock_response = mock_post.return_value
         mock_response.json.return_value = {"project": {"id": "123", "name": "project1"}}
+        mock_response.status_code = 200
 
         result = self.client.create_project(name="project1", email="admin@example.com", description="A test project")
 
@@ -130,6 +135,7 @@ class TestOrganizationClient(unittest.TestCase):
     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"}}
+        mock_response.status_code = 200
 
         usage_limit = {"type": "Requests", "threshold": 1000}
         result = self.client.create_project(
@@ -172,6 +178,7 @@ class TestOrganizationClient(unittest.TestCase):
     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"}}
+        mock_response.status_code = 200
 
         result = self.client.update_project(project_id="123", name="updated_project", description="Updated description")
 
@@ -202,6 +209,7 @@ class TestOrganizationClient(unittest.TestCase):
     def test_delete_project_success(self, mock_delete):
         mock_response = mock_delete.return_value
         mock_response.json.return_value = {"status": "deleted"}
+        mock_response.status_code = 200
 
         result = self.client.delete_project(project_id="123")
 
@@ -226,6 +234,7 @@ class TestOrganizationClient(unittest.TestCase):
     def test_get_project_tokens_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {"tokens": ["token1", "token2"]}
+        mock_response.status_code = 200
 
         result = self.client.get_project_tokens(project_id="123")
 
@@ -252,6 +261,7 @@ class TestOrganizationClient(unittest.TestCase):
     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"}]}
+        mock_response.status_code = 200
 
         result = self.client.export_request_data()
 
@@ -267,6 +277,7 @@ class TestOrganizationClient(unittest.TestCase):
     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"}]}
+        mock_response.status_code = 200
 
         result = self.client.export_request_data(assistant_name="assistant1", status="completed", skip=10, count=5)
 
@@ -298,6 +309,7 @@ class TestOrganizationClient(unittest.TestCase):
     def test_get_memberships_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {
+        
             "count": 1,
             "pages": 1,
             "organizationsMemberships": [
@@ -308,6 +320,7 @@ class TestOrganizationClient(unittest.TestCase):
                 }
             ]
         }
+        mock_response.status_code = 200
 
         result = self.client.get_memberships()
 
@@ -322,6 +335,7 @@ class TestOrganizationClient(unittest.TestCase):
     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": []}
+        mock_response.status_code = 200
 
         result = self.client.get_memberships(
             email="test@example.com",
@@ -361,6 +375,7 @@ class TestOrganizationClient(unittest.TestCase):
     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": [
@@ -373,6 +388,7 @@ class TestOrganizationClient(unittest.TestCase):
                 }
             ]
         }
+        mock_response.status_code = 200
 
         result = self.client.get_project_memberships()
 
@@ -399,10 +415,12 @@ class TestOrganizationClient(unittest.TestCase):
     def test_get_project_roles_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {
+        
             "roles": [
                 {"id": "role-1", "name": "Admin", "externalId": "admin-ext", "type": "backend", "origin": "system"}
             ]
         }
+        mock_response.status_code = 200
 
         result = self.client.get_project_roles(project_id="proj-123")
 
@@ -426,10 +444,12 @@ class TestOrganizationClient(unittest.TestCase):
     def test_get_project_members_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {
+        
             "members": [
                 {"email": "user@example.com", "roles": []}
             ]
         }
+        mock_response.status_code = 200
 
         result = self.client.get_project_members(project_id="proj-123")
 
@@ -453,10 +473,12 @@ class TestOrganizationClient(unittest.TestCase):
     def test_get_organization_members_success(self, mock_get):
         mock_response = mock_get.return_value
         mock_response.json.return_value = {
+        
             "members": [
                 {"email": "user@example.com", "roles": []}
             ]
         }
+        mock_response.status_code = 200
 
         result = self.client.get_organization_members(organization_id="org-123")
 

pygeai/tests/proxy/test_clients.py

@@ -247,6 +247,7 @@ class TestProxyClient(unittest.TestCase):
         """Test successful request."""
         mock_response = Mock()
         mock_response.json.return_value = {"status": "success"}
+        mock_response.status_code = 200
         mock_response.raise_for_status.return_value = None
         mock_session.return_value.request.return_value = mock_response
         
@@ -311,6 +312,7 @@ class TestProxyClient(unittest.TestCase):
         mock_response = Mock()
         mock_response.json.side_effect = ValueError("Invalid JSON")
         mock_response.text = "plain text response"
+        mock_response.status_code = 200
         mock_response.raise_for_status.return_value = None
         mock_session.return_value.request.return_value = mock_response
         

pygeai/tests/proxy/test_integration.py

@@ -230,6 +230,7 @@ class TestProxyIntegration(unittest.IsolatedAsyncioTestCase):
         # Mock response
         mock_response = Mock()
         mock_response.json.return_value = {"status": "success", "data": "test_data"}
+        mock_response.status_code = 200
         mock_response.raise_for_status.return_value = None
         mock_session.return_value.request.return_value = mock_response
         

pygeai/tests/snippets/chat/chat_completion_with_reasoning_effort.py

@@ -0,0 +1,18 @@
+from pygeai.chat.clients import ChatClient
+
+client = ChatClient()
+
+messages = [
+    {
+        "role": "user",
+        "content": "Explain quantum computing in simple terms"
+    }
+]
+
+response = client.chat_completion(
+    model="openai/gpt-4o",
+    messages=messages,
+    reasoning_effort="high"
+)
+
+print(response)

pygeai/tests/snippets/chat/get_response.py

@@ -0,0 +1,15 @@
+from pygeai.chat.clients import ChatClient
+
+client = ChatClient()
+
+model = "openai/o1-pro"
+input_text = "What is the weather like in Paris today?"
+
+response = client.get_response(
+    model=model,
+    input=input_text,
+    temperature=0.7,
+    max_output_tokens=1000
+)
+
+print(response)

pygeai/tests/snippets/chat/get_response_streaming.py

@@ -0,0 +1,20 @@
+from pygeai.chat.clients import ChatClient
+
+client = ChatClient()
+
+model = "openai/o1-pro"
+input_text = "Tell me a short story about a robot"
+
+response = client.get_response(
+    model=model,
+    input=input_text,
+    stream=True,
+    temperature=0.7,
+    max_output_tokens=500
+)
+
+print("Streaming response:")
+for chunk in response:
+    print(chunk, end='', flush=True)
+
+print("\n\nStreaming complete!")

pygeai/tests/snippets/chat/get_response_with_files.py

@@ -0,0 +1,16 @@
+from pygeai.chat.clients import ChatClient
+
+client = ChatClient()
+
+model = "openai/o1-pro"
+input_text = "Please analyze this image and describe what you see"
+files = ["files/image.svg", "files/document.pdf"]
+
+response = client.get_response(
+    model=model,
+    input=input_text,
+    files=files,
+    temperature=0.7
+)
+
+print(response)

pygeai/tests/snippets/chat/get_response_with_tools.py

@@ -0,0 +1,36 @@
+from pygeai.chat.clients import ChatClient
+
+client = ChatClient()
+
+model = "openai/o1-pro"
+input_text = "What is the weather like in Paris today?"
+
+tools = [
+    {
+        "type": "function",
+        "name": "get_weather",
+        "description": "Get current temperature for a given location.",
+        "parameters": {
+            "type": "object",
+            "properties": {
+                "location": {
+                    "type": "string",
+                    "description": "City and country e.g. Paris, France"
+                }
+            },
+            "required": ["location"],
+            "additionalProperties": False
+        }
+    }
+]
+
+response = client.get_response(
+    model=model,
+    input=input_text,
+    tools=tools,
+    tool_choice="auto",
+    temperature=1.0,
+    reasoning={"effort": "medium"}
+)
+
+print(response)

pygeai/tests/snippets/dbg/basic_debugging.py

@@ -0,0 +1,32 @@
+"""
+Basic debugging example with geai-dbg.
+
+This demonstrates setting breakpoints and inspecting variables.
+"""
+from pygeai.dbg.debugger import Debugger
+
+
+def example_function(x, y):
+    """A simple function to debug."""
+    result = x + y
+    print(f"Result: {result}")
+    return result
+
+
+def main():
+    """Main entry point."""
+    a = 10
+    b = 20
+    c = example_function(a, b)
+    print(f"Final value: {c}")
+
+
+if __name__ == "__main__":
+    dbg = Debugger(target=main, module_filter="__main__")
+    dbg.add_breakpoint(module="__main__", function_name="example_function")
+    
+    print("Starting debugger...")
+    print("Type 'h' for help")
+    print("Try commands like: 'p x', 'p y', 'locals', 's', 'n', 'c'")
+    
+    dbg.run()

pygeai/tests/snippets/dbg/breakpoint_management.py

@@ -0,0 +1,48 @@
+"""
+Demonstrates breakpoint management (add, list, remove, enable/disable).
+"""
+from pygeai.dbg.debugger import Debugger
+
+
+def func_a():
+    """Function A."""
+    print("In function A")
+    return "A"
+
+
+def func_b():
+    """Function B."""
+    print("In function B")
+    return "B"
+
+
+def func_c():
+    """Function C."""
+    print("In function C")
+    return "C"
+
+
+def main():
+    """Main entry point."""
+    results = []
+    results.append(func_a())
+    results.append(func_b())
+    results.append(func_c())
+    print(f"Results: {results}")
+
+
+if __name__ == "__main__":
+    dbg = Debugger(target=main, module_filter="__main__")
+    
+    print("Starting debugger...")
+    print("No breakpoints set initially. At first function, try:")
+    print("  'b func_b' - add breakpoint on func_b")
+    print("  'b __main__:func_c' - add breakpoint on specific module:function")
+    print("  'b' - list all breakpoints")
+    print("  'dis func_b' - disable breakpoint on func_b")
+    print("  'en func_b' - enable breakpoint on func_b")
+    print("  'cl func_b' - remove breakpoint on func_b")
+    print("  'c' - continue to next breakpoint")
+    
+    dbg.add_breakpoint(module="__main__", function_name="func_a")
+    dbg.run()