Your Name commited on
Commit
385c775
1 Parent(s): 518385d

支持3.10以下的python版本使用newbing

Browse files
Files changed (2) hide show
  1. config.py +1 -1
  2. request_llm/bridge_newbing.py +17 -70
config.py CHANGED
@@ -65,7 +65,7 @@ API_URL_REDIRECT = {}
65
  CUSTOM_PATH = "/"
66
 
67
  # 如果需要使用newbing,把newbing的长长的cookie放到这里
68
- NEWBING_STYLE = "creative", # ["creative", "balanced", "precise"]
69
  NEWBING_COOKIES = """
70
  your bing cookies here
71
  """
 
65
  CUSTOM_PATH = "/"
66
 
67
  # 如果需要使用newbing,把newbing的长长的cookie放到这里
68
+ NEWBING_STYLE = "creative" # ["creative", "balanced", "precise"]
69
  NEWBING_COOKIES = """
70
  your bing cookies here
71
  """
request_llm/bridge_newbing.py CHANGED
@@ -5,11 +5,7 @@ https://github.com/acheong08/EdgeGPT
5
  ========================================================================
6
  """
7
 
8
- from transformers import AutoModel, AutoTokenizer
9
- import time
10
- import importlib
11
- from toolbox import update_ui, get_conf
12
- from multiprocessing import Process, Pipe
13
  import argparse
14
  import asyncio
15
  import json
@@ -28,13 +24,6 @@ from typing import Union
28
  import certifi
29
  import httpx
30
  import websockets.client as websockets
31
- from prompt_toolkit import PromptSession
32
- from prompt_toolkit.auto_suggest import AutoSuggestFromHistory
33
- from prompt_toolkit.completion import WordCompleter
34
- from prompt_toolkit.history import InMemoryHistory
35
- from prompt_toolkit.key_binding import KeyBindings
36
- from rich.live import Live
37
- from rich.markdown import Markdown
38
 
39
  DELIMITER = "\x1e"
40
 
@@ -146,9 +135,9 @@ class _ChatHubRequest:
146
 
147
  def update(
148
  self,
149
- prompt: str,
150
- conversation_style: CONVERSATION_STYLE_TYPE,
151
- options: list | None = None,
152
  ) -> None:
153
  """
154
  Updates request object
@@ -214,8 +203,8 @@ class _Conversation:
214
 
215
  def __init__(
216
  self,
217
- cookies: dict,
218
- proxy: str | None = None,
219
  ) -> None:
220
  self.struct: dict = {
221
  "conversationId": None,
@@ -271,8 +260,8 @@ class _ChatHub:
271
  Chat API
272
  """
273
 
274
- def __init__(self, conversation: _Conversation) -> None:
275
- self.wss: websockets.WebSocketClientProtocol | None = None
276
  self.request: _ChatHubRequest
277
  self.loop: bool
278
  self.task: asyncio.Task
@@ -351,21 +340,13 @@ class Chatbot:
351
 
352
  def __init__(
353
  self,
354
- cookies: dict = None,
355
- proxy: str | None = None,
356
- cookie_path: str = None,
357
  ) -> None:
358
  if cookies is None:
359
  cookies = {}
360
- if cookie_path is not None:
361
- try:
362
- with open(cookie_path, encoding="utf-8") as f:
363
- self.cookies = json.load(f)
364
- except FileNotFoundError as exc:
365
- raise FileNotFoundError("Cookie file not found") from exc
366
- else:
367
- self.cookies = cookies
368
- self.proxy: str | None = proxy
369
  self.chat_hub: _ChatHub = _ChatHub(
370
  _Conversation(self.cookies, self.proxy),
371
  )
@@ -425,43 +406,6 @@ class Chatbot:
425
  self.chat_hub = _ChatHub(_Conversation(self.cookies))
426
 
427
 
428
- async def _get_input_async(
429
- session: PromptSession = None,
430
- completer: WordCompleter = None,
431
- ) -> str:
432
- """
433
- Multiline input function.
434
- """
435
- return await session.prompt_async(
436
- completer=completer,
437
- multiline=True,
438
- auto_suggest=AutoSuggestFromHistory(),
439
- )
440
-
441
-
442
- def _create_session() -> PromptSession:
443
- kb = KeyBindings()
444
-
445
- @kb.add("enter")
446
- def _(event):
447
- buffer_text = event.current_buffer.text
448
- if buffer_text.startswith("!"):
449
- event.current_buffer.validate_and_handle()
450
- else:
451
- event.current_buffer.insert_text("\n")
452
-
453
- @kb.add("escape")
454
- def _(event):
455
- if event.current_buffer.complete_state:
456
- # event.current_buffer.cancel_completion()
457
- event.current_buffer.text = ""
458
-
459
- return PromptSession(key_bindings=kb, history=InMemoryHistory())
460
-
461
-
462
- def _create_completer(commands: list, pattern_str: str = "$"):
463
- return WordCompleter(words=commands, pattern=re.compile(pattern_str))
464
-
465
 
466
  load_message = ""
467
 
@@ -470,7 +414,10 @@ load_message = ""
470
  第二部分:子进程Worker
471
  ========================================================================
472
  """
473
-
 
 
 
474
  class GetNewBingHandle(Process):
475
  def __init__(self):
476
  super().__init__(daemon=True)
@@ -484,7 +431,6 @@ class GetNewBingHandle(Process):
484
 
485
  def check_dependency(self):
486
  try:
487
- import rich
488
  self.info = "依赖检测通过"
489
  self.success = True
490
  except:
@@ -538,6 +484,7 @@ class GetNewBingHandle(Process):
538
  try:
539
  asyncio.run(self.async_run(question=kwargs['query'], history=kwargs['history']))
540
  except:
 
541
  self.child.send('[Local Message] Newbing失败.')
542
  self.child.send('[Finish]')
543
 
 
5
  ========================================================================
6
  """
7
 
8
+
 
 
 
 
9
  import argparse
10
  import asyncio
11
  import json
 
24
  import certifi
25
  import httpx
26
  import websockets.client as websockets
 
 
 
 
 
 
 
27
 
28
  DELIMITER = "\x1e"
29
 
 
135
 
136
  def update(
137
  self,
138
+ prompt,
139
+ conversation_style,
140
+ options,
141
  ) -> None:
142
  """
143
  Updates request object
 
203
 
204
  def __init__(
205
  self,
206
+ cookies,
207
+ proxy,
208
  ) -> None:
209
  self.struct: dict = {
210
  "conversationId": None,
 
260
  Chat API
261
  """
262
 
263
+ def __init__(self, conversation) -> None:
264
+ self.wss = None
265
  self.request: _ChatHubRequest
266
  self.loop: bool
267
  self.task: asyncio.Task
 
340
 
341
  def __init__(
342
  self,
343
+ cookies,
344
+ proxy
 
345
  ) -> None:
346
  if cookies is None:
347
  cookies = {}
348
+ self.cookies = cookies
349
+ self.proxy = proxy
 
 
 
 
 
 
 
350
  self.chat_hub: _ChatHub = _ChatHub(
351
  _Conversation(self.cookies, self.proxy),
352
  )
 
406
  self.chat_hub = _ChatHub(_Conversation(self.cookies))
407
 
408
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
409
 
410
  load_message = ""
411
 
 
414
  第二部分:子进程Worker
415
  ========================================================================
416
  """
417
+ import time
418
+ import importlib
419
+ from toolbox import update_ui, get_conf, trimmed_format_exc
420
+ from multiprocessing import Process, Pipe
421
  class GetNewBingHandle(Process):
422
  def __init__(self):
423
  super().__init__(daemon=True)
 
431
 
432
  def check_dependency(self):
433
  try:
 
434
  self.info = "依赖检测通过"
435
  self.success = True
436
  except:
 
484
  try:
485
  asyncio.run(self.async_run(question=kwargs['query'], history=kwargs['history']))
486
  except:
487
+ tb_str = '```\n' + trimmed_format_exc() + '```'
488
  self.child.send('[Local Message] Newbing失败.')
489
  self.child.send('[Finish]')
490