-
Notifications
You must be signed in to change notification settings - Fork 2
/
ask_gemini.py
39 lines (31 loc) · 1.16 KB
/
ask_gemini.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
import requests
import json
from dotenv import load_dotenv
import os
import google.generativeai as genai
load_dotenv()
genai.configure(api_key=os.getenv["GEMINI_API_KEY"])
# The Gemini 1.5 models are versatile and work with both text-only and multimodal prompts
model = genai.GenerativeModel('gemini-1.5-flash')
# Function to call the Gemini API from Python
def ask_gemini(prompt):
api_key = os.getenv("GEMINI_API_KEY") # Make sure to set this in your .env file
headers = {
"Authorization": f"Bearer {api_key}",
"Content-Type": "application/json",
}
data = {
"model": "gemini-1.0", # Adjust model name as per Gemini API documentation
"prompt": prompt
}
response = requests.post(
url="https://api.gemini.com/v1/chat/completions", # Adjust the URL as per Gemini API documentation
headers=headers,
json=data
)
if response.status_code == 200:
return response.json()['choices'][0]['message']['content'].strip()
else:
# Handle errors or unexpected response status codes
print(f"Error: Received response code {response.status_code}")
return None