aixtools 0.1.2__tar.gz → 0.1.3__tar.gz

{aixtools-0.1.2 → aixtools-0.1.3}/PKG-INFO

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

{aixtools-0.1.2 → aixtools-0.1.3}/aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.1.2'
-__version_tuple__ = version_tuple = (0, 1, 2)
+__version__ = version = '0.1.3'
+__version_tuple__ = version_tuple = (0, 1, 3)
 
-__commit_id__ = commit_id = 'g8779bb4aa'
+__commit_id__ = commit_id = 'gc117551ae'

{aixtools-0.1.2 → aixtools-0.1.3}/aixtools/mcp/faulty_mcp.py

@@ -49,7 +49,7 @@ class Config:
 
     port: int = 9999
     prob_on_post_404: float = 0.5  # Probability of injecting a 404 error for POST requests
-    prob_on_post_empty_crash: float = 0.3  # Probability of terminating the process on empty request
+    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
@@ -102,9 +102,9 @@ class StarletteErrorMiddleware:  # pylint: disable=too-few-public-methods
         self.app = app
         logger.info("[StarletteErrorMiddleware] Middleware initialized!")
         logger.info("Current configuration:")
-        logger.info("POST 404 probability: %f", config.prob_on_post_404)
-        logger.info("POST empty crash probability: %f", config.prob_on_post_empty_crash)
-        logger.info("DELETE 404 probability: %f", config.prob_on_delete_404)
+        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)
 
@@ -125,22 +125,21 @@ class StarletteErrorMiddleware:  # pylint: disable=too-few-public-methods
             random_number = random()
             logger.info("[StarletteErrorMiddleware] random number: %f", random_number)
             if random_number < config.prob_on_delete_404:
-                logger.info("[StarletteErrorMiddleware] Simulating 404 error for DELETE request")
+                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_on_post_empty_crash:
-                    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:
@@ -280,9 +279,9 @@ if __name__ == "__main__":
         help=f"Probability of injecting a 404 error for POST requests (default: {config.prob_on_post_404})",
     )
     parser.add_argument(
-        "--prob-on-post-empty-crash",
+        "--prob-on-get-crash",
         type=float,
-        help=f"Probability of terminating on empty request (default: {config.prob_on_post_empty_crash})",
+        help=f"Probability of terminating on GET request (default: {config.prob_on_get_crash})",
     )
     parser.add_argument(
         "--prob-on-delete-404",
@@ -311,7 +310,7 @@ if __name__ == "__main__":
     # Update the global configuration with command line arguments
     config.port = args.port or config.port
     _update_config_value("prob_on_post_404")
-    _update_config_value("prob_on_post_empty_crash")
+    _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")

{aixtools-0.1.2 → aixtools-0.1.3}/aixtools.egg-info/PKG-INFO

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