Spaces:

k-mktr
/

gpu-poor-llm-arena

Running

k-mktr commited on Apr 18

Commit

65a6f3b

verified ·

1 Parent(s): f8d0cdb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -115,11 +115,11 @@ def call_ollama_api(model, prompt):
             ]
         # Extract thinking part and main content using regex
-        thinking_match = re.search(r'<think>(.*?)</think>', content, flags=re.DOTALL)
         if thinking_match:
-            thinking_content = thinking_match.group(1).strip()
-            main_content = re.sub(r'<think>.*?</think>', '', content, flags=re.DOTALL).strip()
             logger.info(f"Found thinking content for model: {model}")
             return [

             ]
         # Extract thinking part and main content using regex
+        thinking_match = re.search(r'<(think|thought)>(.*?)</\1>', content, flags=re.DOTALL)
         if thinking_match:
+            thinking_content = thinking_match.group(2).strip()
+            main_content = re.sub(r'<(think|thought)>.*?</\1>', '', content, flags=re.DOTALL).strip()
             logger.info(f"Found thinking content for model: {model}")
             return [