From 1c7ffa66edc41397da8591f6b191d6e0cc5e2fac Mon Sep 17 00:00:00 2001 From: smokestacklightnin <125844868+smokestacklightnin@users.noreply.github.com> Date: Sun, 31 Mar 2024 15:23:53 -0700 Subject: [PATCH] Add temperature option --- ragna/assistants/_ollama.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/ragna/assistants/_ollama.py b/ragna/assistants/_ollama.py index 4c2313f3..a9de8119 100644 --- 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)