H
commited on
Commit
·
6ac2722
1
Parent(s):
194d038
fix botocore (#1414)
Browse files### What problem does this PR solve?
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
- rag/llm/chat_model.py +2 -1
rag/llm/chat_model.py
CHANGED
@@ -539,7 +539,6 @@ class BedrockChat(Base):
|
|
539 |
|
540 |
def __init__(self, key, model_name, **kwargs):
|
541 |
import boto3
|
542 |
-
from botocore.exceptions import ClientError
|
543 |
self.bedrock_ak = eval(key).get('bedrock_ak', '')
|
544 |
self.bedrock_sk = eval(key).get('bedrock_sk', '')
|
545 |
self.bedrock_region = eval(key).get('bedrock_region', '')
|
@@ -548,6 +547,7 @@ class BedrockChat(Base):
|
|
548 |
aws_access_key_id=self.bedrock_ak, aws_secret_access_key=self.bedrock_sk)
|
549 |
|
550 |
def chat(self, system, history, gen_conf):
|
|
|
551 |
if system:
|
552 |
history.insert(0, {"role": "system", "content": system})
|
553 |
for k in list(gen_conf.keys()):
|
@@ -576,6 +576,7 @@ class BedrockChat(Base):
|
|
576 |
return f"ERROR: Can't invoke '{self.model_name}'. Reason: {e}", 0
|
577 |
|
578 |
def chat_streamly(self, system, history, gen_conf):
|
|
|
579 |
if system:
|
580 |
history.insert(0, {"role": "system", "content": system})
|
581 |
for k in list(gen_conf.keys()):
|
|
|
539 |
|
540 |
def __init__(self, key, model_name, **kwargs):
|
541 |
import boto3
|
|
|
542 |
self.bedrock_ak = eval(key).get('bedrock_ak', '')
|
543 |
self.bedrock_sk = eval(key).get('bedrock_sk', '')
|
544 |
self.bedrock_region = eval(key).get('bedrock_region', '')
|
|
|
547 |
aws_access_key_id=self.bedrock_ak, aws_secret_access_key=self.bedrock_sk)
|
548 |
|
549 |
def chat(self, system, history, gen_conf):
|
550 |
+
from botocore.exceptions import ClientError
|
551 |
if system:
|
552 |
history.insert(0, {"role": "system", "content": system})
|
553 |
for k in list(gen_conf.keys()):
|
|
|
576 |
return f"ERROR: Can't invoke '{self.model_name}'. Reason: {e}", 0
|
577 |
|
578 |
def chat_streamly(self, system, history, gen_conf):
|
579 |
+
from botocore.exceptions import ClientError
|
580 |
if system:
|
581 |
history.insert(0, {"role": "system", "content": system})
|
582 |
for k in list(gen_conf.keys()):
|