webscout 6.7__py3-none-any.whl → 6.9__py3-none-any.whl

webscout/Extra/YTToolkit/YTdownloader.py

@@ -12,10 +12,15 @@ from sys import stdout
 import os
 import subprocess
 import sys
+import tempfile
 from webscout.version import __prog__, __version__
-from webscout.zerodir import user_cache_dir
 from webscout.swiftcli import CLI, option, argument, group
 
+# Define cache directory using tempfile
+user_cache_dir = os.path.join(tempfile.gettempdir(), 'webscout')
+if not os.path.exists(user_cache_dir):
+    os.makedirs(user_cache_dir)
+
 logging = LitLogger(name="YTDownloader")
 
 session = requests.session()
@@ -32,7 +37,7 @@ session.headers.update(headers)
 
 get_excep = lambda e: e.args[1] if len(e.args) > 1 else e
 
-appdir = user_cache_dir(__prog__, __prog__)
+appdir = user_cache_dir
 
 if not path.isdir(appdir):
     try:

webscout/Extra/YTToolkit/ytapi/channel.py

@@ -304,4 +304,4 @@ class Channel:
         List[str] | None
             The ids of all playlists or None
         """
-        return dup_filter(Patterns.playlists.findall(channel_playlists(self._target_url)))
+        return dup_filter(Patterns.playlists.findall(channel_playlists(self._target_url)))

webscout/Extra/YTToolkit/ytapi/query.py

@@ -35,3 +35,6 @@ class Search:
     @staticmethod
     def playlists(keywords: str, limit: int = 20) -> Optional[List[str]]:
         return dup_filter(Patterns.playlist_id.findall(find_playlists(keywords)), limit)
+
+if __name__ == "__main__":
+    print(Search.videos("java"))

webscout/Extra/YTToolkit/ytapi/stream.py

@@ -58,3 +58,6 @@ class Video:
             'tags': data[7].split(',') if data[7] else None,
             'description': data[8].replace('\\n', '\n') if data[8] else None
         }
+
+if __name__ == '__main__':
+    print(Video('https://www.youtube.com/watch?v=9bZkp7q19f0').metadata)

webscout/Extra/YTToolkit/ytapi/video.py

@@ -99,4 +99,6 @@ class Video:
             data['genre'] = genre_pattern.search(self._video_data).group(1)
         except AttributeError:
             data['genre'] = None
-        return data
+        return data
+if __name__ == '__main__':
+    print(Video('https://www.youtube.com/watch?v=9bZkp7q19f0').metadata)

webscout/Provider/Cloudflare.py

@@ -1,5 +1,6 @@
 import json
 from uuid import uuid4
+import webscout
 from webscout.AIutel import Optimizers
 from webscout.AIutel import Conversation
 from webscout.AIutel import AwesomePrompts, sanitize_stream
@@ -107,7 +108,7 @@ class Cloudflare(Provider):
             'Sec-Fetch-Dest': 'empty',
             'Sec-Fetch-Mode': 'cors',
             'Sec-Fetch-Site': 'same-origin',
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36 Edg/127.0.0.0',
+            'User-Agent': webscout.LitAgent().random()
         }
 
         self.cookies = {

webscout/Provider/DARKAI.py

@@ -4,7 +4,7 @@ from webscout.AIutel import Optimizers
 from webscout.AIutel import Conversation
 from webscout.AIutel import AwesomePrompts, sanitize_stream
 from webscout.AIbase import Provider
-from webscout import exceptions
+from webscout import exceptions, LitAgent
 import requests
 
 class DARKAI(Provider):
@@ -75,7 +75,7 @@ class DARKAI(Provider):
             "sec-fetch-dest": "empty",
             "sec-fetch-mode": "cors",
             "sec-fetch-site": "cross-site",
-            "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/127.0.0.0 Safari/537.36 Edg/127.0.0.0"
+            "user-agent": LitAgent().random(),
         }
 
         self.__available_optimizers = (

webscout/Provider/Free2GPT.py

@@ -9,7 +9,7 @@ from webscout.AIutel import Conversation
 from webscout.AIutel import AwesomePrompts
 from webscout.AIbase import Provider
 from webscout import exceptions
-
+from webscout import LitAgent
 
 class Free2GPT(Provider):
     """
@@ -48,7 +48,7 @@ class Free2GPT(Provider):
         self.session = requests.Session()
         self.is_conversation = is_conversation
         self.max_tokens_to_sample = max_tokens
-        self.api_endpoint = "https://chat10.free2gpt.xyz/api/generate"
+        self.api_endpoint = "https://chat1.free2gpt.com/api/generate"
         self.stream_chunk_size = 64
         self.timeout = timeout
         self.last_response = {}
@@ -59,15 +59,15 @@ class Free2GPT(Provider):
             "accept-language": "en-US,en;q=0.9,en-IN;q=0.8",
             "content-type": "text/plain;charset=UTF-8",
             "dnt": "1",
-            "origin": "https://chat10.free2gpt.xyz",
-            "referer": "https://chat10.free2gpt.xyz/",
+            "origin": "https://chat1.free2gpt.co",
+            "referer": "https://chat1.free2gpt.co",
             "sec-ch-ua": '"Chromium";v="128", "Not;A=Brand";v="24", "Microsoft Edge";v="128"',
             "sec-ch-ua-mobile": "?0",
             "sec-ch-ua-platform": '"Windows"',
             "sec-fetch-dest": "empty",
             "sec-fetch-mode": "cors",
             "sec-fetch-site": "same-origin",
-            "user-agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/128.0.0.0 Safari/537.36 Edg/128.0.0.0"
+            "user-agent": LitAgent().random(),
         }
 
         self.__available_optimizers = (

webscout/Provider/Marcus.py

@@ -7,7 +7,7 @@ from webscout.AIutel import Conversation
 from webscout.AIutel import AwesomePrompts
 from webscout.AIbase import Provider
 from webscout import exceptions
-
+from webscout import LitAgent as Lit
 
 class Marcus(Provider):
     """
@@ -39,7 +39,7 @@ class Marcus(Provider):
             'accept': '*/*',
             'origin': 'https://www.askmarcus.app',
             'referer': 'https://www.askmarcus.app/chat',
-            'user-agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/131.0.0.0 Safari/537.36'
+            'user-agent': Lit().random(),
         }
         self.__available_optimizers = (
             method
@@ -134,4 +134,4 @@ if __name__ == '__main__':
     ai = Marcus(timeout=30)
     response = ai.chat("Tell me about India", stream=True)
     for chunk in response:
-        print(chunk)
+        print(chunk, end="", flush=True)

webscout/Provider/PI.py

@@ -7,21 +7,19 @@ from webscout.AIutel import Optimizers
 from webscout.AIutel import Conversation
 from webscout.AIutel import AwesomePrompts
 from webscout.AIbase import Provider
-from typing import Dict
+from typing import Dict, Union, Any
 from webscout import LitAgent
-class PiAI(Provider):
-    """PiAI class for interacting with the Pi.ai chat API, extending the Provider class.
+from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
 
-    This class provides methods for sending messages to the Pi.ai chat API and receiving responses,
-    enabling conversational interactions. It supports various configurations such as conversation mode,
-    token limits, and history management.
+class PiAI(Provider):
+    """
+    PiAI is a provider class for interacting with the Pi.ai chat API.
 
     Attributes:
-        scraper (cloudscraper.CloudScraper): The scraper instance for handling HTTP requests.
-        url (str): The API endpoint for the Pi.ai chat service.
-        AVAILABLE_VOICES (Dict[str, int]): A dictionary mapping voice names to their corresponding IDs.
-        headers (Dict[str, str]): The headers to be used in HTTP requests to the API.
+        knowledge_cutoff (str): The knowledge cutoff date for the model
+        AVAILABLE_VOICES (Dict[str, int]): Available voice options for audio responses
     """
+
     def __init__(
         self,
         is_conversation: bool = True,
@@ -33,19 +31,30 @@ class PiAI(Provider):
         proxies: dict = {},
         history_offset: int = 10250,
         act: str = None,
+        logging: bool = False,
     ):
-        """Initializes the PiAI class for interacting with the Pi.ai chat API.
+        """
+        Initializes the PiAI provider with specified parameters.
 
         Args:
-            is_conversation (bool, optional): Flag for enabling conversational mode. Defaults to True.
-            max_tokens (int, optional): Maximum number of tokens to generate in the response. Defaults to 600.
-            timeout (int, optional): Timeout duration for HTTP requests in seconds. Defaults to 30.
-            intro (str, optional): Introductory prompt for the conversation. Defaults to None.
-            filepath (str, optional): Path to a file for storing conversation history. Defaults to None.
-            update_file (bool, optional): Indicates whether to update the file with new prompts and responses. Defaults to True.
-            proxies (dict, optional): Dictionary of HTTP request proxies. Defaults to an empty dictionary.
-            history_offset (int, optional): Number of last messages to retain in conversation history. Defaults to 10250.
-            act (str|int, optional): Key or index for selecting an awesome prompt to use as an intro. Defaults to None.
+            is_conversation (bool): Whether to maintain conversation history
+            max_tokens (int): Maximum number of tokens in response
+            timeout (int): Request timeout in seconds
+            intro (str): Custom introduction message
+            filepath (str): Path to save conversation history
+            update_file (bool): Whether to update conversation history file
+            proxies (dict): Proxy configuration
+            history_offset (int): Conversation history limit
+            act (str): Custom personality/act for the AI
+            logging (bool): Enable debug logging
+
+        Examples:
+            >>> ai = PiAI(logging=True)
+            >>> ai.ask("What's the weather today?", "Alice")
+            Sends a prompt to Pi.ai and returns the response.
+
+            >>> ai.chat("Tell me a joke", voice_name="William")
+            Initiates a chat with Pi.ai using the provided prompt.
         """
         self.scraper = cloudscraper.create_scraper()
         self.url = 'https://pi.ai/api/chat'
@@ -84,7 +93,7 @@ class PiAI(Provider):
         self.max_tokens_to_sample = max_tokens
         self.stream_chunk_size = 64
         self.timeout = timeout
-        self.last_response = {}
+        self.last_response = {} if self.is_conversation else {'text': ""}
         self.conversation_id = None
 
         self.__available_optimizers = (
@@ -105,11 +114,35 @@ class PiAI(Provider):
         )
         self.conversation.history_offset = history_offset
         self.session.proxies = proxies
+
+        # Initialize logger
+        self.logger = LitLogger(name="PiAI", format=LogFormat.MODERN_EMOJI, color_scheme=ColorScheme.CYBERPUNK) if logging else None
+        
+        self.knowledge_cutoff = "December 2023"
+        
         # Initialize conversation ID
         if self.is_conversation:
             self.start_conversation()
 
+        if self.logger:
+            self.logger.debug("PiAI instance initialized")
+
     def start_conversation(self) -> str:
+        """
+        Initializes a new conversation and returns the conversation ID.
+
+        Returns:
+            str: The conversation ID from Pi.ai
+
+        Examples:
+            >>> ai = PiAI()
+            >>> conversation_id = ai.start_conversation()
+            >>> print(conversation_id)
+            'abc123xyz'
+        """
+        if self.logger:
+            self.logger.debug("Starting new conversation")
+            
         response = self.scraper.post(
             "https://pi.ai/api/chat/start",
             headers=self.headers,
@@ -117,8 +150,16 @@ class PiAI(Provider):
             json={},
             timeout=self.timeout
         )
+        
+        if not response.ok and self.logger:
+            self.logger.error(f"Failed to start conversation: {response.status_code}")
+            
         data = response.json()
         self.conversation_id = data['conversations'][0]['sid']
+        
+        if self.logger:
+            self.logger.debug(f"Conversation started with ID: {self.conversation_id}")
+            
         return self.conversation_id
 
     def ask(
@@ -132,26 +173,31 @@ class PiAI(Provider):
         verbose:bool = None,
         output_file:str = None
     ) -> dict:
-        """Interact with the AI by sending a prompt and receiving a response.
+        """
+        Interact with Pi.ai by sending a prompt and receiving a response.
 
         Args:
-            prompt (str): The prompt to be sent to the AI.
-            voice_name (str): The name of the voice to use for audio responses.
-            stream (bool, optional): Flag for streaming response. Defaults to False.
-            raw (bool, optional): If True, returns the raw response as received. Defaults to False.
-            optimizer (str, optional): Name of the prompt optimizer to use - `[code, shell_command]`. Defaults to None.
-            conversationally (bool, optional): If True, chat conversationally when using optimizer. Defaults to False.
-            verbose (bool, optional): If True, provides detailed output. Defaults to None.
-            output_file (str, optional): File path to save the output. Defaults to None.
+            prompt (str): The prompt to be sent to Pi.ai
+            voice_name (str): The name of the voice to use for audio responses
+            stream (bool): Flag for streaming response
+            raw (bool): If True, returns the raw response as received
+            optimizer (str): Name of the prompt optimizer to use
+            conversationally (bool): If True, chat conversationally when using optimizer
+            verbose (bool): If True, provides detailed output
+            output_file (str): File path to save the output
 
         Returns:
-            dict: A dictionary containing the AI's response.
-        ```json
-        {
-            "text": "How may I assist you today?"
-        }
-        ```
+            dict: A dictionary containing the AI's response
+
+        Examples:
+            >>> ai = PiAI(logging=True)
+            >>> response = ai.ask("Hello!", "Alice", verbose=True)
+            >>> print(response['text'])
+            'Hi! How can I help you today?'
         """
+        if self.logger:
+            self.logger.debug(f"ask() called with prompt: {prompt}, voice: {voice_name}")
+
         conversation_prompt = self.conversation.gen_complete_prompt(prompt)
         if optimizer:
             if optimizer in self.__available_optimizers:
@@ -159,6 +205,8 @@ class PiAI(Provider):
                     conversation_prompt if conversationally else prompt
                 )
             else:
+                if self.logger:
+                    self.logger.error(f"Invalid optimizer: {optimizer}")
                 raise Exception(
                     f"Optimizer is not one of {self.__available_optimizers}"
                 )
@@ -187,8 +235,8 @@ class PiAI(Provider):
                             resp = dict(text=streaming_text)
                             self.last_response.update(resp)
                             yield parsed_data if raw else resp
-                    except json.JSONDecodeError:
-                        continue
+                    except:continue
+
             self.conversation.update_chat_history(
                 prompt, self.get_message(self.last_response)
             )
@@ -210,20 +258,30 @@ class PiAI(Provider):
         verbose:bool = True,
         output_file:str = "PiAi.mp3"
     ) -> str:
-        """Generates a response based on the provided prompt.
+        """
+        Generates a response based on the provided prompt.
 
         Args:
-            prompt (str): The input prompt to be sent for generating a response.
-            voice_name (str, optional): The name of the voice to use for the response. Defaults to "Alice".
-            stream (bool, optional): Flag for streaming the response. Defaults to False.
-            optimizer (str, optional): The name of the prompt optimizer to use - `[code, shell_command]`. Defaults to None.
-            conversationally (bool, optional): Indicates whether to chat conversationally when using the optimizer. Defaults to False.
-            verbose (bool, optional): Flag to indicate if verbose output is desired. Defaults to True.
-            output_file (str, optional): The file path where the audio will be saved. Defaults to "PiAi.mp3".
+            prompt (str): Input prompt for generating response
+            voice_name (str): Voice to use for audio response
+            stream (bool): Enable response streaming
+            optimizer (str): Prompt optimizer to use
+            conversationally (bool): Enable conversational mode with optimizer
+            verbose (bool): Enable verbose output
+            output_file (str): Audio output file path
 
         Returns:
-            str: The generated response.
+            str: The generated response
+
+        Examples:
+            >>> ai = PiAI(logging=True)
+            >>> response = ai.chat("Tell me a joke", voice_name="William")
+            >>> print(response)
+            'Why did the scarecrow win an award? Because he was outstanding in his field!'
         """
+        if self.logger:
+            self.logger.debug(f"chat() called with prompt: {prompt}, voice: {voice_name}")
+
         assert (
             voice_name in self.AVAILABLE_VOICES
         ), f"Voice '{voice_name}' not one of [{', '.join(self.AVAILABLE_VOICES.keys())}]"
@@ -271,6 +329,9 @@ class PiAI(Provider):
             verbose (bool): Flag to indicate if verbose output is desired.
             output_file (str): The file path where the audio will be saved.
         """
+        if self.logger:
+            self.logger.debug(f"Downloading audio with voice: {voice_name}")
+
         params = {
             'mode': 'eager',
             'voice': f'voice{self.AVAILABLE_VOICES[voice_name]}',
@@ -278,11 +339,16 @@ class PiAI(Provider):
         }
         try:
             audio_response = self.scraper.get('https://pi.ai/api/chat/voice', params=params, cookies=self.cookies, headers=self.headers, timeout=self.timeout)
+            if not audio_response.ok and self.logger:
+                self.logger.error(f"Audio download failed: {audio_response.status_code}")
+                
             audio_response.raise_for_status()  # Raise an exception for bad status codes
             with open(output_file, "wb") as file:
                 file.write(audio_response.content)
             if verbose:print("\nAudio file downloaded successfully.")
         except requests.exceptions.RequestException as e:
+            if self.logger:
+                self.logger.error(f"Audio download failed: {e}")
             if verbose:print(f"\nFailed to download audio file. Error: {e}")
 
 if __name__ == '__main__':

webscout/Provider/Phind.py

@@ -501,3 +501,9 @@ class Phindv2(Provider):
                 else ""
             )
 
+if __name__ == "__main__":
+    from rich import print
+
+    ai = Phindv2() 
+    print(ai.chat("Tell me a joke"))
+    # Returns the chat response from the Phindv2 API.

webscout/Provider/PizzaGPT.py

@@ -1,17 +1,21 @@
 import requests
-from typing import Any, AsyncGenerator, Dict, Optional
+from typing import Any, AsyncGenerator, Dict, Optional, Union, Generator
 import json
 
 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.AIbase import Provider, AsyncProvider
 from webscout import exceptions
 from webscout import LitAgent as Lit
+from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
 
 class PIZZAGPT(Provider):
     """
-    A class to interact with the PizzaGPT API.
+    PIZZAGPT is a provider class for interacting with the PizzaGPT API.
+
+    Attributes:
+        knowledge_cutoff (str): The knowledge cutoff date for the model
     """
 
     def __init__(
@@ -25,20 +29,18 @@ class PIZZAGPT(Provider):
         proxies: dict = {},
         history_offset: int = 10250,
         act: str = None,
+        logging: bool = False,
     ) -> None:
         """
-        Initializes the PizzaGPT API with given parameters.
+        Initializes the PizzaGPT provider with the specified parameters.
 
-        Args:
-            is_conversation (bool, optional): Flag for chatting conversationally. Defaults to True.
-            max_tokens (int, optional): Maximum number of tokens to be generated upon completion. Defaults to 600.
-            timeout (int, optional): Http request timeout. Defaults to 30.
-            intro (str, optional): Conversation introductory prompt. Defaults to None.
-            filepath (str, optional): Path to file containing conversation history. Defaults to None.
-            update_file (bool, optional): Add new prompts and responses to the file. Defaults to True.
-            proxies (dict, optional): Http request proxies. Defaults to {}.
-            history_offset (int, optional): Limit conversation history to this number of last texts. Defaults to 10250.
-            act (str|int, optional): Awesome prompt key or index. (Used as intro). Defaults to None.
+        Examples:
+            >>> ai = PIZZAGPT(logging=True)
+            >>> ai.ask("What's the weather today?")
+            Sends a prompt to the PizzaGPT API and returns the response.
+
+            >>> ai.chat("Tell me a joke")
+            Initiates a chat with the PizzaGPT API using the provided prompt.
         """
         self.session = requests.Session()
         self.is_conversation = is_conversation
@@ -85,6 +87,10 @@ class PIZZAGPT(Provider):
         )
         self.conversation.history_offset = history_offset
         self.session.proxies = proxies
+        
+        # Initialize logger
+        self.logger = LitLogger(name="PIZZAGPT", format=LogFormat.MODERN_EMOJI, color_scheme=ColorScheme.CYBERPUNK) if logging else None
+
 
     def ask(
         self,
@@ -93,23 +99,16 @@ class PIZZAGPT(Provider):
         raw: bool = False,
         optimizer: str = None,
         conversationally: bool = False,
-    ) -> dict:
-        """Chat with AI
+    ) -> Dict[str, Any]:
+        """
+        Sends a prompt to the PizzaGPT API and returns the response.
 
-        Args:
-            prompt (str): Prompt to be send.
-            stream (bool, optional): Flag for streaming response. Defaults to False.
-            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.
-        Returns:
-           dict : {}
-        ```json
-        {
-           "text" : "How may I assist you today?"
-        }
-        ```
+        Examples:
+            >>> ai = PIZZAGPT()
+            >>> ai.ask("What's the weather today?")
+            Returns the response from the PizzaGPT API.
         """
+
         conversation_prompt = self.conversation.gen_complete_prompt(prompt)
         if optimizer:
             if optimizer in self.__available_optimizers:
@@ -117,27 +116,37 @@ class PIZZAGPT(Provider):
                     conversation_prompt if conversationally else prompt
                 )
             else:
-                raise Exception(
-                    f"Optimizer is not one of {self.__available_optimizers}"
-                )
+                if self.logger:
+                    self.logger.error(f"Invalid optimizer: {optimizer}")
+                raise Exception(f"Optimizer is not one of {self.__available_optimizers}")
 
         self.session.headers.update(self.headers)
         payload = {"question": conversation_prompt}
 
-        response = self.session.post(
-            self.api_endpoint, json=payload, timeout=self.timeout
-        )
-        if not response.ok:
-            raise exceptions.FailedToGenerateResponseError(
-                f"Failed to generate response - ({response.status_code}, {response.reason}) - {response.text}"
+        try:
+            response = self.session.post(
+                self.api_endpoint, json=payload, timeout=self.timeout
             )
+            if self.logger:
+                self.logger.debug(response)
+            if not response.ok:
+                if self.logger:
+                    self.logger.error(f"Failed to generate response: {response.status_code} {response.reason}")
+                raise exceptions.FailedToGenerateResponseError(
+                    f"Failed to generate response - ({response.status_code}, {response.reason}) - {response.text}"
+                )
 
-        resp = response.json()
-        self.last_response.update(dict(text=resp['content']))
-        self.conversation.update_chat_history(
-            prompt, self.get_message(self.last_response)
-        )
-        return self.last_response # Return the updated last_response
+            resp = response.json()
+            if self.logger:
+                self.logger.debug(resp)
+            self.last_response.update(dict(text=resp['content']))
+            self.conversation.update_chat_history(
+                prompt, self.get_message(self.last_response)
+            )
+            return self.last_response
+
+        except Exception as e:
+            raise exceptions.FailedToGenerateResponseError(f"Request failed: {e}")
 
     def chat(
         self,
@@ -146,14 +155,13 @@ class PIZZAGPT(Provider):
         optimizer: str = None,
         conversationally: bool = False,
     ) -> str:
-        """Generate response `str`
-        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.
-        Returns:
-            str: Response generated
+        """
+        Initiates a chat with the PizzaGPT API using the provided prompt.
+
+        Examples:
+            >>> ai = PIZZAGPT()
+            >>> ai.chat("Tell me a joke")
+            Returns the chat response from the PizzaGPT API.
         """
 
         return self.get_message(
@@ -163,6 +171,7 @@ class PIZZAGPT(Provider):
                 conversationally=conversationally,
             )
         )
+
     def get_message(self, response: dict) -> str:
         """Retrieves message only from response
 
@@ -177,7 +186,7 @@ class PIZZAGPT(Provider):
 if __name__ == "__main__":
     from rich import print
 
-    ai = PIZZAGPT() 
+    ai = PIZZAGPT(logging=True) 
     # Stream the response
     response = ai.chat("hi")
     for chunk in response: