fix!: Suffix track methods with metrics

launchdarkly · Nov 22, 2024 · 319f64d · 319f64d
1 parent d2a2ea7
commit 319f64d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ldai/tracker.py b/ldai/tracker.py
@@ -216,7 +216,7 @@ def track_success(self) -> None:
             '$ld:ai:generation', self.context, self.__get_track_data(), 1
         )
 
-    def track_openai(self, func):
+    def track_openai_metrics(self, func):
         """
         Track OpenAI-specific operations.
 
@@ -228,7 +228,7 @@ def track_openai(self, func):
             self.track_tokens(OpenAITokenUsage(result.usage))
         return result
 
-    def track_bedrock_converse(self, res: dict) -> dict:
+    def track_bedrock_converse_metrics(self, res: dict) -> dict:
         """
         Track AWS Bedrock conversation operations.