First Version
This commit is contained in:
@@ -0,0 +1,18 @@
|
||||
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
|
||||
Reference in New Issue
Block a user