webscout 1.4.5__py3-none-any.whl → 2.0__py3-none-any.whl

webscout/Provider/__init__.py

@@ -0,0 +1,57 @@
+# webscout/providers/__init__.py
+
+from .ThinkAnyAI import ThinkAnyAI
+from .Xjai import Xjai
+from .Llama2 import LLAMA2
+from .Llama2 import AsyncLLAMA2
+from .Cohere import Cohere
+from .Reka import REKA
+from .Groq import GROQ
+from .Groq import AsyncGROQ
+from .Openai import OPENAI
+from .Openai import AsyncOPENAI
+from .Leo import LEO
+from .Leo import AsyncLEO
+from .Koboldai import KOBOLDAI
+from .Koboldai import AsyncKOBOLDAI
+from .OpenGPT import OPENGPT 
+from .OpenGPT import AsyncOPENGPT
+from .Perplexity import PERPLEXITY
+from .Blackboxai import BLACKBOXAI 
+from .Blackboxai import AsyncBLACKBOXAI
+from .Phind import PhindSearch 
+from .Phind import AsyncPhindSearch
+from .Yepchat import YEPCHAT
+from .Yepchat import AsyncYEPCHAT
+from .Youchat import YouChat
+from .Gemini import GEMINI
+from .ChatGPTlogin import ChatGPTlogin
+
+__all__ = [
+    'ThinkAnyAI',
+    'Xjai',
+    'LLAMA2', 
+    'AsyncLLAMA2',
+    'Cohere',
+    'REKA',
+    'GROQ',
+    'AsyncGROQ',
+    'OPENAI',
+    'AsyncOPENAI',
+    'LEO',
+    'AsyncLEO',
+    'KOBOLDAI',
+    'AsyncKOBOLDAI',
+    'OPENGPT', 
+    'AsyncOPENGPT',
+    'PERPLEXITY',
+    'BLACKBOXAI', 
+    'AsyncBLACKBOXAI',
+    'PhindSearch', 
+    'AsyncPhindSearch',
+    'YEPCHAT',
+    'AsyncYEPCHAT',
+    'YouChat',
+    'GEMINI',
+    'ChatGPTlogin',
+]

webscout/__init__.py

@@ -2,15 +2,44 @@
 
 Search for anything using the Google, DuckDuckGo, phind.com. Also containes AI models, can transcribe yt videos, temporary email and phone number generation, have TTS support and webai(terminal gpt and open interpeter)
 """
-import g4f
-import logging
-from .webscout_search import WEBS
-from .webscout_search_async import AsyncWEBS
+# webscout/__init__.py
+
+from .webscout_search import WEBS 
+from .webscout_search_async import AsyncWEBS 
 from .version import __version__
-from .DWEBS import DeepWEBS
+from .DWEBS import DeepWEBS 
 from .transcriber import transcriber
 from .voice import play_audio
-
+import g4f
+# Import provider classes for direct access
+from .Provider import (
+    ThinkAnyAI,
+    Xjai,
+    LLAMA2,
+    AsyncLLAMA2,
+    Cohere,
+    REKA,
+    GROQ,
+    AsyncGROQ,
+    OPENAI,
+    AsyncOPENAI,
+    LEO,
+    AsyncLEO,
+    KOBOLDAI,
+    AsyncKOBOLDAI,
+    OPENGPT,
+    AsyncOPENGPT,
+    PERPLEXITY,
+    BLACKBOXAI,
+    AsyncBLACKBOXAI,
+    PhindSearch,
+    AsyncPhindSearch,
+    YEPCHAT,
+    AsyncYEPCHAT,
+    YouChat,
+    GEMINI,
+    ChatGPTlogin,
+)
 
 __repo__ = "https://github.com/OE-LUCIFER/Webscout"
 
@@ -30,7 +59,9 @@ webai = [
     "yepchat",
     "you",
     "xjai",
-    "thinkany"
+    "thinkany",
+    # "chatgptlogin",
+    "auto",
 ]
 
 gpt4free_providers = [
@@ -39,6 +70,45 @@ gpt4free_providers = [
 
 available_providers = webai + gpt4free_providers
 
-__all__ = ["WEBS", "AsyncWEBS", "__version__", "cli"]
+# Add all the provider classes you want to directly import to __all__
+__all__ = [
+    "WEBS", 
+    "AsyncWEBS", 
+    "__version__", 
+    "cli", 
+    "DeepWEBS", 
+    "transcriber", 
+    "play_audio",
+
+    # AI Providers
+    "ThinkAnyAI",
+    "Xjai",
+    "LLAMA2",
+    "AsyncLLAMA2",
+    "Cohere",
+    "REKA",
+    "GROQ",
+    "AsyncGROQ",
+    "OPENAI",
+    "AsyncOPENAI",
+    "LEO",
+    "AsyncLEO",
+    "KOBOLDAI",
+    "AsyncKOBOLDAI",
+    "OPENGPT",
+    "AsyncOPENGPT",
+    "PERPLEXITY",
+    "BLACKBOXAI",
+    "AsyncBLACKBOXAI",
+    "PhindSearch",
+    "AsyncPhindSearch",
+    "YEPCHAT",
+    "AsyncYEPCHAT",
+    "YouChat",
+    "GEMINI",
+    "ChatGPTlogin",
+]
 
-logging.getLogger("webscout").addHandler(logging.NullHandler())
+# Set up basic logger
+import logging
+logging.getLogger("webscout").addHandler(logging.NullHandler()) 

webscout/async_providers.py

@@ -7,7 +7,7 @@ from webscout.AI import AsyncLEO
 from webscout.AI import AsyncKOBOLDAI
 from webscout.AI import AsyncGROQ
 from webscout.AI import AsyncBLACKBOXAI
-from webscout.AI import AsyncGPT4FREE
+from webscout.g4f import AsyncGPT4FREE
 
 mapper: dict[str, object] = {
     "phind": AsyncPhindSearch,
@@ -21,13 +21,3 @@ mapper: dict[str, object] = {
     "groq": AsyncGROQ,
     "openai": AsyncOPENAI,
 }
-
-tgpt_mapper: dict[str, object] = {
-    "phind": AsyncPhindSearch,
-    "opengpt": AsyncOPENGPT,
-    "koboldai": AsyncKOBOLDAI,
-    # "gpt4free": AsyncGPT4FREE,
-    "blackboxai": AsyncBLACKBOXAI,
-    "llama2": AsyncLLAMA2,
-    "yepchat": AsyncYEPCHAT,
-}

webscout/exceptions.py

@@ -10,4 +10,9 @@ class TimeoutE(Exception):
     """Raised for timeout errors during API requests."""
     
 class FailedToGenerateResponseError(Exception):
+    
     """Provider failed to fetch response"""
+class AllProvidersFailure(Exception):
+    """None of the providers generated response successfully"""
+
+    pass

webscout/version.py

@@ -1,2 +1,2 @@
-__version__ = "1.4.5"
+__version__ = "2.0"
 

webscout/webai.py

@@ -25,16 +25,17 @@ from rich.table import Table
 from rich.prompt import Prompt
 from rich.progress import Progress
 from typing import Iterator
-from webscout.AIutel import Optimizers
-from webscout.AIutel import default_path
-from webscout.AIutel import AwesomePrompts
-from webscout.AIutel import RawDog
-from webscout.AIutel import Audio
+from .AIutel import Optimizers
+from .AIutel import default_path
+from .AIutel import AwesomePrompts
+from .AIutel import RawDog
+from .AIutel import Audio
 from webscout import available_providers
 from colorama import Fore
 from colorama import init as init_colorama
 from dotenv import load_dotenv
 import g4f
+import webscout
 import webscout.AIutel
 
 init_colorama(autoreset=True)
@@ -414,7 +415,7 @@ class Main(cmd.Cmd):
                     )
 
             elif provider == "leo":
-                from webscout.AI import LEO
+                from webscout import LEO
 
                 self.bot = LEO(
                     is_conversation=disable_conversation,
@@ -437,7 +438,7 @@ class Main(cmd.Cmd):
                 assert auth, (
                     "OpenAI's API-key is required. " "Use the flag `--key` or `-k`"
                 )
-                from webscout.AI import OPENAI
+                from webscout import OPENAI
 
                 self.bot = OPENAI(
                     api_key=auth,
@@ -456,9 +457,22 @@ class Main(cmd.Cmd):
                     history_offset=history_offset,
                     act=awesome_prompt,
                 )
+            if provider == "auto":
+                from webscout import AUTO
 
+                self.bot = AUTO(
+                    is_conversation=disable_conversation,
+                    max_tokens=max_tokens,
+                    timeout=timeout,
+                    intro=intro,
+                    filepath=filepath,
+                    update_file=update_file,
+                    proxies=proxies,
+                    history_offset=history_offset,
+                    act=awesome_prompt,
+                )
             elif provider == "opengpt":
-                from webscout.AI import OPENGPT
+                from webscout import OPENGPT
 
                 self.bot = OPENGPT(
                     is_conversation=disable_conversation,
@@ -470,9 +484,10 @@ class Main(cmd.Cmd):
                     proxies=proxies,
                     history_offset=history_offset,
                     act=awesome_prompt,
+                    assistant_id="bca37014-6f97-4f2b-8928-81ea8d478d88"
                 )
             elif provider == "thinkany":
-                from webscout.AI import ThinkAnyAI
+                from webscout import ThinkAnyAI
 
                 self.bot = ThinkAnyAI(
                     is_conversation=disable_conversation,
@@ -485,8 +500,22 @@ class Main(cmd.Cmd):
                     history_offset=history_offset,
                     act=awesome_prompt,
                 )
+            # elif provider == "chatgptlogin":
+            #     from webscout import ChatGPTlogin
+
+            #     self.bot = ChatGPTlogin(
+            #         is_conversation=disable_conversation,
+            #         max_tokens=max_tokens,
+            #         timeout=timeout,
+            #         intro=intro,
+            #         filepath=filepath,
+            #         update_file=update_file,
+            #         proxies=proxies,
+            #         history_offset=history_offset,
+            #         act=awesome_prompt,
+            #     )
             elif provider == "yepchat":
-                from webscout.AI import YEPCHAT
+                from webscout import YEPCHAT
 
                 self.bot = YEPCHAT(
                     is_conversation=disable_conversation,
@@ -508,7 +537,7 @@ class Main(cmd.Cmd):
                 assert auth, (
                     "GROQ's API-key is required. " "Use the flag `--key` or `-k`"
                 )
-                from webscout.AI import GROQ
+                from webscout import GROQ
 
 
                 self.bot = GROQ(
@@ -532,7 +561,7 @@ class Main(cmd.Cmd):
                 assert auth, (
                     "Cohere's API-key is required. Use the flag `--key` or `-k`"
                 )
-                from webscout.AI import Cohere
+                from webscout import Cohere
                 self.bot = Cohere(
                     api_key=auth,
                     is_conversation=disable_conversation,
@@ -550,7 +579,7 @@ class Main(cmd.Cmd):
                     act=awesome_prompt,
             )
             elif provider == "reka":
-                from webscout.AI import REKA
+                from webscout import REKA
 
                 self.bot = REKA(
                     api_key=auth,
@@ -568,7 +597,7 @@ class Main(cmd.Cmd):
                 )
 
             elif provider == "koboldai":
-                from webscout.AI import KOBOLDAI
+                from webscout import KOBOLDAI
 
                 self.bot = KOBOLDAI(
                     is_conversation=disable_conversation,
@@ -584,7 +613,7 @@ class Main(cmd.Cmd):
                     act=awesome_prompt,
                 )
             elif provider == "xjai":
-                from webscout.AI import Xjai
+                from webscout import Xjai
 
                 self.bot = Xjai(
                     is_conversation=disable_conversation,
@@ -601,7 +630,7 @@ class Main(cmd.Cmd):
                 )
 
             elif provider == "gemini":
-                from webscout.AI import GEMINI
+                from webscout import GEMINI
 
                 assert auth, (
                     "Path to gemini.google.com.cookies.json file is required. "
@@ -614,7 +643,7 @@ class Main(cmd.Cmd):
                 )
 
             elif provider == "phind":
-                from webscout.AI import PhindSearch
+                from webscout import PhindSearch
 
                 self.bot = PhindSearch(
                     is_conversation=disable_conversation,
@@ -632,7 +661,7 @@ class Main(cmd.Cmd):
 
             elif provider == "blackboxai":
 
-                from webscout.AI import BLACKBOXAI
+                from webscout import BLACKBOXAI
 
                 self.bot = BLACKBOXAI(
                     is_conversation=disable_conversation,
@@ -647,7 +676,7 @@ class Main(cmd.Cmd):
                 )
             elif provider == "you":
 
-                from webscout.AI import YouChat
+                from webscout import YouChat
 
                 self.bot = YouChat(
                     is_conversation=disable_conversation,
@@ -683,7 +712,7 @@ class Main(cmd.Cmd):
 
 
             elif provider == "perplexity":
-                from webscout.AI import PERPLEXITY
+                from webscout import PERPLEXITY
 
                 self.bot = PERPLEXITY(
                     is_conversation=disable_conversation,
@@ -725,7 +754,13 @@ class Main(cmd.Cmd):
         self.__init_time = time.time()
         self.__start_time = time.time()
         self.__end_time = time.time()
-
+        
+    @property
+    def get_provider(self):
+        if self.provider == "auto" and self.bot.provider_name is not None:
+            return self.bot.provider_name
+        else:
+            return self.provider
     @property
     def prompt(self):
         current_time = datetime.datetime.now().strftime("%H:%M:%S")
@@ -740,7 +775,7 @@ class Main(cmd.Cmd):
         if not self.disable_coloring:
             cmd_prompt = (
                 f"╭─[`{Fore.GREEN}{getpass.getuser().capitalize()}@webai]`"
-                f"(`{Fore.YELLOW}{self.provider})`"
+                f"(`{Fore.YELLOW}{self.get_provider})`"
                 f"~[`{Fore.LIGHTWHITE_EX}⏰{Fore.MAGENTA}{current_time}-`"
                 f"{Fore.LIGHTWHITE_EX}💻{Fore.BLUE}{find_range(self.__init_time, time.time(), True)}-`"
                 f"{Fore.LIGHTWHITE_EX}⚡️{Fore.RED}{find_range(self.__start_time, self.__end_time)}s]`"
@@ -753,7 +788,7 @@ class Main(cmd.Cmd):
 
         else:
             return (
-                f"╭─[{getpass.getuser().capitalize()}@webscout]({self.provider})"
+                f"╭─[{getpass.getuser().capitalize()}@webscout]({self.get_provider})"
                 f"~[⏰{current_time}"
                 f"-💻{find_range(self.__init_time, time.time(), True)}"
                 f"-⚡️{find_range(self.__start_time, self.__end_time)}s]"
@@ -1125,7 +1160,7 @@ class Main(cmd.Cmd):
                     busy_bar.stop_spinning()
                     this.stream_output(
                         generated_response,
-                        title="AI Response",
+                        title="Webscout",
                         is_markdown=self.prettify,
                         style=Style(
                             color=self.color,

webscout/webscout_search.py

@@ -4,12 +4,6 @@ from threading import Thread
 import sys
 from types import TracebackType
 from typing import Any, Awaitable, Dict, Optional, Type, Union
-if sys.platform == 'win32':
-    try:
-        asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
-    except AttributeError:
-        # If WindowsSelectorEventLoopPolicy is not available, do nothing
-        pass
 from .webscout_search_async import AsyncWEBS
 
 
@@ -24,7 +18,7 @@ class WEBS(AsyncWEBS):
         proxies: Union[Dict[str, str], str, None] = None,  # deprecated
         timeout: Optional[int] = 10,
     ) -> None:
-        """Initialize the DDGS object.
+        """Initialize the WEBS object.
 
         Args:
             headers (dict, optional): Dictionary of headers for the HTTP client. Defaults to None.
@@ -81,4 +75,4 @@ class WEBS(AsyncWEBS):
         return self._run_async_in_thread(super().maps(*args, **kwargs))
 
     def translate(self, *args: Any, **kwargs: Any) -> Any:
-        return self._run_async_in_thread(super().translate(*args, **kwargs))
+        return self._run_async_in_thread(super().translate(*args, **kwargs))