aixtools 0.1.1__tar.gz → 0.1.3__tar.gz

{aixtools-0.1.1 → aixtools-0.1.3}/.github/workflows/release.yml

@@ -3,7 +3,6 @@ name: Release
 on:
   release:
     types: [published]
-  workflow_dispatch: # Temporarily allow manual triggering
 
 jobs:
   build-and-publish:

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

@@ -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 → 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.1'
-__version_tuple__ = version_tuple = (0, 1, 1)
+__version__ = version = '0.1.3'
+__version_tuple__ = version_tuple = (0, 1, 3)
 
-__commit_id__ = commit_id = 'gcdcf6963f'
+__commit_id__ = commit_id = 'gc117551ae'

{aixtools-0.1.1 → aixtools-0.1.3}/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-0.1.1 → aixtools-0.1.3}/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-0.1.1 → aixtools-0.1.3}/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 → aixtools-0.1.3}/aixtools.egg-info/PKG-INFO

@@ -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