pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2025-12-08 20:42:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor add LLM (#2508)

Browse Source 
### What problem does this PR solve?

#2487

### Type of change

- [x] Refactoring
This commit is contained in:
Kevin Hu
2024-09-20 10:20:35 +08:00
committed by GitHub
parent 4f962d6bff
commit 5968f148bc
2 changed files with 26 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
rag/llm/chat_model.py
View File

@ -458,7 +458,7 @@ class VolcEngineChat(Base):
"""
base_url = base_url if base_url else 'https://ark.cn-beijing.volces.com/api/v3'
ark_api_key = json.loads(key).get('ark_api_key', '')
model_name = json.loads(key).get('ep_id', '')
model_name = json.loads(key).get('ep_id', '') + json.loads(key).get('endpoint_id', '')
super().__init__(ark_api_key, model_name, base_url)