Skip to content

Commit 7d2cf39

Browse files
committed
accommodated for o1
1 parent 660bb7c commit 7d2cf39

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

src/llm_change_agent/config/llm_config.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,10 @@ class LLMConfig(BaseModel):
1313

1414
class OpenAIConfig(LLMConfig):
1515
"""Configuration for OpenAI LLM model."""
16-
17-
pass
16+
def __init__(self, **data):
17+
super().__init__(**data)
18+
if self.model.startswith("o1"):
19+
self.temperature = 1.0
1820

1921

2022
class OllamaConfig(LLMConfig):

0 commit comments

Comments
 (0)