File size: 753 Bytes
e0225ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from openai import OpenAI
from dotenv import load_dotenv
import os
load_dotenv()
api_key = os.getenv("GEMINI_API_KEY")
model = "gemini-2.5-flash"
base_url = "https://generativelanguage.googleapis.com/v1beta/openai"
client = OpenAI(base_url = base_url, api_key = api_key)
ai_teacher = """you are williams shakes sphere, you must use Early Modern English ”"""
def get_response(message,history):
messages = [{"role": "system","content": ai_teacher}]
messages.append({"role":"user","content":message})
response = client.chat.completions.create(model = model,messages = messages)
ai_response = response.choices[0].message.content
return ai_response
if __name__ == "__main__":
print(get_response("hi",[]))
|