main.py aktualisiert
This commit is contained in:
21
main.py
21
main.py
@@ -274,6 +274,7 @@ async def get_models(provider: str, url: str = None):
|
||||
models = [m["name"] for m in data.get("models", [])]
|
||||
|
||||
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
|
||||
import openai
|
||||
client = openai.AsyncOpenAI(api_key=OPENAI_API_KEY)
|
||||
@@ -283,11 +284,21 @@ async def get_models(provider: str, url: str = None):
|
||||
models.sort()
|
||||
|
||||
elif provider == "google":
|
||||
# Hier greift das Backend auf deinen Google API-Key zu
|
||||
import google.generativeai as genai
|
||||
genai.configure(api_key=GOOGLE_API_KEY)
|
||||
# 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 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 = []
|
||||
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}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user