main.py aktualisiert

This commit is contained in:
2026-03-05 23:00:26 +00:00
parent afc7775820
commit 5aa1c5d779

21
main.py
View File

@@ -274,6 +274,7 @@ async def get_models(provider: str, url: str = None):
models = [m["name"] for m in data.get("models", [])] models = [m["name"] for m in data.get("models", [])]
elif provider == "openai": elif provider == "openai":
if not OPENAI_API_KEY or "hier" in OPENAI_API_KEY: return {"models": []}
# Hier greift das Backend auf deinen OpenAI API-Key zu # Hier greift das Backend auf deinen OpenAI API-Key zu
import openai import openai
client = openai.AsyncOpenAI(api_key=OPENAI_API_KEY) client = openai.AsyncOpenAI(api_key=OPENAI_API_KEY)
@@ -283,11 +284,21 @@ async def get_models(provider: str, url: str = None):
models.sort() models.sort()
elif provider == "google": elif provider == "google":
# Hier greift das Backend auf deinen Google API-Key zu # Wir nutzen die neue Library, die du oben importiert hast
import google.generativeai as genai if not GOOGLE_API_KEY:
genai.configure(api_key=GOOGLE_API_KEY) return {"models": ["API-Key fehlt"]}
# Holt alle verfügbaren Modelle, die Text generieren können
models = [m.name.replace("models/", "") for m in genai.list_models() if "generateContent" in m.supported_generation_methods] # 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 = []
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/", ""))
models.sort()
return {"models": models} return {"models": models}