From cb7d17990cf4ed2a1e45b5f92c0209563b87afb4 Mon Sep 17 00:00:00 2001
From: Christian Krinitsin <mail@krinitsin.com>
Date: Thu, 29 May 2025 15:10:41 +0000
Subject: add ollama test script

---
 ollama/llm.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)
 create mode 100755 ollama/llm.py

(limited to 'ollama/llm.py')

diff --git a/ollama/llm.py b/ollama/llm.py
new file mode 100755
index 00000000..acf76421
--- /dev/null
+++ b/ollama/llm.py
@@ -0,0 +1,31 @@
+from ollama import chat, ChatResponse
+from re import sub
+from os import listdir, path
+
+model : str = "deepseek-r1:70b"
+directory : str = "../mailinglist/output_mailinglist"
+
+with open("preambel", "r") as file:
+    preambel = file.read()
+
+for name in listdir(directory):
+    with open(path.join(directory, name)) as file:
+        content = preambel + "\n" + file.read()
+
+    with open("test", "r") as file:
+        content = file.read()
+
+    response : ChatResponse = chat(
+        model=model, messages=[
+            {
+                'role': 'user',
+                'content': content,
+            }
+        ]
+    )
+
+    no_think_response : str = sub(r'<think>(.|\n)*</think>\n\n', '', response.message.content)
+
+    print(no_think_response)
+    print("\n")
+    exit()
-- 
cgit 1.4.1