Fix unittests, set strategy for rate limter

This commit is contained in:
hlohaus
2025-01-28 21:25:26 +01:00
parent 17d2f05d69
commit 2c895831b1
2 changed files with 3 additions and 2 deletions

View File

@@ -40,7 +40,7 @@ class PollinationsAI(AsyncGeneratorProvider, ProviderModelMixin):
default_vision_model = "gpt-4o" default_vision_model = "gpt-4o"
extra_image_models = ["midjourney", "dall-e-3", "flux-pro"] extra_image_models = ["midjourney", "dall-e-3", "flux-pro"]
vision_models = [default_vision_model, "gpt-4o-mini"] vision_models = [default_vision_model, "gpt-4o-mini"]
extra_text_models = ["claude", "claude-email", "deepseek-reasoner"] + vision_models extra_text_models = ["claude", "claude-email", "deepseek-reasoner", "p1"] + vision_models
model_aliases = { model_aliases = {
### Text Models ### ### Text Models ###
"gpt-4o-mini": "openai", "gpt-4o-mini": "openai",

View File

@@ -68,7 +68,8 @@ class Backend_Api(Api):
app=app, app=app,
default_limits=["200 per day", "50 per hour"], default_limits=["200 per day", "50 per hour"],
storage_uri="memory://", storage_uri="memory://",
auto_check=False auto_check=False,
strategy="moving-window",
) )
if has_flask_limiter and app.demo: if has_flask_limiter and app.demo: