aixtools 0.2.14__py3-none-any.whl → 0.2.15__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.2.14'
-__version_tuple__ = version_tuple = (0, 2, 14)
+__version__ = version = '0.2.15'
+__version_tuple__ = version_tuple = (0, 2, 15)
 
 __commit_id__ = commit_id = None

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

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

{aixtools-0.2.14.dist-info → aixtools-0.2.15.dist-info}/RECORD

@@ -1,5 +1,5 @@
 aixtools/__init__.py,sha256=9NGHm7LjsQmsvjTZvw6QFJexSvAU4bCoN_KBk9SCa00,260
-aixtools/_version.py,sha256=lOOOEggb8ag564aDqoOp-6k18Oo-qSEBUsWUUBMG5Qs,706
+aixtools/_version.py,sha256=fu32ckCvDvmdBESK9QY8685INg5f0hTCLuUhaNOVaJQ,706
 aixtools/app.py,sha256=JzQ0nrv_bjDQokllIlGHOV0HEb-V8N6k_nGQH-TEsVU,5227
 aixtools/chainlit.md,sha256=yC37Ly57vjKyiIvK4oUvf4DYxZCwH7iocTlx7bLeGLU,761
 aixtools/context.py,sha256=I_MD40ZnvRm5WPKAKqBUAdXIf8YaurkYUUHSVVy-QvU,598
@@ -52,14 +52,14 @@ aixtools/log_view/node_summary.py,sha256=EJjnBqdBWI-_bI-4nfTxwaost3mtiufb5cK7T54
 aixtools/logfilters/__init__.py,sha256=pTD8ujCqjPWBCeB7yv7lmCtnA2KXOnkIv0HExDagkXs,129
 aixtools/logfilters/context_filter.py,sha256=zR3Bnv3fCqXLeb7bCFTmlnWhC6dFIvUb-u712tOnUPk,2259
 aixtools/logging/__init__.py,sha256=b5oYyGQDUHHxhRtzqKUaQPv8hQeWw54rzDXSV8lDY1w,613
-aixtools/logging/log_objects.py,sha256=ooxa9ue3mFq1tOqAGeRBiM_4TsteVdDzR7qKM-j6Ln8,6904
+aixtools/logging/log_objects.py,sha256=K8p2kHTqr9nntHQ0-jyNrDBoBiqH14gFnD7atAiN_do,6964
 aixtools/logging/logging_config.py,sha256=LvxV3C75-I0096PpcCIbgM-Cp998LzWXeMM14HYbU20,4985
 aixtools/logging/mcp_log_models.py,sha256=7-H2GJXiiyLhpImuyLLftAGG4skxJal8Swax0ob04MY,3463
 aixtools/logging/mcp_logger.py,sha256=d2I5l4t0d6rQH17w23FpE1IUD8Ax-mSaKfByCH86q4I,6257
-aixtools/logging/model_patch_logging.py,sha256=MY2EvR7ZSctC4hJxNMe8iACeVayUJ2V5In2GAnKdgOo,2880
+aixtools/logging/model_patch_logging.py,sha256=CW5-kKI-zNEgZhNV4vx3EQu6fbrEtX7VjA6fE5loRLQ,2916
 aixtools/logging/open_telemetry.py,sha256=fJjF1ou_8GyfNfbyWDQPGK6JAUrUaPwURYPHhXEtDBE,1121
 aixtools/mcp/__init__.py,sha256=Qp4uD1RtypCYgzWrt_ThBVxa5-CFBgcwMWkHbHFUQ54,232
-aixtools/mcp/client.py,sha256=aCDLc_AfrjV8JtZLWnKCTnScvFdNoMJZiGQhBUO8t8w,18289
+aixtools/mcp/client.py,sha256=0UR3KiPCg7WAEf8u4-ewp3H_hiaMwD9KKmx2XRzdA_8,18271
 aixtools/mcp/example_client.py,sha256=QCFGP3NCNJMOKWjUOnFwjnbJhUSb879IA1ZYmwjRnmc,889
 aixtools/mcp/example_server.py,sha256=1SWCyrLWsAnOa81HC4QbPJo_lBVu0b3SZBWI-qDh1vQ,458
 aixtools/mcp/exceptions.py,sha256=9m3hy9fRINbXTSgkE71XW_luklgH6xBRp7VnSRfpyzQ,173
@@ -89,8 +89,8 @@ aixtools/utils/chainlit/cl_agent_show.py,sha256=vaRuowp4BRvhxEr5hw0zHEJ7iaSF_5bo
 aixtools/utils/chainlit/cl_utils.py,sha256=fxaxdkcZg6uHdM8uztxdPowg3a2f7VR7B26VPY4t-3c,5738
 aixtools/vault/__init__.py,sha256=fsr_NuX3GZ9WZ7dGfe0gp_5-z3URxAfwVRXw7Xyc0dU,141
 aixtools/vault/vault.py,sha256=9dZLWdZQk9qN_Q9Djkofw9LUKnJqnrX5H0fGusVLBhA,6037
-aixtools-0.2.14.dist-info/METADATA,sha256=BVCtc_kv3nt7TFaIeOMK8APeIMu1ya4T5wnRdrAqams,27230
-aixtools-0.2.14.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aixtools-0.2.14.dist-info/entry_points.txt,sha256=q8412TG4T0S8K0SKeWp2vkVPIDYQs0jNoHqcQ7qxOiA,155
-aixtools-0.2.14.dist-info/top_level.txt,sha256=wBn-rw9bCtxrR4AYEYgjilNCUVmKY0LWby9Zan2PRJM,9
-aixtools-0.2.14.dist-info/RECORD,,
+aixtools-0.2.15.dist-info/METADATA,sha256=pMsq-5JO6i7tEE3rfFNwk2zD70H7qLLc8fzo1zObSTI,27230
+aixtools-0.2.15.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aixtools-0.2.15.dist-info/entry_points.txt,sha256=q8412TG4T0S8K0SKeWp2vkVPIDYQs0jNoHqcQ7qxOiA,155
+aixtools-0.2.15.dist-info/top_level.txt,sha256=wBn-rw9bCtxrR4AYEYgjilNCUVmKY0LWby9Zan2PRJM,9
+aixtools-0.2.15.dist-info/RECORD,,