webscout 8.3.3__py3-none-any.whl → 8.3.5__py3-none-any.whl

webscout/Bard.py

@@ -81,21 +81,11 @@ class Model(Enum):
         model_header (dict): Additional headers required for the model.
         advanced_only (bool): Whether the model is available only for advanced users.
     """
-    # Updated model definitions based on reference implementation
+    # Only the specified models
     UNSPECIFIED = ("unspecified", {}, False)
-    G_2_0_FLASH = (
-        "gemini-2.0-flash",
-        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"f299729663a2343f"]'},
-        False,
-    )
-    G_2_0_FLASH_THINKING = (
-        "gemini-2.0-flash-thinking",
-        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"7ca48d02d802f20a"]'},
-        False,
-    )
     G_2_5_FLASH = (
         "gemini-2.5-flash",
-        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"35609594dbe934d8"]'},
+        {"x-goog-ext-525001261-jspb": '[1,null,null,null,"71c2d248d3b102ff"]'},
         False,
     )
     G_2_5_PRO = (
@@ -103,16 +93,6 @@ class Model(Enum):
         {"x-goog-ext-525001261-jspb": '[1,null,null,null,"2525e3954d185b3c"]'},
         False,
     )
-    G_2_0_EXP_ADVANCED = (
-        "gemini-2.0-exp-advanced",
-        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"b1e46a6037e6aa9f"]'},
-        True,
-    )
-    G_2_5_EXP_ADVANCED = (
-        "gemini-2.5-exp-advanced",
-        {"x-goog-ext-525001261-jspb": '[null,null,null,null,"203e6bb81620bcfe"]'},
-        True,
-    )
 
     def __init__(self, name, header, advanced_only):
         """

webscout/Provider/AISEARCH/__init__.py

@@ -1,10 +1,11 @@
-from .stellar_search import *
-from .felo_search import *
-from .DeepFind import *
-from .genspark_search import *
-from .monica_search import *
-from .webpilotai_search import *
-from .hika_search import *
-from .scira_search import *
-from .iask_search import *
-from .Perplexity import *
+from .stellar_search import *
+from .felo_search import *
+from .DeepFind import *
+from .genspark_search import *
+from .monica_search import *
+from .webpilotai_search import *
+from .hika_search import *
+from .scira_search import *
+from .iask_search import *
+from .Perplexity import *
+# from .PERPLEXED_search import *

webscout/Provider/AISEARCH/felo_search.py

@@ -69,7 +69,7 @@ class Felo(AISearch):
             "accept-encoding": "gzip, deflate, br, zstd",
             "accept-language": "en-US,en;q=0.9,en-IN;q=0.8",
             "content-type": "application/json",
-            "cookie": "_clck=1gifk45%7C2%7Cfoa%7C0%7C1686; _clsk=1g5lv07%7C1723558310439%7C1%7C1%7Cu.clarity.ms%2Fcollect; _ga=GA1.1.877307181.1723558313; _ga_8SZPRV97HV=GS1.1.1723558313.1.1.1723558341.0.0.0; _ga_Q9Q1E734CC=GS1.1.1723558313.1.1.1723558341.0.0.0",
+            # "cookie": "_clck=1gifk45%7C2%7Cfoa%7C0%7C1686; _clsk=1g5lv07%7C1723558310439%7C1%7C1%7Cu.clarity.ms%2Fcollect; _ga=GA1.1.877307181.1723558313; _ga_8SZPRV97HV=GS1.1.1723558313.1.1.1723558341.0.0.0; _ga_Q9Q1E734CC=GS1.1.1723558313.1.1.1723558341.0.0.0",
             "dnt": "1",
             "origin": "https://felo.ai",
             "referer": "https://felo.ai/",
@@ -141,10 +141,14 @@ class Felo(AISearch):
             "lang": "",
             "agent_lang": "en",
             "search_options": {
-                "langcode": "en-US"
+                "langcode": "en-US",
+                "search_image": True,
+                "search_video": True,
             },
             "search_video": True,
-            "contexts_from": "google"
+            "model": "",
+            "contexts_from": "google",
+            "auto_routing": True,
         }
 
         def for_stream():

webscout/Provider/AISEARCH/scira_search.py

@@ -43,17 +43,32 @@ class Scira(AISearch):
     """
 
     AVAILABLE_MODELS = {
-        "scira-default": "Grok3-mini", # thinking model
-        "scira-grok-3": "Grok3",
-        "scira-anthropic": "Claude 4 Sonnet",
-        "scira-anthropic-thinking": "Claude 4 Sonnet Thinking", # thinking model
-        "scira-vision" : "Grok2-Vision", # vision model
-        "scira-4o": "GPT4o",
-        "scira-qwq": "QWQ-32B",
-        "scira-o4-mini": "o4-mini",
-        "scira-google": "gemini 2.5 flash Thinking", # thinking model
-        "scira-google-pro": "gemini 2.5 pro",
-        "scira-llama-4": "llama 4 Maverick",
+        "scira-default": "grok-3-mini",  # thinking model
+        "scira-x-fast-mini": "grok-3-mini-fast",
+        "scira-x-fast": "grok-3-fast",
+        "scira-nano": "gpt-4.1-nano",
+        "scira-grok-3": "grok-3",
+        "scira-grok-4": "grok-4",
+        "scira-vision": "grok-2-vision-1212",
+        "scira-g2": "grok-2-latest",
+        "scira-4o-mini": "gpt-4o-mini",
+        "scira-o4-mini": "o4-mini-2025-04-16",
+        "scira-o3": "o3",
+        "scira-qwen-32b": "qwen/qwen3-32b",
+        "scira-qwen-30b": "qwen3-30b-a3b",
+        "scira-deepseek-v3": "deepseek-v3-0324",
+        "scira-haiku": "claude-3-5-haiku-20241022",
+        "scira-mistral": "mistral-small-latest",
+        "scira-google-lite": "gemini-2.5-flash-lite-preview-06-17",
+        "scira-google": "gemini-2.5-flash",
+        "scira-google-pro": "gemini-2.5-pro",
+        "scira-anthropic": "claude-sonnet-4-20250514",
+        "scira-anthropic-thinking": "claude-sonnet-4-20250514",
+        "scira-opus": "claude-4-opus-20250514",
+        "scira-opus-pro": "claude-4-opus-20250514",
+        "scira-llama-4": "meta-llama/llama-4-maverick-17b-128e-instruct",
+        "scira-kimi-k2": "kimi-k2-instruct",
+        "kimi-k2-instruct": "scira-kimi-k2",
     }
     def __init__(
         self,

webscout/Provider/AISEARCH/stellar_search.py

@@ -53,7 +53,12 @@ class Stellar(AISearch):
 
     @staticmethod
     def _stellar_extractor(chunk: Union[str, bytes, Dict[str, Any]]) -> Optional[str]:
-        """Extracts content from the Stellar stream format with hex keys and diff arrays. Handles both str and bytes input."""
+        """
+        Extracts content from the Stellar stream format with focused pattern matching.
+        
+        Prioritizes the primary diff pattern to avoid duplication and focuses on
+        incremental content building from stellar.chatastra.ai streaming response.
+        """
         if isinstance(chunk, bytes):
             try:
                 chunk = chunk.decode('utf-8', errors='replace')
@@ -61,14 +66,54 @@ class Stellar(AISearch):
                 return None
         if not isinstance(chunk, str):
             return None
-        # Match patterns like 6e:{"diff":[0," empathy"],"next":"$@6f"}
-        pattern = r'[0-9a-f]+:\{"diff":\[0,"([^"\\]*)"\]'
-        matches = re.findall(pattern, chunk)
-        if matches:
-            extracted_text = ''.join(matches)
-            # Fix escaped newlines
-            extracted_text = extracted_text.replace('\\n', '\n').replace('\\n\\n', '\n\n')
+        
+        # Primary pattern: Hex key diff format (most reliable for streaming)
+        # Matches: 16:{"diff":[0,"AI"],"next":"$@18"}
+        primary_pattern = r'[0-9a-f]+:\{"diff":\[0,"([^"]*?)"\]'
+        primary_matches = re.findall(primary_pattern, chunk)
+        
+        if primary_matches:
+            # Join the matches and clean up
+            extracted_text = ''.join(primary_matches)
+            
+            # Handle escape sequences properly
+            extracted_text = extracted_text.replace('\\n', '\n')
+            extracted_text = extracted_text.replace('\\r', '\r')
+            extracted_text = extracted_text.replace('\\"', '"')
+            extracted_text = extracted_text.replace('\\t', '\t')
+            extracted_text = extracted_text.replace('\\/', '/')
+            extracted_text = extracted_text.replace('\\\\', '\\')
+            
+            # Clean up markdown formatting
+            extracted_text = extracted_text.replace('\\*', '*')
+            extracted_text = extracted_text.replace('\\#', '#')
+            extracted_text = extracted_text.replace('\\[', '[')
+            extracted_text = extracted_text.replace('\\]', ']')
+            extracted_text = extracted_text.replace('\\(', '(')
+            extracted_text = extracted_text.replace('\\)', ')')
+            
             return extracted_text if extracted_text.strip() else None
+        
+        # # Fallback: Look for Ta24 content blocks (complete responses)
+        # if ':Ta24,' in chunk:
+        #     ta24_pattern = r':Ta24,([^}]*?)(?:\d+:|$)'
+        #     ta24_matches = re.findall(ta24_pattern, chunk)
+        #     if ta24_matches:
+        #         extracted_text = ''.join(ta24_matches)
+        #         # Basic cleanup
+        #         extracted_text = extracted_text.replace('\\n', '\n')
+        #         extracted_text = extracted_text.replace('\\"', '"')
+        #         return extracted_text.strip() if extracted_text.strip() else None
+        
+        # # Secondary fallback: Direct diff patterns without hex prefix
+        # fallback_pattern = r'\{"diff":\[0,"([^"]*?)"\]'
+        # fallback_matches = re.findall(fallback_pattern, chunk)
+        # if fallback_matches:
+        #     extracted_text = ''.join(fallback_matches)
+        #     extracted_text = extracted_text.replace('\\n', '\n')
+        #     extracted_text = extracted_text.replace('\\"', '"')
+        #     return extracted_text if extracted_text.strip() else None
+        
         return None
 
     def search(self, prompt: str, stream: bool = False, raw: bool = False) -> Union[SearchResponse, Generator[Union[Dict[str, str], SearchResponse, str], None, None]]:

webscout/Provider/Deepinfra.py

@@ -17,62 +17,72 @@ class DeepInfra(Provider):
     """
 
     AVAILABLE_MODELS = [
-        # "anthropic/claude-3-7-sonnet-latest",  # >>>> NOT WORKING
-        "deepseek-ai/DeepSeek-R1-0528",
-        "deepseek-ai/DeepSeek-R1",
-        "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
-        "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
-        "deepseek-ai/DeepSeek-R1-Turbo",
-        "deepseek-ai/DeepSeek-V3",
+        "anthropic/claude-4-opus",
+        "moonshotai/Kimi-K2-Instruct",
+        "anthropic/claude-4-sonnet",
+        "deepseek-ai/DeepSeek-R1-0528-Turbo",
+        "Qwen/Qwen3-235B-A22B-Thinking-2507",
+        "Qwen/Qwen3-Coder-480B-A35B-Instruct",
+        "Qwen/Qwen3-Coder-480B-A35B-Instruct-Turbo",
+        "Qwen/Qwen3-235B-A22B-Instruct-2507",
+        "Qwen/Qwen3-235B-A22B",
+        "Qwen/Qwen3-30B-A3B",
+        "Qwen/Qwen3-32B",
+        "Qwen/Qwen3-14B",
+        "deepseek-ai/DeepSeek-V3-0324-Turbo",
         "deepseek-ai/DeepSeek-Prover-V2-671B",
-        "google/gemma-2-27b-it",
-        "google/gemma-2-9b-it",
-        "google/gemma-3-12b-it",
+        "meta-llama/Llama-4-Maverick-17B-128E-Instruct-Turbo",
+        "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
+        "meta-llama/Llama-4-Scout-17B-16E-Instruct",
+        "deepseek-ai/DeepSeek-R1-0528",
+        "deepseek-ai/DeepSeek-V3-0324",
+        "mistralai/Mistral-Small-3.1-24B-Instruct-2503",
+        "microsoft/phi-4-reasoning-plus",
+        "Qwen/QwQ-32B",
+        "google/gemini-2.5-flash",
+        "google/gemini-2.5-pro",
         "google/gemma-3-27b-it",
+        "google/gemma-3-12b-it",
         "google/gemma-3-4b-it",
-        "meta-llama/Llama-3.3-70B-Instruct",
+        "microsoft/Phi-4-multimodal-instruct",
+        "deepseek-ai/DeepSeek-R1-Distill-Llama-70B",
+        "deepseek-ai/DeepSeek-V3",
         "meta-llama/Llama-3.3-70B-Instruct-Turbo",
-        "meta-llama/Llama-4-Maverick-17B-128E-Instruct-FP8",
-        "meta-llama/Llama-4-Scout-17B-16E-Instruct",
-        "meta-llama/Llama-Guard-4-12B",
+        "meta-llama/Llama-3.3-70B-Instruct",
+        "microsoft/phi-4",
+        "Gryphe/MythoMax-L2-13b",
+        "NousResearch/Hermes-3-Llama-3.1-405B",
+        "NousResearch/Hermes-3-Llama-3.1-70B",
+        "NovaSky-AI/Sky-T1-32B-Preview",
+        "Qwen/Qwen2.5-72B-Instruct",
+        "Qwen/Qwen2.5-7B-Instruct",
+        "Qwen/Qwen2.5-Coder-32B-Instruct",
+        "Sao10K/L3-8B-Lunaris-v1-Turbo",
+        "Sao10K/L3.1-70B-Euryale-v2.2",
+        "Sao10K/L3.3-70B-Euryale-v2.3",
+        "anthropic/claude-3-7-sonnet-latest",
+        "deepseek-ai/DeepSeek-R1",
+        "deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
+        "deepseek-ai/DeepSeek-R1-Turbo",
+        "google/gemini-2.0-flash-001",
+        "meta-llama/Llama-3.2-11B-Vision-Instruct",
+        "meta-llama/Llama-3.2-1B-Instruct",
+        "meta-llama/Llama-3.2-3B-Instruct",
+        "meta-llama/Llama-3.2-90B-Vision-Instruct",
+        "meta-llama/Meta-Llama-3-70B-Instruct",
+        "meta-llama/Meta-Llama-3-8B-Instruct",
+        "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",
         "meta-llama/Meta-Llama-3.1-8B-Instruct",
         "meta-llama/Meta-Llama-3.1-8B-Instruct-Turbo",
-        "microsoft/Phi-4-multimodal-instruct",
         "microsoft/WizardLM-2-8x22B",
-        "microsoft/phi-4",
-        "microsoft/phi-4-reasoning-plus",
+        "mistralai/Devstral-Small-2505",
+        "mistralai/Devstral-Small-2507",
+        "mistralai/Mistral-7B-Instruct-v0.3",
+        "mistralai/Mistral-Nemo-Instruct-2407",
         "mistralai/Mistral-Small-24B-Instruct-2501",
+        "mistralai/Mistral-Small-3.2-24B-Instruct-2506",
+        "mistralai/Mixtral-8x7B-Instruct-v0.1",
         "nvidia/Llama-3.1-Nemotron-70B-Instruct",
-        "Qwen/QwQ-32B",
-        "Qwen/Qwen2.5-72B-Instruct",
-        "Qwen/Qwen2.5-Coder-32B-Instruct",
-        "Qwen/Qwen3-14B",
-        "Qwen/Qwen3-30B-A3B",
-        "Qwen/Qwen3-32B",
-        "Qwen/Qwen3-235B-A22B",
-        # "google/gemini-1.5-flash",  # >>>> NOT WORKING
-        # "google/gemini-1.5-flash-8b",  # >>>> NOT WORKING
-        # "google/gemini-2.0-flash-001",  # >>>> NOT WORKING
-
-        # "Gryphe/MythoMax-L2-13b",  # >>>> NOT WORKING
-
-        # "meta-llama/Llama-3.2-1B-Instruct",  # >>>> NOT WORKING
-        # "meta-llama/Llama-3.2-3B-Instruct",  # >>>> NOT WORKING
-        # "meta-llama/Llama-3.2-90B-Vision-Instruct",  # >>>> NOT WORKING
-        # "meta-llama/Llama-3.2-11B-Vision-Instruct",  # >>>> NOT WORKING
-        # "meta-llama/Meta-Llama-3-70B-Instruct",  # >>>> NOT WORKING
-        # "meta-llama/Meta-Llama-3-8B-Instruct",  # >>>> NOT WORKING
-        # "meta-llama/Meta-Llama-3.1-70B-Instruct",  # >>>> NOT WORKING
-        # "meta-llama/Meta-Llama-3.1-70B-Instruct-Turbo",  # >>>> NOT WORKING
-        # "meta-llama/Meta-Llama-3.1-405B-Instruct",  # >>>> NOT WORKING
-        # "mistralai/Mixtral-8x7B-Instruct-v0.1",  # >>>> NOT WORKING
-        # "mistralai/Mistral-7B-Instruct-v0.3",  # >>>> NOT WORKING
-        # "mistralai/Mistral-Nemo-Instruct-2407",  # >>>> NOT WORKING
-        # "NousResearch/Hermes-3-Llama-3.1-405B",  # >>>> NOT WORKING
-        # "NovaSky-AI/Sky-T1-32B-Preview",  # >>>> NOT WORKING
-        # "Qwen/Qwen2.5-7B-Instruct",  # >>>> NOT WORKING
-        # "Sao10K/L3.1-70B-Euryale-v2.2",  # >>>> NOT WORKING
-        # "Sao10K/L3.3-70B-Euryale-v2.3",  # >>>> NOT WORKING
     ]
 
     @staticmethod
@@ -84,6 +94,7 @@ class DeepInfra(Provider):
 
     def __init__(
         self,
+        api_key: Optional[str] = None,
         is_conversation: bool = True,
         max_tokens: int = 2049,
         timeout: int = 30,
@@ -107,21 +118,34 @@ class DeepInfra(Provider):
         self.agent = LitAgent()
         # Fingerprint generation might be less relevant with impersonate
         self.fingerprint = self.agent.generate_fingerprint(browser)
-
+        self.api = api_key
         # Use the fingerprint for headers (keep relevant ones)
         self.headers = {
-            "Accept": self.fingerprint["accept"], # Keep Accept
-            "Accept-Language": self.fingerprint["accept_language"], # Keep Accept-Language
+            "Accept": self.fingerprint["accept"],
+            "Accept-Language": self.fingerprint["accept_language"],
             "Content-Type": "application/json",
-            "Cache-Control": "no-cache", # Keep Cache-Control
-            "Origin": "https://deepinfra.com", # Keep Origin
-            "Pragma": "no-cache", # Keep Pragma
-            "Referer": "https://deepinfra.com/", # Keep Referer
-            "Sec-Fetch-Dest": "empty", # Keep Sec-Fetch-*
+            "Cache-Control": "no-cache",
+            "Origin": "https://deepinfra.com",
+            "Pragma": "no-cache",
+            "Referer": "https://deepinfra.com/",
+            "Sec-Fetch-Dest": "empty",
             "Sec-Fetch-Mode": "cors",
             "Sec-Fetch-Site": "same-site",
-            "X-Deepinfra-Source": "web-embed", # Keep custom headers
+            "X-Deepinfra-Source": "web-embed",
+            # Additional headers from LitAgent.generate_fingerprint
+            "User-Agent": self.fingerprint.get("user_agent", ""),
+            "Sec-CH-UA": self.fingerprint.get("sec_ch_ua", ""),
+            "Sec-CH-UA-Mobile": "?0",
+            "Sec-CH-UA-Platform": f'"{self.fingerprint.get("platform", "")}"',
+            "X-Forwarded-For": self.fingerprint.get("x-forwarded-for", ""),
+            "X-Real-IP": self.fingerprint.get("x-real-ip", ""),
+            "X-Client-IP": self.fingerprint.get("x-client-ip", ""),
+            "Forwarded": self.fingerprint.get("forwarded", ""),
+            "X-Forwarded-Proto": self.fingerprint.get("x-forwarded-proto", ""),
+            "X-Request-Id": self.fingerprint.get("x-request-id", ""),
         }
+        if self.api is not None:
+            self.headers["Authorization"] = f"Bearer {self.api}"
 
         # Initialize curl_cffi Session
         self.session = Session()
@@ -321,7 +345,7 @@ if __name__ == "__main__":
 
     for model in DeepInfra.AVAILABLE_MODELS:
         try:
-            test_ai = DeepInfra(model=model, timeout=60)
+            test_ai = DeepInfra(model=model, timeout=60,)
             response = test_ai.chat("Say 'Hello' in one word", stream=True)
             response_text = ""
             for chunk in response:

webscout/Provider/ExaChat.py

@@ -21,9 +21,9 @@ MODEL_CONFIGS = {
             "gemini-2.0-flash",
             "gemini-2.0-flash-exp-image-generation",
             "gemini-2.0-flash-thinking-exp-01-21",
-            "gemini-2.5-pro-exp-03-25",
+            "gemini-2.5-flash-lite-preview-06-17",
             "gemini-2.0-pro-exp-02-05",
-            "gemini-2.5-flash-preview-04-17",
+            "gemini-2.5-flash",
 
         
         ],
@@ -62,7 +62,9 @@ MODEL_CONFIGS = {
         "endpoint": "https://ayle.chat/api/cerebras",
         "models": [
             "llama3.1-8b",
-            "llama-3.3-70b"
+            "llama-3.3-70b",
+            "llama-4-scout-17b-16e-instruct",
+            "qwen-3-32b"
         ],
     },
     "xai": {
@@ -88,9 +90,9 @@ class ExaChat(Provider):
         "gemini-2.0-flash",
         "gemini-2.0-flash-exp-image-generation",
         "gemini-2.0-flash-thinking-exp-01-21",
-        "gemini-2.5-pro-exp-03-25",
         "gemini-2.0-pro-exp-02-05",
-        "gemini-2.5-flash-preview-04-17",
+        "gemini-2.5-flash",
+        "gemini-2.5-flash-lite-preview-06-17",
         
         # OpenRouter Models
         "mistralai/mistral-small-3.1-24b-instruct:free",
@@ -120,6 +122,8 @@ class ExaChat(Provider):
         # Cerebras Models
         "llama3.1-8b",
         "llama-3.3-70b",
+        "llama-4-scout-17b-16e-instruct",
+        "qwen-3-32b",
 
     ]
 

webscout/Provider/Flowith.py

@@ -15,7 +15,7 @@ class Flowith(Provider):
     """
     A provider class for interacting with the Flowith API.
     """
-    AVAILABLE_MODELS = ["gpt-4.1-mini", "deepseek-chat", "deepseek-reasoner", "claude-3.5-haiku", "gemini-2.0-flash", "gemini-2.5-flash", "grok-3-mini"]
+    AVAILABLE_MODELS = ["gpt-4.1-nano", "gpt-4.1-mini", "deepseek-chat", "deepseek-reasoner", "claude-3.5-haiku", "gemini-2.0-flash", "gemini-2.5-flash", "grok-3-mini"]
 
     def __init__(
         self,

webscout/Provider/FreeGemini.py

@@ -83,7 +83,7 @@ class FreeGemini(Provider):
         self.last_response = {}
         self.system_prompt = system_prompt # Stored for consistency
 
-        self.api_endpoint = "https://free-gemini.vercel.app/api/google/v1beta/models/gemini-2.0-flash:streamGenerateContent?alt=sse"
+        self.api_endpoint = "https://free-gemini.vercel.app/api/google/v1beta/models/gemini-2.5-flash:streamGenerateContent?alt=sse"
 
         self.agent = LitAgent()
         self.headers = {
@@ -246,5 +246,5 @@ class FreeGemini(Provider):
 if __name__ == "__main__":
     # Example usage
     free_gemini = FreeGemini()
-    response = free_gemini.chat("What is the capital of France?", stream=False)
+    response = free_gemini.chat("how many r in strawberry", stream=False)
     print(response)  # Should print the response from the API

webscout/Provider/Gemini.py

@@ -10,22 +10,15 @@ from ..Bard import Chatbot, Model
 
 warnings.simplefilter("ignore", category=UserWarning)
 
-# Define model aliases for easy usage
+# Define model aliases for easy usage (only supported models)
 MODEL_ALIASES: Dict[str, Model] = {
     "unspecified": Model.UNSPECIFIED,
-    "gemini-2.0-flash": Model.G_2_0_FLASH,
-    "gemini-2.0-flash-thinking": Model.G_2_0_FLASH_THINKING,
-    "gemini-2.5-pro": Model.G_2_5_PRO,
-    "gemini-2.0-exp-advanced": Model.G_2_0_EXP_ADVANCED,
-    "gemini-2.5-exp-advanced": Model.G_2_5_EXP_ADVANCED,
     "gemini-2.5-flash": Model.G_2_5_FLASH,
+    "gemini-2.5-pro": Model.G_2_5_PRO,
     # Add shorter aliases for convenience
-    "flash": Model.G_2_0_FLASH,
     "flash-2.5": Model.G_2_5_FLASH,
-    "thinking": Model.G_2_0_FLASH_THINKING,
     "pro": Model.G_2_5_PRO,
-    "advanced": Model.G_2_0_EXP_ADVANCED,
-    "advanced-2.5": Model.G_2_5_EXP_ADVANCED,
+    "unspecified": Model.UNSPECIFIED,
 }
 
 # List of available models (friendly names)

webscout/Provider/GeminiProxy.py

@@ -14,10 +14,11 @@ class GeminiProxy(Provider):
     AVAILABLE_MODELS = [
         "gemini-2.0-flash-lite",
         "gemini-2.0-flash",
-        "gemini-2.5-pro-preview-06-05",
-        "gemini-2.5-pro-preview-05-06",
         "gemini-2.5-flash-preview-04-17",
         "gemini-2.5-flash-preview-05-20",
+        "gemini-2.5-flash-lite-preview-06-17",
+        "gemini-2.5-pro",
+        "gemini-2.5-flash",
 
     ]
 
@@ -135,6 +136,31 @@ class GeminiProxy(Provider):
             return str(response)
 
 if __name__ == "__main__":
-    ai = GeminiProxy(timeout=30, model="gemini-2.5-flash-preview-05-20")
-    response = ai.chat("write a poem about AI")
-    print(response)
+    # Ensure curl_cffi is installed
+    print("-" * 80)
+    print(f"{'Model':<50} {'Status':<10} {'Response'}")
+    print("-" * 80)
+    
+    # Test all available models
+    working = 0
+    total = len(GeminiProxy.AVAILABLE_MODELS)
+    
+    for model in GeminiProxy.AVAILABLE_MODELS:
+        try:
+            test_ai = GeminiProxy(model=model, timeout=60)
+            response = test_ai.chat("Say 'Hello' in one word", stream=True)
+            response_text = ""
+            for chunk in response:
+                response_text += chunk
+                print(f"\r{model:<50} {'Testing...':<10}", end="", flush=True)
+            
+            if response_text and len(response_text.strip()) > 0:
+                status = "✓"
+                # Truncate response if too long
+                display_text = response_text.strip()[:50] + "..." if len(response_text.strip()) > 50 else response_text.strip()
+            else:
+                status = "✗"
+                display_text = "Empty or invalid response"
+            print(f"\r{model:<50} {status:<10} {display_text}")
+        except Exception as e:
+            print(f"\r{model:<50} {'✗':<10} {str(e)}")