瀏覽代碼

refac: migration

Timothy J. Baek 11 月之前
父節點
當前提交
d1da3dde65
共有 1 個文件被更改,包括 6 次插入0 次删除
  1. 6 0
      backend/utils/misc.py

+ 6 - 0
backend/utils/misc.py

@@ -193,8 +193,14 @@ def parse_ollama_modelfile(model_text):
     system_desc_match = re.search(
         r'SYSTEM\s+"""(.+?)"""', model_text, re.DOTALL | re.IGNORECASE
     )
+    system_desc_match_single = re.search(
+        r"SYSTEM\s+([^\n]+)", model_text, re.IGNORECASE
+    )
+
     if system_desc_match:
         data["params"]["system"] = system_desc_match.group(1).strip()
+    elif system_desc_match_single:
+        data["params"]["system"] = system_desc_match_single.group(1).strip()
 
     # Parse messages
     messages = []