Преглед на файлове

Fix some security vulnerabilities. (#2160)

### What problem does this PR solve?

Fix some security vulnerabilities

### Type of change

- [x] Performance Improvement

---------

Co-authored-by: Zhedong Cen <cenzhedong2@126.com>
tags/v0.11.0
黄腾 преди 1 година
родител
ревизия
12975cf128
No account linked to committer's email address
променени са 3 файла, в които са добавени 9 реда и са изтрити 9 реда
  1. 1
    1
      api/apps/llm_app.py
  2. 5
    5
      rag/llm/chat_model.py
  3. 3
    3
      rag/llm/embedding_model.py

+ 1
- 1
api/apps/llm_app.py Целия файл

@@ -22,7 +22,7 @@ from api.db.db_models import TenantLLM
from api.utils.api_utils import get_json_result
from rag.llm import EmbeddingModel, ChatModel, RerankModel, CvModel, TTSModel
import requests
import ast

@manager.route('/factories', methods=['GET'])
@login_required

+ 5
- 5
rag/llm/chat_model.py Целия файл

@@ -457,8 +457,8 @@ class VolcEngineChat(Base):
model_name is for display only
"""
base_url = base_url if base_url else 'https://ark.cn-beijing.volces.com/api/v3'
ark_api_key = eval(key).get('ark_api_key', '')
model_name = eval(key).get('ep_id', '')
ark_api_key = json.loads(key).get('ark_api_key', '')
model_name = json.loads(key).get('ep_id', '')
super().__init__(ark_api_key, model_name, base_url)


@@ -602,9 +602,9 @@ class BedrockChat(Base):

def __init__(self, key, model_name, **kwargs):
import boto3
self.bedrock_ak = eval(key).get('bedrock_ak', '')
self.bedrock_sk = eval(key).get('bedrock_sk', '')
self.bedrock_region = eval(key).get('bedrock_region', '')
self.bedrock_ak = json.loads(key).get('bedrock_ak', '')
self.bedrock_sk = json.loads(key).get('bedrock_sk', '')
self.bedrock_region = json.loads(key).get('bedrock_region', '')
self.model_name = model_name
self.client = boto3.client(service_name='bedrock-runtime', region_name=self.bedrock_region,
aws_access_key_id=self.bedrock_ak, aws_secret_access_key=self.bedrock_sk)

+ 3
- 3
rag/llm/embedding_model.py Целия файл

@@ -403,9 +403,9 @@ class BedrockEmbed(Base):
def __init__(self, key, model_name,
**kwargs):
import boto3
self.bedrock_ak = eval(key).get('bedrock_ak', '')
self.bedrock_sk = eval(key).get('bedrock_sk', '')
self.bedrock_region = eval(key).get('bedrock_region', '')
self.bedrock_ak = json.loads(key).get('bedrock_ak', '')
self.bedrock_sk = json.loads(key).get('bedrock_sk', '')
self.bedrock_region = json.loads(key).get('bedrock_region', '')
self.model_name = model_name
self.client = boto3.client(service_name='bedrock-runtime', region_name=self.bedrock_region,
aws_access_key_id=self.bedrock_ak, aws_secret_access_key=self.bedrock_sk)

Loading…
Отказ
Запис