Merge pull request #7920 from envision3d/ollama-structured-output

feat: pass structured output format through to ollama
2024-12-17 13:37:24 -08:00
parent 9531edf6d6 16d900247a
commit a14c06fa0b
2 changed files with 4 additions and 1 deletions
@@ -917,7 +917,7 @@ class ChatMessage(BaseModel):
 class GenerateChatCompletionForm(BaseModel):
    model: str
    messages: list[ChatMessage]
-    format: Optional[str] = None
+    format: Optional[dict] = None
    options: Optional[dict] = None
    template: Optional[str] = None
    stream: Optional[bool] = True