dev: log mistral fc

zhudotexe · Oct 2, 2024 · 3bd0a18 · 3bd0a18
1 parent 9b4e3bd
commit 3bd0a18
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/kani/prompts/impl/mistral.py b/kani/prompts/impl/mistral.py
@@ -1,11 +1,14 @@
 import json
+import logging
 import re
 
 from kani.ai_function import AIFunction
 from kani.engines import Completion, WrapperEngine
 from kani.models import ChatMessage, ChatRole, FunctionCall, ToolCall
 from kani.prompts import ApplyContext, PromptPipeline
 
+log = logging.getLogger(__name__)
+
 
 # ref: https://github.com/mistralai/mistral-common/blob/main/src/mistral_common/tokens/tokenizers/sentencepiece.py
 def maybe_json(content: str):
@@ -199,6 +202,7 @@ def _parse_tool_calls(self, content: str) -> tuple[str, list[ToolCall]]:
         )
         if tool_json is None:
             return content, []
+        log.debug(f"Found tool JSON while parsing: {tool_json.group(1)}")
         actions = json.loads(tool_json.group(1))
 
         # translate back to kani spec