KiWA001 commited on
Commit
2277362
·
1 Parent(s): d999280

fix: register gemini provider in config and engine

Browse files
Files changed (2) hide show
  1. config.py +4 -0
  2. engine.py +6 -1
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 provider disabled (Playwright not installed)")
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] = {}