aixtools 0.1.1__tar.gz → 0.1.2__tar.gz

{aixtools-0.1.1 → aixtools-0.1.2}/.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.2}/PKG-INFO

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

{aixtools-0.1.1 → aixtools-0.1.2}/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.2'
+__version_tuple__ = version_tuple = (0, 1, 2)
 
-__commit_id__ = commit_id = 'gcdcf6963f'
+__commit_id__ = commit_id = 'g8779bb4aa'

{aixtools-0.1.1 → aixtools-0.1.2}/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.2}/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_post_empty_crash: float = 0.3  # Probability of terminating the process on empty 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("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("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,7 +124,7 @@ 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:
+            if random_number < config.prob_on_delete_404:
                 logger.info("[StarletteErrorMiddleware] Simulating 404 error for DELETE request")
                 should_inject_404 = True
 
@@ -137,7 +137,7 @@ class StarletteErrorMiddleware:  # pylint: disable=too-few-public-methods
             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:
+                if random_number < config.prob_on_post_empty_crash:
                     logger.warning("[StarletteErrorMiddleware] Simulating server crash on empty request!")
                     os.kill(os.getpid(), 9)
 
@@ -165,7 +165,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 +275,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-post-empty-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 empty request (default: {config.prob_on_post_empty_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 +310,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_post_empty_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.2}/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.2}/aixtools.egg-info/PKG-INFO

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