From 25870e7523e4427b084aa82d5e08cfdc41807da1 Mon Sep 17 00:00:00 2001 From: Bagus Indrayana Date: Tue, 11 Jul 2023 22:40:29 +0800 Subject: add provider and helper --- testing/binghuan/BingHuan.py | 49 ++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 49 insertions(+) create mode 100644 testing/binghuan/BingHuan.py (limited to 'testing/binghuan/BingHuan.py') diff --git a/testing/binghuan/BingHuan.py b/testing/binghuan/BingHuan.py new file mode 100644 index 00000000..c4a65309 --- /dev/null +++ b/testing/binghuan/BingHuan.py @@ -0,0 +1,49 @@ +import os,sys +import json +import subprocess +# from ...typing import sha256, Dict, get_type_hints + +url = 'https://b.ai-huan.xyz' +model = ['gpt-3.5-turbo', 'gpt-4'] +supports_stream = True +needs_auth = False + +def _create_completion(model: str, messages: list, stream: bool, **kwargs): + path = os.path.dirname(os.path.realpath(__file__)) + config = json.dumps({ + 'messages': messages, + 'model': model}, separators=(',', ':')) + + cmd = ['python', f'{path}/helpers/binghuan.py', config] + + p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) + + for line in iter(p.stdout.readline, b''): + yield line.decode('cp1252') #[:-1] + + +# params = f'g4f.Providers.{os.path.basename(__file__)[:-3]} supports: ' + \ +# '(%s)' % ', '.join( +# [f"{name}: {get_type_hints(_create_completion)[name].__name__}" for name in _create_completion.__code__.co_varnames[:_create_completion.__code__.co_argcount]]) + + +# Temporary For ChatCompletion Class +class ChatCompletion: + @staticmethod + def create(model: str, messages: list, provider: None or str, stream: bool = False, auth: str = False, **kwargs): + kwargs['auth'] = auth + + if provider and needs_auth and not auth: + print( + f'ValueError: {provider} requires authentication (use auth="cookie or token or jwt ..." param)', file=sys.stderr) + sys.exit(1) + + try: + return (_create_completion(model, messages, stream, **kwargs) + if stream else ''.join(_create_completion(model, messages, stream, **kwargs))) + except TypeError as e: + print(e) + arg: str = str(e).split("'")[1] + print( + f"ValueError: {provider} does not support '{arg}' argument", file=sys.stderr) + sys.exit(1) \ No newline at end of file -- cgit v1.2.3 From 659e2f4ff08b46bdcc2420b705eac47da683a3a4 Mon Sep 17 00:00:00 2001 From: Bagus Indrayana Date: Wed, 12 Jul 2023 01:13:24 +0800 Subject: emojin encoding problem idk how to fix it --- testing/binghuan/BingHuan.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'testing/binghuan/BingHuan.py') diff --git a/testing/binghuan/BingHuan.py b/testing/binghuan/BingHuan.py index c4a65309..ce36b1f2 100644 --- a/testing/binghuan/BingHuan.py +++ b/testing/binghuan/BingHuan.py @@ -13,13 +13,13 @@ def _create_completion(model: str, messages: list, stream: bool, **kwargs): config = json.dumps({ 'messages': messages, 'model': model}, separators=(',', ':')) - cmd = ['python', f'{path}/helpers/binghuan.py', config] p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) - + for line in iter(p.stdout.readline, b''): - yield line.decode('cp1252') #[:-1] + yield line.decode('utf-8') + # params = f'g4f.Providers.{os.path.basename(__file__)[:-3]} supports: ' + \ -- cgit v1.2.3 From 23364b479af936f156b14550c918bc7d031c127d Mon Sep 17 00:00:00 2001 From: Bagus Indrayana Date: Wed, 12 Jul 2023 01:19:32 +0800 Subject: update encoding --- testing/binghuan/BingHuan.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'testing/binghuan/BingHuan.py') diff --git a/testing/binghuan/BingHuan.py b/testing/binghuan/BingHuan.py index ce36b1f2..8c859c08 100644 --- a/testing/binghuan/BingHuan.py +++ b/testing/binghuan/BingHuan.py @@ -18,7 +18,7 @@ def _create_completion(model: str, messages: list, stream: bool, **kwargs): p = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.STDOUT) for line in iter(p.stdout.readline, b''): - yield line.decode('utf-8') + yield line.decode('cp1252') -- cgit v1.2.3