Add finish reason

Author: johann-petrak

llms_wrapper/llms.py

@@ -423,6 +423,7 @@ def query(
                 ret["n_prompt_tokens"] = usage.prompt_tokens
                 ret["n_total_tokens"] = usage.total_tokens
             response_message = response['choices'][0]['message']
+            ret["finish_reason"] = response['choices'][0].get('finish_reason', "UNKNOWN")
             ret["answer"] = response_message['content']
             ret["error"] = ""
             ret["ok"] = True

llms_wrapper/version.py

@@ -1,3 +1,3 @@
 import importlib.metadata
-__version__ = "0.1.24"
+__version__ = "0.1.25"
 

notebooks/example-config2.hjson

@@ -22,6 +22,9 @@
     {
       llm: anthropic/claude-3-opus-20240229
     }
+    { 
+      llm: anthropic/claude-3-7-sonnet-20250219
+    }
     {
       llm: mistral/mistral-large-latest
     }
@@ -69,4 +72,4 @@
       api_key_env: MY_DEEPSEEK_API_KEY
     }
   }
-}
+}