openai[patch]: default temp=1 for o1 (#27206)

This commit is contained in:
Bagatur
2024-10-08 15:45:21 -07:00
committed by GitHub
parent b298d0337e
commit ce33c4fa40
2 changed files with 16 additions and 0 deletions

View File

@@ -35,6 +35,13 @@ def test_openai_model_param() -> None:
assert llm.model_name == "foo"
def test_openai_o1_temperature() -> None:
llm = ChatOpenAI(model="o1-preview")
assert llm.temperature == 1
llm = ChatOpenAI(model_name="o1-mini") # type: ignore[call-arg]
assert llm.temperature == 1
def test_function_message_dict_to_function_message() -> None:
content = json.dumps({"result": "Example #1"})
name = "test_function"