MedQA / agent /gemini_agent.py
mgbam's picture
Update agent/gemini_agent.py
87ba598 verified
raw
history blame
300 Bytes
import os, google.generativeai as genai
from config.settings import settings
genai.configure(api_key=settings.gemini_api_key)
def chat_with_gemini(user, prompt):
model = genai.GenerativeModel("gemini-pro")
chat = model.start_chat()
resp = chat.send_message(prompt)
return resp.text