lm-deluge 0.0.81__tar.gz → 0.0.82__tar.gz

{lm_deluge-0.0.81/src/lm_deluge.egg-info → lm_deluge-0.0.82}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lm_deluge
-Version: 0.0.81
+Version: 0.0.82
 Summary: Python utility for using LLM API models.
 Author-email: Benjamin Anderson <ben@trytaylor.ai>
 Requires-Python: >=3.10

{lm_deluge-0.0.81 → lm_deluge-0.0.82}/pyproject.toml

@@ -3,7 +3,7 @@ requires = ["setuptools", "wheel"]
 
 [project]
 name = "lm_deluge"
-version = "0.0.81"
+version = "0.0.82"
 authors = [{ name = "Benjamin Anderson", email = "ben@trytaylor.ai" }]
 description = "Python utility for using LLM API models."
 readme = "README.md"

{lm_deluge-0.0.81 → lm_deluge-0.0.82}/src/lm_deluge/client.py

@@ -103,6 +103,11 @@ class _LLMClient(BaseModel):
     _tracker: StatusTracker | None = PrivateAttr(default=None)
     _capacity_lock: asyncio.Lock = PrivateAttr(default_factory=asyncio.Lock)
 
+    # usage
+    def print_usage(self):
+        if self._tracker:
+            self._tracker.log_usage()
+
     # Progress management for queueing API
     def open(self, total: int | None = None, show_progress: bool = True):
         self._tracker = StatusTracker(

{lm_deluge-0.0.81 → lm_deluge-0.0.82}/src/lm_deluge/tracker.py

@@ -157,19 +157,7 @@ class StatusTracker:
             if response.usage.cache_write_tokens:
                 self.total_cache_write_tokens += response.usage.cache_write_tokens
 
-    def log_final_status(self):
-        # Close progress bar before printing final status
-        self.close_progress_bar()
-
-        if self.num_tasks_failed > 0:
-            print(
-                f"{self.num_tasks_failed} / {self.num_tasks_started} requests failed."
-            )
-        if self.num_rate_limit_errors > 0:
-            print(
-                f"{self.num_rate_limit_errors} rate limit errors received. Consider running at a lower rate."
-            )
-
+    def log_usage(self):
         # Display cumulative usage stats if available
         if (
             self.total_cost > 0
@@ -190,6 +178,21 @@ class StatusTracker:
 
             print("  ", " • ".join(usage_parts))
 
+    def log_final_status(self):
+        # Close progress bar before printing final status
+        self.close_progress_bar()
+
+        if self.num_tasks_failed > 0:
+            print(
+                f"{self.num_tasks_failed} / {self.num_tasks_started} requests failed."
+            )
+        if self.num_rate_limit_errors > 0:
+            print(
+                f"{self.num_rate_limit_errors} rate limit errors received. Consider running at a lower rate."
+            )
+
+        self.log_usage()
+
     @property
     def pbar(self) -> tqdm | None:
         """Backward compatibility property to access progress bar."""

{lm_deluge-0.0.81 → lm_deluge-0.0.82/src/lm_deluge.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lm_deluge
-Version: 0.0.81
+Version: 0.0.82
 Summary: Python utility for using LLM API models.
 Author-email: Benjamin Anderson <ben@trytaylor.ai>
 Requires-Python: >=3.10