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()):
|