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

refine readme (#170)

Browse Source
This commit is contained in:
KevinHuSh
2024-03-29 14:38:15 +08:00
committed by GitHub
parent 05298d5626
commit e876f58b4c
3 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
api/db/services/llm_service.py
View File

@ -84,7 +84,7 @@ class TenantLLMService(CommonService):
if model_config["llm_factory"] not in EmbeddingModel:
return
return EmbeddingModel[model_config["llm_factory"]](
model_config["api_key"], model_config["llm_name"], model_config["api_base"])
model_config["api_key"], model_config["llm_name"], base_url=model_config["api_base"])
if llm_type == LLMType.IMAGE2TEXT.value:
if model_config["llm_factory"] not in CvModel:
@ -98,7 +98,7 @@ class TenantLLMService(CommonService):
if model_config["llm_factory"] not in ChatModel:
return
return ChatModel[model_config["llm_factory"]](
model_config["api_key"], model_config["llm_name"], model_config["api_base"])
model_config["api_key"], model_config["llm_name"], base_url=model_config["api_base"])
@classmethod
@DB.connection_context()