diff options
author | xtekky <98614666+xtekky@users.noreply.github.com> | 2023-07-16 19:50:00 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-07-16 19:50:00 +0200 |
commit | 821c8dcd470456143deaa34ff4edb1ae05f0f147 (patch) | |
tree | b5e6ff34b32d6c29ea6890f06d903daebf94113c /g4f/Provider/Providers/AiService.py | |
parent | Merge pull request #742 from dikos1337/fix-anchors (diff) | |
parent | refactor/move provider from testing folder (diff) | |
download | gpt4free-821c8dcd470456143deaa34ff4edb1ae05f0f147.tar gpt4free-821c8dcd470456143deaa34ff4edb1ae05f0f147.tar.gz gpt4free-821c8dcd470456143deaa34ff4edb1ae05f0f147.tar.bz2 gpt4free-821c8dcd470456143deaa34ff4edb1ae05f0f147.tar.lz gpt4free-821c8dcd470456143deaa34ff4edb1ae05f0f147.tar.xz gpt4free-821c8dcd470456143deaa34ff4edb1ae05f0f147.tar.zst gpt4free-821c8dcd470456143deaa34ff4edb1ae05f0f147.zip |
Diffstat (limited to '')
-rw-r--r-- | g4f/Provider/Providers/AiService.py | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/g4f/Provider/Providers/AiService.py b/g4f/Provider/Providers/AiService.py new file mode 100644 index 00000000..8d475118 --- /dev/null +++ b/g4f/Provider/Providers/AiService.py @@ -0,0 +1,40 @@ +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]])
\ No newline at end of file |