toolchemy 0.2.194__py3-none-any.whl → 0.2.195__py3-none-any.whl

toolchemy/ai/trackers/common.py

@@ -93,6 +93,10 @@ class ITracker(ABC):
     def get_data(self) -> dict:
         pass
 
+    @abstractmethod
+    def get_traces(self, filter_name: str | None = None):
+        pass
+
 
 class TrackerBase(ITracker, ABC):
     def __init__(self, experiment_name: str, with_artifact_logging: bool = True, disabled: bool = False):
@@ -132,6 +136,9 @@ class TrackerBase(ITracker, ABC):
             "tags": self._tags.copy(),
         }
 
+    def get_traces(self, filter_name: str | None = None):
+        raise NotImplementedError()
+
     def _store_param(self, name: str, value: Any):
         if self._disabled:
             raise RuntimeError(f"Disabled trackers cannot store params!")

toolchemy/ai/trackers/mlflow_tracker.py

@@ -195,6 +195,12 @@ class MLFlowTracker(TrackerBase):
         self._store_tag(name, value)
         self._client.set_experiment_tag(self._experiment_id, name, value)
 
+    def get_traces(self, filter_name: str | None = None):
+        filter_string = None
+        if filter_name:
+            filter_string = f"trace.name = '{filter_name}'"
+        return self._client.search_traces(experiment_ids=[self.experiment_id], run_id=self.run_id, filter_string=filter_string)
+
     def _reset_run(self):
         self._active_run = None
         self._active_run_id = None

{toolchemy-0.2.194.dist-info → toolchemy-0.2.195.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: toolchemy
-Version: 0.2.194
+Version: 0.2.195
 Summary: a set of auxiliary programming tools
 License-File: LICENSE
 Author: Cyprian Nosek

{toolchemy-0.2.194.dist-info → toolchemy-0.2.195.dist-info}/RECORD

@@ -10,8 +10,8 @@ toolchemy/ai/clients/pricing.py,sha256=FuAuarQIaLl-e6KLzLbiWawWpqjY_kV4poO_qiTuH
 toolchemy/ai/clients/whisper_client.py,sha256=AQcY2FiGLu0JTYw0zzjjH-7WyZ6NbQWrtxFa6iYhR4M,5252
 toolchemy/ai/prompter.py,sha256=2Vf00d3pHUC1giG39REAn0YhuFUqxHXAQVNr5PCQQJ0,5367
 toolchemy/ai/trackers/__init__.py,sha256=FUv2NuGSmCpDrbE7rmeG0VmQr9rJRC-Y_D0MuQa0e-M,214
-toolchemy/ai/trackers/common.py,sha256=tEYQv1OqcyqMVEfBP0tRse3V9f-gypOrfNBj1U9IcsE,6525
-toolchemy/ai/trackers/mlflow_tracker.py,sha256=pzxIBRdODtA4qp5CezSZHY1b1oMSH_Jpo8xjbA1PDO4,8522
+toolchemy/ai/trackers/common.py,sha256=dtSAJVyBrUZ5vuIKF9N0fargirY61DaJ1pNf6dHRGa0,6712
+toolchemy/ai/trackers/mlflow_tracker.py,sha256=Yv_wnIDRhhco2GblafQ633zISutiPuOLPx8LhPKRbQ0,8822
 toolchemy/ai/trackers/neptune_tracker.py,sha256=GIbkLmHAzzO8Q7FuCmGf--y2NTEfcjDF9CVqqgmWzVg,4712
 toolchemy/db/lightdb.py,sha256=Bp7AimIuOfp2jIORFvBndk8kuKnEV12Ua1hoQXR5brY,9377
 toolchemy/utils/__init__.py,sha256=6Mp65G6Ce_zY86AxvjyzkMjU_e8cbeij7TTHYqLwO0U,574
@@ -29,8 +29,8 @@ toolchemy/utils/utils.py,sha256=JCRNRBJ4vRWzl-GtbetJs8GM72ImVAbDkPIXrvp3IE0,5167
 toolchemy/vision/__init__.py,sha256=p-sQaVd1FUQ1LMGdF51y28Z9IJvMVWaCdvrL0-jKDgc,141
 toolchemy/vision/caption_overlay.py,sha256=fLxHMD7IPSh5XJC5pvdPf1wZvHidlOvnXt_wgW7UJjk,2825
 toolchemy/vision/image.py,sha256=5q7MRK2SEyqWPB62duuzJqGmqBfQP1Tgphu01nP7O8s,2392
-toolchemy-0.2.194.dist-info/METADATA,sha256=nE_yD7yEedvdxxOhPfaWvmV62S1mC4vbx-nH7S1cTQs,980
-toolchemy-0.2.194.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
-toolchemy-0.2.194.dist-info/entry_points.txt,sha256=6l1k81pMAx7Ap1FaX8qW8SDwsONnNLOb28QQD9t6gto,66
-toolchemy-0.2.194.dist-info/licenses/LICENSE,sha256=JY5EPk3yvwZqQHk8JrQ-TFARdyfVTK9uUKV6CuKo_Bs,1062
-toolchemy-0.2.194.dist-info/RECORD,,
+toolchemy-0.2.195.dist-info/METADATA,sha256=0ehp0XRLMcnLUFrV9NaTLJS_om0qe5FPph8kepjDnh8,980
+toolchemy-0.2.195.dist-info/WHEEL,sha256=3ny-bZhpXrU6vSQ1UPG34FoxZBp3lVcvK0LkgUz6VLk,88
+toolchemy-0.2.195.dist-info/entry_points.txt,sha256=6l1k81pMAx7Ap1FaX8qW8SDwsONnNLOb28QQD9t6gto,66
+toolchemy-0.2.195.dist-info/licenses/LICENSE,sha256=JY5EPk3yvwZqQHk8JrQ-TFARdyfVTK9uUKV6CuKo_Bs,1062
+toolchemy-0.2.195.dist-info/RECORD,,