layout refine (#115)

2024-03-08 18:59:53 +08:00
parent fa171dfe6c
commit bcb58b7e71
5 changed files with 19 additions and 6 deletions
--- a/api/db/services/llm_service.py
+++ b/api/db/services/llm_service.py
@@ -49,7 +49,9 @@ class TenantLLMService(CommonService):
            LLMFactories.logo,
            LLMFactories.tags,
            cls.model.model_type,
-            cls.model.llm_name]
+            cls.model.llm_name,
+            cls.model.used_tokens
+        ]
        objs = cls.model.select(*fields).join(LLMFactories, on=(cls.model.llm_factory == LLMFactories.name)).where(
            cls.model.tenant_id == tenant_id).dicts()