webscout 7.7__py3-none-any.whl → 7.8__py3-none-any.whl

webscout/Provider/OLLAMA.py

@@ -79,7 +79,7 @@ class OLLAMA(Provider):
         conversationally: bool = False,
         tools: Optional[List[Dict]] = None,
         images: Optional[List[str]] = None,
-    ) -> dict | AsyncGenerator:
+    ) -> Union[dict, AsyncGenerator]:
         """Chat with AI
 
         Args:
@@ -91,7 +91,7 @@ class OLLAMA(Provider):
             tools (List[Dict], optional): List of tools/functions to use. Defaults to None.
             images (List[str], optional): List of image paths or base64 encoded images. Defaults to None.
         Returns:
-           dict|AsyncGenerator : ai content
+           Union[dict, AsyncGenerator] : ai content
         """
         conversation_prompt = self.conversation.gen_complete_prompt(prompt)
         if optimizer:
@@ -150,7 +150,7 @@ class OLLAMA(Provider):
         conversationally: bool = False,
         tools: Optional[List[Dict]] = None,
         images: Optional[List[str]] = None,
-    ) -> dict | AsyncGenerator:
+    ) -> Union[dict, AsyncGenerator]:
         """Async version of ask method"""
         conversation_prompt = self.conversation.gen_complete_prompt(prompt)
         if optimizer:
@@ -208,7 +208,7 @@ class OLLAMA(Provider):
         conversationally: bool = False,
         tools: Optional[List[Dict]] = None,
         images: Optional[List[str]] = None,
-    ) -> str | AsyncGenerator:
+    ) -> Union[str, AsyncGenerator]:
         """Generate response `str`
         Args:
             prompt (str): Prompt to be send.
@@ -218,7 +218,7 @@ class OLLAMA(Provider):
             tools (List[Dict], optional): List of tools/functions to use. Defaults to None.
             images (List[str], optional): List of image paths or base64 encoded images. Defaults to None.
         Returns:
-            str: Response generated
+            Union[str, AsyncGenerator]: Response generated
         """
         def for_stream():
             for response in self.ask(
@@ -249,7 +249,7 @@ class OLLAMA(Provider):
         conversationally: bool = False,
         tools: Optional[List[Dict]] = None,
         images: Optional[List[str]] = None,
-    ) -> str | AsyncGenerator:
+    ) -> Union[str, AsyncGenerator]:
         """Async version of chat method"""
         async def for_stream():
             async for response in await self.aask(
@@ -289,7 +289,7 @@ class OLLAMA(Provider):
         prompt: str,
         stream: bool = False,
         **kwargs
-    ) -> dict | AsyncGenerator:
+    ) -> Union[dict, AsyncGenerator]:
         """Generate text using the model"""
         try:
             if stream:
@@ -305,7 +305,7 @@ class OLLAMA(Provider):
         prompt: str,
         stream: bool = False,
         **kwargs
-    ) -> dict | AsyncGenerator:
+    ) -> Union[dict, AsyncGenerator]:
         """Async version of generate method"""
         try:
             if stream:

webscout/Provider/Openai.py

@@ -5,7 +5,7 @@ from ..AIutel import AwesomePrompts, sanitize_stream
 from ..AIbase import  Provider, AsyncProvider
 
 from webscout import exceptions
-from typing import Any, AsyncGenerator, Dict
+from typing import Any, AsyncGenerator, Dict, List, Optional, Union
 import requests
 import httpx
 #----------------------------------------------------------OpenAI-----------------------------------
@@ -332,7 +332,8 @@ class AsyncOPENAI(AsyncProvider):
         raw: bool = False,
         optimizer: str = None,
         conversationally: bool = False,
-    ) -> dict | AsyncGenerator:
+        tools: Optional[List[Dict[str, Any]]] = None,
+    ) -> Union[dict, AsyncGenerator]:
         """Chat with AI asynchronously.
 
         Args:
@@ -341,6 +342,7 @@ class AsyncOPENAI(AsyncProvider):
             raw (bool, optional): Stream back raw response as received. Defaults to False.
             optimizer (str, optional): Prompt optimizer name - `[code, shell_command]`. Defaults to None.
             conversationally (bool, optional): Chat conversationally when using optimizer. Defaults to False.
+            tools (Optional[List[Dict[str, Any]]], optional): List of tools to be used. Defaults to None.
         Returns:
            dict|AsyncGenerator : ai content.
         ```json
@@ -444,13 +446,15 @@ class AsyncOPENAI(AsyncProvider):
         stream: bool = False,
         optimizer: str = None,
         conversationally: bool = False,
-    ) -> str | AsyncGenerator:
+        tools: Optional[List[Dict[str, Any]]] = None,
+    ) -> Union[str, AsyncGenerator]:
         """Generate response `str` asynchronously.
         Args:
             prompt (str): Prompt to be send.
             stream (bool, optional): Flag for streaming response. Defaults to False.
             optimizer (str, optional): Prompt optimizer name - `[code, shell_command]`. Defaults to None.
             conversationally (bool, optional): Chat conversationally when using optimizer. Defaults to False.
+            tools (Optional[List[Dict[str, Any]]], optional): List of tools to be used. Defaults to None.
         Returns:
             str|AsyncGenerator: Response generated
         """

webscout/Provider/PI.py

@@ -9,7 +9,7 @@ from webscout.AIutel import Conversation
 from webscout.AIutel import AwesomePrompts
 from webscout.AIbase import Provider
 from typing import Dict, Union, Any, Optional
-from webscout import LitAgent
+from webscout.litagent import LitAgent
 
 class PiAI(Provider):
     """

webscout/Provider/Perplexitylabs.py

@@ -13,7 +13,7 @@ from webscout.AIutel import Conversation
 from webscout.AIutel import AwesomePrompts, sanitize_stream
 from webscout.AIbase import Provider
 from webscout import exceptions
-from webscout import LitAgent
+from webscout.litagent import LitAgent
 
 class PerplexityLabs(Provider):
     """

webscout/Provider/Phind.py

@@ -8,7 +8,7 @@ from webscout.AIutel import AwesomePrompts, sanitize_stream
 from webscout.AIbase import  Provider
 
 from webscout import exceptions
-from typing import Any, AsyncGenerator, Dict
+from typing import Union, Any, AsyncGenerator, Dict
 
 
 #------------------------------------------------------phind-------------------------------------------------------------

webscout/Provider/PizzaGPT.py

@@ -5,7 +5,7 @@ from typing import Any, Dict, Optional, Union, Generator
 from webscout.AIutel import Optimizers, Conversation, AwesomePrompts
 from webscout.AIbase import Provider
 from webscout import exceptions
-from webscout import LitAgent as Lit
+from webscout.litagent import LitAgent as Lit
 
 class PIZZAGPT(Provider):
     """

webscout/Provider/QwenLM.py

@@ -1,13 +1,10 @@
-import requests
 import json
-from typing import Any, Dict, Generator, Optional
-import uuid
-import re
+from typing import Union, Any, Dict, Generator, Optional
 
 import cloudscraper
 
 from webscout.AIutel import Optimizers, Conversation, AwesomePrompts
-from webscout.AIbase import Provider, AsyncProvider
+from webscout.AIbase import Provider
 from webscout import exceptions
 
 class QwenLM(Provider):
@@ -126,7 +123,7 @@ class QwenLM(Provider):
         raw: bool = False,
         optimizer: Optional[str] = None,
         conversationally: bool = False,
-    ) -> Dict[str, Any] | Generator[Dict[str, Any], None, None]:
+    ) -> Union[Dict[str, Any], Generator[Any, None, None]]:
         """Chat with AI."""
 
         conversation_prompt = self.conversation.gen_complete_prompt(prompt)
@@ -229,7 +226,7 @@ class QwenLM(Provider):
         stream: bool = False,
         optimizer: Optional[str] = None,
         conversationally: bool = False,
-    ) -> str | Generator[str, None, None]:
+    ) -> Union[str, Generator[str, None, None]]:
         """Generate response string from chat."""
 
         def for_stream() -> Generator[str, None, None]:

webscout/Provider/TTI/FreeAIPlayground/async_freeaiplayground.py

@@ -21,7 +21,9 @@ class AsyncFreeAIImager(AsyncImageProvider):
         "Flux Pro Ultra Raw",
         "Flux Schnell",
         "Flux Realism",
-        "grok-2-aurora"
+        "grok-2-aurora",
+        "Flux Dev",
+
     ]
 
     def __init__(

webscout/Provider/TTI/FreeAIPlayground/sync_freeaiplayground.py

@@ -1,6 +1,6 @@
 import requests
 import os
-from typing import List
+from typing import Union, List
 from string import punctuation
 from random import choice
 from random import randint
@@ -21,7 +21,8 @@ class FreeAIImager(ImageProvider):
         "Flux Pro Ultra Raw",
         "Flux Schnell",
         "Flux Realism",
-        "grok-2-aurora"
+        "grok-2-aurora",
+        "Flux Dev",
     ]
 
     def __init__(
@@ -29,7 +30,6 @@ class FreeAIImager(ImageProvider):
         model: str = "dall-e-3",  # Updated default model
         timeout: int = 60,
         proxies: dict = {},
-        logging: bool = True
     ):
         """Initialize your FreeAIPlayground provider with custom settings! ⚙️
 

webscout/Provider/TTI/ImgSys/__init__.py

@@ -0,0 +1,23 @@
+"""ImgSys Provider Package - Generate images from multiple providers! 🎨
+
+Examples:
+    >>> # Synchronous usage
+    >>> from webscout import ImgSys
+    >>> provider = ImgSys()
+    >>> images = provider.generate("A cool cyberpunk city")
+    >>> provider.save(images, dir="my_images")
+    >>> 
+    >>> # Asynchronous usage
+    >>> import asyncio
+    >>> from webscout import AsyncImgSys
+    >>> async def main():
+    ...     provider = AsyncImgSys()
+    ...     images = await provider.generate("A cool cyberpunk city")
+    ...     await provider.save(images, dir="my_images")
+    >>> asyncio.run(main())
+"""
+
+from .sync_imgsys import ImgSys
+from .async_imgsys import AsyncImgSys
+
+__all__ = ["ImgSys", "AsyncImgSys"] 

webscout/Provider/TTI/ImgSys/async_imgsys.py

@@ -0,0 +1,202 @@
+"""ImgSys Asynchronous Provider - Generate images from multiple providers! 🔥
+
+Examples:
+    >>> import asyncio
+    >>> from webscout import AsyncImgSys
+    >>> 
+    >>> async def main():
+    ...     provider = AsyncImgSys()
+    ...     # Generate images
+    ...     images = await provider.generate("A cool cyberpunk city at night")
+    ...     await provider.save(images, dir="my_images")
+    >>> asyncio.run(main())
+"""
+
+import aiohttp
+import os
+import time
+import asyncio
+from typing import List, Optional, Union
+from aiohttp import ClientError
+from pathlib import Path
+
+from webscout.AIbase import ImageProvider
+from webscout.litagent import LitAgent
+
+# Get a fresh user agent! 🔄
+agent = LitAgent()
+
+class AsyncImgSys(ImageProvider):
+    """Your homie for generating fire images using imgsys.org! 🎨
+
+    This provider generates images from multiple providers, with built-in retry logic
+    and error handling to make sure you get your images no cap! 💯
+
+    Examples:
+        >>> import asyncio
+        >>> from webscout import AsyncImgSys
+        >>> async def main():
+        ...     provider = AsyncImgSys()
+        ...     # Generate images
+        ...     images = await provider.generate("A futuristic city")
+        ...     await provider.save(images, "city.jpg")
+        >>> asyncio.run(main())
+    """
+
+    def __init__(
+        self, 
+        timeout: int = 60, 
+        proxies: Optional[dict] = None
+    ):
+        """Initialize your AsyncImgSys provider with custom settings
+
+        Examples:
+            >>> provider = AsyncImgSys(timeout=30)
+            >>> provider = AsyncImgSys(proxies={"http": "http://proxy:8080"})
+
+        Args:
+            timeout (int): HTTP request timeout in seconds (default: 60)
+            proxies (dict, optional): Proxy configuration for requests
+        """
+        self.request_id_endpoint = "https://imgsys.org/api/initiate"
+        self.image_response_endpoint = "https://imgsys.org/api/get"
+        self.image_provider_endpoint = "https://imgsys.org/api/submit"
+        
+        self.headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+            "User-Agent": agent.random(),
+        }
+        self.timeout = aiohttp.ClientTimeout(total=timeout)
+        self.proxies = proxies
+        self.prompt: str = "AI-generated image - webscout"
+        self.image_extension: str = "jpeg"
+
+    async def generate(
+        self,
+        prompt: str,
+        max_retries: int = 3,
+        retry_delay: int = 5,
+    ) -> List[bytes]:
+        """Generate some fire images from your prompt! 🎨
+
+        Args:
+            prompt (str): Your image description
+            max_retries (int): Max retry attempts if something fails (default: 3)
+            retry_delay (int): Seconds to wait between retries (default: 5)
+
+        Returns:
+            List[bytes]: Your generated images as bytes
+
+        Raises:
+            ValueError: If the inputs ain't valid
+            ClientError: If the API calls fail after retries
+        """
+        # Input validation
+        if not prompt:
+            raise ValueError("Yo fam, the prompt can't be empty! 🤔")
+
+        self.prompt = prompt
+        response = []
+        
+        async with aiohttp.ClientSession(headers=self.headers, timeout=self.timeout) as session:
+            # Get request ID
+            data = {"prompt": prompt}
+            async with session.post(self.request_id_endpoint, json=data) as resp:
+                resp.raise_for_status()
+                request_id = (await resp.json())["requestId"]
+
+            # Poll for results
+            for attempt in range(max_retries):
+                try:
+                    # Get image URLs
+                    async with session.get(
+                        f"{self.image_response_endpoint}?requestId={request_id}"
+                    ) as resp:
+                        resp.raise_for_status()
+                        image_data = await resp.json()
+
+                    if "results" in image_data and len(image_data["results"]) >= 2:
+                        # Get provider names
+                        async with session.post(
+                            self.image_provider_endpoint,
+                            json={"requestId": request_id, "preference": 0}
+                        ) as resp:
+                            resp.raise_for_status()
+                            provider_data = await resp.json()
+
+                        # Download images
+                        for i, url in enumerate(image_data["results"][:2]):
+                            async with session.get(url) as resp:
+                                resp.raise_for_status()
+                                response.append(await resp.read())
+                        
+                        break
+                    else:
+                        if attempt == max_retries - 1:
+                            raise ClientError("Failed to get image results after max retries")
+                        await asyncio.sleep(retry_delay)
+
+                except ClientError as e:
+                    if attempt == max_retries - 1:
+                        raise
+                    await asyncio.sleep(retry_delay)
+
+        return response
+
+    async def save(
+        self,
+        response: List[bytes],
+        name: Optional[str] = None,
+        dir: Optional[Union[str, Path]] = None,
+        filenames_prefix: str = "",
+    ) -> List[str]:
+        """Save your fire generated images! 💾
+
+        Examples:
+            >>> import asyncio
+            >>> from webscout import AsyncImgSys
+            >>> async def main():
+            ...     provider = AsyncImgSys()
+            ...     images = await provider.generate("Cool art")
+            ...     # Save with default settings
+            ...     paths = await provider.save(images)
+            ...     # Save with custom name and directory
+            ...     paths = await provider.save(
+            ...         images,
+            ...         name="my_art",
+            ...         dir="my_images",
+            ...         filenames_prefix="test_"
+            ...     )
+            >>> asyncio.run(main())
+
+        Args:
+            response (List[bytes]): Your generated images
+            name (Optional[str]): Custom name for your images
+            dir (Optional[Union[str, Path]]): Where to save the images (default: current directory)
+            filenames_prefix (str): Prefix for your image files
+
+        Returns:
+            List[str]: Paths to your saved images
+        """
+        save_dir = dir if dir else os.getcwd()
+        if not os.path.exists(save_dir):
+            os.makedirs(save_dir)
+
+        saved_paths = []
+        timestamp = int(time.time())
+        
+        for i, image_bytes in enumerate(response):
+            if name:
+                filename = f"{filenames_prefix}{name}_{i}.{self.image_extension}"
+            else:
+                filename = f"{filenames_prefix}imgsys_{timestamp}_{i}.{self.image_extension}"
+            
+            filepath = os.path.join(save_dir, filename)
+            
+            async with aiohttp.AsyncFile(filepath, "wb") as f:
+                await f.write(image_bytes)
+            
+            saved_paths.append(filepath)
+
+        return saved_paths 

webscout/Provider/TTI/ImgSys/sync_imgsys.py

@@ -0,0 +1,195 @@
+"""ImgSys Synchronous Provider - Generate images from multiple providers! 🔥
+
+Examples:
+    >>> from webscout import ImgSys
+    >>> provider = ImgSys()
+    >>> 
+    >>> # Generate images
+    >>> images = provider.generate("A cool cyberpunk city at night")
+    >>> provider.save(images, dir="my_images")
+"""
+
+import requests
+import os
+import time
+from typing import List, Optional, Union
+from requests.exceptions import RequestException
+from pathlib import Path
+
+from webscout.AIbase import ImageProvider
+from webscout.litagent import LitAgent
+
+# Get a fresh user agent! 🔄
+agent = LitAgent()
+
+class ImgSys(ImageProvider):
+    """Your homie for generating fire images using imgsys.org! 🎨
+
+    This provider generates images from multiple providers, with built-in retry logic
+    and error handling to make sure you get your images no cap! 💯
+
+    Examples:
+        >>> provider = ImgSys()
+        >>> # Generate images
+        >>> images = provider.generate("A futuristic city")
+        >>> provider.save(images, "city.jpg")
+    """
+
+    def __init__(
+        self, 
+        timeout: int = 60, 
+        proxies: Optional[dict] = None
+    ):
+        """Initialize your ImgSys provider with custom settings
+
+        Examples:
+            >>> provider = ImgSys(timeout=30)
+            >>> provider = ImgSys(proxies={"http": "http://proxy:8080"})
+
+        Args:
+            timeout (int): HTTP request timeout in seconds (default: 60)
+            proxies (dict, optional): Proxy configuration for requests
+        """
+        self.request_id_endpoint = "https://imgsys.org/api/initiate"
+        self.image_response_endpoint = "https://imgsys.org/api/get"
+        self.image_provider_endpoint = "https://imgsys.org/api/submit"
+        
+        self.headers = {
+            "Accept": "application/json",
+            "Content-Type": "application/json",
+            "User-Agent": agent.random(),
+        }
+        self.session = requests.Session()
+        self.session.headers.update(self.headers)
+        if proxies:
+            self.session.proxies.update(proxies)
+            
+        self.timeout = timeout
+        self.prompt: str = "AI-generated image - webscout"
+        self.image_extension: str = "jpeg"
+
+    def generate(
+        self,
+        prompt: str,
+        max_retries: int = 3,
+        retry_delay: int = 5,
+    ) -> List[bytes]:
+        """Generate some fire images from your prompt! 🎨
+
+        Args:
+            prompt (str): Your image description
+            max_retries (int): Max retry attempts if something fails (default: 3)
+            retry_delay (int): Seconds to wait between retries (default: 5)
+
+        Returns:
+            List[bytes]: Your generated images as bytes
+
+        Raises:
+            ValueError: If the inputs ain't valid
+            RequestException: If the API calls fail after retries
+        """
+        # Input validation
+        if not prompt:
+            raise ValueError("Yo fam, the prompt can't be empty! 🤔")
+
+        self.prompt = prompt
+        response = []
+        
+        # Get request ID
+        data = {"prompt": prompt}
+        resp = self.session.post(self.request_id_endpoint, json=data, timeout=self.timeout)
+        resp.raise_for_status()
+        request_id = resp.json()["requestId"]
+
+        # Poll for results
+        for attempt in range(max_retries):
+            try:
+                # Get image URLs
+                resp = self.session.get(
+                    f"{self.image_response_endpoint}?requestId={request_id}",
+                    timeout=self.timeout
+                )
+                resp.raise_for_status()
+                image_data = resp.json()
+
+                if "results" in image_data and len(image_data["results"]) >= 2:
+                    # Get provider names
+                    provider_resp = self.session.post(
+                        self.image_provider_endpoint,
+                        json={"requestId": request_id, "preference": 0},
+                        timeout=self.timeout
+                    )
+                    provider_resp.raise_for_status()
+                    provider_data = provider_resp.json()
+
+                    # Download images
+                    for i, url in enumerate(image_data["results"][:2]):
+                        img_resp = self.session.get(url, timeout=self.timeout)
+                        img_resp.raise_for_status()
+                        response.append(img_resp.content)
+                    
+                    break
+                else:
+                    if attempt == max_retries - 1:
+                        raise RequestException("Failed to get image results after max retries")
+                    time.sleep(retry_delay)
+
+            except RequestException as e:
+                if attempt == max_retries - 1:
+                    raise
+                time.sleep(retry_delay)
+
+        return response
+
+    def save(
+        self,
+        response: List[bytes],
+        name: Optional[str] = None,
+        dir: Optional[Union[str, Path]] = None,
+        filenames_prefix: str = "",
+    ) -> List[str]:
+        """Save your fire generated images! 💾
+
+        Examples:
+            >>> provider = ImgSys()
+            >>> images = provider.generate("Cool art")
+            >>> # Save with default settings
+            >>> paths = provider.save(images)
+            >>> # Save with custom name and directory
+            >>> paths = provider.save(
+            ...     images,
+            ...     name="my_art",
+            ...     dir="my_images",
+            ...     filenames_prefix="test_"
+            ... )
+
+        Args:
+            response (List[bytes]): Your generated images
+            name (Optional[str]): Custom name for your images
+            dir (Optional[Union[str, Path]]): Where to save the images (default: current directory)
+            filenames_prefix (str): Prefix for your image files
+
+        Returns:
+            List[str]: Paths to your saved images
+        """
+        save_dir = dir if dir else os.getcwd()
+        if not os.path.exists(save_dir):
+            os.makedirs(save_dir)
+
+        saved_paths = []
+        timestamp = int(time.time())
+        
+        for i, image_bytes in enumerate(response):
+            if name:
+                filename = f"{filenames_prefix}{name}_{i}.{self.image_extension}"
+            else:
+                filename = f"{filenames_prefix}imgsys_{timestamp}_{i}.{self.image_extension}"
+            
+            filepath = os.path.join(save_dir, filename)
+            
+            with open(filepath, "wb") as f:
+                f.write(image_bytes)
+            
+            saved_paths.append(filepath)
+
+        return saved_paths 

webscout/Provider/TTI/__init__.py

@@ -7,4 +7,6 @@ from .artbit import *
 from .talkai import *
 from .piclumen import *
 from .MagicStudio import *
-from .fastflux import *
+from .fastflux import *
+from .pixelmuse import *
+from .ImgSys import *

webscout/Provider/TTI/artbit/async_artbit.py

@@ -31,7 +31,7 @@ import aiohttp
 import aiofiles
 import asyncio
 import os
-from typing import List
+from typing import Union, List
 from webscout.AIbase import AsyncImageProvider
 from webscout.litagent import LitAgent
 

webscout/Provider/TTI/artbit/sync_artbit.py

@@ -25,7 +25,7 @@ Examples:
 import cloudscraper
 import os
 import requests
-from typing import List
+from typing import Union, List
 from webscout.AIbase import ImageProvider
 from webscout.litagent import LitAgent