返回提交历史
Modified
g4f/Provider/Yupp.py
+4
-2
XFEstudio/gpt4free
Add MissingAuthError for unconfigured Yupp accounts
8225e99a
代码差异
1 个文件
+4
-2
@@ -9,7 +9,7 @@ import requests
9
9
10
10
from ..providers.base_provider import AbstractProvider, ProviderModelMixin
11
11
from ..providers.response import Reasoning, PlainTextResponse, PreviewResponse, JsonConversation, ImageResponse, ProviderInfo
12
from ..errors import RateLimitError, ProviderException
12
from ..errors import RateLimitError, ProviderException, MissingAuthError
13
13
from ..cookies import get_cookies
14
14
from ..tools.auth import AuthManager
15
15
from .yupp.models import YuppModelManager
@@ -174,6 +174,8 @@ class Yupp(AbstractProvider, ProviderModelMixin):
174
174
api_key = AuthManager.load_api_key(cls)
175
175
if not api_key:
176
176
api_key = get_cookies("yupp.ai", False).get("__Secure-yupp.session-token")
177
if not api_key:
178
raise MissingAuthError("No Yupp accounts configured. Set YUPP_API_KEY environment variable.")
177
179
manager = YuppModelManager(api_key=api_key)
178
180
models = manager.client.fetch_models()
179
181
if models:
@@ -204,7 +206,7 @@ class Yupp(AbstractProvider, ProviderModelMixin):
204
206
elif YUPP_ACCOUNTS:
205
207
log_debug(f"Yupp provider using existing accounts: {len(YUPP_ACCOUNTS)}")
206
208
else:
207
raise ProviderException("No Yupp accounts configured. Set YUPP_API_KEY environment variable.")
209
raise MissingAuthError("No Yupp accounts configured. Set YUPP_API_KEY environment variable.")
208
210
209
211
if messages is None:
210
212
messages = []