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

fix bug of inserting cites (#76)

Browse Source
This commit is contained in:
KevinHuSh
2024-02-27 17:51:54 +08:00
committed by GitHub
parent 4568a4b2cb
commit 1567e881de
5 changed files with 31 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
api/apps/user_app.py
View File

@ -208,9 +208,9 @@ def user_register(user_id, user):
for llm in LLMService.query(fid=LLM_FACTORY):
tenant_llm.append({"tenant_id": user_id, "llm_factory": LLM_FACTORY, "llm_name": llm.llm_name, "model_type":llm.model_type, "api_key": API_KEY})
if not UserService.save(**user):return
TenantService.save(**tenant)
UserTenantService.save(**usr_tenant)
if not UserService.insert(**user):return
TenantService.insert(**tenant)
UserTenantService.insert(**usr_tenant)
TenantLLMService.insert_many(tenant_llm)
return UserService.query(email=user["email"])