Spaces:
Sleeping
Sleeping
| 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?", [])) |