Skip to content

Commit 7be4d27

Browse files
Dan O'Brienkeelerm84
andauthored
Update ldai/tracker.py
Co-authored-by: Matthew M. Keeler <[email protected]>
1 parent a1f9490 commit 7be4d27

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

ldai/tracker.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -43,9 +43,10 @@ def track_openai(self, func, *args, **kwargs):
4343
return result
4444

4545
def track_bedrock_converse(self, res: dict) -> dict:
46-
if res.get('$metadata', {}).get('httpStatusCode') == 200:
46+
status_code = res.get('$metadata', {}).get('httpStatusCode', 0)
47+
if status_code == 200:
4748
self.track_success()
48-
elif res.get('$metadata', {}).get('httpStatusCode') and res['$metadata']['httpStatusCode'] >= 400:
49+
elif status_code >= 400:
4950
# Potentially add error tracking in the future.
5051
pass
5152
if res.get('metrics', {}).get('latencyMs'):

0 commit comments

Comments
 (0)