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", [])]
|
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}
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user