Spaces:
Running
Running
fix: register gemini provider in config and engine
Browse files
config.py
CHANGED
|
@@ -24,6 +24,7 @@ MODEL_RANKING = [
|
|
| 24 |
("gpt-4", "g4f", "gpt-4"),
|
| 25 |
("gpt-4o-mini", "g4f", "gpt-4o-mini"),
|
| 26 |
("glm-5", "zai", "glm-5"),
|
|
|
|
| 27 |
("gpt-oss-20b", "pollinations", "openai"),
|
| 28 |
("mistral-small-3.2", "pollinations", "mistral"),
|
| 29 |
|
|
@@ -74,6 +75,9 @@ PROVIDER_MODELS = {
|
|
| 74 |
"zai": [
|
| 75 |
"glm-5",
|
| 76 |
],
|
|
|
|
|
|
|
|
|
|
| 77 |
"pollinations": [
|
| 78 |
"gpt-oss-20b",
|
| 79 |
"mistral-small-3.2-24b",
|
|
|
|
| 24 |
("gpt-4", "g4f", "gpt-4"),
|
| 25 |
("gpt-4o-mini", "g4f", "gpt-4o-mini"),
|
| 26 |
("glm-5", "zai", "glm-5"),
|
| 27 |
+
("gemini-1.5-pro", "gemini", "gemini-1.5-pro"),
|
| 28 |
("gpt-oss-20b", "pollinations", "openai"),
|
| 29 |
("mistral-small-3.2", "pollinations", "mistral"),
|
| 30 |
|
|
|
|
| 75 |
"zai": [
|
| 76 |
"glm-5",
|
| 77 |
],
|
| 78 |
+
"gemini": [
|
| 79 |
+
"gemini-1.5-pro",
|
| 80 |
+
],
|
| 81 |
"pollinations": [
|
| 82 |
"gpt-oss-20b",
|
| 83 |
"mistral-small-3.2-24b",
|
engine.py
CHANGED
|
@@ -17,6 +17,7 @@ from supabase import create_client, Client
|
|
| 17 |
from providers.base import BaseProvider
|
| 18 |
from providers.g4f_provider import G4FProvider
|
| 19 |
from providers.pollinations_provider import PollinationsProvider
|
|
|
|
| 20 |
from providers.zai_provider import ZaiProvider
|
| 21 |
from config import MODEL_RANKING, PROVIDER_MODELS, SUPABASE_URL, SUPABASE_KEY
|
| 22 |
from models import ModelInfo
|
|
@@ -51,8 +52,12 @@ class AIEngine:
|
|
| 51 |
if ZaiProvider.is_available():
|
| 52 |
self._providers["zai"] = ZaiProvider()
|
| 53 |
logger.info("✅ Z.ai provider enabled (Playwright available)")
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
else:
|
| 55 |
-
logger.warning("⚠️ Z.ai
|
| 56 |
# Success Tracker: Key = "provider/model_id"
|
| 57 |
# Value = {success, failure, consecutive_failures, avg_time_ms, total_time_ms, count_samples}
|
| 58 |
self._stats: dict[str, dict] = {}
|
|
|
|
| 17 |
from providers.base import BaseProvider
|
| 18 |
from providers.g4f_provider import G4FProvider
|
| 19 |
from providers.pollinations_provider import PollinationsProvider
|
| 20 |
+
from providers.gemini_provider import GeminiProvider
|
| 21 |
from providers.zai_provider import ZaiProvider
|
| 22 |
from config import MODEL_RANKING, PROVIDER_MODELS, SUPABASE_URL, SUPABASE_KEY
|
| 23 |
from models import ModelInfo
|
|
|
|
| 52 |
if ZaiProvider.is_available():
|
| 53 |
self._providers["zai"] = ZaiProvider()
|
| 54 |
logger.info("✅ Z.ai provider enabled (Playwright available)")
|
| 55 |
+
|
| 56 |
+
# Gemini also uses Playwright, so we enable it here too
|
| 57 |
+
self._providers["gemini"] = GeminiProvider()
|
| 58 |
+
logger.info("✅ Gemini provider enabled")
|
| 59 |
else:
|
| 60 |
+
logger.warning("⚠️ Z.ai/Gemini providers disabled (Playwright not installed)")
|
| 61 |
# Success Tracker: Key = "provider/model_id"
|
| 62 |
# Value = {success, failure, consecutive_failures, avg_time_ms, total_time_ms, count_samples}
|
| 63 |
self._stats: dict[str, dict] = {}
|