Skip to content

Commit 319f64d

Browse files
committed
fix!: Suffix track methods with metrics
1 parent d2a2ea7 commit 319f64d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

ldai/tracker.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -216,7 +216,7 @@ def track_success(self) -> None:
216216
'$ld:ai:generation', self.context, self.__get_track_data(), 1
217217
)
218218

219-
def track_openai(self, func):
219+
def track_openai_metrics(self, func):
220220
"""
221221
Track OpenAI-specific operations.
222222
@@ -228,7 +228,7 @@ def track_openai(self, func):
228228
self.track_tokens(OpenAITokenUsage(result.usage))
229229
return result
230230

231-
def track_bedrock_converse(self, res: dict) -> dict:
231+
def track_bedrock_converse_metrics(self, res: dict) -> dict:
232232
"""
233233
Track AWS Bedrock conversation operations.
234234

0 commit comments

Comments
 (0)