webscout 8.3.7__py3-none-any.whl → 2025.10.11__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 (273) hide show
  1. webscout/AIauto.py +250 -250
  2. webscout/AIbase.py +379 -379
  3. webscout/AIutel.py +60 -60
  4. webscout/Bard.py +1012 -1012
  5. webscout/Bing_search.py +417 -417
  6. webscout/DWEBS.py +529 -529
  7. webscout/Extra/Act.md +309 -309
  8. webscout/Extra/GitToolkit/__init__.py +10 -10
  9. webscout/Extra/GitToolkit/gitapi/README.md +110 -110
  10. webscout/Extra/GitToolkit/gitapi/__init__.py +11 -11
  11. webscout/Extra/GitToolkit/gitapi/repository.py +195 -195
  12. webscout/Extra/GitToolkit/gitapi/user.py +96 -96
  13. webscout/Extra/GitToolkit/gitapi/utils.py +61 -61
  14. webscout/Extra/YTToolkit/README.md +375 -375
  15. webscout/Extra/YTToolkit/YTdownloader.py +956 -956
  16. webscout/Extra/YTToolkit/__init__.py +2 -2
  17. webscout/Extra/YTToolkit/transcriber.py +475 -475
  18. webscout/Extra/YTToolkit/ytapi/README.md +44 -44
  19. webscout/Extra/YTToolkit/ytapi/__init__.py +6 -6
  20. webscout/Extra/YTToolkit/ytapi/channel.py +307 -307
  21. webscout/Extra/YTToolkit/ytapi/errors.py +13 -13
  22. webscout/Extra/YTToolkit/ytapi/extras.py +118 -118
  23. webscout/Extra/YTToolkit/ytapi/https.py +88 -88
  24. webscout/Extra/YTToolkit/ytapi/patterns.py +61 -61
  25. webscout/Extra/YTToolkit/ytapi/playlist.py +58 -58
  26. webscout/Extra/YTToolkit/ytapi/pool.py +7 -7
  27. webscout/Extra/YTToolkit/ytapi/query.py +39 -39
  28. webscout/Extra/YTToolkit/ytapi/stream.py +62 -62
  29. webscout/Extra/YTToolkit/ytapi/utils.py +62 -62
  30. webscout/Extra/YTToolkit/ytapi/video.py +232 -232
  31. webscout/Extra/autocoder/__init__.py +9 -9
  32. webscout/Extra/autocoder/autocoder.py +1105 -1105
  33. webscout/Extra/autocoder/autocoder_utiles.py +332 -332
  34. webscout/Extra/gguf.md +429 -429
  35. webscout/Extra/gguf.py +1213 -1213
  36. webscout/Extra/tempmail/README.md +487 -487
  37. webscout/Extra/tempmail/__init__.py +27 -27
  38. webscout/Extra/tempmail/async_utils.py +140 -140
  39. webscout/Extra/tempmail/base.py +160 -160
  40. webscout/Extra/tempmail/cli.py +186 -186
  41. webscout/Extra/tempmail/emailnator.py +84 -84
  42. webscout/Extra/tempmail/mail_tm.py +360 -360
  43. webscout/Extra/tempmail/temp_mail_io.py +291 -291
  44. webscout/Extra/weather.md +281 -281
  45. webscout/Extra/weather.py +193 -193
  46. webscout/Litlogger/README.md +10 -10
  47. webscout/Litlogger/__init__.py +15 -15
  48. webscout/Litlogger/formats.py +13 -13
  49. webscout/Litlogger/handlers.py +121 -121
  50. webscout/Litlogger/levels.py +13 -13
  51. webscout/Litlogger/logger.py +134 -134
  52. webscout/Provider/AISEARCH/Perplexity.py +332 -332
  53. webscout/Provider/AISEARCH/README.md +279 -279
  54. webscout/Provider/AISEARCH/__init__.py +16 -1
  55. webscout/Provider/AISEARCH/felo_search.py +206 -206
  56. webscout/Provider/AISEARCH/genspark_search.py +323 -323
  57. webscout/Provider/AISEARCH/hika_search.py +185 -185
  58. webscout/Provider/AISEARCH/iask_search.py +410 -410
  59. webscout/Provider/AISEARCH/monica_search.py +219 -219
  60. webscout/Provider/AISEARCH/scira_search.py +316 -316
  61. webscout/Provider/AISEARCH/stellar_search.py +177 -177
  62. webscout/Provider/AISEARCH/webpilotai_search.py +255 -255
  63. webscout/Provider/Aitopia.py +314 -314
  64. webscout/Provider/Apriel.py +306 -0
  65. webscout/Provider/ChatGPTClone.py +236 -236
  66. webscout/Provider/ChatSandbox.py +343 -343
  67. webscout/Provider/Cloudflare.py +324 -324
  68. webscout/Provider/Cohere.py +208 -208
  69. webscout/Provider/Deepinfra.py +370 -366
  70. webscout/Provider/ExaAI.py +260 -260
  71. webscout/Provider/ExaChat.py +308 -308
  72. webscout/Provider/Flowith.py +221 -221
  73. webscout/Provider/GMI.py +293 -0
  74. webscout/Provider/Gemini.py +164 -164
  75. webscout/Provider/GeminiProxy.py +167 -167
  76. webscout/Provider/GithubChat.py +371 -372
  77. webscout/Provider/Groq.py +800 -800
  78. webscout/Provider/HeckAI.py +383 -383
  79. webscout/Provider/Jadve.py +282 -282
  80. webscout/Provider/K2Think.py +307 -307
  81. webscout/Provider/Koboldai.py +205 -205
  82. webscout/Provider/LambdaChat.py +423 -423
  83. webscout/Provider/Nemotron.py +244 -244
  84. webscout/Provider/Netwrck.py +248 -248
  85. webscout/Provider/OLLAMA.py +395 -395
  86. webscout/Provider/OPENAI/Cloudflare.py +393 -393
  87. webscout/Provider/OPENAI/FalconH1.py +451 -451
  88. webscout/Provider/OPENAI/FreeGemini.py +296 -296
  89. webscout/Provider/OPENAI/K2Think.py +431 -431
  90. webscout/Provider/OPENAI/NEMOTRON.py +240 -240
  91. webscout/Provider/OPENAI/PI.py +427 -427
  92. webscout/Provider/OPENAI/README.md +959 -959
  93. webscout/Provider/OPENAI/TogetherAI.py +345 -345
  94. webscout/Provider/OPENAI/TwoAI.py +465 -465
  95. webscout/Provider/OPENAI/__init__.py +33 -18
  96. webscout/Provider/OPENAI/base.py +248 -248
  97. webscout/Provider/OPENAI/chatglm.py +528 -0
  98. webscout/Provider/OPENAI/chatgpt.py +592 -592
  99. webscout/Provider/OPENAI/chatgptclone.py +521 -521
  100. webscout/Provider/OPENAI/chatsandbox.py +202 -202
  101. webscout/Provider/OPENAI/deepinfra.py +318 -314
  102. webscout/Provider/OPENAI/e2b.py +1665 -1665
  103. webscout/Provider/OPENAI/exaai.py +420 -420
  104. webscout/Provider/OPENAI/exachat.py +452 -452
  105. webscout/Provider/OPENAI/friendli.py +232 -232
  106. webscout/Provider/OPENAI/{refact.py → gmi.py} +324 -274
  107. webscout/Provider/OPENAI/groq.py +364 -364
  108. webscout/Provider/OPENAI/heckai.py +314 -314
  109. webscout/Provider/OPENAI/llmchatco.py +337 -337
  110. webscout/Provider/OPENAI/netwrck.py +355 -355
  111. webscout/Provider/OPENAI/oivscode.py +290 -290
  112. webscout/Provider/OPENAI/opkfc.py +518 -518
  113. webscout/Provider/OPENAI/pydantic_imports.py +1 -1
  114. webscout/Provider/OPENAI/scirachat.py +535 -535
  115. webscout/Provider/OPENAI/sonus.py +308 -308
  116. webscout/Provider/OPENAI/standardinput.py +442 -442
  117. webscout/Provider/OPENAI/textpollinations.py +340 -340
  118. webscout/Provider/OPENAI/toolbaz.py +419 -416
  119. webscout/Provider/OPENAI/typefully.py +362 -362
  120. webscout/Provider/OPENAI/utils.py +295 -295
  121. webscout/Provider/OPENAI/venice.py +436 -436
  122. webscout/Provider/OPENAI/wisecat.py +387 -387
  123. webscout/Provider/OPENAI/writecream.py +166 -166
  124. webscout/Provider/OPENAI/x0gpt.py +378 -378
  125. webscout/Provider/OPENAI/yep.py +389 -389
  126. webscout/Provider/OpenGPT.py +230 -230
  127. webscout/Provider/Openai.py +243 -243
  128. webscout/Provider/PI.py +405 -405
  129. webscout/Provider/Perplexitylabs.py +430 -430
  130. webscout/Provider/QwenLM.py +272 -272
  131. webscout/Provider/STT/__init__.py +16 -1
  132. webscout/Provider/Sambanova.py +257 -257
  133. webscout/Provider/StandardInput.py +309 -309
  134. webscout/Provider/TTI/README.md +82 -82
  135. webscout/Provider/TTI/__init__.py +33 -18
  136. webscout/Provider/TTI/aiarta.py +413 -413
  137. webscout/Provider/TTI/base.py +136 -136
  138. webscout/Provider/TTI/bing.py +243 -243
  139. webscout/Provider/TTI/gpt1image.py +149 -149
  140. webscout/Provider/TTI/imagen.py +196 -196
  141. webscout/Provider/TTI/infip.py +211 -211
  142. webscout/Provider/TTI/magicstudio.py +232 -232
  143. webscout/Provider/TTI/monochat.py +219 -219
  144. webscout/Provider/TTI/piclumen.py +214 -214
  145. webscout/Provider/TTI/pixelmuse.py +232 -232
  146. webscout/Provider/TTI/pollinations.py +232 -232
  147. webscout/Provider/TTI/together.py +288 -288
  148. webscout/Provider/TTI/utils.py +12 -12
  149. webscout/Provider/TTI/venice.py +367 -367
  150. webscout/Provider/TTS/README.md +192 -192
  151. webscout/Provider/TTS/__init__.py +33 -18
  152. webscout/Provider/TTS/parler.py +110 -110
  153. webscout/Provider/TTS/streamElements.py +333 -333
  154. webscout/Provider/TTS/utils.py +280 -280
  155. webscout/Provider/TeachAnything.py +237 -237
  156. webscout/Provider/TextPollinationsAI.py +310 -310
  157. webscout/Provider/TogetherAI.py +356 -356
  158. webscout/Provider/TwoAI.py +312 -312
  159. webscout/Provider/TypliAI.py +311 -311
  160. webscout/Provider/UNFINISHED/ChatHub.py +208 -208
  161. webscout/Provider/UNFINISHED/ChutesAI.py +313 -313
  162. webscout/Provider/UNFINISHED/GizAI.py +294 -294
  163. webscout/Provider/UNFINISHED/Marcus.py +198 -198
  164. webscout/Provider/UNFINISHED/Qodo.py +477 -477
  165. webscout/Provider/UNFINISHED/VercelAIGateway.py +338 -338
  166. webscout/Provider/UNFINISHED/XenAI.py +324 -324
  167. webscout/Provider/UNFINISHED/Youchat.py +330 -330
  168. webscout/Provider/UNFINISHED/liner.py +334 -0
  169. webscout/Provider/UNFINISHED/liner_api_request.py +262 -262
  170. webscout/Provider/UNFINISHED/puterjs.py +634 -634
  171. webscout/Provider/UNFINISHED/samurai.py +223 -223
  172. webscout/Provider/UNFINISHED/test_lmarena.py +119 -119
  173. webscout/Provider/Venice.py +250 -250
  174. webscout/Provider/VercelAI.py +256 -256
  175. webscout/Provider/WiseCat.py +231 -231
  176. webscout/Provider/WrDoChat.py +366 -366
  177. webscout/Provider/__init__.py +33 -18
  178. webscout/Provider/ai4chat.py +174 -174
  179. webscout/Provider/akashgpt.py +331 -331
  180. webscout/Provider/cerebras.py +446 -446
  181. webscout/Provider/chatglm.py +394 -301
  182. webscout/Provider/cleeai.py +211 -211
  183. webscout/Provider/elmo.py +282 -282
  184. webscout/Provider/geminiapi.py +208 -208
  185. webscout/Provider/granite.py +261 -261
  186. webscout/Provider/hermes.py +263 -263
  187. webscout/Provider/julius.py +223 -223
  188. webscout/Provider/learnfastai.py +309 -309
  189. webscout/Provider/llama3mitril.py +214 -214
  190. webscout/Provider/llmchat.py +243 -243
  191. webscout/Provider/llmchatco.py +290 -290
  192. webscout/Provider/meta.py +801 -801
  193. webscout/Provider/oivscode.py +309 -309
  194. webscout/Provider/scira_chat.py +383 -383
  195. webscout/Provider/searchchat.py +292 -292
  196. webscout/Provider/sonus.py +258 -258
  197. webscout/Provider/toolbaz.py +370 -367
  198. webscout/Provider/turboseek.py +273 -273
  199. webscout/Provider/typefully.py +207 -207
  200. webscout/Provider/yep.py +372 -372
  201. webscout/__init__.py +30 -31
  202. webscout/__main__.py +5 -5
  203. webscout/auth/api_key_manager.py +189 -189
  204. webscout/auth/config.py +175 -175
  205. webscout/auth/models.py +185 -185
  206. webscout/auth/routes.py +664 -664
  207. webscout/auth/simple_logger.py +236 -236
  208. webscout/cli.py +523 -523
  209. webscout/conversation.py +438 -438
  210. webscout/exceptions.py +361 -361
  211. webscout/litagent/Readme.md +298 -298
  212. webscout/litagent/__init__.py +28 -28
  213. webscout/litagent/agent.py +581 -581
  214. webscout/litagent/constants.py +59 -59
  215. webscout/litprinter/__init__.py +58 -58
  216. webscout/models.py +181 -181
  217. webscout/optimizers.py +419 -419
  218. webscout/prompt_manager.py +288 -288
  219. webscout/sanitize.py +1078 -1078
  220. webscout/scout/README.md +401 -401
  221. webscout/scout/__init__.py +8 -8
  222. webscout/scout/core/__init__.py +6 -6
  223. webscout/scout/core/crawler.py +297 -297
  224. webscout/scout/core/scout.py +706 -706
  225. webscout/scout/core/search_result.py +95 -95
  226. webscout/scout/core/text_analyzer.py +62 -62
  227. webscout/scout/core/text_utils.py +277 -277
  228. webscout/scout/core/web_analyzer.py +51 -51
  229. webscout/scout/element.py +599 -599
  230. webscout/scout/parsers/__init__.py +69 -69
  231. webscout/scout/parsers/html5lib_parser.py +172 -172
  232. webscout/scout/parsers/html_parser.py +236 -236
  233. webscout/scout/parsers/lxml_parser.py +178 -178
  234. webscout/scout/utils.py +37 -37
  235. webscout/swiftcli/Readme.md +323 -323
  236. webscout/swiftcli/__init__.py +95 -95
  237. webscout/swiftcli/core/__init__.py +7 -7
  238. webscout/swiftcli/core/cli.py +308 -308
  239. webscout/swiftcli/core/context.py +104 -104
  240. webscout/swiftcli/core/group.py +241 -241
  241. webscout/swiftcli/decorators/__init__.py +28 -28
  242. webscout/swiftcli/decorators/command.py +221 -221
  243. webscout/swiftcli/decorators/options.py +220 -220
  244. webscout/swiftcli/decorators/output.py +302 -302
  245. webscout/swiftcli/exceptions.py +21 -21
  246. webscout/swiftcli/plugins/__init__.py +9 -9
  247. webscout/swiftcli/plugins/base.py +135 -135
  248. webscout/swiftcli/plugins/manager.py +269 -269
  249. webscout/swiftcli/utils/__init__.py +59 -59
  250. webscout/swiftcli/utils/formatting.py +252 -252
  251. webscout/swiftcli/utils/parsing.py +267 -267
  252. webscout/update_checker.py +117 -117
  253. webscout/version.py +1 -1
  254. webscout/webscout_search.py +1183 -1183
  255. webscout/webscout_search_async.py +649 -649
  256. webscout/yep_search.py +346 -346
  257. webscout/zeroart/README.md +89 -89
  258. webscout/zeroart/__init__.py +134 -134
  259. webscout/zeroart/base.py +66 -66
  260. webscout/zeroart/effects.py +100 -100
  261. webscout/zeroart/fonts.py +1238 -1238
  262. {webscout-8.3.7.dist-info → webscout-2025.10.11.dist-info}/METADATA +937 -937
  263. webscout-2025.10.11.dist-info/RECORD +300 -0
  264. webscout/Provider/AISEARCH/DeepFind.py +0 -254
  265. webscout/Provider/OPENAI/Qwen3.py +0 -303
  266. webscout/Provider/OPENAI/qodo.py +0 -630
  267. webscout/Provider/OPENAI/xenai.py +0 -514
  268. webscout/tempid.py +0 -134
  269. webscout-8.3.7.dist-info/RECORD +0 -301
  270. {webscout-8.3.7.dist-info → webscout-2025.10.11.dist-info}/WHEEL +0 -0
  271. {webscout-8.3.7.dist-info → webscout-2025.10.11.dist-info}/entry_points.txt +0 -0
  272. {webscout-8.3.7.dist-info → webscout-2025.10.11.dist-info}/licenses/LICENSE.md +0 -0
  273. {webscout-8.3.7.dist-info → webscout-2025.10.11.dist-info}/top_level.txt +0 -0
webscout/AIbase.py CHANGED
@@ -1,379 +1,379 @@
1
- from abc import ABC, abstractmethod
2
- from pathlib import Path
3
- from typing import AsyncGenerator, Dict, List, Union, Generator, Optional, Any
4
- from typing_extensions import TypeAlias
5
-
6
- # Type aliases for better readability
7
- Response: TypeAlias = dict[str, Union[str, bool, None]]
8
- AsyncImageData: TypeAlias = Union[bytes, str, AsyncGenerator[bytes, None]]
9
-
10
- class SearchResponse:
11
- """A wrapper class for search API responses.
12
-
13
- This class automatically converts response objects to their text representation
14
- when printed or converted to string.
15
-
16
- Attributes:
17
- text (str): The text content of the response
18
-
19
- Example:
20
- >>> response = SearchResponse("Hello, world!")
21
- >>> print(response)
22
- Hello, world!
23
- >>> str(response)
24
- 'Hello, world!'
25
- """
26
- def __init__(self, text: str):
27
- self.text = text
28
-
29
- def __str__(self):
30
- return self.text
31
-
32
- def __repr__(self):
33
- return self.text
34
-
35
- class AIProviderError(Exception):
36
- pass
37
-
38
- class Provider(ABC):
39
-
40
- @abstractmethod
41
- def ask(
42
- self,
43
- prompt: str,
44
- stream: bool = False,
45
- raw: bool = False,
46
- optimizer: Optional[str] = None,
47
- conversationally: bool = False,
48
- ) -> Response:
49
- raise NotImplementedError("Method needs to be implemented in subclass")
50
-
51
- @abstractmethod
52
- def chat(
53
- self,
54
- prompt: str,
55
- stream: bool = False,
56
- optimizer: Optional[str] = None,
57
- conversationally: bool = False,
58
- ) -> str:
59
- raise NotImplementedError("Method needs to be implemented in subclass")
60
-
61
- @abstractmethod
62
- def get_message(self, response: Response) -> str:
63
- raise NotImplementedError("Method needs to be implemented in subclass")
64
-
65
- class AsyncProvider(ABC):
66
-
67
- @abstractmethod
68
- async def ask(
69
- self,
70
- prompt: str,
71
- stream: bool = False,
72
- raw: bool = False,
73
- optimizer: Optional[str] = None,
74
- conversationally: bool = False,
75
- ) -> Response:
76
- raise NotImplementedError("Method needs to be implemented in subclass")
77
-
78
- @abstractmethod
79
- async def chat(
80
- self,
81
- prompt: str,
82
- stream: bool = False,
83
- optimizer: Optional[str] = None,
84
- conversationally: bool = False,
85
- ) -> str:
86
- raise NotImplementedError("Method needs to be implemented in subclass")
87
-
88
- @abstractmethod
89
- async def get_message(self, response: Response) -> str:
90
- raise NotImplementedError("Method needs to be implemented in subclass")
91
-
92
- class TTSProvider(ABC):
93
-
94
- @abstractmethod
95
- def tts(self, text: str, voice: str = None, verbose: bool = False) -> str:
96
- """Convert text to speech and save to a temporary file.
97
-
98
- Args:
99
- text (str): The text to convert to speech
100
- voice (str, optional): The voice to use. Defaults to provider's default voice.
101
- verbose (bool, optional): Whether to print debug information. Defaults to False.
102
-
103
- Returns:
104
- str: Path to the generated audio file
105
- """
106
- raise NotImplementedError("Method needs to be implemented in subclass")
107
-
108
- def save_audio(self, audio_file: str, destination: str = None, verbose: bool = False) -> str:
109
- """Save audio to a specific destination.
110
-
111
- Args:
112
- audio_file (str): Path to the source audio file
113
- destination (str, optional): Destination path. Defaults to current directory with timestamp.
114
- verbose (bool, optional): Whether to print debug information. Defaults to False.
115
-
116
- Returns:
117
- str: Path to the saved audio file
118
- """
119
- import shutil
120
- import os
121
- from pathlib import Path
122
- import time
123
-
124
- source_path = Path(audio_file)
125
-
126
- if not source_path.exists():
127
- raise FileNotFoundError(f"Audio file not found: {audio_file}")
128
-
129
- if destination is None:
130
- # Create a default destination with timestamp in current directory
131
- timestamp = int(time.time())
132
- destination = os.path.join(os.getcwd(), f"tts_audio_{timestamp}{source_path.suffix}")
133
-
134
- # Ensure the destination directory exists
135
- os.makedirs(os.path.dirname(os.path.abspath(destination)), exist_ok=True)
136
-
137
- # Copy the file
138
- shutil.copy2(source_path, destination)
139
-
140
- if verbose:
141
- print(f"[debug] Audio saved to {destination}")
142
-
143
- return destination
144
-
145
- def stream_audio(self, text: str, voice: str = None, chunk_size: int = 1024, verbose: bool = False) -> Generator[bytes, None, None]:
146
- """Stream audio in chunks.
147
-
148
- Args:
149
- text (str): The text to convert to speech
150
- voice (str, optional): The voice to use. Defaults to provider's default voice.
151
- chunk_size (int, optional): Size of audio chunks to yield. Defaults to 1024.
152
- verbose (bool, optional): Whether to print debug information. Defaults to False.
153
-
154
- Yields:
155
- Generator[bytes, None, None]: Audio data chunks
156
- """
157
- # Generate the audio file
158
- audio_file = self.tts(text, voice=voice, verbose=verbose)
159
-
160
- # Stream the file in chunks
161
- with open(audio_file, 'rb') as f:
162
- while chunk := f.read(chunk_size):
163
- yield chunk
164
-
165
- class AsyncTTSProvider(ABC):
166
-
167
- @abstractmethod
168
- async def tts(self, text: str, voice: str = None, verbose: bool = False) -> str:
169
- """Convert text to speech and save to a temporary file asynchronously.
170
-
171
- Args:
172
- text (str): The text to convert to speech
173
- voice (str, optional): The voice to use. Defaults to provider's default voice.
174
- verbose (bool, optional): Whether to print debug information. Defaults to False.
175
-
176
- Returns:
177
- str: Path to the generated audio file
178
- """
179
- raise NotImplementedError("Method needs to be implemented in subclass")
180
-
181
-
182
- class STTProvider(ABC):
183
- """Abstract base class for Speech-to-Text providers."""
184
-
185
- @abstractmethod
186
- def transcribe(self, audio_path: Union[str, Path], **kwargs) -> Dict[str, Any]:
187
- """Transcribe audio file to text.
188
-
189
- Args:
190
- audio_path (Union[str, Path]): Path to the audio file
191
- **kwargs: Additional provider-specific parameters
192
-
193
- Returns:
194
- Dict[str, Any]: Transcription result in OpenAI Whisper format
195
- """
196
- raise NotImplementedError("Method needs to be implemented in subclass")
197
-
198
- @abstractmethod
199
- def transcribe_from_url(self, audio_url: str, **kwargs) -> Dict[str, Any]:
200
- """Transcribe audio from URL to text.
201
-
202
- Args:
203
- audio_url (str): URL of the audio file
204
- **kwargs: Additional provider-specific parameters
205
-
206
- Returns:
207
- Dict[str, Any]: Transcription result in OpenAI Whisper format
208
- """
209
- raise NotImplementedError("Method needs to be implemented in subclass")
210
-
211
-
212
- class AsyncSTTProvider(ABC):
213
- """Abstract base class for asynchronous Speech-to-Text providers."""
214
-
215
- @abstractmethod
216
- async def transcribe(self, audio_path: Union[str, Path], **kwargs) -> Dict[str, Any]:
217
- """Transcribe audio file to text asynchronously.
218
-
219
- Args:
220
- audio_path (Union[str, Path]): Path to the audio file
221
- **kwargs: Additional provider-specific parameters
222
-
223
- Returns:
224
- Dict[str, Any]: Transcription result in OpenAI Whisper format
225
- """
226
- raise NotImplementedError("Method needs to be implemented in subclass")
227
-
228
- @abstractmethod
229
- async def transcribe_from_url(self, audio_url: str, **kwargs) -> Dict[str, Any]:
230
- """Transcribe audio from URL to text asynchronously.
231
-
232
- Args:
233
- audio_url (str): URL of the audio file
234
- **kwargs: Additional provider-specific parameters
235
-
236
- Returns:
237
- Dict[str, Any]: Transcription result in OpenAI Whisper format
238
- """
239
- raise NotImplementedError("Method needs to be implemented in subclass")
240
-
241
- async def save_audio(self, audio_file: str, destination: str = None, verbose: bool = False) -> str:
242
- """Save audio to a specific destination asynchronously.
243
-
244
- Args:
245
- audio_file (str): Path to the source audio file
246
- destination (str, optional): Destination path. Defaults to current directory with timestamp.
247
- verbose (bool, optional): Whether to print debug information. Defaults to False.
248
-
249
- Returns:
250
- str: Path to the saved audio file
251
- """
252
- import shutil
253
- import os
254
- from pathlib import Path
255
- import time
256
- import asyncio
257
-
258
- source_path = Path(audio_file)
259
-
260
- if not source_path.exists():
261
- raise FileNotFoundError(f"Audio file not found: {audio_file}")
262
-
263
- if destination is None:
264
- # Create a default destination with timestamp in current directory
265
- timestamp = int(time.time())
266
- destination = os.path.join(os.getcwd(), f"tts_audio_{timestamp}{source_path.suffix}")
267
-
268
- # Ensure the destination directory exists
269
- os.makedirs(os.path.dirname(os.path.abspath(destination)), exist_ok=True)
270
-
271
- # Copy the file using asyncio to avoid blocking
272
- await asyncio.to_thread(shutil.copy2, source_path, destination)
273
-
274
- if verbose:
275
- print(f"[debug] Audio saved to {destination}")
276
-
277
- return destination
278
-
279
- async def stream_audio(self, text: str, voice: str = None, chunk_size: int = 1024, verbose: bool = False) -> AsyncGenerator[bytes, None]:
280
- """Stream audio in chunks asynchronously.
281
-
282
- Args:
283
- text (str): The text to convert to speech
284
- voice (str, optional): The voice to use. Defaults to provider's default voice.
285
- chunk_size (int, optional): Size of audio chunks to yield. Defaults to 1024.
286
- verbose (bool, optional): Whether to print debug information. Defaults to False.
287
-
288
- Yields:
289
- AsyncGenerator[bytes, None]: Audio data chunks
290
- """
291
- import aiofiles
292
-
293
- # Generate the audio file
294
- audio_file = await self.tts(text, voice=voice, verbose=verbose)
295
-
296
- # Stream the file in chunks
297
- async with aiofiles.open(audio_file, 'rb') as f:
298
- while chunk := await f.read(chunk_size):
299
- yield chunk
300
-
301
- class AISearch(ABC):
302
- """Abstract base class for AI-powered search providers.
303
-
304
- This class defines the interface for AI search providers that can perform
305
- web searches and return AI-generated responses based on search results.
306
-
307
- All search providers should inherit from this class and implement the
308
- required methods.
309
- """
310
-
311
- @abstractmethod
312
- def search(
313
- self,
314
- prompt: str,
315
- stream: bool = False,
316
- raw: bool = False,
317
- ) -> Union[SearchResponse, Generator[Union[Dict[str, str], SearchResponse], None, None]]:
318
- """Search using the provider's API and get AI-generated responses.
319
-
320
- This method sends a search query to the provider and returns the AI-generated response.
321
- It supports both streaming and non-streaming modes, as well as raw response format.
322
-
323
- Args:
324
- prompt (str): The search query or prompt to send to the API.
325
- stream (bool, optional): If True, yields response chunks as they arrive.
326
- If False, returns complete response. Defaults to False.
327
- raw (bool, optional): If True, returns raw response dictionaries.
328
- If False, returns SearchResponse objects that convert to text automatically.
329
- Defaults to False.
330
-
331
- Returns:
332
- Union[SearchResponse, Generator[Union[Dict[str, str], SearchResponse], None, None]]:
333
- - If stream=False: Returns complete response as SearchResponse object
334
- - If stream=True: Yields response chunks as either Dict or SearchResponse objects
335
-
336
- Raises:
337
- APIConnectionError: If the API request fails
338
- """
339
- raise NotImplementedError("Method needs to be implemented in subclass")
340
-
341
- class AsyncAISearch(ABC):
342
- """Abstract base class for asynchronous AI-powered search providers.
343
-
344
- This class defines the interface for asynchronous AI search providers that can perform
345
- web searches and return AI-generated responses based on search results.
346
-
347
- All asynchronous search providers should inherit from this class and implement the
348
- required methods.
349
- """
350
-
351
- @abstractmethod
352
- async def search(
353
- self,
354
- prompt: str,
355
- stream: bool = False,
356
- raw: bool = False,
357
- ) -> Union[SearchResponse, AsyncGenerator[Union[Dict[str, str], SearchResponse], None]]:
358
- """Search using the provider's API and get AI-generated responses asynchronously.
359
-
360
- This method sends a search query to the provider and returns the AI-generated response.
361
- It supports both streaming and non-streaming modes, as well as raw response format.
362
-
363
- Args:
364
- prompt (str): The search query or prompt to send to the API.
365
- stream (bool, optional): If True, yields response chunks as they arrive.
366
- If False, returns complete response. Defaults to False.
367
- raw (bool, optional): If True, returns raw response dictionaries.
368
- If False, returns SearchResponse objects that convert to text automatically.
369
- Defaults to False.
370
-
371
- Returns:
372
- Union[SearchResponse, AsyncGenerator[Union[Dict[str, str], SearchResponse], None]]:
373
- - If stream=False: Returns complete response as SearchResponse object
374
- - If stream=True: Yields response chunks as either Dict or SearchResponse objects
375
-
376
- Raises:
377
- APIConnectionError: If the API request fails
378
- """
379
- raise NotImplementedError("Method needs to be implemented in subclass")
1
+ from abc import ABC, abstractmethod
2
+ from pathlib import Path
3
+ from typing import AsyncGenerator, Dict, List, Union, Generator, Optional, Any
4
+ from typing_extensions import TypeAlias
5
+
6
+ # Type aliases for better readability
7
+ Response: TypeAlias = dict[str, Union[str, bool, None]]
8
+ AsyncImageData: TypeAlias = Union[bytes, str, AsyncGenerator[bytes, None]]
9
+
10
+ class SearchResponse:
11
+ """A wrapper class for search API responses.
12
+
13
+ This class automatically converts response objects to their text representation
14
+ when printed or converted to string.
15
+
16
+ Attributes:
17
+ text (str): The text content of the response
18
+
19
+ Example:
20
+ >>> response = SearchResponse("Hello, world!")
21
+ >>> print(response)
22
+ Hello, world!
23
+ >>> str(response)
24
+ 'Hello, world!'
25
+ """
26
+ def __init__(self, text: str):
27
+ self.text = text
28
+
29
+ def __str__(self):
30
+ return self.text
31
+
32
+ def __repr__(self):
33
+ return self.text
34
+
35
+ class AIProviderError(Exception):
36
+ pass
37
+
38
+ class Provider(ABC):
39
+
40
+ @abstractmethod
41
+ def ask(
42
+ self,
43
+ prompt: str,
44
+ stream: bool = False,
45
+ raw: bool = False,
46
+ optimizer: Optional[str] = None,
47
+ conversationally: bool = False,
48
+ ) -> Response:
49
+ raise NotImplementedError("Method needs to be implemented in subclass")
50
+
51
+ @abstractmethod
52
+ def chat(
53
+ self,
54
+ prompt: str,
55
+ stream: bool = False,
56
+ optimizer: Optional[str] = None,
57
+ conversationally: bool = False,
58
+ ) -> str:
59
+ raise NotImplementedError("Method needs to be implemented in subclass")
60
+
61
+ @abstractmethod
62
+ def get_message(self, response: Response) -> str:
63
+ raise NotImplementedError("Method needs to be implemented in subclass")
64
+
65
+ class AsyncProvider(ABC):
66
+
67
+ @abstractmethod
68
+ async def ask(
69
+ self,
70
+ prompt: str,
71
+ stream: bool = False,
72
+ raw: bool = False,
73
+ optimizer: Optional[str] = None,
74
+ conversationally: bool = False,
75
+ ) -> Response:
76
+ raise NotImplementedError("Method needs to be implemented in subclass")
77
+
78
+ @abstractmethod
79
+ async def chat(
80
+ self,
81
+ prompt: str,
82
+ stream: bool = False,
83
+ optimizer: Optional[str] = None,
84
+ conversationally: bool = False,
85
+ ) -> str:
86
+ raise NotImplementedError("Method needs to be implemented in subclass")
87
+
88
+ @abstractmethod
89
+ async def get_message(self, response: Response) -> str:
90
+ raise NotImplementedError("Method needs to be implemented in subclass")
91
+
92
+ class TTSProvider(ABC):
93
+
94
+ @abstractmethod
95
+ def tts(self, text: str, voice: str = None, verbose: bool = False) -> str:
96
+ """Convert text to speech and save to a temporary file.
97
+
98
+ Args:
99
+ text (str): The text to convert to speech
100
+ voice (str, optional): The voice to use. Defaults to provider's default voice.
101
+ verbose (bool, optional): Whether to print debug information. Defaults to False.
102
+
103
+ Returns:
104
+ str: Path to the generated audio file
105
+ """
106
+ raise NotImplementedError("Method needs to be implemented in subclass")
107
+
108
+ def save_audio(self, audio_file: str, destination: str = None, verbose: bool = False) -> str:
109
+ """Save audio to a specific destination.
110
+
111
+ Args:
112
+ audio_file (str): Path to the source audio file
113
+ destination (str, optional): Destination path. Defaults to current directory with timestamp.
114
+ verbose (bool, optional): Whether to print debug information. Defaults to False.
115
+
116
+ Returns:
117
+ str: Path to the saved audio file
118
+ """
119
+ import shutil
120
+ import os
121
+ from pathlib import Path
122
+ import time
123
+
124
+ source_path = Path(audio_file)
125
+
126
+ if not source_path.exists():
127
+ raise FileNotFoundError(f"Audio file not found: {audio_file}")
128
+
129
+ if destination is None:
130
+ # Create a default destination with timestamp in current directory
131
+ timestamp = int(time.time())
132
+ destination = os.path.join(os.getcwd(), f"tts_audio_{timestamp}{source_path.suffix}")
133
+
134
+ # Ensure the destination directory exists
135
+ os.makedirs(os.path.dirname(os.path.abspath(destination)), exist_ok=True)
136
+
137
+ # Copy the file
138
+ shutil.copy2(source_path, destination)
139
+
140
+ if verbose:
141
+ print(f"[debug] Audio saved to {destination}")
142
+
143
+ return destination
144
+
145
+ def stream_audio(self, text: str, voice: str = None, chunk_size: int = 1024, verbose: bool = False) -> Generator[bytes, None, None]:
146
+ """Stream audio in chunks.
147
+
148
+ Args:
149
+ text (str): The text to convert to speech
150
+ voice (str, optional): The voice to use. Defaults to provider's default voice.
151
+ chunk_size (int, optional): Size of audio chunks to yield. Defaults to 1024.
152
+ verbose (bool, optional): Whether to print debug information. Defaults to False.
153
+
154
+ Yields:
155
+ Generator[bytes, None, None]: Audio data chunks
156
+ """
157
+ # Generate the audio file
158
+ audio_file = self.tts(text, voice=voice, verbose=verbose)
159
+
160
+ # Stream the file in chunks
161
+ with open(audio_file, 'rb') as f:
162
+ while chunk := f.read(chunk_size):
163
+ yield chunk
164
+
165
+ class AsyncTTSProvider(ABC):
166
+
167
+ @abstractmethod
168
+ async def tts(self, text: str, voice: str = None, verbose: bool = False) -> str:
169
+ """Convert text to speech and save to a temporary file asynchronously.
170
+
171
+ Args:
172
+ text (str): The text to convert to speech
173
+ voice (str, optional): The voice to use. Defaults to provider's default voice.
174
+ verbose (bool, optional): Whether to print debug information. Defaults to False.
175
+
176
+ Returns:
177
+ str: Path to the generated audio file
178
+ """
179
+ raise NotImplementedError("Method needs to be implemented in subclass")
180
+
181
+
182
+ class STTProvider(ABC):
183
+ """Abstract base class for Speech-to-Text providers."""
184
+
185
+ @abstractmethod
186
+ def transcribe(self, audio_path: Union[str, Path], **kwargs) -> Dict[str, Any]:
187
+ """Transcribe audio file to text.
188
+
189
+ Args:
190
+ audio_path (Union[str, Path]): Path to the audio file
191
+ **kwargs: Additional provider-specific parameters
192
+
193
+ Returns:
194
+ Dict[str, Any]: Transcription result in OpenAI Whisper format
195
+ """
196
+ raise NotImplementedError("Method needs to be implemented in subclass")
197
+
198
+ @abstractmethod
199
+ def transcribe_from_url(self, audio_url: str, **kwargs) -> Dict[str, Any]:
200
+ """Transcribe audio from URL to text.
201
+
202
+ Args:
203
+ audio_url (str): URL of the audio file
204
+ **kwargs: Additional provider-specific parameters
205
+
206
+ Returns:
207
+ Dict[str, Any]: Transcription result in OpenAI Whisper format
208
+ """
209
+ raise NotImplementedError("Method needs to be implemented in subclass")
210
+
211
+
212
+ class AsyncSTTProvider(ABC):
213
+ """Abstract base class for asynchronous Speech-to-Text providers."""
214
+
215
+ @abstractmethod
216
+ async def transcribe(self, audio_path: Union[str, Path], **kwargs) -> Dict[str, Any]:
217
+ """Transcribe audio file to text asynchronously.
218
+
219
+ Args:
220
+ audio_path (Union[str, Path]): Path to the audio file
221
+ **kwargs: Additional provider-specific parameters
222
+
223
+ Returns:
224
+ Dict[str, Any]: Transcription result in OpenAI Whisper format
225
+ """
226
+ raise NotImplementedError("Method needs to be implemented in subclass")
227
+
228
+ @abstractmethod
229
+ async def transcribe_from_url(self, audio_url: str, **kwargs) -> Dict[str, Any]:
230
+ """Transcribe audio from URL to text asynchronously.
231
+
232
+ Args:
233
+ audio_url (str): URL of the audio file
234
+ **kwargs: Additional provider-specific parameters
235
+
236
+ Returns:
237
+ Dict[str, Any]: Transcription result in OpenAI Whisper format
238
+ """
239
+ raise NotImplementedError("Method needs to be implemented in subclass")
240
+
241
+ async def save_audio(self, audio_file: str, destination: str = None, verbose: bool = False) -> str:
242
+ """Save audio to a specific destination asynchronously.
243
+
244
+ Args:
245
+ audio_file (str): Path to the source audio file
246
+ destination (str, optional): Destination path. Defaults to current directory with timestamp.
247
+ verbose (bool, optional): Whether to print debug information. Defaults to False.
248
+
249
+ Returns:
250
+ str: Path to the saved audio file
251
+ """
252
+ import shutil
253
+ import os
254
+ from pathlib import Path
255
+ import time
256
+ import asyncio
257
+
258
+ source_path = Path(audio_file)
259
+
260
+ if not source_path.exists():
261
+ raise FileNotFoundError(f"Audio file not found: {audio_file}")
262
+
263
+ if destination is None:
264
+ # Create a default destination with timestamp in current directory
265
+ timestamp = int(time.time())
266
+ destination = os.path.join(os.getcwd(), f"tts_audio_{timestamp}{source_path.suffix}")
267
+
268
+ # Ensure the destination directory exists
269
+ os.makedirs(os.path.dirname(os.path.abspath(destination)), exist_ok=True)
270
+
271
+ # Copy the file using asyncio to avoid blocking
272
+ await asyncio.to_thread(shutil.copy2, source_path, destination)
273
+
274
+ if verbose:
275
+ print(f"[debug] Audio saved to {destination}")
276
+
277
+ return destination
278
+
279
+ async def stream_audio(self, text: str, voice: str = None, chunk_size: int = 1024, verbose: bool = False) -> AsyncGenerator[bytes, None]:
280
+ """Stream audio in chunks asynchronously.
281
+
282
+ Args:
283
+ text (str): The text to convert to speech
284
+ voice (str, optional): The voice to use. Defaults to provider's default voice.
285
+ chunk_size (int, optional): Size of audio chunks to yield. Defaults to 1024.
286
+ verbose (bool, optional): Whether to print debug information. Defaults to False.
287
+
288
+ Yields:
289
+ AsyncGenerator[bytes, None]: Audio data chunks
290
+ """
291
+ import aiofiles
292
+
293
+ # Generate the audio file
294
+ audio_file = await self.tts(text, voice=voice, verbose=verbose)
295
+
296
+ # Stream the file in chunks
297
+ async with aiofiles.open(audio_file, 'rb') as f:
298
+ while chunk := await f.read(chunk_size):
299
+ yield chunk
300
+
301
+ class AISearch(ABC):
302
+ """Abstract base class for AI-powered search providers.
303
+
304
+ This class defines the interface for AI search providers that can perform
305
+ web searches and return AI-generated responses based on search results.
306
+
307
+ All search providers should inherit from this class and implement the
308
+ required methods.
309
+ """
310
+
311
+ @abstractmethod
312
+ def search(
313
+ self,
314
+ prompt: str,
315
+ stream: bool = False,
316
+ raw: bool = False,
317
+ ) -> Union[SearchResponse, Generator[Union[Dict[str, str], SearchResponse], None, None]]:
318
+ """Search using the provider's API and get AI-generated responses.
319
+
320
+ This method sends a search query to the provider and returns the AI-generated response.
321
+ It supports both streaming and non-streaming modes, as well as raw response format.
322
+
323
+ Args:
324
+ prompt (str): The search query or prompt to send to the API.
325
+ stream (bool, optional): If True, yields response chunks as they arrive.
326
+ If False, returns complete response. Defaults to False.
327
+ raw (bool, optional): If True, returns raw response dictionaries.
328
+ If False, returns SearchResponse objects that convert to text automatically.
329
+ Defaults to False.
330
+
331
+ Returns:
332
+ Union[SearchResponse, Generator[Union[Dict[str, str], SearchResponse], None, None]]:
333
+ - If stream=False: Returns complete response as SearchResponse object
334
+ - If stream=True: Yields response chunks as either Dict or SearchResponse objects
335
+
336
+ Raises:
337
+ APIConnectionError: If the API request fails
338
+ """
339
+ raise NotImplementedError("Method needs to be implemented in subclass")
340
+
341
+ class AsyncAISearch(ABC):
342
+ """Abstract base class for asynchronous AI-powered search providers.
343
+
344
+ This class defines the interface for asynchronous AI search providers that can perform
345
+ web searches and return AI-generated responses based on search results.
346
+
347
+ All asynchronous search providers should inherit from this class and implement the
348
+ required methods.
349
+ """
350
+
351
+ @abstractmethod
352
+ async def search(
353
+ self,
354
+ prompt: str,
355
+ stream: bool = False,
356
+ raw: bool = False,
357
+ ) -> Union[SearchResponse, AsyncGenerator[Union[Dict[str, str], SearchResponse], None]]:
358
+ """Search using the provider's API and get AI-generated responses asynchronously.
359
+
360
+ This method sends a search query to the provider and returns the AI-generated response.
361
+ It supports both streaming and non-streaming modes, as well as raw response format.
362
+
363
+ Args:
364
+ prompt (str): The search query or prompt to send to the API.
365
+ stream (bool, optional): If True, yields response chunks as they arrive.
366
+ If False, returns complete response. Defaults to False.
367
+ raw (bool, optional): If True, returns raw response dictionaries.
368
+ If False, returns SearchResponse objects that convert to text automatically.
369
+ Defaults to False.
370
+
371
+ Returns:
372
+ Union[SearchResponse, AsyncGenerator[Union[Dict[str, str], SearchResponse], None]]:
373
+ - If stream=False: Returns complete response as SearchResponse object
374
+ - If stream=True: Yields response chunks as either Dict or SearchResponse objects
375
+
376
+ Raises:
377
+ APIConnectionError: If the API request fails
378
+ """
379
+ raise NotImplementedError("Method needs to be implemented in subclass")