webscout 7.1__py3-none-any.whl → 7.2__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Potentially problematic release.


This version of webscout might be problematic. Click here for more details.

Files changed (144) hide show
  1. webscout/AIauto.py +191 -191
  2. webscout/AIbase.py +122 -122
  3. webscout/AIutel.py +440 -440
  4. webscout/Bard.py +343 -161
  5. webscout/DWEBS.py +489 -492
  6. webscout/Extra/YTToolkit/YTdownloader.py +995 -995
  7. webscout/Extra/YTToolkit/__init__.py +2 -2
  8. webscout/Extra/YTToolkit/transcriber.py +476 -479
  9. webscout/Extra/YTToolkit/ytapi/channel.py +307 -307
  10. webscout/Extra/YTToolkit/ytapi/playlist.py +58 -58
  11. webscout/Extra/YTToolkit/ytapi/pool.py +7 -7
  12. webscout/Extra/YTToolkit/ytapi/utils.py +62 -62
  13. webscout/Extra/YTToolkit/ytapi/video.py +103 -103
  14. webscout/Extra/autocoder/__init__.py +9 -9
  15. webscout/Extra/autocoder/autocoder_utiles.py +199 -199
  16. webscout/Extra/autocoder/rawdog.py +5 -7
  17. webscout/Extra/autollama.py +230 -230
  18. webscout/Extra/gguf.py +3 -3
  19. webscout/Extra/weather.py +171 -171
  20. webscout/LLM.py +442 -442
  21. webscout/Litlogger/__init__.py +67 -681
  22. webscout/Litlogger/core/__init__.py +6 -0
  23. webscout/Litlogger/core/level.py +20 -0
  24. webscout/Litlogger/core/logger.py +123 -0
  25. webscout/Litlogger/handlers/__init__.py +12 -0
  26. webscout/Litlogger/handlers/console.py +50 -0
  27. webscout/Litlogger/handlers/file.py +143 -0
  28. webscout/Litlogger/handlers/network.py +174 -0
  29. webscout/Litlogger/styles/__init__.py +7 -0
  30. webscout/Litlogger/styles/colors.py +231 -0
  31. webscout/Litlogger/styles/formats.py +377 -0
  32. webscout/Litlogger/styles/text.py +87 -0
  33. webscout/Litlogger/utils/__init__.py +6 -0
  34. webscout/Litlogger/utils/detectors.py +154 -0
  35. webscout/Litlogger/utils/formatters.py +200 -0
  36. webscout/Provider/AISEARCH/DeepFind.py +250 -250
  37. webscout/Provider/Blackboxai.py +3 -3
  38. webscout/Provider/ChatGPTGratis.py +226 -0
  39. webscout/Provider/Cloudflare.py +3 -4
  40. webscout/Provider/DeepSeek.py +218 -0
  41. webscout/Provider/Deepinfra.py +3 -3
  42. webscout/Provider/Free2GPT.py +131 -124
  43. webscout/Provider/Gemini.py +100 -115
  44. webscout/Provider/Glider.py +3 -3
  45. webscout/Provider/Groq.py +5 -1
  46. webscout/Provider/Jadve.py +3 -3
  47. webscout/Provider/Marcus.py +191 -192
  48. webscout/Provider/Netwrck.py +3 -3
  49. webscout/Provider/PI.py +2 -2
  50. webscout/Provider/PizzaGPT.py +2 -3
  51. webscout/Provider/QwenLM.py +311 -0
  52. webscout/Provider/TTI/AiForce/__init__.py +22 -22
  53. webscout/Provider/TTI/AiForce/async_aiforce.py +257 -257
  54. webscout/Provider/TTI/AiForce/sync_aiforce.py +242 -242
  55. webscout/Provider/TTI/Nexra/__init__.py +22 -22
  56. webscout/Provider/TTI/Nexra/async_nexra.py +286 -286
  57. webscout/Provider/TTI/Nexra/sync_nexra.py +258 -258
  58. webscout/Provider/TTI/PollinationsAI/__init__.py +23 -23
  59. webscout/Provider/TTI/PollinationsAI/async_pollinations.py +330 -330
  60. webscout/Provider/TTI/PollinationsAI/sync_pollinations.py +285 -285
  61. webscout/Provider/TTI/artbit/__init__.py +22 -22
  62. webscout/Provider/TTI/artbit/async_artbit.py +184 -184
  63. webscout/Provider/TTI/artbit/sync_artbit.py +176 -176
  64. webscout/Provider/TTI/blackbox/__init__.py +4 -4
  65. webscout/Provider/TTI/blackbox/async_blackbox.py +212 -212
  66. webscout/Provider/TTI/blackbox/sync_blackbox.py +199 -199
  67. webscout/Provider/TTI/deepinfra/__init__.py +4 -4
  68. webscout/Provider/TTI/deepinfra/async_deepinfra.py +227 -227
  69. webscout/Provider/TTI/deepinfra/sync_deepinfra.py +199 -199
  70. webscout/Provider/TTI/huggingface/__init__.py +22 -22
  71. webscout/Provider/TTI/huggingface/async_huggingface.py +199 -199
  72. webscout/Provider/TTI/huggingface/sync_huggingface.py +195 -195
  73. webscout/Provider/TTI/imgninza/__init__.py +4 -4
  74. webscout/Provider/TTI/imgninza/async_ninza.py +214 -214
  75. webscout/Provider/TTI/imgninza/sync_ninza.py +209 -209
  76. webscout/Provider/TTI/talkai/__init__.py +4 -4
  77. webscout/Provider/TTI/talkai/async_talkai.py +229 -229
  78. webscout/Provider/TTI/talkai/sync_talkai.py +207 -207
  79. webscout/Provider/TTS/deepgram.py +182 -182
  80. webscout/Provider/TTS/elevenlabs.py +136 -136
  81. webscout/Provider/TTS/gesserit.py +150 -150
  82. webscout/Provider/TTS/murfai.py +138 -138
  83. webscout/Provider/TTS/parler.py +133 -134
  84. webscout/Provider/TTS/streamElements.py +360 -360
  85. webscout/Provider/TTS/utils.py +280 -280
  86. webscout/Provider/TTS/voicepod.py +116 -116
  87. webscout/Provider/TextPollinationsAI.py +2 -3
  88. webscout/Provider/WiseCat.py +193 -0
  89. webscout/Provider/__init__.py +144 -134
  90. webscout/Provider/cerebras.py +242 -227
  91. webscout/Provider/chatglm.py +204 -204
  92. webscout/Provider/dgaf.py +2 -3
  93. webscout/Provider/gaurish.py +2 -3
  94. webscout/Provider/geminiapi.py +208 -208
  95. webscout/Provider/granite.py +223 -0
  96. webscout/Provider/hermes.py +218 -218
  97. webscout/Provider/llama3mitril.py +179 -179
  98. webscout/Provider/llamatutor.py +3 -3
  99. webscout/Provider/llmchat.py +2 -3
  100. webscout/Provider/meta.py +794 -794
  101. webscout/Provider/multichat.py +331 -331
  102. webscout/Provider/typegpt.py +359 -359
  103. webscout/Provider/yep.py +2 -2
  104. webscout/__main__.py +5 -5
  105. webscout/cli.py +319 -319
  106. webscout/conversation.py +241 -242
  107. webscout/exceptions.py +328 -328
  108. webscout/litagent/__init__.py +28 -28
  109. webscout/litagent/agent.py +2 -3
  110. webscout/litprinter/__init__.py +0 -58
  111. webscout/scout/__init__.py +8 -8
  112. webscout/scout/core.py +884 -884
  113. webscout/scout/element.py +459 -459
  114. webscout/scout/parsers/__init__.py +69 -69
  115. webscout/scout/parsers/html5lib_parser.py +172 -172
  116. webscout/scout/parsers/html_parser.py +236 -236
  117. webscout/scout/parsers/lxml_parser.py +178 -178
  118. webscout/scout/utils.py +38 -38
  119. webscout/swiftcli/__init__.py +811 -811
  120. webscout/update_checker.py +2 -12
  121. webscout/version.py +1 -1
  122. webscout/webscout_search.py +5 -4
  123. webscout/zeroart/__init__.py +54 -54
  124. webscout/zeroart/base.py +60 -60
  125. webscout/zeroart/effects.py +99 -99
  126. webscout/zeroart/fonts.py +816 -816
  127. {webscout-7.1.dist-info → webscout-7.2.dist-info}/METADATA +4 -3
  128. webscout-7.2.dist-info/RECORD +217 -0
  129. webstoken/__init__.py +30 -30
  130. webstoken/classifier.py +189 -189
  131. webstoken/keywords.py +216 -216
  132. webstoken/language.py +128 -128
  133. webstoken/ner.py +164 -164
  134. webstoken/normalizer.py +35 -35
  135. webstoken/processor.py +77 -77
  136. webstoken/sentiment.py +206 -206
  137. webstoken/stemmer.py +73 -73
  138. webstoken/tagger.py +60 -60
  139. webstoken/tokenizer.py +158 -158
  140. webscout-7.1.dist-info/RECORD +0 -198
  141. {webscout-7.1.dist-info → webscout-7.2.dist-info}/LICENSE.md +0 -0
  142. {webscout-7.1.dist-info → webscout-7.2.dist-info}/WHEEL +0 -0
  143. {webscout-7.1.dist-info → webscout-7.2.dist-info}/entry_points.txt +0 -0
  144. {webscout-7.1.dist-info → webscout-7.2.dist-info}/top_level.txt +0 -0
@@ -5,7 +5,7 @@ from typing import Any, Dict, Generator, Optional
5
5
  from webscout.AIutel import Optimizers, Conversation, AwesomePrompts
6
6
  from webscout.AIbase import Provider
7
7
  from webscout import exceptions
8
- from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
8
+ from webscout.Litlogger import Logger, LogFormat
9
9
  from webscout import LitAgent as Lit
10
10
 
11
11
  class GliderAI(Provider):
@@ -39,10 +39,10 @@ class GliderAI(Provider):
39
39
  if model not in self.AVAILABLE_MODELS:
40
40
  raise ValueError(f"Invalid model: {model}. Choose from: {', '.join(self.AVAILABLE_MODELS)}")
41
41
 
42
- self.logger = LitLogger(
42
+ self.logger = Logger(
43
43
  name="GliderAI",
44
44
  format=LogFormat.MODERN_EMOJI,
45
- color_scheme=ColorScheme.CYBERPUNK
45
+
46
46
  ) if logging else None
47
47
 
48
48
  if self.logger:
webscout/Provider/Groq.py CHANGED
@@ -18,12 +18,16 @@ class GROQ(Provider):
18
18
  AVAILABLE_MODELS = [
19
19
  # "whisper-large-v3",
20
20
  "llama3-70b-8192",
21
+ "qwen-2.5-32b",
22
+ "qwen-2.5-coder-32b",
23
+ "deepseek-r1-distill-qwen-32b",
24
+ "deepseek-r1-distill-llama-70b",
21
25
  "llama-3.2-3b-preview",
22
26
  "gemma2-9b-it",
23
27
  "llama-3.2-11b-vision-preview",
24
28
  "llama3-8b-8192",
25
29
  "llama-3.3-70b-versatile",
26
- "deepseek-r1-distill-llama-70b",
30
+ "llama-3.2-11b-vision-preview",
27
31
  # "distil-whisper-large-v3-en",
28
32
  "mixtral-8x7b-32768",
29
33
  "llama-3.3-70b-specdec",
@@ -8,7 +8,7 @@ from webscout.AIutel import Optimizers, Conversation, AwesomePrompts
8
8
  from webscout.AIbase import Provider
9
9
  from webscout import exceptions
10
10
  from webscout.litagent import LitAgent
11
- from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
11
+ from webscout.Litlogger import Logger, LogFormat
12
12
 
13
13
  class JadveOpenAI(Provider):
14
14
  """
@@ -53,10 +53,10 @@ class JadveOpenAI(Provider):
53
53
  if model not in self.AVAILABLE_MODELS:
54
54
  raise ValueError(f"Invalid model: {model}. Choose from: {self.AVAILABLE_MODELS}")
55
55
 
56
- self.logger = LitLogger(
56
+ self.logger = Logger(
57
57
  name="JadveOpenAI",
58
58
  format=LogFormat.MODERN_EMOJI,
59
- color_scheme=ColorScheme.CYBERPUNK
59
+
60
60
  ) if logging else None
61
61
 
62
62
  if self.logger:
@@ -1,192 +1,191 @@
1
- import requests
2
- import json
3
- from typing import Any, Dict, Optional, Generator
4
-
5
- from webscout.AIutel import Optimizers
6
- from webscout.AIutel import Conversation
7
- from webscout.AIutel import AwesomePrompts
8
- from webscout.AIbase import Provider
9
- from webscout import exceptions
10
- from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
11
- from webscout import LitAgent as Lit
12
-
13
- class Marcus(Provider):
14
- """
15
- This class provides methods for interacting with the AskMarcus API.
16
- Improved to match webscout provider standards with comprehensive logging.
17
- """
18
-
19
- def __init__(
20
- self,
21
- is_conversation: bool = True,
22
- max_tokens: int = 2048,
23
- timeout: int = 30,
24
- intro: str = None,
25
- filepath: str = None,
26
- update_file: bool = True,
27
- proxies: dict = {},
28
- history_offset: int = 10250,
29
- act: str = None,
30
- logging: bool = False
31
- ):
32
- """Initializes the Marcus API with logging capabilities."""
33
- self.logger = LitLogger(
34
- name="Marcus",
35
- format=LogFormat.MODERN_EMOJI,
36
- color_scheme=ColorScheme.CYBERPUNK
37
- ) if logging else None
38
-
39
- if self.logger:
40
- self.logger.info("Initializing Marcus API")
41
-
42
- self.session = requests.Session()
43
- self.is_conversation = is_conversation
44
- self.max_tokens_to_sample = max_tokens
45
- self.api_endpoint = "https://www.askmarcus.app/api/response"
46
- self.timeout = timeout
47
- self.last_response = {}
48
-
49
- self.headers = {
50
- 'content-type': 'application/json',
51
- 'accept': '*/*',
52
- 'origin': 'https://www.askmarcus.app',
53
- 'referer': 'https://www.askmarcus.app/chat',
54
- 'user-agent': Lit().random(),
55
- }
56
-
57
- self.__available_optimizers = (
58
- method
59
- for method in dir(Optimizers)
60
- if callable(getattr(Optimizers, method)) and not method.startswith("__")
61
- )
62
-
63
- Conversation.intro = (
64
- AwesomePrompts().get_act(
65
- act, raise_not_found=True, default=None, case_insensitive=True
66
- )
67
- if act
68
- else intro or Conversation.intro
69
- )
70
-
71
- self.conversation = Conversation(
72
- is_conversation, self.max_tokens_to_sample, filepath, update_file
73
- )
74
- self.conversation.history_offset = history_offset
75
- self.session.proxies = proxies
76
-
77
- if self.logger:
78
- self.logger.info("Marcus API initialized successfully")
79
-
80
- def ask(
81
- self,
82
- prompt: str,
83
- stream: bool = False,
84
- raw: bool = False,
85
- optimizer: str = None,
86
- conversationally: bool = False,
87
- ) -> Dict[str, Any] | Generator[str, None, None]:
88
- """Sends a prompt to the AskMarcus API and returns the response with logging."""
89
- if self.logger:
90
- self.logger.debug(f"Processing request - Prompt: {prompt[:50]}...")
91
- self.logger.debug(f"Stream: {stream}, Optimizer: {optimizer}")
92
-
93
- conversation_prompt = self.conversation.gen_complete_prompt(prompt)
94
- if optimizer:
95
- if optimizer in self.__available_optimizers:
96
- conversation_prompt = getattr(Optimizers, optimizer)(
97
- conversation_prompt if conversationally else prompt
98
- )
99
- if self.logger:
100
- self.logger.debug(f"Applied optimizer: {optimizer}")
101
- else:
102
- if self.logger:
103
- self.logger.error(f"Invalid optimizer requested: {optimizer}")
104
- raise exceptions.FailedToGenerateResponseError(
105
- f"Optimizer is not one of {self.__available_optimizers}"
106
- )
107
-
108
- data = {"message": conversation_prompt}
109
-
110
- def for_stream():
111
- try:
112
- if self.logger:
113
- self.logger.debug("Initiating streaming request to API")
114
-
115
- with requests.post(
116
- self.api_endpoint,
117
- headers=self.headers,
118
- json=data,
119
- stream=True,
120
- timeout=self.timeout
121
- ) as response:
122
- response.raise_for_status()
123
-
124
- if self.logger:
125
- self.logger.info(f"API connection established successfully. Status: {response.status_code}")
126
-
127
- for line in response.iter_lines():
128
- if line:
129
- yield line.decode('utf-8')
130
-
131
- self.conversation.update_chat_history(
132
- prompt, self.get_message(self.last_response)
133
- )
134
-
135
- except requests.exceptions.RequestException as e:
136
- if self.logger:
137
- self.logger.error(f"API request failed: {str(e)}")
138
- raise exceptions.ProviderConnectionError(f"Error connecting to Marcus: {str(e)}")
139
-
140
- def for_non_stream():
141
- if self.logger:
142
- self.logger.debug("Processing non-streaming request")
143
-
144
- full_response = ""
145
- for line in for_stream():
146
- full_response += line
147
- self.last_response = {"text": full_response}
148
-
149
- if self.logger:
150
- self.logger.debug("Response processing completed")
151
-
152
- return self.last_response
153
-
154
- return for_stream() if stream else for_non_stream()
155
-
156
- def chat(
157
- self,
158
- prompt: str,
159
- stream: bool = False,
160
- optimizer: str = None,
161
- conversationally: bool = False,
162
- ) -> str | Generator[str, None, None]:
163
- """Generates a response from the AskMarcus API with logging."""
164
- if self.logger:
165
- self.logger.debug(f"Chat request initiated - Prompt: {prompt[:50]}...")
166
-
167
- def for_stream():
168
- for response_chunk in self.ask(
169
- prompt, stream=True, optimizer=optimizer, conversationally=conversationally
170
- ):
171
- yield response_chunk
172
-
173
- def for_non_stream():
174
- response = self.ask(
175
- prompt, stream=False, optimizer=optimizer, conversationally=conversationally
176
- )
177
- return self.get_message(response)
178
-
179
- return for_stream() if stream else for_non_stream()
180
-
181
- def get_message(self, response: Dict[str, Any]) -> str:
182
- """Extracts the message from the API response."""
183
- assert isinstance(response, dict), "Response should be of dict data-type only"
184
- return response.get("text", "")
185
-
186
- if __name__ == "__main__":
187
- from rich import print
188
- # Enable logging for testing
189
- ai = Marcus(logging=True)
190
- response = ai.chat(input(">>> "), stream=True)
191
- for chunk in response:
192
- print(chunk, end="", flush=True)
1
+ import requests
2
+ import json
3
+ from typing import Any, Dict, Optional, Generator
4
+
5
+ from webscout.AIutel import Optimizers
6
+ from webscout.AIutel import Conversation
7
+ from webscout.AIutel import AwesomePrompts
8
+ from webscout.AIbase import Provider
9
+ from webscout import exceptions
10
+ from webscout.Litlogger import Logger, LogFormat
11
+ from webscout import LitAgent as Lit
12
+
13
+ class Marcus(Provider):
14
+ """
15
+ This class provides methods for interacting with the AskMarcus API.
16
+ Improved to match webscout provider standards with comprehensive logging.
17
+ """
18
+
19
+ def __init__(
20
+ self,
21
+ is_conversation: bool = True,
22
+ max_tokens: int = 2048,
23
+ timeout: int = 30,
24
+ intro: str = None,
25
+ filepath: str = None,
26
+ update_file: bool = True,
27
+ proxies: dict = {},
28
+ history_offset: int = 10250,
29
+ act: str = None,
30
+ logging: bool = False
31
+ ):
32
+ """Initializes the Marcus API with logging capabilities."""
33
+ self.logger = Logger(
34
+ name="Marcus",
35
+ format=LogFormat.MODERN_EMOJI,
36
+ ) if logging else None
37
+
38
+ if self.logger:
39
+ self.logger.info("Initializing Marcus API")
40
+
41
+ self.session = requests.Session()
42
+ self.is_conversation = is_conversation
43
+ self.max_tokens_to_sample = max_tokens
44
+ self.api_endpoint = "https://www.askmarcus.app/api/response"
45
+ self.timeout = timeout
46
+ self.last_response = {}
47
+
48
+ self.headers = {
49
+ 'content-type': 'application/json',
50
+ 'accept': '*/*',
51
+ 'origin': 'https://www.askmarcus.app',
52
+ 'referer': 'https://www.askmarcus.app/chat',
53
+ 'user-agent': Lit().random(),
54
+ }
55
+
56
+ self.__available_optimizers = (
57
+ method
58
+ for method in dir(Optimizers)
59
+ if callable(getattr(Optimizers, method)) and not method.startswith("__")
60
+ )
61
+
62
+ Conversation.intro = (
63
+ AwesomePrompts().get_act(
64
+ act, raise_not_found=True, default=None, case_insensitive=True
65
+ )
66
+ if act
67
+ else intro or Conversation.intro
68
+ )
69
+
70
+ self.conversation = Conversation(
71
+ is_conversation, self.max_tokens_to_sample, filepath, update_file
72
+ )
73
+ self.conversation.history_offset = history_offset
74
+ self.session.proxies = proxies
75
+
76
+ if self.logger:
77
+ self.logger.info("Marcus API initialized successfully")
78
+
79
+ def ask(
80
+ self,
81
+ prompt: str,
82
+ stream: bool = False,
83
+ raw: bool = False,
84
+ optimizer: str = None,
85
+ conversationally: bool = False,
86
+ ) -> Dict[str, Any] | Generator[str, None, None]:
87
+ """Sends a prompt to the AskMarcus API and returns the response with logging."""
88
+ if self.logger:
89
+ self.logger.debug(f"Processing request - Prompt: {prompt[:50]}...")
90
+ self.logger.debug(f"Stream: {stream}, Optimizer: {optimizer}")
91
+
92
+ conversation_prompt = self.conversation.gen_complete_prompt(prompt)
93
+ if optimizer:
94
+ if optimizer in self.__available_optimizers:
95
+ conversation_prompt = getattr(Optimizers, optimizer)(
96
+ conversation_prompt if conversationally else prompt
97
+ )
98
+ if self.logger:
99
+ self.logger.debug(f"Applied optimizer: {optimizer}")
100
+ else:
101
+ if self.logger:
102
+ self.logger.error(f"Invalid optimizer requested: {optimizer}")
103
+ raise exceptions.FailedToGenerateResponseError(
104
+ f"Optimizer is not one of {self.__available_optimizers}"
105
+ )
106
+
107
+ data = {"message": conversation_prompt}
108
+
109
+ def for_stream():
110
+ try:
111
+ if self.logger:
112
+ self.logger.debug("Initiating streaming request to API")
113
+
114
+ with requests.post(
115
+ self.api_endpoint,
116
+ headers=self.headers,
117
+ json=data,
118
+ stream=True,
119
+ timeout=self.timeout
120
+ ) as response:
121
+ response.raise_for_status()
122
+
123
+ if self.logger:
124
+ self.logger.info(f"API connection established successfully. Status: {response.status_code}")
125
+
126
+ for line in response.iter_lines():
127
+ if line:
128
+ yield line.decode('utf-8')
129
+
130
+ self.conversation.update_chat_history(
131
+ prompt, self.get_message(self.last_response)
132
+ )
133
+
134
+ except requests.exceptions.RequestException as e:
135
+ if self.logger:
136
+ self.logger.error(f"API request failed: {str(e)}")
137
+ raise exceptions.ProviderConnectionError(f"Error connecting to Marcus: {str(e)}")
138
+
139
+ def for_non_stream():
140
+ if self.logger:
141
+ self.logger.debug("Processing non-streaming request")
142
+
143
+ full_response = ""
144
+ for line in for_stream():
145
+ full_response += line
146
+ self.last_response = {"text": full_response}
147
+
148
+ if self.logger:
149
+ self.logger.debug("Response processing completed")
150
+
151
+ return self.last_response
152
+
153
+ return for_stream() if stream else for_non_stream()
154
+
155
+ def chat(
156
+ self,
157
+ prompt: str,
158
+ stream: bool = False,
159
+ optimizer: str = None,
160
+ conversationally: bool = False,
161
+ ) -> str | Generator[str, None, None]:
162
+ """Generates a response from the AskMarcus API with logging."""
163
+ if self.logger:
164
+ self.logger.debug(f"Chat request initiated - Prompt: {prompt[:50]}...")
165
+
166
+ def for_stream():
167
+ for response_chunk in self.ask(
168
+ prompt, stream=True, optimizer=optimizer, conversationally=conversationally
169
+ ):
170
+ yield response_chunk
171
+
172
+ def for_non_stream():
173
+ response = self.ask(
174
+ prompt, stream=False, optimizer=optimizer, conversationally=conversationally
175
+ )
176
+ return self.get_message(response)
177
+
178
+ return for_stream() if stream else for_non_stream()
179
+
180
+ def get_message(self, response: Dict[str, Any]) -> str:
181
+ """Extracts the message from the API response."""
182
+ assert isinstance(response, dict), "Response should be of dict data-type only"
183
+ return response.get("text", "")
184
+
185
+ if __name__ == "__main__":
186
+ from rich import print
187
+ # Enable logging for testing
188
+ ai = Marcus(logging=True)
189
+ response = ai.chat(input(">>> "), stream=True)
190
+ for chunk in response:
191
+ print(chunk, end="", flush=True)
@@ -8,7 +8,7 @@ from datetime import date
8
8
  from webscout.AIutel import Optimizers, Conversation, AwesomePrompts
9
9
  from webscout.AIbase import Provider
10
10
  from webscout import exceptions
11
- from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
11
+ from webscout.Litlogger import Logger, LogFormat
12
12
  from webscout.litagent import LitAgent
13
13
 
14
14
  class Netwrck(Provider):
@@ -49,10 +49,10 @@ class Netwrck(Provider):
49
49
  ):
50
50
  """Initializes the Netwrck API client."""
51
51
  # Initialize logger first for initialization logging
52
- self.logger = LitLogger(
52
+ self.logger = Logger(
53
53
  name="Netwrck",
54
54
  format=LogFormat.MODERN_EMOJI,
55
- color_scheme=ColorScheme.CYBERPUNK
55
+
56
56
  ) if logging else None
57
57
 
58
58
  if model not in self.AVAILABLE_MODELS:
webscout/Provider/PI.py CHANGED
@@ -10,7 +10,7 @@ from webscout.AIutel import AwesomePrompts
10
10
  from webscout.AIbase import Provider
11
11
  from typing import Dict, Union, Any, Optional
12
12
  from webscout import LitAgent
13
- from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
13
+ from webscout.Litlogger import Logger, LogFormat
14
14
 
15
15
  class PiAI(Provider):
16
16
  """
@@ -96,7 +96,7 @@ class PiAI(Provider):
96
96
  self.conversation.history_offset = history_offset
97
97
  self.session.proxies = proxies
98
98
 
99
- self.logger = LitLogger(name="PiAI", format=LogFormat.MODERN_EMOJI, color_scheme=ColorScheme.CYBERPUNK) if logging else None
99
+ self.logger = Logger(name="PiAI", format=LogFormat.MODERN_EMOJI) if logging else None
100
100
 
101
101
  self.knowledge_cutoff = "December 2023"
102
102
 
@@ -6,7 +6,7 @@ from webscout.AIutel import Optimizers, Conversation, AwesomePrompts
6
6
  from webscout.AIbase import Provider
7
7
  from webscout import exceptions
8
8
  from webscout import LitAgent as Lit
9
- from webscout.Litlogger import LitLogger, LogFormat, ColorScheme
9
+ from webscout.Litlogger import Logger, LogFormat
10
10
 
11
11
  class PIZZAGPT(Provider):
12
12
  """
@@ -68,10 +68,9 @@ class PIZZAGPT(Provider):
68
68
  self.conversation.history_offset = history_offset
69
69
  self.session.proxies = proxies
70
70
 
71
- self.logger = LitLogger(
71
+ self.logger = Logger(
72
72
  name="PIZZAGPT",
73
73
  format=LogFormat.MODERN_EMOJI,
74
- color_scheme=ColorScheme.CYBERPUNK
75
74
  ) if logging else None
76
75
 
77
76
  if self.logger: