webscout 7.5__py3-none-any.whl → 7.6__py3-none-any.whl

webscout/Provider/Netwrck.py

@@ -16,22 +16,22 @@ class Netwrck(Provider):
     """
     greeting = """Hello! I'm a helpful assistant. How can I help you today?"""
 
-    AVAILABLE_MODELS = {
-        "lumimaid": "neversleep/llama-3-lumimaid-8b:extended",
-        "grok": "x-ai/grok-2",
-        "claude": "anthropic/claude-3-7-sonnet-20250219",
-        "euryale": "sao10k/l3-euryale-70b",
-        "gpt4mini": "openai/gpt-4o-mini",
-        "mythomax": "gryphe/mythomax-l2-13b",
-        "gemini": "google/gemini-pro-1.5",
-        "nemotron": "nvidia/llama-3.1-nemotron-70b-instruct",
-        "deepseek-r1": "deepseek/deepseek-r1",
-        "deepseek": "deepseek/deepseek-chat",
-    }
+    AVAILABLE_MODELS = [
+        "neversleep/llama-3-lumimaid-8b:extended",
+        "x-ai/grok-2",
+        "anthropic/claude-3-7-sonnet-20250219",
+        "sao10k/l3-euryale-70b",
+        "openai/gpt-4o-mini",
+        "gryphe/mythomax-l2-13b",
+        "google/gemini-pro-1.5",
+        "nvidia/llama-3.1-nemotron-70b-instruct",
+        "deepseek-r1",
+        "deepseek",
+    ]
 
     def __init__(
         self,
-        model: str = "claude",
+        model: str = "anthropic/claude-3-7-sonnet-20250219",
         is_conversation: bool = True,
         max_tokens: int = 4096,
         timeout: int = 30,
@@ -47,10 +47,10 @@ class Netwrck(Provider):
     ):
         """Initializes the Netwrck API client."""
         if model not in self.AVAILABLE_MODELS:
-            raise ValueError(f"Invalid model: {model}. Choose from: {list(self.AVAILABLE_MODELS.keys())}")
+            raise ValueError(f"Invalid model: {model}. Choose from: {self.AVAILABLE_MODELS}")
 
         self.model = model
-        self.model_name = self.AVAILABLE_MODELS[model]
+        self.model_name = model  # Use the model name directly since it's already in the correct format
         self.system_prompt = system_prompt
         self.session = requests.Session()
         self.is_conversation = is_conversation
@@ -198,7 +198,30 @@ class Netwrck(Provider):
         return response["text"].replace('\\n', '\n').replace('\\n\\n', '\n\n')
 
 if __name__ == "__main__":
-    from rich import print
-
-    netwrck = Netwrck(model="claude")
-    print(netwrck.chat("Hello! How are you?"))
+    print("-" * 80)
+    print(f"{'Model':<50} {'Status':<10} {'Response'}")
+    print("-" * 80)
+    
+    # Test all available models
+    working = 0
+    total = len(Netwrck.AVAILABLE_MODELS)
+    
+    for model in Netwrck.AVAILABLE_MODELS:
+        try:
+            test_ai = Netwrck(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)}")

webscout/Provider/PI.py

@@ -18,8 +18,9 @@ class PiAI(Provider):
     Attributes:
         knowledge_cutoff (str): The knowledge cutoff date for the model
         AVAILABLE_VOICES (Dict[str, int]): Available voice options for audio responses
+        AVAILABLE_MODELS (List[str]): Available model options for the API
     """
-
+    AVAILABLE_MODELS = ["inflection_3_pi"]
     AVAILABLE_VOICES: Dict[str, int] = {
         "voice1": 1,
         "voice2": 2,
@@ -44,7 +45,8 @@ class PiAI(Provider):
         act: str = None,
         voice: bool = False,
         voice_name: str = "voice3",
-        output_file: str = "PiAI.mp3"
+        output_file: str = "PiAI.mp3",
+        model: str = "inflection_3_pi",
     ):
         """
         Initializes PiAI with voice support.

webscout/Provider/Perplexitylabs.py

@@ -386,10 +386,30 @@ class PerplexityLabs(Provider):
 
 
 if __name__ == "__main__":
-    from rich import print
+    print("-" * 80)
+    print(f"{'Model':<50} {'Status':<10} {'Response'}")
+    print("-" * 80)
     
-    # Example usage
-    ai = PerplexityLabs(timeout=60, model="r1-1776")
-
-    for chunk in ai.chat("Explain the concept of neural networks", stream=True):
-        print(chunk, end="", flush=True)
+    # Test all available models
+    working = 0
+    total = len(PerplexityLabs.AVAILABLE_MODELS)
+    
+    for model in PerplexityLabs.AVAILABLE_MODELS:
+        try:
+            test_ai = PerplexityLabs(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)}")

webscout/Provider/PizzaGPT.py

@@ -6,14 +6,14 @@ from webscout.AIutel import Optimizers, Conversation, AwesomePrompts
 from webscout.AIbase import Provider
 from webscout import exceptions
 from webscout import LitAgent as Lit
-from webscout.Litlogger import Logger, LogFormat
 
 class PIZZAGPT(Provider):
     """
     PIZZAGPT is a provider class for interacting with the PizzaGPT API.
     Supports web search integration and handles responses using regex.
     """
-
+    AVAILABLE_MODELS = ["gpt-4o-mini"]
+    
     def __init__(
         self,
         is_conversation: bool = True,
@@ -25,10 +25,12 @@ class PIZZAGPT(Provider):
         proxies: dict = {},
         history_offset: int = 10250,
         act: str = None,
-        logging: bool = False,
         model: str = "gpt-4o-mini"
     ) -> None:
         """Initialize PizzaGPT with enhanced configuration options."""
+        if model not in self.AVAILABLE_MODELS:
+            raise ValueError(f"Invalid model: {model}. Choose from: {self.AVAILABLE_MODELS}")
+            
         self.session = requests.Session()
         self.is_conversation = is_conversation
         self.max_tokens_to_sample = max_tokens
@@ -45,7 +47,9 @@ class PIZZAGPT(Provider):
             "origin": "https://www.pizzagpt.it",
             "referer": "https://www.pizzagpt.it/en",
             "user-agent": Lit().random(),
-            "x-secret": "Marinara"
+            "x-secret": "Marinara",
+            "sec-ch-ua": '"Chromium";v="134", "Not:A-Brand";v="24"',
+            "sec-ch-ua-platform": '"Windows"'
         }
 
         self.__available_optimizers = (
@@ -67,28 +71,15 @@ class PIZZAGPT(Provider):
         )
         self.conversation.history_offset = history_offset
         self.session.proxies = proxies
-        
-        self.logger = Logger(
-            name="PIZZAGPT",
-            format=LogFormat.MODERN_EMOJI,
-        ) if logging else None
-
-        if self.logger:
-            self.logger.info(f"PIZZAGPT initialized with model: {self.model}")
 
     def _extract_content(self, text: str) -> Dict[str, Any]:
         """
         Extract content from response text using regex.
         """
-        if self.logger:
-            self.logger.debug("Extracting content from response text")
-            
         try:
             # Look for content pattern
             content_match = re.search(r'"content"\s*:\s*"(.*?)"(?=\s*[,}])', text, re.DOTALL)
             if not content_match:
-                if self.logger:
-                    self.logger.error("Content pattern not found in response")
                 raise exceptions.FailedToGenerateResponseError("Content not found in response")
                 
             content = content_match.group(1)
@@ -108,8 +99,6 @@ class PIZZAGPT(Provider):
             }
             
         except Exception as e:
-            if self.logger:
-                self.logger.error(f"Failed to extract content: {str(e)}")
             raise exceptions.FailedToGenerateResponseError(f"Failed to extract content: {str(e)}")
 
     def ask(
@@ -124,21 +113,13 @@ class PIZZAGPT(Provider):
         """
         Send a prompt to PizzaGPT API with optional web search capability.
         """
-        if self.logger:
-            self.logger.debug(f"Processing request - Prompt: {prompt[:50]}...")
-            self.logger.debug(f"Web search enabled: {web_search}")
-
         conversation_prompt = self.conversation.gen_complete_prompt(prompt)
         if optimizer:
             if optimizer in self.__available_optimizers:
                 conversation_prompt = getattr(Optimizers, optimizer)(
                     conversation_prompt if conversationally else prompt
                 )
-                if self.logger:
-                    self.logger.debug(f"Applied optimizer: {optimizer}")
             else:
-                if self.logger:
-                    self.logger.error(f"Invalid optimizer: {optimizer}")
                 raise Exception(f"Optimizer is not one of {self.__available_optimizers}")
 
         payload = {
@@ -147,9 +128,6 @@ class PIZZAGPT(Provider):
             "searchEnabled": web_search
         }
 
-        if self.logger:
-            self.logger.debug(f"Sending payload: {json.dumps(payload, indent=2)}")
-
         try:
             response = self.session.post(
                 self.api_endpoint,
@@ -157,26 +135,17 @@ class PIZZAGPT(Provider):
                 timeout=self.timeout
             )
             
-            if self.logger:
-                self.logger.debug(f"Response status: {response.status_code}")
-            
             if not response.ok:
-                if self.logger:
-                    self.logger.error(f"API request failed: {response.status_code} - {response.reason}")
                 raise exceptions.FailedToGenerateResponseError(
                     f"Failed to generate response - ({response.status_code}, {response.reason})"
                 )
 
             response_text = response.text
             if not response_text:
-                if self.logger:
-                    self.logger.error("Empty response received from API")
                 raise exceptions.FailedToGenerateResponseError("Empty response received from API")
 
             try:
                 resp = self._extract_content(response_text)
-                if self.logger:
-                    self.logger.debug("Response parsed successfully")
                     
                 self.last_response.update(dict(text=resp['content']))
                 self.conversation.update_chat_history(
@@ -185,14 +154,9 @@ class PIZZAGPT(Provider):
                 return self.last_response
 
             except Exception as e:
-                if self.logger:
-                    self.logger.error(f"Failed to parse response: {str(e)}")
-                    self.logger.debug(f"Raw response text: {response_text[:500]}")
                 raise exceptions.FailedToGenerateResponseError(f"Failed to parse response: {str(e)}")
 
         except requests.exceptions.RequestException as e:
-            if self.logger:
-                self.logger.error(f"Request failed: {str(e)}")
             raise exceptions.FailedToGenerateResponseError(f"Request failed: {str(e)}")
 
     def chat(
@@ -206,9 +170,6 @@ class PIZZAGPT(Provider):
         """
         Chat with PizzaGPT with optional web search capability.
         """
-        if self.logger:
-            self.logger.debug(f"Chat request initiated with web_search={web_search}")
-            
         try:
             response = self.ask(
                 prompt,
@@ -218,8 +179,6 @@ class PIZZAGPT(Provider):
             )
             return self.get_message(response)
         except Exception as e:
-            if self.logger:
-                self.logger.error(f"Chat failed: {str(e)}")
             raise
 
     def get_message(self, response: dict) -> str:
@@ -231,9 +190,9 @@ if __name__ == "__main__":
     from rich import print
     
     # Example usage with web search enabled
-    ai = PIZZAGPT(logging=True)
+    ai = PIZZAGPT()
     try:
-        response = ai.chat("Who is Founder and CEO of HelpingAI??", web_search=True)
+        response = ai.chat("hi")
         print(response)
     except Exception as e:
         print(f"Error: {str(e)}")

webscout/Provider/TTI/AiForce/async_aiforce.py

@@ -9,15 +9,8 @@ from aiohttp import ClientError
 from pathlib import Path
 
 from webscout.AIbase import AsyncImageProvider
-from webscout.Litlogger import Logger, LogFormat
 from webscout.litagent import LitAgent
 
-# Initialize our fire logger and agent 🔥
-logger = Logger(
-    "AsyncAiForce",
-    format=LogFormat.MODERN_EMOJI,
-
-)
 agent = LitAgent()
 
 class AsyncAiForceimager(AsyncImageProvider):
@@ -46,8 +39,6 @@ class AsyncAiForceimager(AsyncImageProvider):
     AVAILABLE_MODELS = [
         "stable-diffusion-xl-lightning",
         "stable-diffusion-xl-base",
-        "Flux-1.1-Pro",
-        "ideogram",
         "flux",
         "flux-realism",
         "flux-anime",
@@ -58,13 +49,12 @@ class AsyncAiForceimager(AsyncImageProvider):
         "any-dark"
     ]
 
-    def __init__(self, timeout: int = 60, proxies: dict = {}, logging: bool = True):
+    def __init__(self, timeout: int = 60, proxies: dict = {}):
         """Initialize your async AiForce provider with custom settings! ⚙️
 
         Args:
             timeout (int): Request timeout in seconds (default: 60)
             proxies (dict): Proxy settings for requests (default: {})
-            logging (bool): Enable fire logging (default: True)
         """
         self.api_endpoint = "https://api.airforce/imagine2"
         self.headers = {
@@ -77,16 +67,13 @@ class AsyncAiForceimager(AsyncImageProvider):
         self.proxies = proxies
         self.prompt: str = "AI-generated image - webscout"
         self.image_extension: str = "png"
-        self.logging = logging
-        if self.logging:
-            logger.info("AsyncAiForce provider initialized! 🚀")
 
     async def generate(
         self, 
         prompt: str, 
         amount: int = 1, 
         additives: bool = True,
-        model: str = "Flux-1.1-Pro", 
+        model: str = "flux-3d", 
         width: int = 768, 
         height: int = 768, 
         seed: Optional[int] = None,
@@ -104,7 +91,7 @@ class AsyncAiForceimager(AsyncImageProvider):
             ...     images = await provider.generate(
             ...         prompt="Epic dragon",
             ...         amount=3,
-            ...         model="Flux-1.1-Pro"
+            ...         model="flux-3d"
             ...     )
 
         Args:
@@ -143,9 +130,6 @@ class AsyncAiForceimager(AsyncImageProvider):
         self.prompt = prompt
         response = []
         
-        if self.logging:
-            logger.info(f"Generating {amount} images with {model}... 🎨")
-        
         async with aiohttp.ClientSession(headers=self.headers) as session:
             for i in range(amount):
                 url = f"{self.api_endpoint}?model={model}&prompt={prompt}&size={width}:{height}"
@@ -157,21 +141,13 @@ class AsyncAiForceimager(AsyncImageProvider):
                         async with session.get(url, timeout=self.timeout, proxy=self.proxies.get('http')) as resp:
                             resp.raise_for_status()
                             response.append(await resp.read())
-                            if self.logging:
-                                logger.success(f"Generated image {i + 1}/{amount}! 🎨")
                             break
                     except ClientError as e:
                         if attempt == max_retries - 1:
-                            if self.logging:
-                                logger.error(f"Failed to generate image after {max_retries} attempts: {e} 😢")
                             raise
                         else:
-                            if self.logging:
-                                logger.warning(f"Attempt {attempt + 1} failed. Retrying in {retry_delay} seconds... 🔄")
                             await asyncio.sleep(retry_delay)
 
-        if self.logging:
-            logger.success("Images generated successfully! 🎉")
         return response
 
     async def save(
@@ -209,16 +185,11 @@ class AsyncAiForceimager(AsyncImageProvider):
         save_dir = dir if dir else os.getcwd()
         if not os.path.exists(save_dir):
             os.makedirs(save_dir)
-            if self.logging:
-                logger.info(f"Created directory: {save_dir} 📁")
 
         name = self.prompt if name is None else name
         saved_paths = []
         timestamp = int(time.time())
         
-        if self.logging:
-            logger.info(f"Saving {len(response)} images... 💾")
-        
         async def save_single_image(image_bytes: bytes, index: int) -> str:
             filename = f"{filenames_prefix}{name}_{index}.{self.image_extension}"
             filepath = os.path.join(save_dir, filename)
@@ -228,8 +199,6 @@ class AsyncAiForceimager(AsyncImageProvider):
                 with open(filepath, "wb") as f:
                     f.write(image_bytes)
             
-            if self.logging:
-                logger.success(f"Saved image to: {filepath} 💾")
             return filepath
 
         # Handle both List[bytes] and AsyncGenerator
@@ -240,8 +209,6 @@ class AsyncAiForceimager(AsyncImageProvider):
 
         tasks = [save_single_image(img, i) for i, img in enumerate(image_list)]
         saved_paths = await asyncio.gather(*tasks)
-        if self.logging:
-            logger.success(f"Images saved successfully! Check {dir} 🎉")
         return saved_paths
 
 if __name__ == "__main__":
@@ -252,6 +219,6 @@ if __name__ == "__main__":
             paths = await bot.save(resp)
             print(paths)
         except Exception as e:
-            logger.error(f"An error occurred: {e} 😢")
+            print(f"An error occurred: {e}")
 
     asyncio.run(main())

webscout/Provider/TTI/AiForce/sync_aiforce.py

@@ -8,15 +8,10 @@ from requests.exceptions import RequestException
 from pathlib import Path
 
 from webscout.AIbase import ImageProvider
-from webscout.Litlogger import Logger, LogFormat
+
 from webscout.litagent import LitAgent
 
-# Initialize our fire logger and agent 🔥
-logger = Logger(
-    "AiForce",
-    format=LogFormat.MODERN_EMOJI,
 
-)
 agent = LitAgent()
 
 class AiForceimager(ImageProvider):
@@ -43,8 +38,6 @@ class AiForceimager(ImageProvider):
     AVAILABLE_MODELS = [
         "stable-diffusion-xl-lightning",
         "stable-diffusion-xl-base",
-        "Flux-1.1-Pro",
-        "ideogram",
         "flux",
         "flux-realism",
         "flux-anime",
@@ -55,13 +48,12 @@ class AiForceimager(ImageProvider):
         "any-dark"
     ]
 
-    def __init__(self, timeout: int = 60, proxies: dict = {}, logging: bool = True):
+    def __init__(self, timeout: int = 60, proxies: dict = {}):
         """Initialize your AiForce provider with custom settings! ⚙️
 
         Args:
             timeout (int): Request timeout in seconds (default: 60)
             proxies (dict): Proxy settings for requests (default: {})
-            logging (bool): Enable fire logging (default: True)
         """
         self.api_endpoint = "https://api.airforce/imagine2"
         self.headers = {
@@ -76,16 +68,14 @@ class AiForceimager(ImageProvider):
         self.timeout = timeout
         self.prompt: str = "AI-generated image - webscout"
         self.image_extension: str = "png"
-        self.logging = logging
-        if self.logging:
-            logger.info("AiForce provider initialized! 🚀")
+
 
     def generate(
         self, 
         prompt: str, 
         amount: int = 1, 
         additives: bool = True,
-        model: str = "Flux-1.1-Pro", 
+        model: str = "flux-3d", 
         width: int = 768, 
         height: int = 768, 
         seed: Optional[int] = None,
@@ -139,8 +129,6 @@ class AiForceimager(ImageProvider):
         )
 
         self.prompt = prompt
-        if self.logging:
-            logger.info(f"Generating {amount} images with {model}... 🎨")
         response = []
         for _ in range(amount):
             url = f"{self.api_endpoint}?model={model}&prompt={prompt}&size={width}:{height}"
@@ -152,21 +140,13 @@ class AiForceimager(ImageProvider):
                     resp = self.session.get(url, timeout=self.timeout)
                     resp.raise_for_status()
                     response.append(resp.content)
-                    if self.logging:
-                        logger.success(f"Generated image {_ + 1}/{amount}! 🎨")
                     break
                 except RequestException as e:
                     if attempt == max_retries - 1:
-                        if self.logging:
-                            logger.error(f"Failed to generate image after {max_retries} attempts: {e} 😢")
                         raise
                     else:
-                        if self.logging:
-                            logger.warning(f"Attempt {attempt + 1} failed. Retrying in {retry_delay} seconds... 🔄")
                         time.sleep(retry_delay)
 
-        if self.logging:
-            logger.success("Images generated successfully! 🎉")
         return response
 
     def save(
@@ -203,12 +183,8 @@ class AiForceimager(ImageProvider):
         save_dir = dir if dir else os.getcwd()
         if not os.path.exists(save_dir):
             os.makedirs(save_dir)
-            if self.logging:
-                logger.info(f"Created directory: {save_dir} 📁")
 
         name = self.prompt if name is None else name
-        if self.logging:
-            logger.info(f"Saving {len(response)} images... 💾")
         filenames = []
         count = 0
 
@@ -225,18 +201,45 @@ class AiForceimager(ImageProvider):
 
             with open(absolute_path_to_file, "wb") as fh:
                 fh.write(image)
-            if self.logging:
-                logger.success(f"Saved image to: {absolute_path_to_file} 💾")
 
-        if self.logging:
-            logger.success(f"Images saved successfully! Check {dir} 🎉")
         return filenames
 
 if __name__ == "__main__":
     bot = AiForceimager()
-    try:
-        resp = bot.generate("A shiny red sports car speeding down a scenic mountain road", 1)
-        print(bot.save(resp))
-    except Exception as e:
-        if bot.logging:
-            logger.error(f"An error occurred: {e} 😢")
+    test_prompt = "A shiny red sports car speeding down a scenic mountain road"
+    
+    print(f"Testing all available models with prompt: '{test_prompt}'")
+    print("-" * 50)
+    
+    # Create a directory for test images if it doesn't exist
+    test_dir = "model_test_images"
+    if not os.path.exists(test_dir):
+        os.makedirs(test_dir)
+        
+    for model in bot.AVAILABLE_MODELS:
+        print(f"Testing model: {model}")
+        try:
+            # Generate an image with the current model
+            resp = bot.generate(
+                prompt=test_prompt,
+                amount=1,
+                model=model,
+                width=768,
+                height=768
+            )
+            
+            # Save the image with model name as prefix
+            saved_paths = bot.save(
+                resp, 
+                name=f"{model}_test", 
+                dir=test_dir, 
+                filenames_prefix=f"{model}_"
+            )
+            
+            print(f"✓ Success! Saved image: {saved_paths[0]}")
+        except Exception as e:
+            print(f"✗ Failed with model {model}: {str(e)}")
+        
+        print("-" * 30)
+    
+    print("All model tests completed!")

webscout/Provider/TTI/FreeAIPlayground/__init__.py

@@ -1,9 +1,9 @@
-"""
-FreeAI Provider Package
-Provides access to various AI models for image generation including DALL-E 3 and Flux models
-"""
-
-from .sync_freeaiplayground import FreeAIImager
-from .async_freeaiplayground import AsyncFreeAIImager
-
-__all__ = ['FreeAIImager', 'AsyncFreeAIImager']
+"""
+FreeAI Provider Package
+Provides access to various AI models for image generation including DALL-E 3 and Flux models
+"""
+
+from .sync_freeaiplayground import FreeAIImager
+from .async_freeaiplayground import AsyncFreeAIImager
+
+__all__ = ['FreeAIImager', 'AsyncFreeAIImager']