summaryrefslogtreecommitdiffstats
path: root/g4f/api/__init__.py
diff options
context:
space:
mode:
authorcatmeowjiao <meowjiao321@163.com>2023-10-27 11:33:47 +0200
committercatmeowjiao <meowjiao321@163.com>2023-10-27 11:33:47 +0200
commit325f8558da81fd7e6dacc0ed8e50859e285f116c (patch)
treeb8e030a40f7d4afa9cd5c472d57b23ed2ed54b87 /g4f/api/__init__.py
parentadd proxy support (diff)
downloadgpt4free-325f8558da81fd7e6dacc0ed8e50859e285f116c.tar
gpt4free-325f8558da81fd7e6dacc0ed8e50859e285f116c.tar.gz
gpt4free-325f8558da81fd7e6dacc0ed8e50859e285f116c.tar.bz2
gpt4free-325f8558da81fd7e6dacc0ed8e50859e285f116c.tar.lz
gpt4free-325f8558da81fd7e6dacc0ed8e50859e285f116c.tar.xz
gpt4free-325f8558da81fd7e6dacc0ed8e50859e285f116c.tar.zst
gpt4free-325f8558da81fd7e6dacc0ed8e50859e285f116c.zip
Diffstat (limited to '')
-rw-r--r--g4f/api/__init__.py22
1 files changed, 19 insertions, 3 deletions
diff --git a/g4f/api/__init__.py b/g4f/api/__init__.py
index b78768bd..fec5606f 100644
--- a/g4f/api/__init__.py
+++ b/g4f/api/__init__.py
@@ -105,9 +105,25 @@ class Api:
logger.info(f'model: {model}, stream: {stream}, request: {messages[-1]["content"]}')
- response = self.engine.ChatCompletion.create(model=model,
- stream=stream, messages=messages,
- ignored=self.list_ignored_providers)
+ config = None
+ proxy = None
+
+ try:
+ config = json.load(open("config.json","r",encoding="utf-8"))
+ proxy = config["proxy"]
+
+ except Exception:
+ pass
+
+ if proxy != None:
+ response = self.engine.ChatCompletion.create(model=model,
+ stream=stream, messages=messages,
+ ignored=self.list_ignored_providers,
+ proxy=proxy)
+ else:
+ response = self.engine.ChatCompletion.create(model=model,
+ stream=stream, messages=messages,
+ ignored=self.list_ignored_providers)
completion_id = ''.join(random.choices(string.ascii_letters + string.digits, k=28))
completion_timestamp = int(time.time())