|
|
from openai import OpenAI |
|
|
from dotenv import load_dotenv |
|
|
import os |
|
|
from prompts import ai_Travel_Assistant |
|
|
|
|
|
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(api_key=api_key, base_url=base_url) |
|
|
|
|
|
ai_Travel_Assistant = ai_Travel_Assistant |
|
|
|
|
|
|
|
|
def get_response(message, history): |
|
|
messages = [{"role": "system", "content": ai_Travel_Assistant}] |
|
|
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 |
|
|
|
|
|
|
|
|
|
|
|
|
