修复计量错误

LlmKira · Mar 3, 2023 · 2173fb8 · 2173fb8
1 parent 16a68b2
commit 2173fb8
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/llm_kira/client/Optimizer.py b/src/llm_kira/client/Optimizer.py
@@ -181,6 +181,10 @@ def run(self) -> List[Interaction]:
         _knowledge_token_limit = int(self.token_limit * self.reference_ratio)
         _interaction_token_limit = self.token_limit - _knowledge_token_limit
 
+        # Cost - Head = Body
+        _desc_cost = self.tokenizer(str(self.desc))
+        _interaction_token_limit = _interaction_token_limit - _desc_cost
+
         # Desc
         _returner = [Interaction(single=True, ask=PromptItem(start="system", text=self.desc))]