Fix not initialized response latencies (All-Hands-AI#5679)

kernelai · Dec 22, 2024 · 4a8bf3d · 4a8bf3d
1 parent 2cfbd26
commit 4a8bf3d
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/openhands/llm/metrics.py b/openhands/llm/metrics.py
@@ -46,8 +46,14 @@ def costs(self) -> list[Cost]:
 
     @property
     def response_latencies(self) -> list[ResponseLatency]:
+        if not hasattr(self, '_response_latencies'):
+            self._response_latencies = []
         return self._response_latencies
 
+    @response_latencies.setter
+    def response_latencies(self, value: list[ResponseLatency]) -> None:
+        self._response_latencies = value
+
     def add_cost(self, value: float) -> None:
         if value < 0:
             raise ValueError('Added cost cannot be negative.')