mirrors/gpt4free
1
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-12-06 02:30:41 -08:00
Code Issues Projects Releases Packages Wiki Activity

Fix unittests

Browse source
This commit is contained in:
hlohaus 2025-05-15 02:36:49 +02:00
parent e2bb1b1f0d
commit 8132cf6abc
1 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

9
g4f/Provider/TypeGPT.py
View file

@ -3,6 +3,7 @@ from __future__ import annotations
import requests import requests
from .template import OpenaiTemplate from .template import OpenaiTemplate
from .. import debug
class TypeGPT(OpenaiTemplate): class TypeGPT(OpenaiTemplate):
label = "TypeGpt" label = "TypeGpt"
@ -36,6 +37,10 @@ class TypeGPT(OpenaiTemplate):
@classmethod @classmethod
def get_models(cls, **kwargs): def get_models(cls, **kwargs):
if not cls.models: if not cls.models:
cls.models = requests.get(f"{cls.url}/api/config").json()["customModels"].split(",") try:
cls.models = [model.split("@")[0].strip("+") for model in cls.models if not model.startswith("-") and model not in cls.image_models] cls.models = requests.get(f"{cls.url}/api/config").json()["customModels"].split(",")
cls.models = [model.split("@")[0].strip("+") for model in cls.models if not model.startswith("-") and model not in cls.image_models]
except Exception as e:
cls.models = cls.fallback_models
debug.log(f"Error fetching models: {e}")
return cls.models return cls.models