Azure openai fixes (#2428)

This commit is contained in:
Parshva Daftari
2025-03-25 00:34:21 +05:30
committed by GitHub
parent 2b49c9eedd
commit 953a5a4a2d
4 changed files with 17 additions and 6 deletions

View File

@@ -80,13 +80,21 @@ class AzureOpenAILLM(LLMBase):
Returns:
str: The generated response.
"""
params = {
common_params = {
"model": self.config.model,
"messages": messages,
"temperature": self.config.temperature,
"max_tokens": self.config.max_tokens,
"top_p": self.config.top_p,
}
if self.config.model in {"o3-mini", "o1-preview", "o1"}:
params = common_params
else:
params = {
**common_params,
"temperature": self.config.temperature,
"max_tokens": self.config.max_tokens,
"top_p": self.config.top_p,
}
if response_format:
params["response_format"] = response_format
if tools: # TODO: Remove tools if no issues found with new memory addition logic