Add temperature option

Quansight · Mar 31, 2024 · 1c7ffa6 · 1c7ffa6
1 parent 99f42d3
commit 1c7ffa6
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ragna/assistants/_ollama.py b/ragna/assistants/_ollama.py
@@ -56,7 +56,8 @@ async def _call_api(
                 ],
                 "model": self._MODEL,
                 "stream": True,
-            },  # TODO: Add optional parameters for the model
+                "temperature": 0.0,
+            },
         ) as response:
             await self._assert_api_call_is_success(response)