aixtools 0.1.1__py3-none-any.whl → 0.1.3__py3-none-any.whl

aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.1'
-__version_tuple__ = version_tuple = (0, 1, 1)
+__version__ = version = '0.1.3'
+__version_tuple__ = version_tuple = (0, 1, 3)
 
 __commit_id__ = commit_id = None

aixtools/agents/agent_batch.py

@@ -5,7 +5,7 @@ Batch processing functionality for running multiple agent queries in parallel.
 import asyncio
 from typing import Any
 
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 from aixtools.agents.agent import get_agent, run_agent
 
@@ -13,10 +13,7 @@ from aixtools.agents.agent import get_agent, run_agent
 class AgentQueryParams(BaseModel):
     """Parameters for configuring agent queries in batch processing."""
 
-    class Config:  # pylint: disable=too-few-public-methods
-        """Configuration for the model."""
-
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(arbitrary_types_allowed=True)
 
     id: str = ""  # Unique identifier for the query
     prompt: str | list[str]

aixtools/mcp/faulty_mcp.py

@@ -48,11 +48,11 @@ class Config:
     """Global configuration for the faulty MCP server."""
 
     port: int = 9999
-    prob_throw_in_list_handle: float = 0.5  # Probability of throwing an exception in list tools handling
-    prob_delete_404: float = 0.5  # Probability of injecting a 404 error for DELETE requests
-    prob_general_404: float = 0.5  # Probability of injecting a 404 error for other requests
-    prob_terminate_on_empty_request: float = 0.3  # Probability of terminating the process on empty request
-    prob_terminate_in_list_handle: float = 0.3  # Probability of terminating the process in list tools handling
+    prob_on_post_404: float = 0.5  # Probability of injecting a 404 error for POST requests
+    prob_on_get_crash: float = 0.3  # Probability of terminating the process on GET request
+    prob_on_delete_404: float = 0.5  # Probability of injecting a 404 error for DELETE requests
+    prob_in_list_tools_throw: float = 0.5  # Probability of throwing an exception in list tools handling
+    prob_in_list_tools_crash: float = 0.3  # Probability of terminating the process in list tools handling
 
 
 # Global configuration
@@ -69,11 +69,11 @@ class McpErrorMiddleware(McpMiddleware):
         if context.method == "tools/list":
             random_number = random()
             logger.info("[McpErrorMiddleware] random number: %f", random_number)
-            if random_number < config.prob_terminate_in_list_handle:
+            if random_number < config.prob_in_list_tools_crash:
                 logger.warning("[McpErrorMiddleware] Simulating server crash!")
                 os.kill(os.getpid(), 9)
 
-            if random_number < config.prob_throw_in_list_handle:
+            if random_number < config.prob_in_list_tools_throw:
                 exception_class = choice(
                     [
                         ValidationError,
@@ -102,11 +102,11 @@ class StarletteErrorMiddleware:  # pylint: disable=too-few-public-methods
         self.app = app
         logger.info("[StarletteErrorMiddleware] Middleware initialized!")
         logger.info("Current configuration:")
-        logger.info("Exception in list tools handling probability: %f", config.prob_throw_in_list_handle)
-        logger.info("DELETE 404 probability: %f", config.prob_delete_404)
-        logger.info("General 404 probability: %f", config.prob_general_404)
-        logger.info("Terminate on empty request probability: %f", config.prob_terminate_on_empty_request)
-        logger.info("Terminate in list handle probability: %f", config.prob_terminate_in_list_handle)
+        logger.info("HTTP 404 on POST request probability: %f", config.prob_on_post_404)
+        logger.info("Terminate on GET request probability: %f", config.prob_on_get_crash)
+        logger.info("HTTP 404 on DELETE request probability: %f", config.prob_on_delete_404)
+        logger.info("Exception in list tools handling probability: %f", config.prob_in_list_tools_throw)
+        logger.info("Terminate in list handle probability: %f", config.prob_in_list_tools_crash)
 
     async def __call__(self, scope: Scope, receive: Receive, send: Send):  # noqa: PLR0915 # pylint: disable=too-many-statements
         # Log all the condition variables for debugging
@@ -124,23 +124,22 @@ class StarletteErrorMiddleware:  # pylint: disable=too-few-public-methods
         if http_method == "DELETE":
             random_number = random()
             logger.info("[StarletteErrorMiddleware] random number: %f", random_number)
-            if random_number < config.prob_delete_404:
-                logger.info("[StarletteErrorMiddleware] Simulating 404 error for DELETE request")
+            if random_number < config.prob_on_delete_404:
+                logger.info("[StarletteErrorMiddleware] Simulating 404 error on DELETE request")
                 should_inject_404 = True
 
+        if http_method == "GET":
+            random_number = random()
+            logger.info("[StarletteErrorMiddleware] random number: %f", random_number)
+            if random_number < config.prob_on_get_crash:
+                logger.warning("[StarletteErrorMiddleware] Simulating server crash on GET request!")
+                os.kill(os.getpid(), 9)
+
         async def logging_receive():
             nonlocal should_inject_404
             message = await receive()
             logger.info("[StarletteErrorMiddleware] Received message: %s", str(message))
 
-            # Check for empty request and possibly terminate the process
-            if message["type"] == "http.request" and message["body"] == b"" and not message.get("more_body", False):
-                random_number = random()
-                logger.info("[StarletteErrorMiddleware] Empty request received, random number: %f", random_number)
-                if random_number < config.prob_terminate_on_empty_request:
-                    logger.warning("[StarletteErrorMiddleware] Simulating server crash on empty request!")
-                    os.kill(os.getpid(), 9)
-
             if message["type"] == "http.request":  # pylint: disable=too-many-nested-blocks
                 body = message.get("body", b"")
                 if body:
@@ -165,7 +164,7 @@ class StarletteErrorMiddleware:  # pylint: disable=too-few-public-methods
                                 # Check if we should inject 404
                                 random_number = random()
                                 logger.info("[StarletteErrorMiddleware] random number: %f", random_number)
-                                if random_number < config.prob_general_404:
+                                if random_number < config.prob_on_post_404:
                                     should_inject_404 = True
                                     logger.info("[StarletteErrorMiddleware] %s - will inject 404!", method_name)
                         except (UnicodeDecodeError, json.JSONDecodeError) as e:
@@ -275,29 +274,29 @@ if __name__ == "__main__":
         help=f"Port to run the server on (default: {config.port})",
     )
     parser.add_argument(
-        "--prob-throw-in-list-handle",
+        "--prob-on-post-404",
         type=float,
-        help=f"Probability of exception in list tools handling (default: {config.prob_throw_in_list_handle})",
+        help=f"Probability of injecting a 404 error for POST requests (default: {config.prob_on_post_404})",
     )
     parser.add_argument(
-        "--prob-delete-404",
+        "--prob-on-get-crash",
         type=float,
-        help=f"Probability of injecting a 404 error for DELETE requests (default: {config.prob_delete_404})",
+        help=f"Probability of terminating on GET request (default: {config.prob_on_get_crash})",
     )
     parser.add_argument(
-        "--prob-general-404",
+        "--prob-on-delete-404",
         type=float,
-        help=f"Probability of injecting a 404 error for other requests (default: {config.prob_general_404})",
+        help=f"Probability of injecting a 404 error for DELETE requests (default: {config.prob_on_delete_404})",
     )
     parser.add_argument(
-        "--prob-terminate-on-empty-request",
+        "--prob-in-list-tools-throw",
         type=float,
-        help=f"Probability of terminating on empty request (default: {config.prob_terminate_on_empty_request})",
+        help=f"Probability of exception in list tools handling (default: {config.prob_in_list_tools_throw})",
     )
     parser.add_argument(
-        "--prob-terminate-in-list-handle",
+        "--prob-in-list-tools-crash",
         type=float,
-        help=f"Probability of terminating in list tools handling (default: {config.prob_terminate_in_list_handle})",
+        help=f"Probability of terminating in list tools handling (default: {config.prob_in_list_tools_crash})",
     )
 
     args = parser.parse_args()
@@ -310,11 +309,11 @@ if __name__ == "__main__":
 
     # Update the global configuration with command line arguments
     config.port = args.port or config.port
-    _update_config_value("prob_throw_in_list_handle")
-    _update_config_value("prob_delete_404")
-    _update_config_value("prob_general_404")
-    _update_config_value("prob_terminate_on_empty_request")
-    _update_config_value("prob_terminate_in_list_handle")
+    _update_config_value("prob_on_post_404")
+    _update_config_value("prob_on_get_crash")
+    _update_config_value("prob_on_delete_404")
+    _update_config_value("prob_in_list_tools_throw")
+    _update_config_value("prob_in_list_tools_crash")
 
     # Run the server
     run_server_on_port()

aixtools/model_patch/model_patch.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from pydantic import BaseModel
+from pydantic import BaseModel, ConfigDict
 
 from aixtools.logging.logging_config import get_logger
 
@@ -19,8 +19,7 @@ class ModelRawRequestResult(BaseModel):
     request_id: str  # Unique request ID
     result: Any  # Method's result
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(arbitrary_types_allowed=True)
 
 
 class ModelRawRequestYieldItem(BaseModel):
@@ -29,8 +28,7 @@ class ModelRawRequestYieldItem(BaseModel):
     item_num: int  # Item number in the stream
     item: Any  # Yielded item
 
-    class Config:
-        arbitrary_types_allowed = True
+    model_config = ConfigDict(arbitrary_types_allowed=True)
 
 
 def get_request_fn(model):

{aixtools-0.1.1.dist-info → aixtools-0.1.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aixtools
-Version: 0.1.1
+Version: 0.1.3
 Summary: Tools for AI exploration and debugging
 Requires-Python: >=3.11.2
 Description-Content-Type: text/markdown

{aixtools-0.1.1.dist-info → aixtools-0.1.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 aixtools/__init__.py,sha256=9NGHm7LjsQmsvjTZvw6QFJexSvAU4bCoN_KBk9SCa00,260
-aixtools/_version.py,sha256=m8HxkqoKGw_wAJtc4ZokpJKNLXqp4zwnNhbnfDtro7w,704
+aixtools/_version.py,sha256=q5nF98G8SoVeJqaknL0xdyxtv0egsqb0fK06_84Izu8,704
 aixtools/app.py,sha256=JzQ0nrv_bjDQokllIlGHOV0HEb-V8N6k_nGQH-TEsVU,5227
 aixtools/chainlit.md,sha256=yC37Ly57vjKyiIvK4oUvf4DYxZCwH7iocTlx7bLeGLU,761
 aixtools/context.py,sha256=XuvSGjG8f-QHBJlI_yCPdjYo4rm_WrswmUE8GjLoRqI,491
@@ -22,7 +22,7 @@ aixtools/a2a/app.py,sha256=VQAKJHYJJr5-qJRLXvRgc2pZtH_SPXvB293nJaEDXTw,5071
 aixtools/a2a/utils.py,sha256=EHr3IyyBJn23ni-JcfAf6i3VpQmPs0g1TSnAZazvY_8,4039
 aixtools/agents/__init__.py,sha256=MAW196S2_G7uGqv-VNjvlOETRfuV44WlU1leO7SiR0A,282
 aixtools/agents/agent.py,sha256=E1zu70t53RqIbcLI_R09wUtsiYZR1bTnElCQ5PrsrKw,6127
-aixtools/agents/agent_batch.py,sha256=QHRXh7X1VdzLX9JcgIIvU06E46orhv3R7uJyj9TfIlo,2322
+aixtools/agents/agent_batch.py,sha256=0Zu9yNCRPAQZPjXQ-dIUAmP1uGTVbxVt7xvnMpoJMjU,2251
 aixtools/db/__init__.py,sha256=b8vRhme3egV-aUZbAntnOaDkSXB8UT0Xy5oqQhU_z0Q,399
 aixtools/db/database.py,sha256=caWe95GlxZYlxn2ubDmR-_cQUW0ulkpR3BHunKIaOsw,3369
 aixtools/db/vector_db.py,sha256=be4JGyXj3o8VEfy9L6SO1aAoDET_zazMJkYfjlYHTYQ,4133
@@ -46,8 +46,8 @@ aixtools/mcp/__init__.py,sha256=tLo2KZ1Ojo-rgEEJBGtZfUw-iOoopWoHDnYQTq3IzfE,163
 aixtools/mcp/example_client.py,sha256=QCFGP3NCNJMOKWjUOnFwjnbJhUSb879IA1ZYmwjRnmc,889
 aixtools/mcp/example_server.py,sha256=1SWCyrLWsAnOa81HC4QbPJo_lBVu0b3SZBWI-qDh1vQ,458
 aixtools/mcp/fast_mcp_log.py,sha256=XYOS406dVjn5YTHyGRsRvVNQ0SKlRObfrKj6EeLFjHg,1057
-aixtools/mcp/faulty_mcp.py,sha256=7wPjw0w0eSKpwuqkIYIRCD9iSGGb5IDQpCgcpMOLBmg,13044
-aixtools/model_patch/model_patch.py,sha256=My1BHp9pIdnEdJ_yXNnwCgrLRJ04g9CDG7-HUfC4IjY,1795
+aixtools/mcp/faulty_mcp.py,sha256=IzayVeElQZRw-sw_QwOwHHh2PuK2T9NKikEi_tt8SPU,12759
+aixtools/model_patch/model_patch.py,sha256=JT-oHubIn2LeoSwWbzEQ5vLH7crJmFUecHyQfaAFHa0,1813
 aixtools/server/__init__.py,sha256=rwPx020YpOzCnrxA80Lc4yLLcIp-Mpe9hNqVO9wDPv0,448
 aixtools/server/app_mounter.py,sha256=0tJ0tC140ezAjnYdlhpLJQjY-TO8NVw7D8LseYCCVY8,3336
 aixtools/server/path.py,sha256=SaIJxvmhJy3kzx5zJ6d4cKP6kKu2wFFciQkOLGTA4gg,3056
@@ -78,8 +78,8 @@ scripts/log_view.sh,sha256=bp8oXFRRbbHpyvHAN85wfDHTVK7vMJOYsBx_-bgECQc,511
 scripts/run_example_mcp_server.sh,sha256=f7m7h7O_wo6-nAsYlOXVWIASCOh3Qbuu0XWizlxMhl8,355
 scripts/run_faulty_mcp_server.sh,sha256=u_-8NbPDnJQt6IinNSjh8tc2ed-_MjGyipJXrUXaGR8,291
 scripts/run_server.sh,sha256=5iiB9bB5M2MuOgxVQqu7Oa_tBVtJpt0uB4z9uLu2J50,720
-aixtools-0.1.1.dist-info/METADATA,sha256=nuPN9YtQDx4QOYIGC4flAoG-rt4j0o4faWr9Zmsl__4,10109
-aixtools-0.1.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aixtools-0.1.1.dist-info/entry_points.txt,sha256=dHoutULEZx7xXSqJrZdViSVjfInJibfLibi2nRXL3SE,56
-aixtools-0.1.1.dist-info/top_level.txt,sha256=IPyw70hj9gVDyugaIr3LRlanLYXzostW4azlTgdlALo,34
-aixtools-0.1.1.dist-info/RECORD,,
+aixtools-0.1.3.dist-info/METADATA,sha256=EjMCZ1tNjNpNF95J0h08jI3Rb19Phe9-fDei7S1Syg0,10109
+aixtools-0.1.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aixtools-0.1.3.dist-info/entry_points.txt,sha256=dHoutULEZx7xXSqJrZdViSVjfInJibfLibi2nRXL3SE,56
+aixtools-0.1.3.dist-info/top_level.txt,sha256=IPyw70hj9gVDyugaIr3LRlanLYXzostW4azlTgdlALo,34
+aixtools-0.1.3.dist-info/RECORD,,