Upload 2 files

llmdolphin.py

@@ -6,6 +6,7 @@ from llama_cpp_agent.providers import LlamaCppPythonProvider
 from llama_cpp_agent.chat_history import BasicChatHistory
 from llama_cpp_agent.chat_history.messages import Roles
 from ja_to_danbooru.ja_to_danbooru import jatags_to_danbooru_tags
+import timeout_decorator
 
 
 llm_models_dir = "./llm_models"
@@ -429,6 +430,7 @@ def dolphin_respond(
         yield [(outputs, None)]
 
 
+@timeout_decorator.timeout(15, use_signals=False)
 def dolphin_parse(
     history: list[tuple[str, str]],
 ):
@@ -526,6 +528,7 @@ def dolphin_respond_auto(
         yield [(outputs, None)]
 
 
+@timeout_decorator.timeout(15, use_signals=False)
 def dolphin_parse_simple(
     message: str,
     history: list[tuple[str, str]],

requirements.txt

@@ -14,4 +14,5 @@ httpx==0.13.3
 httpcore
 googletrans==4.0.0rc1
 git+https://github.com/huggingface/diffusers
-rapidfuzz
+rapidfuzz
+timeout-decorator