(Hotfix): Fix logs and filestore init for llm registry (#10470)

This commit is contained in:
Rohit Malhotra
2025-08-18 16:57:08 -04:00
committed by GitHub
parent 39333dd5de
commit 875036d920
2 changed files with 11 additions and 4 deletions

View File

@@ -39,6 +39,10 @@ class ConversationStats:
pickled = pickle.dumps(self.service_to_metrics)
serialized_metrics = base64.b64encode(pickled).decode('utf-8')
self.file_store.write(self.metrics_path, serialized_metrics)
logger.info(
'Saved converation stats',
extra={'conversation_id': self.conversation_id},
)
def maybe_restore_metrics(self):
if not self.file_store or not self.conversation_id:
@@ -56,9 +60,6 @@ class ConversationStats:
total_metrics = Metrics()
for metrics in self.service_to_metrics.values():
total_metrics.merge(metrics)
logger.info(f'metrics by all services: {self.service_to_metrics}')
logger.info(f'combined metrics\n\n{total_metrics}')
return total_metrics
def get_metrics_for_service(self, service_id: str) -> Metrics: