simoncwang's picture
new multi-agent
b540717
raw
history blame contribute delete
487 Bytes
from smolagents import OpenAIServerModel
import os
from dotenv import load_dotenv
load_dotenv()
from openai import OpenAI
# patch issue with missing stop parameter for o4-mini model
class PatchedOpenAIServerModel(OpenAIServerModel):
def _prepare_completion_kwargs(self, *args, **kwargs):
completion_kwargs = super()._prepare_completion_kwargs(*args, **kwargs)
if 'stop' in completion_kwargs:
del completion_kwargs['stop']
return completion_kwargs