From 55ab511784bcfaaf1b9f2e07ecc2349e36302510 Mon Sep 17 00:00:00 2001
From: Luke R <luke.robles@iterable.com>
Date: Thu, 13 Mar 2025 14:33:24 -0700
Subject: [PATCH] Add model name to llm output

---
 app/core_utils.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/app/core_utils.py b/app/core_utils.py
index 49b90f0c..3d9d414e 100755
--- a/app/core_utils.py
+++ b/app/core_utils.py
@@ -174,7 +174,10 @@ async def send_to_llm(ctx, message):
         response = requests.post(url, json=payload, headers=headers)
         answer = response.json()["message"]["content"]
 
-        return remove_between(answer, "<think>", "</think>")
+        return (
+            remove_between(answer, "<think>", "</think>")
+            + f"\n-# Generated using {llm_rules.get('model')}"
+        )
     except Exception as e:
         print(e)
         return "Somethings wrong, maybe the LLM crashed"