Fix startup error on import (#292)

trustgraph-ai · Jan 29, 2025 · 701ec1e · 701ec1e
1 parent 1543a06
commit 701ec1e
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/trustgraph-bedrock/trustgraph/model/text_completion/bedrock/llm.py b/trustgraph-bedrock/trustgraph/model/text_completion/bedrock/llm.py
@@ -5,7 +5,6 @@
 """
 
 import boto3
-from botocore.errorfactory import ThrottlingException
 import json
 from prometheus_client import Histogram
 import os
@@ -259,7 +258,7 @@ def handle(self, msg):
 
             print("Done.", flush=True)
 
-        except ThrottlingException:
+        except self.bedrock.exceptions.ThrottlingException:
 
             print("Send rate limit response...", flush=True)