AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-10 07:19:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

Ignore tenant not found error while increasing token usage. (#4950)

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

#4940

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Kevin Hu 2025-02-14 11:10:49 +08:00 committed by GitHub
parent dce7053c24
commit 849d9eb463
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -173,7 +173,8 @@ class TenantLLMService(CommonService):
def increase_usage(cls, tenant_id, llm_type, used_tokens, llm_name=None): def increase_usage(cls, tenant_id, llm_type, used_tokens, llm_name=None):
e, tenant = TenantService.get_by_id(tenant_id) e, tenant = TenantService.get_by_id(tenant_id)
if not e: if not e:
raise LookupError("Tenant not found") logging.error(f"Tenant not found: {tenant_id}")
return 0
llm_map = { llm_map = {
LLMType.EMBEDDING.value: tenant.embd_id, LLMType.EMBEDDING.value: tenant.embd_id,
@ -186,7 +187,8 @@ class TenantLLMService(CommonService):
mdlnm = llm_map.get(llm_type) mdlnm = llm_map.get(llm_type)
if mdlnm is None: if mdlnm is None:
raise ValueError("LLM type error") logging.error(f"LLM type error: {llm_type}")
return 0
llm_name, llm_factory = TenantLLMService.split_model_name_and_factory(mdlnm) llm_name, llm_factory = TenantLLMService.split_model_name_and_factory(mdlnm)