File size: 487 Bytes
b540717 ab3aca2 b540717 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
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 |