AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-13 21:25:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix data init error (#820)

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

#810 

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
KevinHuSh 2024-05-17 14:33:19 +08:00 committed by GitHub
parent 4ca176bd41
commit 9e5d501e83
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/db/init_data.py
View File

@ -391,9 +391,9 @@ def init_llm_factory():
TenantLLMService.filter_update([TenantLLMService.model.llm_factory == "QAnything"], {"llm_factory": "Youdao"}) TenantLLMService.filter_update([TenantLLMService.model.llm_factory == "QAnything"], {"llm_factory": "Youdao"})
## insert openai two embedding models to the current openai user. ## insert openai two embedding models to the current openai user.
print("Start to insert 2 OpenAI embedding models...") print("Start to insert 2 OpenAI embedding models...")
tenant_ids = set([row.tenant_id for row in TenantLLMService.get_openai_models()]) tenant_ids = set([row["tenant_id"] for row in TenantLLMService.get_openai_models()])
for tid in tenant_ids: for tid in tenant_ids:
for row in TenantLLMService.get_openai_models(llm_factory="OpenAI", tenant_id=tid): for row in TenantLLMService.query(llm_factory="OpenAI", tenant_id=tid):
row = row.to_dict() row = row.to_dict()
row["model_type"] = LLMType.EMBEDDING.value row["model_type"] = LLMType.EMBEDDING.value
row["llm_name"] = "text-embedding-3-small" row["llm_name"] = "text-embedding-3-small"