webscout 8.3.6__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 (304) hide show
  1. webscout/AIauto.py +250 -250
  2. webscout/AIbase.py +379 -379
  3. webscout/AIutel.py +60 -58
  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 +33 -11
  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 -314
  61. webscout/Provider/AISEARCH/stellar_search.py +177 -177
  62. webscout/Provider/AISEARCH/webpilotai_search.py +255 -255
  63. webscout/Provider/Aitopia.py +314 -315
  64. webscout/Provider/Andi.py +3 -3
  65. webscout/Provider/Apriel.py +306 -0
  66. webscout/Provider/ChatGPTClone.py +236 -236
  67. webscout/Provider/ChatSandbox.py +343 -342
  68. webscout/Provider/Cloudflare.py +324 -324
  69. webscout/Provider/Cohere.py +208 -207
  70. webscout/Provider/Deepinfra.py +370 -369
  71. webscout/Provider/ExaAI.py +260 -260
  72. webscout/Provider/ExaChat.py +308 -387
  73. webscout/Provider/Flowith.py +221 -221
  74. webscout/Provider/GMI.py +293 -0
  75. webscout/Provider/Gemini.py +164 -162
  76. webscout/Provider/GeminiProxy.py +167 -166
  77. webscout/Provider/GithubChat.py +371 -370
  78. webscout/Provider/Groq.py +800 -800
  79. webscout/Provider/HeckAI.py +383 -379
  80. webscout/Provider/Jadve.py +282 -297
  81. webscout/Provider/K2Think.py +308 -0
  82. webscout/Provider/Koboldai.py +206 -384
  83. webscout/Provider/LambdaChat.py +423 -425
  84. webscout/Provider/Nemotron.py +244 -245
  85. webscout/Provider/Netwrck.py +248 -247
  86. webscout/Provider/OLLAMA.py +395 -394
  87. webscout/Provider/OPENAI/Cloudflare.py +394 -395
  88. webscout/Provider/OPENAI/FalconH1.py +452 -457
  89. webscout/Provider/OPENAI/FreeGemini.py +297 -299
  90. webscout/Provider/OPENAI/{monochat.py → K2Think.py} +432 -329
  91. webscout/Provider/OPENAI/NEMOTRON.py +241 -244
  92. webscout/Provider/OPENAI/PI.py +428 -427
  93. webscout/Provider/OPENAI/README.md +959 -959
  94. webscout/Provider/OPENAI/TogetherAI.py +345 -345
  95. webscout/Provider/OPENAI/TwoAI.py +466 -467
  96. webscout/Provider/OPENAI/__init__.py +33 -59
  97. webscout/Provider/OPENAI/ai4chat.py +313 -303
  98. webscout/Provider/OPENAI/base.py +249 -269
  99. webscout/Provider/OPENAI/chatglm.py +528 -0
  100. webscout/Provider/OPENAI/chatgpt.py +593 -588
  101. webscout/Provider/OPENAI/chatgptclone.py +521 -524
  102. webscout/Provider/OPENAI/chatsandbox.py +202 -177
  103. webscout/Provider/OPENAI/deepinfra.py +319 -315
  104. webscout/Provider/OPENAI/e2b.py +1665 -1665
  105. webscout/Provider/OPENAI/exaai.py +420 -420
  106. webscout/Provider/OPENAI/exachat.py +452 -452
  107. webscout/Provider/OPENAI/friendli.py +232 -232
  108. webscout/Provider/OPENAI/{refact.py → gmi.py} +324 -274
  109. webscout/Provider/OPENAI/groq.py +364 -364
  110. webscout/Provider/OPENAI/heckai.py +314 -311
  111. webscout/Provider/OPENAI/llmchatco.py +337 -337
  112. webscout/Provider/OPENAI/netwrck.py +355 -354
  113. webscout/Provider/OPENAI/oivscode.py +290 -290
  114. webscout/Provider/OPENAI/opkfc.py +518 -518
  115. webscout/Provider/OPENAI/pydantic_imports.py +1 -1
  116. webscout/Provider/OPENAI/scirachat.py +535 -529
  117. webscout/Provider/OPENAI/sonus.py +308 -308
  118. webscout/Provider/OPENAI/standardinput.py +442 -442
  119. webscout/Provider/OPENAI/textpollinations.py +340 -348
  120. webscout/Provider/OPENAI/toolbaz.py +419 -413
  121. webscout/Provider/OPENAI/typefully.py +362 -362
  122. webscout/Provider/OPENAI/utils.py +295 -295
  123. webscout/Provider/OPENAI/venice.py +436 -436
  124. webscout/Provider/OPENAI/wisecat.py +387 -387
  125. webscout/Provider/OPENAI/writecream.py +166 -166
  126. webscout/Provider/OPENAI/x0gpt.py +378 -378
  127. webscout/Provider/OPENAI/yep.py +389 -389
  128. webscout/Provider/OpenGPT.py +230 -230
  129. webscout/Provider/Openai.py +244 -496
  130. webscout/Provider/PI.py +405 -404
  131. webscout/Provider/Perplexitylabs.py +430 -431
  132. webscout/Provider/QwenLM.py +272 -254
  133. webscout/Provider/STT/__init__.py +32 -2
  134. webscout/Provider/{Llama3.py → Sambanova.py} +257 -258
  135. webscout/Provider/StandardInput.py +309 -309
  136. webscout/Provider/TTI/README.md +82 -82
  137. webscout/Provider/TTI/__init__.py +33 -12
  138. webscout/Provider/TTI/aiarta.py +413 -413
  139. webscout/Provider/TTI/base.py +136 -136
  140. webscout/Provider/TTI/bing.py +243 -243
  141. webscout/Provider/TTI/gpt1image.py +149 -149
  142. webscout/Provider/TTI/imagen.py +196 -196
  143. webscout/Provider/TTI/infip.py +211 -211
  144. webscout/Provider/TTI/magicstudio.py +232 -232
  145. webscout/Provider/TTI/monochat.py +219 -219
  146. webscout/Provider/TTI/piclumen.py +214 -214
  147. webscout/Provider/TTI/pixelmuse.py +232 -232
  148. webscout/Provider/TTI/pollinations.py +232 -232
  149. webscout/Provider/TTI/together.py +288 -288
  150. webscout/Provider/TTI/utils.py +12 -12
  151. webscout/Provider/TTI/venice.py +367 -367
  152. webscout/Provider/TTS/README.md +192 -192
  153. webscout/Provider/TTS/__init__.py +33 -10
  154. webscout/Provider/TTS/parler.py +110 -110
  155. webscout/Provider/TTS/streamElements.py +333 -333
  156. webscout/Provider/TTS/utils.py +280 -280
  157. webscout/Provider/TeachAnything.py +237 -236
  158. webscout/Provider/TextPollinationsAI.py +311 -318
  159. webscout/Provider/TogetherAI.py +356 -357
  160. webscout/Provider/TwoAI.py +313 -569
  161. webscout/Provider/TypliAI.py +312 -311
  162. webscout/Provider/UNFINISHED/ChatHub.py +208 -208
  163. webscout/Provider/UNFINISHED/ChutesAI.py +313 -313
  164. webscout/Provider/{GizAI.py → UNFINISHED/GizAI.py} +294 -294
  165. webscout/Provider/{Marcus.py → UNFINISHED/Marcus.py} +198 -198
  166. webscout/Provider/{Qodo.py → UNFINISHED/Qodo.py} +477 -477
  167. webscout/Provider/UNFINISHED/VercelAIGateway.py +338 -338
  168. webscout/Provider/{XenAI.py → UNFINISHED/XenAI.py} +324 -324
  169. webscout/Provider/UNFINISHED/Youchat.py +330 -330
  170. webscout/Provider/UNFINISHED/liner.py +334 -0
  171. webscout/Provider/UNFINISHED/liner_api_request.py +262 -262
  172. webscout/Provider/UNFINISHED/puterjs.py +634 -634
  173. webscout/Provider/UNFINISHED/samurai.py +223 -223
  174. webscout/Provider/UNFINISHED/test_lmarena.py +119 -119
  175. webscout/Provider/Venice.py +251 -250
  176. webscout/Provider/VercelAI.py +256 -255
  177. webscout/Provider/WiseCat.py +232 -231
  178. webscout/Provider/WrDoChat.py +367 -366
  179. webscout/Provider/__init__.py +33 -86
  180. webscout/Provider/ai4chat.py +174 -174
  181. webscout/Provider/akashgpt.py +331 -334
  182. webscout/Provider/cerebras.py +446 -340
  183. webscout/Provider/chatglm.py +394 -214
  184. webscout/Provider/cleeai.py +211 -212
  185. webscout/Provider/deepseek_assistant.py +1 -1
  186. webscout/Provider/elmo.py +282 -282
  187. webscout/Provider/geminiapi.py +208 -208
  188. webscout/Provider/granite.py +261 -261
  189. webscout/Provider/hermes.py +263 -265
  190. webscout/Provider/julius.py +223 -222
  191. webscout/Provider/learnfastai.py +309 -309
  192. webscout/Provider/llama3mitril.py +214 -214
  193. webscout/Provider/llmchat.py +243 -243
  194. webscout/Provider/llmchatco.py +290 -290
  195. webscout/Provider/meta.py +801 -801
  196. webscout/Provider/oivscode.py +309 -309
  197. webscout/Provider/scira_chat.py +384 -457
  198. webscout/Provider/searchchat.py +292 -291
  199. webscout/Provider/sonus.py +258 -258
  200. webscout/Provider/toolbaz.py +370 -364
  201. webscout/Provider/turboseek.py +274 -265
  202. webscout/Provider/typefully.py +208 -207
  203. webscout/Provider/x0gpt.py +1 -0
  204. webscout/Provider/yep.py +372 -371
  205. webscout/__init__.py +30 -31
  206. webscout/__main__.py +5 -5
  207. webscout/auth/api_key_manager.py +189 -189
  208. webscout/auth/config.py +175 -175
  209. webscout/auth/models.py +185 -185
  210. webscout/auth/routes.py +664 -664
  211. webscout/auth/simple_logger.py +236 -236
  212. webscout/cli.py +523 -523
  213. webscout/conversation.py +438 -438
  214. webscout/exceptions.py +361 -361
  215. webscout/litagent/Readme.md +298 -298
  216. webscout/litagent/__init__.py +28 -28
  217. webscout/litagent/agent.py +581 -581
  218. webscout/litagent/constants.py +59 -59
  219. webscout/litprinter/__init__.py +58 -58
  220. webscout/models.py +181 -181
  221. webscout/optimizers.py +419 -419
  222. webscout/prompt_manager.py +288 -288
  223. webscout/sanitize.py +1078 -1078
  224. webscout/scout/README.md +401 -401
  225. webscout/scout/__init__.py +8 -8
  226. webscout/scout/core/__init__.py +6 -6
  227. webscout/scout/core/crawler.py +297 -297
  228. webscout/scout/core/scout.py +706 -706
  229. webscout/scout/core/search_result.py +95 -95
  230. webscout/scout/core/text_analyzer.py +62 -62
  231. webscout/scout/core/text_utils.py +277 -277
  232. webscout/scout/core/web_analyzer.py +51 -51
  233. webscout/scout/element.py +599 -599
  234. webscout/scout/parsers/__init__.py +69 -69
  235. webscout/scout/parsers/html5lib_parser.py +172 -172
  236. webscout/scout/parsers/html_parser.py +236 -236
  237. webscout/scout/parsers/lxml_parser.py +178 -178
  238. webscout/scout/utils.py +37 -37
  239. webscout/swiftcli/Readme.md +323 -323
  240. webscout/swiftcli/__init__.py +95 -95
  241. webscout/swiftcli/core/__init__.py +7 -7
  242. webscout/swiftcli/core/cli.py +308 -308
  243. webscout/swiftcli/core/context.py +104 -104
  244. webscout/swiftcli/core/group.py +241 -241
  245. webscout/swiftcli/decorators/__init__.py +28 -28
  246. webscout/swiftcli/decorators/command.py +221 -221
  247. webscout/swiftcli/decorators/options.py +220 -220
  248. webscout/swiftcli/decorators/output.py +302 -302
  249. webscout/swiftcli/exceptions.py +21 -21
  250. webscout/swiftcli/plugins/__init__.py +9 -9
  251. webscout/swiftcli/plugins/base.py +135 -135
  252. webscout/swiftcli/plugins/manager.py +269 -269
  253. webscout/swiftcli/utils/__init__.py +59 -59
  254. webscout/swiftcli/utils/formatting.py +252 -252
  255. webscout/swiftcli/utils/parsing.py +267 -267
  256. webscout/update_checker.py +117 -117
  257. webscout/version.py +1 -1
  258. webscout/webscout_search.py +1183 -1183
  259. webscout/webscout_search_async.py +649 -649
  260. webscout/yep_search.py +346 -346
  261. webscout/zeroart/README.md +89 -89
  262. webscout/zeroart/__init__.py +134 -134
  263. webscout/zeroart/base.py +66 -66
  264. webscout/zeroart/effects.py +100 -100
  265. webscout/zeroart/fonts.py +1238 -1238
  266. {webscout-8.3.6.dist-info → webscout-2025.10.11.dist-info}/METADATA +937 -936
  267. webscout-2025.10.11.dist-info/RECORD +300 -0
  268. webscout/Provider/AISEARCH/DeepFind.py +0 -254
  269. webscout/Provider/AllenAI.py +0 -440
  270. webscout/Provider/Blackboxai.py +0 -793
  271. webscout/Provider/FreeGemini.py +0 -250
  272. webscout/Provider/GptOss.py +0 -207
  273. webscout/Provider/Hunyuan.py +0 -283
  274. webscout/Provider/Kimi.py +0 -445
  275. webscout/Provider/MCPCore.py +0 -322
  276. webscout/Provider/MiniMax.py +0 -207
  277. webscout/Provider/OPENAI/BLACKBOXAI.py +0 -1045
  278. webscout/Provider/OPENAI/MiniMax.py +0 -298
  279. webscout/Provider/OPENAI/Qwen3.py +0 -304
  280. webscout/Provider/OPENAI/autoproxy.py +0 -1067
  281. webscout/Provider/OPENAI/copilot.py +0 -321
  282. webscout/Provider/OPENAI/gptoss.py +0 -288
  283. webscout/Provider/OPENAI/kimi.py +0 -469
  284. webscout/Provider/OPENAI/mcpcore.py +0 -431
  285. webscout/Provider/OPENAI/multichat.py +0 -378
  286. webscout/Provider/OPENAI/qodo.py +0 -630
  287. webscout/Provider/OPENAI/xenai.py +0 -514
  288. webscout/Provider/Reka.py +0 -214
  289. webscout/Provider/UNFINISHED/fetch_together_models.py +0 -90
  290. webscout/Provider/asksteve.py +0 -220
  291. webscout/Provider/copilot.py +0 -441
  292. webscout/Provider/freeaichat.py +0 -294
  293. webscout/Provider/koala.py +0 -182
  294. webscout/Provider/lmarena.py +0 -198
  295. webscout/Provider/monochat.py +0 -275
  296. webscout/Provider/multichat.py +0 -375
  297. webscout/Provider/scnet.py +0 -244
  298. webscout/Provider/talkai.py +0 -194
  299. webscout/tempid.py +0 -128
  300. webscout-8.3.6.dist-info/RECORD +0 -327
  301. {webscout-8.3.6.dist-info → webscout-2025.10.11.dist-info}/WHEEL +0 -0
  302. {webscout-8.3.6.dist-info → webscout-2025.10.11.dist-info}/entry_points.txt +0 -0
  303. {webscout-8.3.6.dist-info → webscout-2025.10.11.dist-info}/licenses/LICENSE.md +0 -0
  304. {webscout-8.3.6.dist-info → webscout-2025.10.11.dist-info}/top_level.txt +0 -0
@@ -1,390 +1,390 @@
1
- import time
2
- import uuid
3
- import cloudscraper # Import cloudscraper
4
- import json
5
- from typing import List, Dict, Optional, Union, Generator, Any
6
-
7
- # Import base classes and utility structures
8
- from webscout.Provider.OPENAI.base import OpenAICompatibleProvider, BaseChat, BaseCompletions
9
- from webscout.Provider.OPENAI.utils import (
10
- ChatCompletionChunk, ChatCompletion, Choice, ChoiceDelta,
11
- ChatCompletionMessage, CompletionUsage, get_system_prompt, count_tokens # Import count_tokens
12
- )
13
-
14
- # Attempt to import LitAgent, fallback if not available
15
- try:
16
- from webscout.litagent import LitAgent
17
- except ImportError:
18
- # Define a dummy LitAgent if webscout is not installed or accessible
19
- class LitAgent:
20
- def generate_fingerprint(self, browser: str = "chrome") -> Dict[str, Any]:
21
- print("Warning: LitAgent not found. Using default minimal headers.")
22
- return {
23
- "accept": "*/*",
24
- "accept_language": "en-US,en;q=0.9",
25
- "platform": "Windows",
26
- "sec_ch_ua": '"Not/A)Brand";v="99", "Google Chrome";v="127", "Chromium";v="127"',
27
- "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",
28
- "browser_type": browser,
29
- }
30
-
31
- # --- YEPCHAT Client ---
32
-
33
- # ANSI escape codes for formatting
34
- BOLD = "\033[1m"
35
- RED = "\033[91m"
36
- RESET = "\033[0m"
37
-
38
- class Completions(BaseCompletions):
39
- def __init__(self, client: 'YEPCHAT'):
40
- self._client = client
41
-
42
- def create(
43
- self,
44
- *,
45
- model: str,
46
- messages: List[Dict[str, str]],
47
- max_tokens: Optional[int] = 1280,
48
- stream: bool = False,
49
- temperature: Optional[float] = 0.6,
50
- top_p: Optional[float] = 0.7,
51
- system_prompt: Optional[str] = None, # Added for consistency, but will be ignored
52
- timeout: Optional[int] = None,
53
- proxies: Optional[Dict[str, str]] = None,
54
- **kwargs: Any
55
- ) -> Union[ChatCompletion, Generator[ChatCompletionChunk, None, None]]:
56
- """
57
- Creates a model response for the given chat conversation using YEPCHAT API.
58
- Mimics openai.chat.completions.create
59
- Note: YEPCHAT does not support system messages. They will be ignored.
60
- """
61
- # Only accept and use the raw model name (no prefix logic)
62
- model_raw = model
63
- # Validate model
64
- if model_raw not in self._client.AVAILABLE_MODELS:
65
- raise ValueError(
66
- f"Invalid model: {model}. Choose from: {self._client.AVAILABLE_MODELS}"
67
- )
68
-
69
- # Filter out system messages and warn the user if any are present
70
- filtered_messages = []
71
- has_system_message = False
72
- if get_system_prompt(messages) or system_prompt: # Check both message list and explicit param
73
- has_system_message = True
74
-
75
- for msg in messages:
76
- if msg["role"] == "system":
77
- continue # Skip system messages
78
- filtered_messages.append(msg)
79
-
80
- if has_system_message:
81
- # Print warning in bold red
82
- print(f"{BOLD}{RED}Warning: YEPCHAT does not support system messages, they will be ignored.{RESET}")
83
-
84
- # If no messages left after filtering, raise an error
85
- if not filtered_messages:
86
- raise ValueError("At least one user or assistant message is required for YEPCHAT.")
87
-
88
- payload = {
89
- "stream": stream,
90
- "max_tokens": max_tokens,
91
- "top_p": top_p,
92
- "temperature": temperature,
93
- "messages": filtered_messages, # Use filtered messages
94
- "model": model_raw, # Send only the raw model name to the API
95
- }
96
-
97
- # Add any extra kwargs to the payload
98
- payload.update(kwargs)
99
-
100
- request_id = f"chatcmpl-{uuid.uuid4()}"
101
- created_time = int(time.time())
102
-
103
- if stream:
104
- return self._create_stream(request_id, created_time, model, payload, timeout, proxies)
105
- else:
106
- return self._create_non_stream(request_id, created_time, model, payload, timeout, proxies)
107
-
108
- def _create_stream(
109
- self, request_id: str, created_time: int, model: str, payload: Dict[str, Any], timeout: Optional[int] = None, proxies: Optional[Dict[str, str]] = None
110
- ) -> Generator[ChatCompletionChunk, None, None]:
111
- try:
112
- response = self._client.session.post(
113
- self._client.api_endpoint,
114
- headers=self._client.headers,
115
- cookies=self._client.cookies,
116
- json=payload,
117
- stream=True,
118
- timeout=timeout or self._client.timeout,
119
- proxies=proxies or getattr(self._client, "proxies", None)
120
- )
121
-
122
- if not response.ok:
123
- raise IOError(
124
- f"YEPCHAT API Error: {response.status_code} {response.reason} - {response.text}"
125
- )
126
-
127
- # Track tokens for streaming
128
- prompt_tokens = count_tokens([m.get('content', '') for m in payload.get('messages', [])])
129
- completion_tokens = 0
130
- total_tokens = prompt_tokens
131
-
132
- for line in response.iter_lines(decode_unicode=True):
133
- if line:
134
- line = line.strip()
135
- if line.startswith("data: "):
136
- json_str = line[6:]
137
- if json_str == "[DONE]":
138
- break
139
- try:
140
- data = json.loads(json_str)
141
- choice_data = data.get('choices', [{}])[0]
142
- delta_data = choice_data.get('delta', {})
143
- finish_reason = choice_data.get('finish_reason')
144
- content = delta_data.get('content')
145
- role = delta_data.get('role', None)
146
-
147
- # Count tokens for this chunk
148
- chunk_tokens = count_tokens(content) if content else 0
149
- completion_tokens += chunk_tokens
150
- total_tokens = prompt_tokens + completion_tokens
151
-
152
- if content is not None or role is not None:
153
- delta = ChoiceDelta(content=content, role=role)
154
- choice = Choice(index=0, delta=delta, finish_reason=finish_reason)
155
- chunk = ChatCompletionChunk(
156
- id=request_id,
157
- choices=[choice],
158
- created=created_time,
159
- model=model,
160
- )
161
- # Set usage directly on the chunk object
162
- chunk.usage = {
163
- "prompt_tokens": prompt_tokens,
164
- "completion_tokens": completion_tokens,
165
- "total_tokens": total_tokens,
166
- "estimated_cost": None
167
- }
168
- # Yield the chunk with usage information
169
- yield chunk
170
-
171
- except json.JSONDecodeError:
172
- print(f"Warning: Could not decode JSON line: {json_str}")
173
- continue
174
-
175
- # Yield final chunk with finish reason if not already sent
176
- delta = ChoiceDelta()
177
- choice = Choice(index=0, delta=delta, finish_reason="stop")
178
- chunk = ChatCompletionChunk(
179
- id=request_id,
180
- choices=[choice],
181
- created=created_time,
182
- model=model,
183
- )
184
- # Set usage directly on the chunk object
185
- chunk.usage = {
186
- "prompt_tokens": prompt_tokens,
187
- "completion_tokens": completion_tokens,
188
- "total_tokens": total_tokens,
189
- "estimated_cost": None
190
- }
191
- yield chunk
192
-
193
- except cloudscraper.exceptions.CloudflareChallengeError as e:
194
- pass
195
-
196
- def _create_non_stream(
197
- self, request_id: str, created_time: int, model: str, payload: Dict[str, Any], timeout: Optional[int] = None, proxies: Optional[Dict[str, str]] = None
198
- ) -> ChatCompletion:
199
- full_response_content = ""
200
- finish_reason = "stop"
201
- try:
202
- # Make a non-streaming request to the API
203
- payload_copy = payload.copy()
204
- payload_copy["stream"] = False
205
- response = self._client.session.post(
206
- self._client.api_endpoint,
207
- headers=self._client.headers,
208
- cookies=self._client.cookies,
209
- json=payload_copy,
210
- timeout=timeout or self._client.timeout,
211
- proxies=proxies or getattr(self._client, "proxies", None)
212
- )
213
- if not response.ok:
214
- raise IOError(
215
- f"YEPCHAT API Error: {response.status_code} {response.reason} - {response.text}"
216
- )
217
- data = response.json()
218
- if 'choices' in data and len(data['choices']) > 0:
219
- # YEPCHAT non-streaming returns message content in choices[0]['message']['content']
220
- full_response_content = data['choices'][0].get('message', {}).get('content', '')
221
- finish_reason = data['choices'][0].get('finish_reason', 'stop')
222
- else:
223
- full_response_content = ''
224
- finish_reason = 'stop'
225
- except Exception as e:
226
- print(f"Error obtaining non-stream response from YEPCHAT: {e}")
227
- finish_reason = "error"
228
-
229
- message = ChatCompletionMessage(
230
- role="assistant",
231
- content=full_response_content
232
- )
233
- choice = Choice(
234
- index=0,
235
- message=message,
236
- finish_reason=finish_reason
237
- )
238
- # Use count_tokens to compute usage
239
- prompt_tokens = count_tokens([m.get('content', '') for m in payload.get('messages', [])])
240
- completion_tokens = count_tokens(full_response_content)
241
- usage = CompletionUsage(
242
- prompt_tokens=prompt_tokens,
243
- completion_tokens=completion_tokens,
244
- total_tokens=prompt_tokens + completion_tokens
245
- )
246
- completion = ChatCompletion(
247
- id=request_id,
248
- choices=[choice],
249
- created=created_time,
250
- model=model,
251
- usage=usage,
252
- )
253
- return completion
254
-
255
- class Chat(BaseChat):
256
- def __init__(self, client: 'YEPCHAT'):
257
- self.completions = Completions(client)
258
-
259
- class YEPCHAT(OpenAICompatibleProvider):
260
- """
261
- OpenAI-compatible client for YEPCHAT API.
262
-
263
- Usage:
264
- client = YEPCHAT()
265
- response = client.chat.completions.create(
266
- model="DeepSeek-R1-Distill-Qwen-32B",
267
- messages=[{"role": "user", "content": "Hello!"}]
268
- )
269
- print(response.choices[0].message.content)
270
- """
271
- _base_models = ["DeepSeek-R1-Distill-Qwen-32B", "Mixtral-8x7B-Instruct-v0.1"]
272
-
273
- # Create AVAILABLE_MODELS as a list of base model names (no prefix)
274
- AVAILABLE_MODELS = _base_models
275
-
276
- def __init__(
277
- self,
278
- browser: str = "chrome"
279
- ):
280
- """
281
- Initialize the YEPCHAT client.
282
-
283
- Args:
284
- browser: Browser name for LitAgent to generate User-Agent.
285
- """
286
- self.timeout = None
287
- self.api_endpoint = "https://api.yep.com/v1/chat/completions"
288
- self.session = cloudscraper.create_scraper() # Use cloudscraper
289
-
290
- # Initialize LitAgent for user agent generation and fingerprinting
291
- try:
292
- agent = LitAgent()
293
- fingerprint = agent.generate_fingerprint(browser=browser)
294
- except Exception as e:
295
- print(f"Warning: Failed to generate fingerprint with LitAgent: {e}. Using fallback.")
296
- # Fallback fingerprint data
297
- fingerprint = {
298
- "accept": "*/*",
299
- "accept_language": "en-US,en;q=0.9",
300
- "sec_ch_ua": '"Not/A)Brand";v="99", "Google Chrome";v="127", "Chromium";v="127"',
301
- "platform": "Windows",
302
- "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"
303
- }
304
-
305
- # Initialize headers using the fingerprint
306
- self.headers = {
307
- "Accept": fingerprint["accept"],
308
- "Accept-Encoding": "gzip, deflate, br, zstd",
309
- "Accept-Language": fingerprint["accept_language"],
310
- "Content-Type": "application/json; charset=utf-8",
311
- "DNT": "1",
312
- "Origin": "https://yep.com",
313
- "Referer": "https://yep.com/",
314
- "Sec-CH-UA": fingerprint["sec_ch_ua"] or '"Not)A;Brand";v="99", "Microsoft Edge";v="127", "Chromium";v="127"',
315
- "Sec-CH-UA-Mobile": "?0",
316
- "Sec-CH-UA-Platform": f'"{fingerprint["platform"]}"',
317
- "User-Agent": fingerprint["user_agent"],
318
- "x-forwarded-for": fingerprint["x-forwarded-for"],
319
- "x-real-ip": fingerprint["x-real-ip"],
320
- "x-client-ip": fingerprint["x-client-ip"],
321
- "forwarded": fingerprint["forwarded"],
322
- "x-forwarded-proto": "https",
323
- "x-request-id": fingerprint["x-request-id"],
324
- }
325
- self.session.headers.update(self.headers)
326
-
327
- # Generate cookies (consider if these need refreshing or specific values)
328
- self.cookies = {"__Host-session": uuid.uuid4().hex, '__cf_bm': uuid.uuid4().hex}
329
-
330
- # Initialize the chat interface
331
- self.chat = Chat(self)
332
-
333
- @property
334
- def models(self):
335
- class _ModelList:
336
- def list(inner_self):
337
- return YEPCHAT.AVAILABLE_MODELS
338
- return _ModelList()
339
-
340
- def convert_model_name(self, model: str) -> str:
341
- """
342
- Ensures the model name is valid for YEPCHAT.
343
- Returns the validated model name or raises an error if invalid.
344
- """
345
- if model in self.AVAILABLE_MODELS:
346
- return model
347
- else:
348
- # Raise error instead of defaulting, as model is mandatory in create()
349
- raise ValueError(f"Model '{model}' not supported by YEPCHAT. Available: {self.AVAILABLE_MODELS}")
350
-
351
- # Example usage (optional, for testing)
352
- if __name__ == '__main__':
353
- print("Testing YEPCHAT OpenAI-Compatible Client...")
354
-
355
- # Test Non-Streaming
356
- try:
357
- print("\n--- Non-Streaming Test (DeepSeek) ---")
358
- client = YEPCHAT()
359
- response = client.chat.completions.create(
360
- model="DeepSeek-R1-Distill-Qwen-32B",
361
- messages=[
362
- {"role": "user", "content": "Say 'Hello World'"}
363
- ],
364
- stream=False
365
- )
366
- print("Response:", response.choices[0].message.content)
367
- print("Usage:", response.usage) # Will show 0 tokens
368
- except Exception as e:
369
- print(f"Non-Streaming Test Failed: {e}")
370
-
371
- # Test Streaming
372
- try:
373
- print("\n--- Streaming Test (Mixtral) ---")
374
- client_stream = YEPCHAT()
375
- stream = client_stream.chat.completions.create(
376
- model="Mixtral-8x7B-Instruct-v0.1",
377
- messages=[
378
- {"role": "user", "content": "Write a short sentence about AI."}
379
- ],
380
- stream=True
381
- )
382
- print("Streaming Response:")
383
- for chunk in stream:
384
- content = chunk.choices[0].delta.content
385
- if content:
386
- print(content, end="", flush=True)
387
- print() # Add a newline at the end
388
-
389
- except Exception as e:
1
+ import time
2
+ import uuid
3
+ import cloudscraper # Import cloudscraper
4
+ import json
5
+ from typing import List, Dict, Optional, Union, Generator, Any
6
+
7
+ # Import base classes and utility structures
8
+ from webscout.Provider.OPENAI.base import OpenAICompatibleProvider, BaseChat, BaseCompletions
9
+ from webscout.Provider.OPENAI.utils import (
10
+ ChatCompletionChunk, ChatCompletion, Choice, ChoiceDelta,
11
+ ChatCompletionMessage, CompletionUsage, get_system_prompt, count_tokens # Import count_tokens
12
+ )
13
+
14
+ # Attempt to import LitAgent, fallback if not available
15
+ try:
16
+ from webscout.litagent import LitAgent
17
+ except ImportError:
18
+ # Define a dummy LitAgent if webscout is not installed or accessible
19
+ class LitAgent:
20
+ def generate_fingerprint(self, browser: str = "chrome") -> Dict[str, Any]:
21
+ print("Warning: LitAgent not found. Using default minimal headers.")
22
+ return {
23
+ "accept": "*/*",
24
+ "accept_language": "en-US,en;q=0.9",
25
+ "platform": "Windows",
26
+ "sec_ch_ua": '"Not/A)Brand";v="99", "Google Chrome";v="127", "Chromium";v="127"',
27
+ "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",
28
+ "browser_type": browser,
29
+ }
30
+
31
+ # --- YEPCHAT Client ---
32
+
33
+ # ANSI escape codes for formatting
34
+ BOLD = "\033[1m"
35
+ RED = "\033[91m"
36
+ RESET = "\033[0m"
37
+
38
+ class Completions(BaseCompletions):
39
+ def __init__(self, client: 'YEPCHAT'):
40
+ self._client = client
41
+
42
+ def create(
43
+ self,
44
+ *,
45
+ model: str,
46
+ messages: List[Dict[str, str]],
47
+ max_tokens: Optional[int] = 1280,
48
+ stream: bool = False,
49
+ temperature: Optional[float] = 0.6,
50
+ top_p: Optional[float] = 0.7,
51
+ system_prompt: Optional[str] = None, # Added for consistency, but will be ignored
52
+ timeout: Optional[int] = None,
53
+ proxies: Optional[Dict[str, str]] = None,
54
+ **kwargs: Any
55
+ ) -> Union[ChatCompletion, Generator[ChatCompletionChunk, None, None]]:
56
+ """
57
+ Creates a model response for the given chat conversation using YEPCHAT API.
58
+ Mimics openai.chat.completions.create
59
+ Note: YEPCHAT does not support system messages. They will be ignored.
60
+ """
61
+ # Only accept and use the raw model name (no prefix logic)
62
+ model_raw = model
63
+ # Validate model
64
+ if model_raw not in self._client.AVAILABLE_MODELS:
65
+ raise ValueError(
66
+ f"Invalid model: {model}. Choose from: {self._client.AVAILABLE_MODELS}"
67
+ )
68
+
69
+ # Filter out system messages and warn the user if any are present
70
+ filtered_messages = []
71
+ has_system_message = False
72
+ if get_system_prompt(messages) or system_prompt: # Check both message list and explicit param
73
+ has_system_message = True
74
+
75
+ for msg in messages:
76
+ if msg["role"] == "system":
77
+ continue # Skip system messages
78
+ filtered_messages.append(msg)
79
+
80
+ if has_system_message:
81
+ # Print warning in bold red
82
+ print(f"{BOLD}{RED}Warning: YEPCHAT does not support system messages, they will be ignored.{RESET}")
83
+
84
+ # If no messages left after filtering, raise an error
85
+ if not filtered_messages:
86
+ raise ValueError("At least one user or assistant message is required for YEPCHAT.")
87
+
88
+ payload = {
89
+ "stream": stream,
90
+ "max_tokens": max_tokens,
91
+ "top_p": top_p,
92
+ "temperature": temperature,
93
+ "messages": filtered_messages, # Use filtered messages
94
+ "model": model_raw, # Send only the raw model name to the API
95
+ }
96
+
97
+ # Add any extra kwargs to the payload
98
+ payload.update(kwargs)
99
+
100
+ request_id = f"chatcmpl-{uuid.uuid4()}"
101
+ created_time = int(time.time())
102
+
103
+ if stream:
104
+ return self._create_stream(request_id, created_time, model, payload, timeout, proxies)
105
+ else:
106
+ return self._create_non_stream(request_id, created_time, model, payload, timeout, proxies)
107
+
108
+ def _create_stream(
109
+ self, request_id: str, created_time: int, model: str, payload: Dict[str, Any], timeout: Optional[int] = None, proxies: Optional[Dict[str, str]] = None
110
+ ) -> Generator[ChatCompletionChunk, None, None]:
111
+ try:
112
+ response = self._client.session.post(
113
+ self._client.api_endpoint,
114
+ headers=self._client.headers,
115
+ cookies=self._client.cookies,
116
+ json=payload,
117
+ stream=True,
118
+ timeout=timeout or self._client.timeout,
119
+ proxies=proxies or getattr(self._client, "proxies", None)
120
+ )
121
+
122
+ if not response.ok:
123
+ raise IOError(
124
+ f"YEPCHAT API Error: {response.status_code} {response.reason} - {response.text}"
125
+ )
126
+
127
+ # Track tokens for streaming
128
+ prompt_tokens = count_tokens([m.get('content', '') for m in payload.get('messages', [])])
129
+ completion_tokens = 0
130
+ total_tokens = prompt_tokens
131
+
132
+ for line in response.iter_lines(decode_unicode=True):
133
+ if line:
134
+ line = line.strip()
135
+ if line.startswith("data: "):
136
+ json_str = line[6:]
137
+ if json_str == "[DONE]":
138
+ break
139
+ try:
140
+ data = json.loads(json_str)
141
+ choice_data = data.get('choices', [{}])[0]
142
+ delta_data = choice_data.get('delta', {})
143
+ finish_reason = choice_data.get('finish_reason')
144
+ content = delta_data.get('content')
145
+ role = delta_data.get('role', None)
146
+
147
+ # Count tokens for this chunk
148
+ chunk_tokens = count_tokens(content) if content else 0
149
+ completion_tokens += chunk_tokens
150
+ total_tokens = prompt_tokens + completion_tokens
151
+
152
+ if content is not None or role is not None:
153
+ delta = ChoiceDelta(content=content, role=role)
154
+ choice = Choice(index=0, delta=delta, finish_reason=finish_reason)
155
+ chunk = ChatCompletionChunk(
156
+ id=request_id,
157
+ choices=[choice],
158
+ created=created_time,
159
+ model=model,
160
+ )
161
+ # Set usage directly on the chunk object
162
+ chunk.usage = {
163
+ "prompt_tokens": prompt_tokens,
164
+ "completion_tokens": completion_tokens,
165
+ "total_tokens": total_tokens,
166
+ "estimated_cost": None
167
+ }
168
+ # Yield the chunk with usage information
169
+ yield chunk
170
+
171
+ except json.JSONDecodeError:
172
+ print(f"Warning: Could not decode JSON line: {json_str}")
173
+ continue
174
+
175
+ # Yield final chunk with finish reason if not already sent
176
+ delta = ChoiceDelta()
177
+ choice = Choice(index=0, delta=delta, finish_reason="stop")
178
+ chunk = ChatCompletionChunk(
179
+ id=request_id,
180
+ choices=[choice],
181
+ created=created_time,
182
+ model=model,
183
+ )
184
+ # Set usage directly on the chunk object
185
+ chunk.usage = {
186
+ "prompt_tokens": prompt_tokens,
187
+ "completion_tokens": completion_tokens,
188
+ "total_tokens": total_tokens,
189
+ "estimated_cost": None
190
+ }
191
+ yield chunk
192
+
193
+ except cloudscraper.exceptions.CloudflareChallengeError as e:
194
+ pass
195
+
196
+ def _create_non_stream(
197
+ self, request_id: str, created_time: int, model: str, payload: Dict[str, Any], timeout: Optional[int] = None, proxies: Optional[Dict[str, str]] = None
198
+ ) -> ChatCompletion:
199
+ full_response_content = ""
200
+ finish_reason = "stop"
201
+ try:
202
+ # Make a non-streaming request to the API
203
+ payload_copy = payload.copy()
204
+ payload_copy["stream"] = False
205
+ response = self._client.session.post(
206
+ self._client.api_endpoint,
207
+ headers=self._client.headers,
208
+ cookies=self._client.cookies,
209
+ json=payload_copy,
210
+ timeout=timeout or self._client.timeout,
211
+ proxies=proxies or getattr(self._client, "proxies", None)
212
+ )
213
+ if not response.ok:
214
+ raise IOError(
215
+ f"YEPCHAT API Error: {response.status_code} {response.reason} - {response.text}"
216
+ )
217
+ data = response.json()
218
+ if 'choices' in data and len(data['choices']) > 0:
219
+ # YEPCHAT non-streaming returns message content in choices[0]['message']['content']
220
+ full_response_content = data['choices'][0].get('message', {}).get('content', '')
221
+ finish_reason = data['choices'][0].get('finish_reason', 'stop')
222
+ else:
223
+ full_response_content = ''
224
+ finish_reason = 'stop'
225
+ except Exception as e:
226
+ print(f"Error obtaining non-stream response from YEPCHAT: {e}")
227
+ finish_reason = "error"
228
+
229
+ message = ChatCompletionMessage(
230
+ role="assistant",
231
+ content=full_response_content
232
+ )
233
+ choice = Choice(
234
+ index=0,
235
+ message=message,
236
+ finish_reason=finish_reason
237
+ )
238
+ # Use count_tokens to compute usage
239
+ prompt_tokens = count_tokens([m.get('content', '') for m in payload.get('messages', [])])
240
+ completion_tokens = count_tokens(full_response_content)
241
+ usage = CompletionUsage(
242
+ prompt_tokens=prompt_tokens,
243
+ completion_tokens=completion_tokens,
244
+ total_tokens=prompt_tokens + completion_tokens
245
+ )
246
+ completion = ChatCompletion(
247
+ id=request_id,
248
+ choices=[choice],
249
+ created=created_time,
250
+ model=model,
251
+ usage=usage,
252
+ )
253
+ return completion
254
+
255
+ class Chat(BaseChat):
256
+ def __init__(self, client: 'YEPCHAT'):
257
+ self.completions = Completions(client)
258
+
259
+ class YEPCHAT(OpenAICompatibleProvider):
260
+ """
261
+ OpenAI-compatible client for YEPCHAT API.
262
+
263
+ Usage:
264
+ client = YEPCHAT()
265
+ response = client.chat.completions.create(
266
+ model="DeepSeek-R1-Distill-Qwen-32B",
267
+ messages=[{"role": "user", "content": "Hello!"}]
268
+ )
269
+ print(response.choices[0].message.content)
270
+ """
271
+ _base_models = ["DeepSeek-R1-Distill-Qwen-32B", "Mixtral-8x7B-Instruct-v0.1"]
272
+
273
+ # Create AVAILABLE_MODELS as a list of base model names (no prefix)
274
+ AVAILABLE_MODELS = _base_models
275
+
276
+ def __init__(
277
+ self,
278
+ browser: str = "chrome"
279
+ ):
280
+ """
281
+ Initialize the YEPCHAT client.
282
+
283
+ Args:
284
+ browser: Browser name for LitAgent to generate User-Agent.
285
+ """
286
+ self.timeout = None
287
+ self.api_endpoint = "https://api.yep.com/v1/chat/completions"
288
+ self.session = cloudscraper.create_scraper() # Use cloudscraper
289
+
290
+ # Initialize LitAgent for user agent generation and fingerprinting
291
+ try:
292
+ agent = LitAgent()
293
+ fingerprint = agent.generate_fingerprint(browser=browser)
294
+ except Exception as e:
295
+ print(f"Warning: Failed to generate fingerprint with LitAgent: {e}. Using fallback.")
296
+ # Fallback fingerprint data
297
+ fingerprint = {
298
+ "accept": "*/*",
299
+ "accept_language": "en-US,en;q=0.9",
300
+ "sec_ch_ua": '"Not/A)Brand";v="99", "Google Chrome";v="127", "Chromium";v="127"',
301
+ "platform": "Windows",
302
+ "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"
303
+ }
304
+
305
+ # Initialize headers using the fingerprint
306
+ self.headers = {
307
+ "Accept": fingerprint["accept"],
308
+ "Accept-Encoding": "gzip, deflate, br, zstd",
309
+ "Accept-Language": fingerprint["accept_language"],
310
+ "Content-Type": "application/json; charset=utf-8",
311
+ "DNT": "1",
312
+ "Origin": "https://yep.com",
313
+ "Referer": "https://yep.com/",
314
+ "Sec-CH-UA": fingerprint["sec_ch_ua"] or '"Not)A;Brand";v="99", "Microsoft Edge";v="127", "Chromium";v="127"',
315
+ "Sec-CH-UA-Mobile": "?0",
316
+ "Sec-CH-UA-Platform": f'"{fingerprint["platform"]}"',
317
+ "User-Agent": fingerprint["user_agent"],
318
+ "x-forwarded-for": fingerprint["x-forwarded-for"],
319
+ "x-real-ip": fingerprint["x-real-ip"],
320
+ "x-client-ip": fingerprint["x-client-ip"],
321
+ "forwarded": fingerprint["forwarded"],
322
+ "x-forwarded-proto": "https",
323
+ "x-request-id": fingerprint["x-request-id"],
324
+ }
325
+ self.session.headers.update(self.headers)
326
+
327
+ # Generate cookies (consider if these need refreshing or specific values)
328
+ self.cookies = {"__Host-session": uuid.uuid4().hex, '__cf_bm': uuid.uuid4().hex}
329
+
330
+ # Initialize the chat interface
331
+ self.chat = Chat(self)
332
+
333
+ @property
334
+ def models(self):
335
+ class _ModelList:
336
+ def list(inner_self):
337
+ return YEPCHAT.AVAILABLE_MODELS
338
+ return _ModelList()
339
+
340
+ def convert_model_name(self, model: str) -> str:
341
+ """
342
+ Ensures the model name is valid for YEPCHAT.
343
+ Returns the validated model name or raises an error if invalid.
344
+ """
345
+ if model in self.AVAILABLE_MODELS:
346
+ return model
347
+ else:
348
+ # Raise error instead of defaulting, as model is mandatory in create()
349
+ raise ValueError(f"Model '{model}' not supported by YEPCHAT. Available: {self.AVAILABLE_MODELS}")
350
+
351
+ # Example usage (optional, for testing)
352
+ if __name__ == '__main__':
353
+ print("Testing YEPCHAT OpenAI-Compatible Client...")
354
+
355
+ # Test Non-Streaming
356
+ try:
357
+ print("\n--- Non-Streaming Test (DeepSeek) ---")
358
+ client = YEPCHAT()
359
+ response = client.chat.completions.create(
360
+ model="DeepSeek-R1-Distill-Qwen-32B",
361
+ messages=[
362
+ {"role": "user", "content": "Say 'Hello World'"}
363
+ ],
364
+ stream=False
365
+ )
366
+ print("Response:", response.choices[0].message.content)
367
+ print("Usage:", response.usage) # Will show 0 tokens
368
+ except Exception as e:
369
+ print(f"Non-Streaming Test Failed: {e}")
370
+
371
+ # Test Streaming
372
+ try:
373
+ print("\n--- Streaming Test (Mixtral) ---")
374
+ client_stream = YEPCHAT()
375
+ stream = client_stream.chat.completions.create(
376
+ model="Mixtral-8x7B-Instruct-v0.1",
377
+ messages=[
378
+ {"role": "user", "content": "Write a short sentence about AI."}
379
+ ],
380
+ stream=True
381
+ )
382
+ print("Streaming Response:")
383
+ for chunk in stream:
384
+ content = chunk.choices[0].delta.content
385
+ if content:
386
+ print(content, end="", flush=True)
387
+ print() # Add a newline at the end
388
+
389
+ except Exception as e:
390
390
  print(f"Streaming Test Failed: {e}")