From e20207101abe5045be06cdd1da36c321747c82a8 Mon Sep 17 00:00:00 2001
From: KevinHuSh <kevinhu.sh@gmail.com>
Date: Fri, 12 Apr 2024 09:08:08 +0800
Subject: [PATCH] fix wrong log printting (#330)
### What problem does this PR solve?
Issue link:#325
### Type of change
- [x] Bug Fix (non-breaking change which fixes an issue)
---
api/db/services/llm_service.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/api/db/services/llm_service.py b/api/db/services/llm_service.py
index 0e9e774..b14972d 100644
--- a/api/db/services/llm_service.py
+++ b/api/db/services/llm_service.py
@@ -136,7 +136,7 @@ class LLMBundle(object):
def encode(self, texts: list, batch_size=32):
emd, used_tokens = self.mdl.encode(texts, batch_size)
- if TenantLLMService.increase_usage(
+ if not TenantLLMService.increase_usage(
self.tenant_id, self.llm_type, used_tokens):
database_logger.error(
"Can't update token usage for {}/EMBEDDING".format(self.tenant_id))
@@ -144,7 +144,7 @@ class LLMBundle(object):
def encode_queries(self, query: str):
emd, used_tokens = self.mdl.encode_queries(query)
- if TenantLLMService.increase_usage(
+ if not TenantLLMService.increase_usage(
self.tenant_id, self.llm_type, used_tokens):
database_logger.error(
"Can't update token usage for {}/EMBEDDING".format(self.tenant_id))
--
GitLab