aixtools 0.2.13__tar.gz → 0.2.15__tar.gz

{aixtools-0.2.13 → aixtools-0.2.15}/PKG-INFO

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

{aixtools-0.2.13 → aixtools-0.2.15}/aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.2.13'
-__version_tuple__ = version_tuple = (0, 2, 13)
+__version__ = version = '0.2.15'
+__version_tuple__ = version_tuple = (0, 2, 15)
 
-__commit_id__ = commit_id = 'gb809b4cc2'
+__commit_id__ = commit_id = 'g03781cdd0'

{aixtools-0.2.13 → aixtools-0.2.15}/aixtools/logging/log_objects.py

@@ -107,11 +107,11 @@ def safe_deepcopy(obj):
     return None  # fallback for non-serializable, non-introspectable objects
 
 
-def save_objects_to_logfile(objects: list, log_dir=LOGS_DIR):
+async def save_objects_to_logfile(objects: list, log_dir=LOGS_DIR):
     """Save the objects to a (pickle) log file"""
     with ObjectLogger(log_dir=log_dir) as object_logger:
         for obj in objects:
-            object_logger.log(obj)
+            await object_logger.log(obj)
 
 
 class BaseLogger:
@@ -144,7 +144,7 @@ class ObjectLogger(BaseLogger):
         log_dir=LOGS_DIR,
         verbose: bool = True,
         debug: bool | None = None,
-        parent_logger: Union["ObjectLogger", NoneType] = None,
+        parent_logger: Union["BaseLogger", NoneType] = None,
     ):
         self.verbose = verbose
         self.debug = (
@@ -195,8 +195,9 @@ class ObjectLogger(BaseLogger):
                 elif self.verbose:
                     print(obj, flush=True)
                 obj_to_save = safe_deepcopy(obj)
-                pickle.dump(obj_to_save, self.file)
-                self.file.flush()  # ensure it's written immediately
+                if self.file is not None:
+                    pickle.dump(obj_to_save, self.file)
+                    self.file.flush()  # ensure it's written immediately
             except Exception as e:  # pylint: disable=broad-exception-caught
                 logger.error("Failed to log object: %s", e)
                 logger.error(traceback.format_exc())

{aixtools-0.2.13 → aixtools-0.2.15}/aixtools/logging/model_patch_logging.py

@@ -27,16 +27,16 @@ def log_async_method(fn, agent_logger):
         # Log request
         uuid = str(uuid4())  # Create a unique ID for this request
         log_object = ModelRawRequest(method_name=fn.__name__, request_id=uuid, args=args, kwargs=kwargs)
-        agent_logger.log(log_object)
+        await agent_logger.log(log_object)
         # Invoke the original method
         try:
             result = await fn(*args, **kwargs)
             # Log results
             log_object = ModelRawRequestResult(method_name=fn.__name__, request_id=uuid, result=result)
-            agent_logger.log(log_object)
+            await agent_logger.log(log_object)
         except Exception as e:
             # Log exception
-            agent_logger.log(e)
+            await agent_logger.log(e)
             raise e
         return result
 
@@ -52,7 +52,7 @@ def log_async_stream(fn, agent_logger):
         # Log request
         uuid = str(uuid4())  # Create a unique ID for this request
         log_object = ModelRawRequest(method_name=fn.__name__, request_id=uuid, args=args, kwargs=kwargs)
-        agent_logger.log(log_object)
+        await agent_logger.log(log_object)
         # Invoke the original method
         async with fn(*args, **kwargs) as stream:
 
@@ -64,12 +64,12 @@ def log_async_stream(fn, agent_logger):
                         log_object = ModelRawRequestYieldItem(
                             method_name=fn.__name__, request_id=uuid, item=item, item_num=item_num
                         )
-                        agent_logger.log(log_object)
+                        await agent_logger.log(log_object)
                         item_num += 1
                         yield item
                 except Exception as e:
                     # Log exception
-                    agent_logger.log(e)
+                    await agent_logger.log(e)
                     raise e
 
             yield gen()

{aixtools-0.2.13 → aixtools-0.2.15}/aixtools/mcp/__init__.py

@@ -2,8 +2,10 @@
 Model Context Protocol (MCP) implementation for AI agent communication.
 """
 
+from aixtools.mcp.exceptions import AixToolError
 from aixtools.mcp.fast_mcp_log import FastMcpLog
 
 __all__ = [
     "FastMcpLog",
+    "AixToolError",
 ]

{aixtools-0.2.13 → aixtools-0.2.15}/aixtools/mcp/client.py

@@ -52,7 +52,7 @@ async def default_mcp_log_handler(message: LogMessage):
 def get_mcp_client(
     url: str | None = None,
     command: str | None = None,
-    args: list[str] = None,
+    args: list[str] | None = None,
     log_handler: callable = default_mcp_log_handler,  # type: ignore
 ) -> MCPServerStreamableHTTP | MCPServerStdio:
     """
@@ -67,7 +67,7 @@ def get_mcp_client(
     if args is None:
         args = []
     if url:
-        return MCPServerStreamableHTTP(url="http://127.0.0.1:8089/mcp/", log_handler=log_handler)
+        return MCPServerStreamableHTTP(url=url, log_handler=log_handler)
     if command:
         return MCPServerStdio(command=command, args=args, log_handler=log_handler)
     raise ValueError("Either url or command must be provided to create MCP client.")

aixtools-0.2.15/aixtools/mcp/exceptions.py

@@ -0,0 +1,5 @@
+"""Custom exceptions for MCP services."""
+
+
+class AixToolError(Exception):
+    """Retryable tool error. Raise this when a tool fails but the operation should be retried."""

{aixtools-0.2.13 → aixtools-0.2.15}/aixtools.egg-info/SOURCES.txt

@@ -64,6 +64,7 @@ aixtools/mcp/__init__.py
 aixtools/mcp/client.py
 aixtools/mcp/example_client.py
 aixtools/mcp/example_server.py
+aixtools/mcp/exceptions.py
 aixtools/mcp/fast_mcp_log.py
 aixtools/mcp/faulty_mcp.py
 aixtools/model_patch/model_patch.py