bug: [7] multiple models

tvaroska · Dec 10, 2024 · 754156c · 754156c
1 parent 4c127f9
commit 754156c
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/promptgit/prompt.py b/promptgit/prompt.py
@@ -144,6 +144,9 @@ def parse_children(children):
             return response
 
         fields = parse_children(mistletoe.Document(content).children)
+        # Multiple models - each one at one line
+        if 'models' in fields:
+            fields['models'] = fields['models'].split('\n')
 
         return cls.model_validate(fields)