from avaliar.models.base import AvaliarBaseLLM
from openai import OpenAI
class MyModel(AvaliarBaseLLM):
def __init__(self):
self.client = OpenAI()
def generate(self, prompt: str) -> str:
response = self.client.chat.completions.create(
model="gpt-4o",
messages=[{"role": "user", "content": prompt}],
)
return response.choices[0].message.content
def generate_samples(
self, prompt: str, n: int, temperature: float
) -> list[str]:
responses = []
for _ in range(n):
response = self.client.chat.completions.create(
model="gpt-4o",
messages=[{"role": "user", "content": prompt}],
temperature=temperature,
)
responses.append(response.choices[0].message.content)
return responses