chatbot-with-memory / chatbot_with_memory.py
Stagiaire-18-HNAI's picture
Upload 4 files
9828d3a verified
from openai import OpenAI
from dotenv import load_dotenv
import os
from prompts import ai_motivational_speaker
load_dotenv()
api_key = os.getenv("GEMINI_API_KEY")
model = "gemini-2.5-flash-lite"
base_url = "https://generativelanguage.googleapis.com/v1beta/openai/"
client = OpenAI(base_url=base_url, api_key=api_key)
ai_motivational_speaker=ai_motivational_speaker
def get_response(message, history):
messages = [{"role":"system", "content":ai_motivational_speaker}]
messages.extend(history)
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
print(get_response("how to be patient?", []))