From 570593150673558ebf4c9e270d2e921b942fcbfb Mon Sep 17 00:00:00 2001 From: Bagus Indrayana Date: Wed, 12 Jul 2023 01:43:02 +0800 Subject: add provider --- testing/aiservice/AiService.py | 62 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 62 insertions(+) create mode 100644 testing/aiservice/AiService.py diff --git a/testing/aiservice/AiService.py b/testing/aiservice/AiService.py new file mode 100644 index 00000000..287a39ef --- /dev/null +++ b/testing/aiservice/AiService.py @@ -0,0 +1,62 @@ +import os,sys +import requests +# from ...typing import get_type_hints + +url = "https://aiservice.vercel.app/api/chat/answer" +model = ['gpt-3.5-turbo'] +supports_stream = False +needs_auth = False + + +def _create_completion(model: str, messages: list, stream: bool, **kwargs): + base = '' + for message in messages: + base += '%s: %s\n' % (message['role'], message['content']) + base += 'assistant:' + + headers = { + "accept": "*/*", + "content-type": "text/plain;charset=UTF-8", + "sec-fetch-dest": "empty", + "sec-fetch-mode": "cors", + "sec-fetch-site": "same-origin", + "Referer": "https://aiservice.vercel.app/chat", + } + data = { + "input": base + } + response = requests.post(url, headers=headers, json=data) + if response.status_code == 200: + _json = response.json() + yield _json['data'] + else: + print(f"Error Occurred::{response.status_code}") + return None + + + +# 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 633f2312dc6b19c6f8b510c9ca89a8defe1a0f13 Mon Sep 17 00:00:00 2001 From: Bagus Indrayana Date: Wed, 12 Jul 2023 01:44:20 +0800 Subject: add testing --- testing/aiservice/testing.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) create mode 100644 testing/aiservice/testing.py diff --git a/testing/aiservice/testing.py b/testing/aiservice/testing.py new file mode 100644 index 00000000..5cb6c5ef --- /dev/null +++ b/testing/aiservice/testing.py @@ -0,0 +1,30 @@ +from AiService import ChatCompletion + +# Test 1 +response = ChatCompletion.create(model="gpt-3.5-turbo", + provider="AiService", + stream=False, + messages=[{'role': 'user', 'content': 'who are you?'}]) + +print(response) + +# Test 2 +response = ChatCompletion.create(model="gpt-3.5-turbo", + provider="AiService", + stream=False, + messages=[{'role': 'user', 'content': 'what you can do?'}]) + +print(response) + + +# Test 3 +response = ChatCompletion.create(model="gpt-3.5-turbo", + provider="AiService", + stream=False, + messages=[ + {'role': 'user', 'content': 'now your name is Bob'}, + {'role': 'assistant', 'content': 'Hello Im Bob, you asistant'}, + {'role': 'user', 'content': 'what your name again?'}, + ]) + +print(response) \ No newline at end of file -- cgit v1.2.3 From c4915ba9c147b0c33330aa555219451d198e7747 Mon Sep 17 00:00:00 2001 From: Bagus Indrayana Date: Wed, 12 Jul 2023 01:45:14 +0800 Subject: add readme --- testing/aiservice/README.md | 2 ++ 1 file changed, 2 insertions(+) create mode 100644 testing/aiservice/README.md diff --git a/testing/aiservice/README.md b/testing/aiservice/README.md new file mode 100644 index 00000000..83b06481 --- /dev/null +++ b/testing/aiservice/README.md @@ -0,0 +1,2 @@ +https://github.com/xtekky/gpt4free/issues/40#issuecomment-1629152431 +probably gpt-3.5 \ No newline at end of file -- cgit v1.2.3