From bfbf8d82e0f592a4580c860116ab0496bad0e389 Mon Sep 17 00:00:00 2001
From: Karl <karl@k-world.me.uk>
Date: Fri, 27 Jun 2025 16:38:25 +0100
Subject: [PATCH] try cleanup generated prompt if any other text

---
 libs/ollama.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/libs/ollama.py b/libs/ollama.py
index 18b6be3..6ae4161 100644
--- a/libs/ollama.py
+++ b/libs/ollama.py
@@ -3,6 +3,7 @@ import logging
 import litellm
 import nest_asyncio
 from libs.generic import load_recent_prompts, load_config
+import re
 nest_asyncio.apply()
 
 logging.basicConfig(level=logging.INFO)
@@ -74,5 +75,10 @@ def create_prompt_on_openwebui(prompt: str, topic: str = "random") -> str:
     #     api_key=user_config["openwebui"]["api_key"],
     # )
     # prompt = response["choices"][0]["message"]["content"].strip('"')
+    match = re.search(r'"([^"]+)"', prompt)
+    if not match:
+        match = re.search(r":\s*\n*\s*(.+)", prompt) 
+    if match:
+        prompt = match.group(1)
     logging.debug(prompt)
-    return prompt.split(": ")[-1]
\ No newline at end of file
+    return prompt
\ No newline at end of file