Rag_basic/utils/api_llm.py

13 lines
476 B
Python

import google.generativeai as genai
import time
genai.configure(api_key="AIzaSyDWqNUBKhaZjbFI8CW52_hKr46JtWABkGU")
model = genai.GenerativeModel("models/gemini-2.0-flash-001")
start_time = time.time()
response = model.generate_content("Xin chào, bạn là ai?")
print(response.text)
print("Thời gian thực thi: %.2f giây" % (time.time() - start_time))
# for m in genai.list_models():
# if "generateContent" in m.supported_generation_methods:
# print(m.name)