clean up

ServiceNow · Jan 7, 2025 · 8825033 · 8825033
1 parent 3996d9d
commit 8825033
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tapeagents/llms.py b/tapeagents/llms.py
@@ -680,8 +680,7 @@ def batch_generate(self, prompts: list[Prompt]) -> list[LLMCall]:
             stream=self.stream,
             verify=False,
         )
-        time_send_request = time.time() - start_send_request
-        self._stats["time_send_request"].append(time_send_request)
+        self._stats["time_send_request"].append(time.time() - start_send_request)
         if not r.ok:
             logger.error(f"Failed to get completion: {r.text}")
             r.raise_for_status()