From d1da3dde656ccaa8c905e3e873e8e318de20a3ed Mon Sep 17 00:00:00 2001 From: "Timothy J. Baek" Date: Sat, 8 Jun 2024 13:45:33 -0700 Subject: [PATCH] refac: migration --- backend/utils/misc.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/backend/utils/misc.py b/backend/utils/misc.py index 9069857b7f..4bb5ddf56d 100644 --- a/backend/utils/misc.py +++ b/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 = []