Skip to content

GitLab

Commit d1da3dde authored by Timothy J. Baek's avatar Timothy J. Baek
Browse files

refac: migration

parent 3499ec3f
Hide whitespace changes
Inline Side-by-side
Showing with 6 additions and 0 deletions
backend/utils/misc.py
View file @ d1da3dde
...@@ -193,8 +193,14 @@ def parse_ollama_modelfile(model_text): ...@@ -193,8 +193,14 @@ def parse_ollama_modelfile(model_text):
system_desc_match = re.search( system_desc_match = re.search(
r'SYSTEM\s+"""(.+?)"""', model_text, re.DOTALL | re.IGNORECASE 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: if system_desc_match:
data["params"]["system"] = system_desc_match.group(1).strip() 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 # Parse messages
messages = [] messages = []
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment