浏览代码

fix botocore (#1414)

### What problem does this PR solve?


### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.8.0
H 1年前
父节点
当前提交
61557a101a
没有帐户链接到提交者的电子邮件
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2
    1
      rag/llm/chat_model.py

+ 2
- 1
rag/llm/chat_model.py 查看文件



def __init__(self, key, model_name, **kwargs): def __init__(self, key, model_name, **kwargs):
import boto3 import boto3
from botocore.exceptions import ClientError
self.bedrock_ak = eval(key).get('bedrock_ak', '') self.bedrock_ak = eval(key).get('bedrock_ak', '')
self.bedrock_sk = eval(key).get('bedrock_sk', '') self.bedrock_sk = eval(key).get('bedrock_sk', '')
self.bedrock_region = eval(key).get('bedrock_region', '') self.bedrock_region = eval(key).get('bedrock_region', '')
aws_access_key_id=self.bedrock_ak, aws_secret_access_key=self.bedrock_sk) aws_access_key_id=self.bedrock_ak, aws_secret_access_key=self.bedrock_sk)


def chat(self, system, history, gen_conf): def chat(self, system, history, gen_conf):
from botocore.exceptions import ClientError
if system: if system:
history.insert(0, {"role": "system", "content": system}) history.insert(0, {"role": "system", "content": system})
for k in list(gen_conf.keys()): for k in list(gen_conf.keys()):
return f"ERROR: Can't invoke '{self.model_name}'. Reason: {e}", 0 return f"ERROR: Can't invoke '{self.model_name}'. Reason: {e}", 0


def chat_streamly(self, system, history, gen_conf): def chat_streamly(self, system, history, gen_conf):
from botocore.exceptions import ClientError
if system: if system:
history.insert(0, {"role": "system", "content": system}) history.insert(0, {"role": "system", "content": system})
for k in list(gen_conf.keys()): for k in list(gen_conf.keys()):

正在加载...
取消
保存