webscout 1.3.0__py3-none-any.whl → 1.3.1__py3-none-any.whl

webscout/LLM.py

@@ -8,7 +8,7 @@ class LLM:
         self.model = model
         self.conversation_history = [{"role": "system", "content": system_message}]
 
-    def mistral_chat(self, messages: List[Dict[str, str]]) -> Union[str, None]:
+    def chat(self, messages: List[Dict[str, str]]) -> Union[str, None]:
         url = "https://api.deepinfra.com/v1/openai/chat/completions"
         headers = {
             'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36',
@@ -43,26 +43,3 @@ class LLM:
             return result.json()['choices'][0]['message']['content']
         except:
             return None
-
-    def chat(self):
-        while True:
-            prompt = input("👦: ")
-            user_message = {"role": "user", "content": prompt}
-            self.conversation_history.append(user_message)
-            try:
-                resp = self.mistral_chat(self.conversation_history)
-                print(f"🤖: {resp}")
-                self.conversation_history.append({"role": "assistant", "content": resp})
-            except Exception as e:
-                print(f"🤖: Oops, something went wrong: {e}! Looks like even AI needs some oiling sometimes.")
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description='LLM CLI', epilog='To use a specific model, run:\n'
-                                                                     'python -m webscout.LLM model_name\n'
-                                                                     'Replace "model_name" with the name of the model you wish to use It supports ALL text generation models on deepinfra.com.')
-    parser.add_argument('model', type=str, help='Model to use for text generation. Specify the full model name, e.g., "mistralai/Mistral-7B-Instruct-v0.1".')
-    parser.add_argument('--system-message', type=str, default="You are a Helpful AI.", help='Custom system prompt for the AI.')
-    args = parser.parse_args()
-
-    LLM = LLM(args.model, args.system_message)
-    LLM.chat()

webscout/__init__.py

@@ -11,6 +11,7 @@ from .version import __version__
 from .DWEBS import DeepWEBS
 from .transcriber import transcriber
 from .voice import play_audio
+from .LLM import LLM
 
 
 __all__ = ["WEBS", "AsyncWEBS", "__version__", "cli"]

webscout/version.py

@@ -1,2 +1,2 @@
-__version__ = "1.3.0"
+__version__ = "1.3.1"
 

{webscout-1.3.0.dist-info → webscout-1.3.1.dist-info}/METADATA

@@ -1,7 +1,7 @@
 Metadata-Version: 2.1
 Name: webscout
-Version: 1.3.0
-Summary: Search for words, documents, images, videos, news, maps and text translation using the Google, DuckDuckGo.com, yep.com, phind.com, you.com, etc Also containes AI models and now can transcribe yt videos
+Version: 1.3.1
+Summary: Search for words, documents, images, videos, news, maps and text translation using the Google, DuckDuckGo.com, yep.com, phind.com, you.com, etc Also containes AI models, can transcribe yt videos and have TTS support
 Author: OEvortex
 Author-email: helpingai5@gmail.com
 License: HelpingAI Simplified Universal License
@@ -99,6 +99,7 @@ Search for words, documents, images, videos, news, maps and text translation usi
     - [9. `KOBOLDIA` -](#9-koboldia--)
   - [usage of special .LLM file from webscout (webscout.LLM)](#usage-of-special-llm-file-from-webscout-webscoutllm)
     - [`LLM`](#llm)
+    - [`LLM` with internet](#llm-with-internet)
 
 ## Install
 ```python
@@ -704,11 +705,102 @@ print(message)
 ```python
 from webscout.LLM import LLM
 
-def chat(model_name, system_message="You are Jarvis"):# system prompt
-    AI = LLM(model_name, system_message)
-    AI.chat()
+# Read the system message from the file
+with open('system.txt', 'r') as file:
+    system_message = file.read()
+
+# Initialize the LLM class with the model name and system message
+llm = LLM(model="microsoft/WizardLM-2-8x22B", system_message=system_message)
+
+while True:
+    # Get the user input
+    user_input = input("User: ")
+
+    # Define the messages to be sent
+    messages = [
+        {"role": "user", "content": user_input}
+    ]
+
+    # Use the mistral_chat method to get the response
+    response = llm.chat(messages)
+
+    # Print the response
+    print("AI: ", response)
+```
+### `LLM` with internet
+```python
+from __future__ import annotations
+from typing import List, Optional
+
+from webscout import LLM
+from webscout import WEBS
+import warnings
+
+system_message: str = (
+    "As AI, you possess internet access and are capable of executing real-time web searches based on user inputs. "
+    "You shall utilize this capability to enrich conversations, offer informed insights, and augment your ability to "
+    "respond accurately and thoroughly. However, refrain from stating 'You have provided a list of strings,' ensuring "
+    "seamless interactions with users. Embrace a responsive demeanor, harnessing available online resources to address "
+    "queries, share pertinent content, and facilitate meaningful exchanges. By doing so, you create value through "
+    "connection and engagement, ultimately enhancing overall user satisfaction and experience. Additionally, "
+    "continue upholding the principles of respect, impartiality, and intellectual integrity throughout all interactions."
+)
+
+# Ignore the specific UserWarning
+warnings.filterwarnings("ignore", category=UserWarning, module="curl_cffi.aio", lineno=205)
+LLM = LLM(model="meta-llama/Meta-Llama-3-70B-Instruct", system_message=system_message)
+
+
+def chat(
+    user_input: str, webs: WEBS, max_results: int = 10
+) -> Optional[str]:
+    """
+    Chat function to perform a web search based on the user input and generate a response using the LLM model.
+
+    Parameters
+    ----------
+    user_input : str
+        The user input to be used for the web search
+    webs : WEBS
+        The web search instance to be used to perform the search
+    max_results : int, optional
+        The maximum number of search results to include in the response, by default 10
+
+    Returns
+    -------
+    Optional[str]
+        The response generated by the LLM model, or None if there is no response
+    """
+    # Perform a web search based on the user input
+    search_results: List[str] = []
+    for r in webs.text(
+        user_input, region="wt-wt", safesearch="off", timelimit="y", max_results=max_results
+    ):
+        search_results.append(str(r))  # Convert each result to a string
+
+    # Define the messages to be sent, including the user input, search results, and system message
+    messages = [
+        {"role": "user", "content": user_input + "\n" + "websearch results are:" + "\n".join(search_results)},
+    ]
+
+    # Use the chat method to get the response
+    response = LLM.chat(messages)
+
+    return response
+
 
 if __name__ == "__main__":
-    model_name = "mistralai/Mistral-7B-Instruct-v0.2" # name of the model you wish to use It supports ALL text generation models on deepinfra.com.
-    chat(model_name)
+    while True:
+        # Get the user input
+        user_input = input("User: ")
+
+        # Perform a web search based on the user input
+        with WEBS() as webs:
+            response = chat(user_input, webs)
+
+        # Print the response
+        if response:
+            print("AI:", response)
+        else:
+            print("No response")
 ```

{webscout-1.3.0.dist-info → webscout-1.3.1.dist-info}/RECORD

@@ -15,21 +15,21 @@ webscout/AIbase.py,sha256=vQi2ougu5bG-QdmoYmxCQsOg7KTEgG7EF6nZh5qqUGw,2343
 webscout/AIutel.py,sha256=fNN4mmjXcxjJGq2CVJP1MU2oQ78p8OyExQBjVif6e-k,24123
 webscout/DWEBS.py,sha256=QT-7-dUgWhQ_H7EVZD53AVyXxyskoPMKCkFIpzkN56Q,7332
 webscout/HelpingAI.py,sha256=YeZw0zYVHMcBFFPNdd3_Ghpm9ebt_EScQjHO_IIs4lg,8103
-webscout/LLM.py,sha256=XByJPiATLA_57FBWKw18Xx_PGRCPOj-GJE96aQH1k2Y,3309
-webscout/__init__.py,sha256=1DBQX84kGCCzGNFV2MtNqASb_2WlM7BuM0enCqnOx8I,550
+webscout/LLM.py,sha256=CiDz0okZNEoXuxMwadZnwRGSLpqk2zg0vzvXSxQZjcE,1910
+webscout/__init__.py,sha256=rgmTILV0qx0x2PVdMq7flk5nas102sQN5z8p_OZaTzg,572
 webscout/__main__.py,sha256=ZtTRgsRjUi2JOvYFLF1ZCh55Sdoz94I-BS-TlJC7WDU,126
 webscout/cli.py,sha256=F888fdrFUQgczMBN4yMOSf6Nh-IbvkqpPhDsbnA2FtQ,17059
 webscout/exceptions.py,sha256=4AOO5wexeL96nvUS-badcckcwrPS7UpZyAgB9vknHZE,276
 webscout/models.py,sha256=5iQIdtedT18YuTZ3npoG7kLMwcrKwhQ7928dl_7qZW0,692
 webscout/transcriber.py,sha256=EddvTSq7dPJ42V3pQVnGuEiYQ7WjJ9uyeR9kMSxN7uY,20622
 webscout/utils.py,sha256=c_98M4oqpb54pUun3fpGGlCerFD6ZHUbghyp5b7Mwgo,2605
-webscout/version.py,sha256=R-_i2z0hXcrgq3LLNeJVLbwdS54wc3y_4KzSPxzw8j0,25
+webscout/version.py,sha256=xBny7vFjm_5sB7eDRX5tmYecHaNfJlEkwuSeLudsFXo,25
 webscout/voice.py,sha256=1Ids_2ToPBMX0cH_UyPMkY_6eSE9H4Gazrl0ujPmFag,941
 webscout/webscout_search.py,sha256=3_lli-hDb8_kCGwscK29xuUcOS833ROgpNhDzrxh0dk,3085
 webscout/webscout_search_async.py,sha256=Y5frH0k3hLqBCR-8dn7a_b7EvxdYxn6wHiKl3jWosE0,40670
-webscout-1.3.0.dist-info/LICENSE.md,sha256=mRVwJuT4SXC5O93BFdsfWBjlXjGn2Np90Zm5SocUzM0,3150
-webscout-1.3.0.dist-info/METADATA,sha256=E9A3XzQIkmBA6C6E0PXYyFxdqqPCH2PHYSVSKUJczcc,28258
-webscout-1.3.0.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-webscout-1.3.0.dist-info/entry_points.txt,sha256=8-93eRslYrzTHs5E-6yFRJrve00C9q-SkXJD113jzRY,197
-webscout-1.3.0.dist-info/top_level.txt,sha256=OD5YKy6Y3hldL7SmuxsiEDxAG4LgdSSWwzYk22MF9fk,18
-webscout-1.3.0.dist-info/RECORD,,
+webscout-1.3.1.dist-info/LICENSE.md,sha256=mRVwJuT4SXC5O93BFdsfWBjlXjGn2Np90Zm5SocUzM0,3150
+webscout-1.3.1.dist-info/METADATA,sha256=Lqt3sYRj9TpssiZ_mD3z0Sh_UrdVPPm5L9o0hPAIFWw,31547
+webscout-1.3.1.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+webscout-1.3.1.dist-info/entry_points.txt,sha256=8-93eRslYrzTHs5E-6yFRJrve00C9q-SkXJD113jzRY,197
+webscout-1.3.1.dist-info/top_level.txt,sha256=OD5YKy6Y3hldL7SmuxsiEDxAG4LgdSSWwzYk22MF9fk,18
+webscout-1.3.1.dist-info/RECORD,,