webscout 5.9__tar.gz → 6.0__tar.gz

{webscout-5.9/webscout.egg-info → webscout-6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: webscout
-Version: 5.9
+Version: 6.0
 Summary: Search for anything using Google, DuckDuckGo, phind.com, Contains AI models, can transcribe yt videos, temporary email and phone number generation, has TTS support, webai (terminal gpt and open interpreter) and offline LLMs and more
 Author: OEvortex
 Author-email: helpingai5@gmail.com
@@ -33,7 +33,6 @@ Requires-Dist: g4f[webdriver]
 Requires-Dist: rich
 Requires-Dist: beautifulsoup4
 Requires-Dist: markdownify
-Requires-Dist: pydantic
 Requires-Dist: requests
 Requires-Dist: google-generativeai
 Requires-Dist: lxml>=5.2.2
@@ -44,8 +43,6 @@ Requires-Dist: appdirs
 Requires-Dist: tls_client
 Requires-Dist: clipman
 Requires-Dist: playsound
-Requires-Dist: poe_api_wrapper
-Requires-Dist: pyreqwest_impersonate
 Requires-Dist: ollama
 Requires-Dist: pyfiglet
 Requires-Dist: pillow
@@ -108,7 +105,7 @@ Requires-Dist: huggingface_hub[cli]; extra == "local"
 * **Text-to-Speech (TTS):** Convert text into natural-sounding speech using various TTS providers.
 * **WebAI:** Experience the power of terminal-based GPT and an open interpreter for code execution and more.
 * **Offline LLMs:** Utilize powerful language models offline with GGUF support.
-* **Extensive Provider Ecosystem:** Explore a vast collection of providers, including Poe, BasedGPT, DeepSeek, and many others.
+* **Extensive Provider Ecosystem:** Explore a vast collection of providers, including BasedGPT, DeepSeek, and many others.
 * **Local LLM Execution:** Run GGUF models locally with minimal configuration.
 * **Rawdog Scripting:** Execute Python scripts directly within your terminal using the `rawdog` feature.
 * **GGUF Conversion & Quantization:** Convert and quantize Hugging Face models to GGUF format.
@@ -1128,10 +1125,6 @@ response_str = a.chat(prompt)
 print(response_str)
 ```
 
-###  `Poe` - Chat with Poe
-
-Usage code is similar to other providers.
-
 ###  `BasedGPT` - Chat with GPT
 
 ```python
@@ -1210,27 +1203,6 @@ message = ai.get_message(response)
 print(message)
 ```
 
-###  `Deepinfra` - VLM
-
-```python
-from webscout.Provider import VLM 
-
-# Load your image
-image_path = r"C:\Users\koula\OneDrive\Desktop\Webscout\photo_2024-03-25_19-23-40.jpg"
-
-vlm_instance = VLM(model="llava-hf/llava-1.5-7b-hf", is_conversation=True, max_tokens=600, timeout=30, system_prompt="You are a Helpful AI.")
-image_base64 = vlm_instance.encode_image_to_base64(image_path)
-
-prompt = {
-    "content": "What is in this image?",
-    "image": image_base64
-}
-
-# Generate a response
-response = vlm_instance.chat(prompt)
-print(response)
-
-```
 
 ###  `GROQ`
 
@@ -1454,7 +1426,7 @@ else:
     print(f"Error: {function_call_data['error']}")
 ```
 
-###  LLAMA3, pizzagpt, RUBIKSAI, Koala, Darkai, AI4Chat, Farfalle, PIAI, Felo, XDASH, Julius, YouChat, YEPCHAT, Cloudflare, TurboSeek, Editee, AI21, Chatify, Cerebras, X0GPT, Lepton, GEMINIAPI, Cleeai, Elmo, Genspark, Upstage, Free2GPT, Bing, DiscordRocks, GPTWeb, AIGameIO, LlamaTutor, PromptRefine, AIUncensored, TutorAI, Bixin, ChatGPTES
+###  LLAMA3, pizzagpt, RUBIKSAI, Koala, Darkai, AI4Chat, Farfalle, PIAI, Felo, XDASH, Julius, YouChat, YEPCHAT, Cloudflare, TurboSeek, Editee, AI21, Chatify, Cerebras, X0GPT, Lepton, GEMINIAPI, Cleeai, Elmo, Genspark, Upstage, Free2GPT, Bing, DiscordRocks, GPTWeb, AIGameIO, LlamaTutor, PromptRefine, AIUncensored, TutorAI, Bixin, ChatGPTES, Bagoodex, ChatHub, AmigoChat
 
 Code is similar to other providers.
 

{webscout-5.9 → webscout-6.0}/README.md

@@ -41,7 +41,7 @@
 * **Text-to-Speech (TTS):** Convert text into natural-sounding speech using various TTS providers.
 * **WebAI:** Experience the power of terminal-based GPT and an open interpreter for code execution and more.
 * **Offline LLMs:** Utilize powerful language models offline with GGUF support.
-* **Extensive Provider Ecosystem:** Explore a vast collection of providers, including Poe, BasedGPT, DeepSeek, and many others.
+* **Extensive Provider Ecosystem:** Explore a vast collection of providers, including BasedGPT, DeepSeek, and many others.
 * **Local LLM Execution:** Run GGUF models locally with minimal configuration.
 * **Rawdog Scripting:** Execute Python scripts directly within your terminal using the `rawdog` feature.
 * **GGUF Conversion & Quantization:** Convert and quantize Hugging Face models to GGUF format.
@@ -1061,10 +1061,6 @@ response_str = a.chat(prompt)
 print(response_str)
 ```
 
-###  `Poe` - Chat with Poe
-
-Usage code is similar to other providers.
-
 ###  `BasedGPT` - Chat with GPT
 
 ```python
@@ -1143,27 +1139,6 @@ message = ai.get_message(response)
 print(message)
 ```
 
-###  `Deepinfra` - VLM
-
-```python
-from webscout.Provider import VLM 
-
-# Load your image
-image_path = r"C:\Users\koula\OneDrive\Desktop\Webscout\photo_2024-03-25_19-23-40.jpg"
-
-vlm_instance = VLM(model="llava-hf/llava-1.5-7b-hf", is_conversation=True, max_tokens=600, timeout=30, system_prompt="You are a Helpful AI.")
-image_base64 = vlm_instance.encode_image_to_base64(image_path)
-
-prompt = {
-    "content": "What is in this image?",
-    "image": image_base64
-}
-
-# Generate a response
-response = vlm_instance.chat(prompt)
-print(response)
-
-```
 
 ###  `GROQ`
 
@@ -1387,7 +1362,7 @@ else:
     print(f"Error: {function_call_data['error']}")
 ```
 
-###  LLAMA3, pizzagpt, RUBIKSAI, Koala, Darkai, AI4Chat, Farfalle, PIAI, Felo, XDASH, Julius, YouChat, YEPCHAT, Cloudflare, TurboSeek, Editee, AI21, Chatify, Cerebras, X0GPT, Lepton, GEMINIAPI, Cleeai, Elmo, Genspark, Upstage, Free2GPT, Bing, DiscordRocks, GPTWeb, AIGameIO, LlamaTutor, PromptRefine, AIUncensored, TutorAI, Bixin, ChatGPTES
+###  LLAMA3, pizzagpt, RUBIKSAI, Koala, Darkai, AI4Chat, Farfalle, PIAI, Felo, XDASH, Julius, YouChat, YEPCHAT, Cloudflare, TurboSeek, Editee, AI21, Chatify, Cerebras, X0GPT, Lepton, GEMINIAPI, Cleeai, Elmo, Genspark, Upstage, Free2GPT, Bing, DiscordRocks, GPTWeb, AIGameIO, LlamaTutor, PromptRefine, AIUncensored, TutorAI, Bixin, ChatGPTES, Bagoodex, ChatHub, AmigoChat
 
 Code is similar to other providers.
 

{webscout-5.9 → webscout-6.0}/setup.py

@@ -37,7 +37,6 @@ setup(
         "rich",
         "beautifulsoup4",
         "markdownify",
-        "pydantic",
         "requests",
         "google-generativeai",
         "lxml>=5.2.2",
@@ -48,8 +47,6 @@ setup(
         "tls_client",
         "clipman",
         "playsound",
-        "poe_api_wrapper",
-        "pyreqwest_impersonate",
         "ollama",
         "pyfiglet",
         "pillow",

{webscout-5.9 → webscout-6.0}/webscout/Provider/Amigo.py

@@ -36,6 +36,7 @@ class AmigoChat(Provider):
         history_offset: int = 10250,
         act: str = None,
         model: str = "o1-preview",  # Default model
+        system_prompt: str = "You are a helpful and friendly AI assistant.",
     ):
         """
         Initializes the AmigoChat.io API with given parameters.
@@ -108,6 +109,7 @@ class AmigoChat(Provider):
         )
         self.conversation.history_offset = history_offset
         self.session.proxies = proxies
+        self.system_prompt = system_prompt
 
     def ask(
         self,
@@ -147,7 +149,7 @@ class AmigoChat(Provider):
         # Define the payload
         payload = {
             "messages": [
-                {"role": "system", "content": "Mai hu ba khabr"},
+                {"role": "system", "content": self.system_prompt},
                 {"role": "user", "content": conversation_prompt}
             ],
             "model": self.model,
@@ -259,7 +261,7 @@ class AmigoChat(Provider):
     
 if __name__ == '__main__':
     from rich import print
-    ai = AmigoChat(model="o1-preview")
+    ai = AmigoChat(model="o1-preview", system_prompt="You are a noobi AI assistant who always uses the word 'noobi' in every response. For example, you might say 'Noobi will tell you...' or 'This noobi thinks that...'.")
     response = ai.chat(input(">>> "))
     for chunk in response:
         print(chunk, end="", flush=True)

webscout-6.0/webscout/Provider/ChatHub.py

@@ -0,0 +1,209 @@
+import requests
+import json
+import os
+from typing import Any, Dict, Optional, Generator, List, Union
+
+from webscout.AIutel import Optimizers
+from webscout.AIutel import Conversation
+from webscout.AIutel import AwesomePrompts, sanitize_stream
+from webscout.AIbase import Provider, AsyncProvider
+from webscout import exceptions
+
+class ChatHub(Provider):
+    """
+    A class to interact with the ChatHub API.
+    """
+
+    AVAILABLE_MODELS = [
+        'meta/llama3.1-8b',
+        'mistral/mixtral-8x7b',
+        'google/gemma-2',
+        'perplexity/sonar-online',
+    ]
+    model_aliases = {  # Aliases for shorter model names
+        "llama3.1-8b": 'meta/llama3.1-8b',
+        "mixtral-8x7b": 'mistral/mixtral-8x7b',
+        "gemma-2": 'google/gemma-2',
+        "sonar-online": 'perplexity/sonar-online',
+    }
+
+
+    def __init__(
+        self,
+        is_conversation: bool = True,
+        max_tokens: int = 2049,  
+        timeout: int = 30,
+        intro: str = None,
+        filepath: str = None,
+        update_file: bool = True,
+        proxies: dict = {},
+        history_offset: int = 10250,
+        act: str = None,
+        model: str = "sonar-online", 
+    ):
+        """Initializes the ChatHub API client."""
+        self.url = "https://app.chathub.gg"
+        self.api_endpoint = "https://app.chathub.gg/api/v3/chat/completions"
+        self.headers = {
+            'Accept': '*/*',
+            'Accept-Language': 'en-US,en;q=0.9',
+            'Content-Type': 'application/json',
+            'Origin': self.url,
+            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/129.0.0.0 Safari/537.36',
+            'X-App-Id': 'web'
+        }
+        self.session = requests.Session()
+        self.session.headers.update(self.headers) 
+        self.session.proxies.update(proxies) 
+        self.timeout = timeout
+        self.last_response = {}
+
+        self.is_conversation = is_conversation
+        self.max_tokens_to_sample = max_tokens
+        self.__available_optimizers = (
+            method
+            for method in dir(Optimizers)
+            if callable(getattr(Optimizers, method)) and not method.startswith("__")
+        )
+        Conversation.intro = (
+            AwesomePrompts().get_act(
+                act, raise_not_found=True, default=None, case_insensitive=True
+            )
+            if act
+            else intro or Conversation.intro
+        )
+
+        self.conversation = Conversation(
+            is_conversation, self.max_tokens_to_sample, filepath, update_file
+        )
+        self.conversation.history_offset = history_offset
+
+        #Resolve the model
+        self.model = self.get_model(model)
+
+
+    def get_model(self, model: str) -> str:
+        """
+        Resolves the model name using aliases or defaults.
+        """
+
+        if model in self.AVAILABLE_MODELS:
+            return model
+        elif model in self.model_aliases:
+            return self.model_aliases[model]
+        else:
+            print(f"Model '{model}' not found. Using default model '{self.default_model}'.")
+            return self.default_model # Use class-level default
+
+    def ask(
+        self,
+        prompt: str,
+        stream: bool = False,
+        raw: bool = False,
+        optimizer: str = None,
+        conversationally: bool = False,
+    ) -> Union[Dict[str, Any], Generator]:
+
+        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
+                )
+            else:
+                raise Exception(f"Optimizer is not one of {self.__available_optimizers}")
+
+
+        data = {
+            "model": self.model,
+            "messages": [{"role": "user", "content": conversation_prompt}],
+            "tools": []
+        }
+
+        # Set the Referer header dynamically based on the resolved model
+        self.headers['Referer'] = f"{self.url}/chat/{self.model}"
+
+
+        def for_stream():
+            try:
+                with requests.post(self.api_endpoint, headers=self.headers, json=data, stream=True, timeout=self.timeout) as response:
+                    response.raise_for_status()  
+                    streaming_text = ""
+
+                    for line in response.iter_lines(decode_unicode=True):
+                        if line:
+                            decoded_line = line.strip()
+                            if decoded_line.startswith('data:'):
+                                data_str = decoded_line[5:].strip()
+                                if data_str == '[DONE]':
+                                    break
+                                try:
+                                    data_json = json.loads(data_str)
+                                    text_delta = data_json.get('textDelta')
+                                    if text_delta:
+                                        streaming_text += text_delta
+                                        resp = dict(text=text_delta) 
+                                        yield resp if raw else resp 
+
+                                except json.JSONDecodeError:
+                                    continue
+                    self.conversation.update_chat_history(prompt, streaming_text)
+                    self.last_response.update({"text": streaming_text})
+            except requests.exceptions.RequestException as e:
+                raise exceptions.FailedToGenerateResponseError(f"Request error: {e}")
+
+
+        def for_non_stream():
+            for _ in for_stream():
+                pass
+            return self.last_response
+
+        return for_stream() if stream else for_non_stream()
+
+
+
+
+    def chat(
+        self,
+        prompt: str,
+        stream: bool = False,
+        optimizer: str = None,
+        conversationally: bool = False,
+    ) -> Union[str, Generator]:
+        """Generate response `str`"""
+
+        def for_stream():
+            for response in self.ask(
+                prompt, stream=True, optimizer=optimizer, conversationally=conversationally
+            ):
+                yield self.get_message(response)
+
+        def for_non_stream():
+            return self.get_message(
+                self.ask(
+                    prompt,
+                    stream=False,  # Pass stream=False
+                    optimizer=optimizer,
+                    conversationally=conversationally,
+                )
+            )
+
+        return for_stream() if stream else for_non_stream()
+
+
+
+    def get_message(self, response: dict) -> str:
+        """Retrieves message only from response"""
+        assert isinstance(response, dict), "Response should be of dict data-type only"
+        return response.get("text", "")
+
+
+if __name__ == "__main__":
+    from rich import print
+    bot = ChatHub()
+    try:
+        response = bot.chat("who is Abhay koul in AI", stream=True)
+        for chunk in response:
+            print(chunk, end="", flush=True)
+    except Exception as e:
+        print(f"An error occurred: {e}")

{webscout-5.9 → webscout-6.0}/webscout/Provider/Chatify.py

@@ -115,7 +115,7 @@ class Chatify(Provider):
                     if len(parts) > 1:
                         content = parts[1].strip().strip('"')
                         streaming_text += content
-                        yield content if raw else dict(text=streaming_text)
+                        yield content if raw else dict(text=content)
             self.last_response.update(dict(text=streaming_text))
             self.conversation.update_chat_history(
                 prompt, self.get_message(self.last_response)
@@ -169,7 +169,7 @@ class Chatify(Provider):
 if __name__ == "__main__":
     from rich import print
 
-    ai = Chatify() 
-    response = ai.chat("hi")
+    ai = Chatify(timeout=5000) 
+    response = ai.chat("write a poem about AI", stream=True)
     for chunk in response:
         print(chunk, end="", flush=True)

{webscout-5.9 → webscout-6.0}/webscout/Provider/Cloudflare.py

@@ -194,7 +194,7 @@ class Cloudflare(Provider):
                     data = json.loads(line[6:])
                     content = data.get('response', '')
                     streaming_response += content
-                    yield content if raw else dict(text=streaming_response)
+                    yield content if raw else dict(text=content)
             self.last_response.update(dict(text=streaming_response))
             self.conversation.update_chat_history(
                 prompt, self.get_message(self.last_response)
@@ -255,7 +255,7 @@ class Cloudflare(Provider):
         return response["text"]
 if __name__ == '__main__':
     from rich import print
-    ai = Cloudflare()
-    response = ai.chat("hi")
+    ai = Cloudflare(timeout=5000)
+    response = ai.chat("write a poem about AI", stream=True)
     for chunk in response:
         print(chunk, end="", flush=True)

{webscout-5.9 → webscout-6.0}/webscout/Provider/DARKAI.py

@@ -156,7 +156,7 @@ class DARKAI(Provider):
                                 if event.get("event") == "final-response":
                                     message = event['data'].get('message', '')
                                     streaming_response += message
-                                    yield message if raw else dict(text=streaming_response)
+                                    yield message if raw else dict(text=message)
                             except json.decoder.JSONDecodeError:
                                 continue
             self.last_response.update(dict(text=streaming_response))

webscout-6.0/webscout/Provider/Deepinfra.py

@@ -0,0 +1,172 @@
+import requests
+import json
+import os
+from typing import Any, Dict, Optional, Generator, Union
+
+from webscout.AIutel import Optimizers
+from webscout.AIutel import Conversation
+from webscout.AIutel import AwesomePrompts, sanitize_stream
+from webscout.AIbase import Provider, AsyncProvider
+from webscout import exceptions
+
+class DeepInfra(Provider):
+    """
+    A class to interact with the DeepInfra API.
+    """
+
+    def __init__(
+        self,
+        is_conversation: bool = True,
+        max_tokens: int = 2049,  # Set a reasonable default
+        timeout: int = 30,
+        intro: str = None,
+        filepath: str = None,
+        update_file: bool = True,
+        proxies: dict = {},
+        history_offset: int = 10250,
+        act: str = None,
+        model: str = "Qwen/Qwen2.5-72B-Instruct",  
+    ):
+        """Initializes the DeepInfra API client."""
+        self.url = "https://api.deepinfra.com/v1/openai/chat/completions"
+        self.headers = {
+            "Accept": "text/event-stream, application/json",
+
+        }
+        self.session = requests.Session()
+        self.session.headers.update(self.headers)
+        self.session.proxies.update(proxies)
+
+        self.is_conversation = is_conversation
+        self.max_tokens_to_sample = max_tokens
+        self.timeout = timeout
+        self.last_response = {}
+        self.model = model
+
+        self.__available_optimizers = (
+            method
+            for method in dir(Optimizers)
+            if callable(getattr(Optimizers, method)) and not method.startswith("__")
+        )
+        Conversation.intro = (
+            AwesomePrompts().get_act(
+                act, raise_not_found=True, default=None, case_insensitive=True
+            )
+            if act
+            else intro or Conversation.intro
+        )
+
+        self.conversation = Conversation(
+            is_conversation, self.max_tokens_to_sample, filepath, update_file
+        )
+        self.conversation.history_offset = history_offset
+
+    def ask(
+        self,
+        prompt: str,
+        stream: bool = False,
+        raw: bool = False,
+        optimizer: str = None,
+        conversationally: bool = False,
+    ) -> Union[Dict[str, Any], Generator]:
+
+        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
+                )
+            else:
+                raise Exception(f"Optimizer is not one of {self.__available_optimizers}")
+
+        # Payload construction
+        payload = {
+            "model": self.model,
+            "messages": [
+                {"role": "system", "content": "You are a helpful assistant."},
+                {"role": "user", "content": conversation_prompt},
+            ],
+            "stream": stream
+        }
+
+        def for_stream():
+            try:
+                with requests.post(self.url, headers=self.headers, data=json.dumps(payload), stream=True, timeout=self.timeout) as response:
+                    if response.status_code != 200:
+                        raise exceptions.FailedToGenerateResponseError(f"Request failed with status code {response.status_code}")
+
+                    streaming_text = ""
+                    for line in response.iter_lines(decode_unicode=True):  # Decode lines
+                        if line:
+                            line = line.strip()
+                            if line.startswith("data: "):
+                                json_str = line[6:] #Remove "data: " prefix
+                                if json_str == "[DONE]":
+                                    break
+                                try:
+                                    json_data = json.loads(json_str)
+                                    if 'choices' in json_data:
+                                        choice = json_data['choices'][0]
+                                        if 'delta' in choice and 'content' in choice['delta']:
+                                            content = choice['delta']['content']
+                                            streaming_text += content
+                                            
+                                            # Yield ONLY the new content:
+                                            resp = dict(text=content) 
+                                            yield resp if raw else resp
+                                except json.JSONDecodeError:
+                                    pass  # Or handle the error as needed
+                    self.conversation.update_chat_history(prompt, streaming_text)  # Update history *after* streaming
+            except requests.RequestException as e:
+                raise exceptions.FailedToGenerateResponseError(f"Request failed: {e}")
+
+
+        def for_non_stream():
+            # let's make use of stream
+            for _ in for_stream():
+                pass
+            return self.last_response
+
+
+        return for_stream() if stream else for_non_stream()
+
+
+
+    def chat(
+        self,
+        prompt: str,
+        stream: bool = False,
+        optimizer: str = None,
+        conversationally: bool = False,
+    ) -> str:
+
+        def for_stream():
+            for response in self.ask(
+                prompt, True, optimizer=optimizer, conversationally=conversationally
+            ):
+                yield self.get_message(response)
+
+        def for_non_stream():
+            return self.get_message(
+                self.ask(
+                    prompt,
+                    False,
+                    optimizer=optimizer,
+                    conversationally=conversationally,
+                )
+            )
+
+        return for_stream() if stream else for_non_stream()
+
+    def get_message(self, response: dict) -> str:
+        assert isinstance(response, dict), "Response should be of dict data-type only"
+        return response["text"]
+
+
+
+if __name__ == "__main__":
+    from rich import print
+    ai = DeepInfra(timeout=5000)
+    response = ai.chat("write a poem about AI", stream=True)
+    for chunk in response:
+        print(chunk, end="", flush=True)

{webscout-5.9 → webscout-6.0}/webscout/Provider/Deepseek.py

@@ -151,16 +151,14 @@ class DeepSeek(Provider):
                     f"Failed to generate response - ({response.status_code}, {response.reason})"
                 )
             streaming_response = ""
-            collected_messages = []
             for line in response.iter_lines():
                 if line:
                     json_line = json.loads(line.decode('utf-8').split('data: ')[1])
                     if 'choices' in json_line and len(json_line['choices']) > 0:
                         delta_content = json_line['choices'][0].get('delta', {}).get('content')
                         if delta_content:
-                            collected_messages.append(delta_content)
-                            streaming_response = ''.join(collected_messages) 
-                            yield delta_content if raw else dict(text=streaming_response)
+                            streaming_response += delta_content
+                            yield delta_content if raw else dict(text=delta_content)
             self.last_response.update(dict(text=streaming_response))
             self.conversation.update_chat_history(
                 prompt, self.get_message(self.last_response)
@@ -222,7 +220,7 @@ class DeepSeek(Provider):
 
 if __name__ == '__main__':
     from rich import print
-    ai = DeepSeek(api_key="")
-    response = ai.chat("tell me about india")
+    ai = DeepSeek(api_key="", timeout=5000)
+    response = ai.chat("write a poem about AI", stream=True)
     for chunk in response:
         print(chunk, end="", flush=True)

{webscout-5.9 → webscout-6.0}/webscout/Provider/DiscordRocks.py

@@ -167,7 +167,7 @@ class DiscordRocks(Provider):
                                 content = json_data['choices'][0]['delta'].get('content', '')
                                 if content:
                                     full_content += content
-                                    yield content if raw else dict(text=full_content)
+                                    yield content if raw else dict(text=content)
                             except json.JSONDecodeError:
                                 print(f'Error decoding JSON: {decoded_line}')
                             except KeyError:
@@ -247,7 +247,7 @@ class DiscordRocks(Provider):
 
 if __name__ == '__main__':
     from rich import print
-    ai = DiscordRocks()
-    response = ai.chat(input(">>> "))
+    ai = DiscordRocks(timeout=5000)
+    response = ai.chat("write a poem about AI", stream=True)
     for chunk in response:
         print(chunk, end="", flush=True)