main.py aktualisiert
This commit is contained in:
13
main.py
13
main.py
@@ -284,19 +284,18 @@ async def get_models(provider: str, url: str = None):
|
||||
models.sort()
|
||||
|
||||
elif provider == "google":
|
||||
# Wir nutzen die neue Library, die du oben importiert hast
|
||||
if not GOOGLE_API_KEY:
|
||||
return {"models": ["API-Key fehlt"]}
|
||||
|
||||
# Wir erstellen kurz einen Client für die Abfrage
|
||||
client = genai.Client(api_key=GOOGLE_API_KEY)
|
||||
|
||||
# In der neuen Library heißt es client.models.list()
|
||||
models = []
|
||||
|
||||
# Im neuen SDK (google-genai) heißt das Feld 'supported_actions'
|
||||
for m in client.models.list():
|
||||
# Wir filtern nur Modelle, die Text generieren können
|
||||
if "generateContent" in m.supported_generation_methods:
|
||||
models.append(m.name.replace("models/", ""))
|
||||
if 'generateContent' in m.supported_actions:
|
||||
# Wir nehmen den Namen und entfernen das 'models/' Präfix
|
||||
model_name = m.name.replace("models/", "")
|
||||
models.append(model_name)
|
||||
|
||||
models.sort()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user