aixtools 0.1.2__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.2'
-__version_tuple__ = version_tuple = (0, 1, 2)
+__version__ = version = '0.1.3'
+__version_tuple__ = version_tuple = (0, 1, 3)
 
 __commit_id__ = commit_id = None

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.dist-info → aixtools-0.1.3.dist-info}/METADATA

@@ -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.dist-info → aixtools-0.1.3.dist-info}/RECORD

@@ -1,5 +1,5 @@
 aixtools/__init__.py,sha256=9NGHm7LjsQmsvjTZvw6QFJexSvAU4bCoN_KBk9SCa00,260
-aixtools/_version.py,sha256=Ok5oAXdWgR9aghaFXTafTeDW6sYO3uVe6d2Nket57R4,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
@@ -46,7 +46,7 @@ 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=_qqiZs6SYa_baqLzBwV9ImqSjiW54bn2mbQqpy6Gb_w,12969
+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
@@ -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.2.dist-info/METADATA,sha256=3bcgV9cnwdS8dqv1K78MVnwYmXEuV5dY0QDS5pSLzaM,10109
-aixtools-0.1.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aixtools-0.1.2.dist-info/entry_points.txt,sha256=dHoutULEZx7xXSqJrZdViSVjfInJibfLibi2nRXL3SE,56
-aixtools-0.1.2.dist-info/top_level.txt,sha256=IPyw70hj9gVDyugaIr3LRlanLYXzostW4azlTgdlALo,34
-aixtools-0.1.2.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,,