aliakbar
18 lines
696 B
Python
18 lines
696 B
Python
from openai import OpenAI
|
|
|
|
|
|
class LLMInterface:
|
|
def __init__(self, personality):
|
|
api_key = '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'
|
|
self.client = OpenAI(api_key=api_key)
|
|
self.personality = personality
|
|
|
|
def get_reponse(self, query):
|
|
completion = self.client.chat.completions.create(
|
|
model="gpt-4o-mini",
|
|
messages=[
|
|
{"role": "system", "content": self.personality},
|
|
{"role": "user", "content": query}
|
|
]
|
|
)
|
|
return completion.choices[0].message.content |