bugfix: allow sending temperature=0 values with a stricter `if temper…

…ature is None` comparison when falling back to NOT_GIVEN
GooeyAI · Dec 6, 2024 · 04b1c1f · 04b1c1f
1 parent bca5888
commit 04b1c1f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/daras_ai_v2/language_model.py b/daras_ai_v2/language_model.py
@@ -1052,7 +1052,7 @@ def _run_openai_chat(
                 max_completion_tokens=max_completion_tokens,
                 stop=stop or NOT_GIVEN,
                 n=num_outputs,
-                temperature=temperature or NOT_GIVEN,
+                temperature=temperature if temperature is not None else NOT_GIVEN,
                 frequency_penalty=frequency_penalty,
                 presence_penalty=presence_penalty,
                 tools=[tool.spec for tool in tools] if tools else NOT_GIVEN,