Log model used

mahaloz · Nov 7, 2024 · 0f98eca · 0f98eca
1 parent 2ebe2a6
commit 0f98eca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/dailalib/__init__.py b/dailalib/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "3.10.2"
+__version__ = "3.10.3"
 
 import os
 # stop LiteLLM from querying at all to the remote server

diff --git a/dailalib/api/litellm/prompts/prompt.py b/dailalib/api/litellm/prompts/prompt.py
@@ -83,9 +83,7 @@ def _query_model(ai_api=self.ai_api, function=function, dec_text=dec_text, **_kw
                 few_shot=bool(self.ai_api.prompt_style == PromptType.FEW_SHOT),
             )
             self.last_rendered_template = query_text
-            #ai_api.info(f"Prompting using model: {self.ai_api.model}...")
-            #ai_api.info(f"Prompting with style: {self.ai_api.prompt_style}...")
-            #ai_api.info(f"Prompting with: {query_text}")
+            ai_api.info(f"Prompting using: model={self.ai_api.model} and style={self.ai_api.prompt_style}")
 
             ai_api.on_query(self.name, self.ai_api.model, self.ai_api.prompt_style, function, dec_text)
             response += self.ai_api.query_model(query_text)