From 9e31b7f47e8bc8b2d5e44623c8f68639712e54e0 Mon Sep 17 00:00:00 2001
From: SpudGunMan <kellykeeton@yahoo.com>
Date: Wed, 29 Jan 2025 20:01:28 -0800
Subject: [PATCH] deepseek compatibility

deepseek
---
 modules/llm.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/modules/llm.py b/modules/llm.py
index 3c40600..183c2f6 100644
--- a/modules/llm.py
+++ b/modules/llm.py
@@ -211,6 +211,10 @@ def llm_query(input, nodeID=0, location_name=None):
             if result.status_code == 200:
                 result_json = result.json()
                 result = result_json.get("response", "")
+
+                # deepseek-r1 has added <think> </think> tags to the response
+                if "<think>" in result:
+                    result = result.split("</think>")[1]
             else:
                 raise Exception(f"HTTP Error: {result.status_code}")