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()
|
models.sort()
|
||||||
|
|
||||||
elif provider == "google":
|
elif provider == "google":
|
||||||
# Wir nutzen die neue Library, die du oben importiert hast
|
|
||||||
if not GOOGLE_API_KEY:
|
if not GOOGLE_API_KEY:
|
||||||
return {"models": ["API-Key fehlt"]}
|
return {"models": ["API-Key fehlt"]}
|
||||||
|
|
||||||
# Wir erstellen kurz einen Client für die Abfrage
|
|
||||||
client = genai.Client(api_key=GOOGLE_API_KEY)
|
client = genai.Client(api_key=GOOGLE_API_KEY)
|
||||||
|
|
||||||
# In der neuen Library heißt es client.models.list()
|
|
||||||
models = []
|
models = []
|
||||||
|
|
||||||
|
# Im neuen SDK (google-genai) heißt das Feld 'supported_actions'
|
||||||
for m in client.models.list():
|
for m in client.models.list():
|
||||||
# Wir filtern nur Modelle, die Text generieren können
|
if 'generateContent' in m.supported_actions:
|
||||||
if "generateContent" in m.supported_generation_methods:
|
# Wir nehmen den Namen und entfernen das 'models/' Präfix
|
||||||
models.append(m.name.replace("models/", ""))
|
model_name = m.name.replace("models/", "")
|
||||||
|
models.append(model_name)
|
||||||
|
|
||||||
models.sort()
|
models.sort()
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user