diff options
author | Civitasv <hscivitasv@gmail.com> | 2023-04-30 07:19:46 +0200 |
---|---|---|
committer | Civitasv <hscivitasv@gmail.com> | 2023-04-30 07:19:46 +0200 |
commit | 70acea30a58c55288c1d31c630801e9498b24114 (patch) | |
tree | deeff36849acce2ea0bffba5c1d319adc396ce3a /gpt4free/forefront/__init__.py | |
parent | Merge branch 'main' of https://github.com/xtekky/gpt4free (diff) | |
download | gpt4free-70acea30a58c55288c1d31c630801e9498b24114.tar gpt4free-70acea30a58c55288c1d31c630801e9498b24114.tar.gz gpt4free-70acea30a58c55288c1d31c630801e9498b24114.tar.bz2 gpt4free-70acea30a58c55288c1d31c630801e9498b24114.tar.lz gpt4free-70acea30a58c55288c1d31c630801e9498b24114.tar.xz gpt4free-70acea30a58c55288c1d31c630801e9498b24114.tar.zst gpt4free-70acea30a58c55288c1d31c630801e9498b24114.zip |
Diffstat (limited to 'gpt4free/forefront/__init__.py')
-rw-r--r-- | gpt4free/forefront/__init__.py | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/gpt4free/forefront/__init__.py b/gpt4free/forefront/__init__.py index f0ca1a15..aa78cfa7 100644 --- a/gpt4free/forefront/__init__.py +++ b/gpt4free/forefront/__init__.py @@ -98,12 +98,15 @@ class StreamingCompletion: action_type='new', default_persona='607e41fe-95be-497e-8e97-010a59b2e2c0', # default model='gpt-4', + proxy=None ) -> Generator[ForeFrontResponse, None, None]: if not token: raise Exception('Token is required!') if not chat_id: chat_id = str(uuid4()) + proxies = { 'http': 'http://' + proxy, 'https': 'http://' + proxy } if proxy else None + headers = { 'authority': 'chat-server.tenant-forefront-default.knative.chi.coreweave.com', 'accept': '*/*', @@ -135,6 +138,7 @@ class StreamingCompletion: for chunk in post( 'https://chat-server.tenant-forefront-default.knative.chi.coreweave.com/chat', headers=headers, + proxies=proxies, json=json_data, stream=True, ).iter_lines(): @@ -169,6 +173,7 @@ class Completion: action_type='new', default_persona='607e41fe-95be-497e-8e97-010a59b2e2c0', # default model='gpt-4', + proxy=None ) -> ForeFrontResponse: text = '' final_response = None @@ -179,6 +184,7 @@ class Completion: action_type=action_type, default_persona=default_persona, model=model, + proxy=proxy ): if response: final_response = response |