mirror of
https://github.com/Rarebuffalo/securelens-backend.git
synced 2026-06-19 07:00:30 +00:00
updated model for suitable testing
This commit is contained in:
11
list_models.py
Normal file
11
list_models.py
Normal file
@@ -0,0 +1,11 @@
|
||||
import os
|
||||
from google import genai
|
||||
from dotenv import load_dotenv
|
||||
|
||||
load_dotenv('/home/Krishna-Singh/securelens-backend/.env')
|
||||
client = genai.Client(api_key=os.environ.get("GEMINI_API_KEY"))
|
||||
|
||||
print("Supported Models:")
|
||||
for model in client.models.list():
|
||||
if 'generateContent' in model.supported_actions:
|
||||
print(f"- {model.name}")
|
||||
Reference in New Issue
Block a user