liqian 2 years ago
parent
commit
9242263573
2 changed files with 5 additions and 2 deletions
  1. 1 0
      config.py
  2. 4 2
      gpt_process.py

+ 1 - 0
config.py

@@ -16,6 +16,7 @@ class BaseConfig(object):
     # gpt配置
     GPT_HOST = 'https://api.openai.com/v1/chat/completions'
     GPT_OPENAI_API_KEY = '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'
+    GPT_URL = 'http://aigc.piaoquantv.com/aigc-server/aigc/content'
     # 代理地址
     PROXIES = {
         'http': 'http://127.0.0.1:4780',

+ 4 - 2
gpt_process.py

@@ -12,6 +12,7 @@ log_ = Log()
 
 
 def request_gpt(prompt):
+    """
     headers = {
         'Content-Type': 'application/json',
         'Authorization': f'Bearer {config_.GPT_OPENAI_API_KEY}',
@@ -23,12 +24,13 @@ def request_gpt(prompt):
         'messages': [
             {
                 'role': 'user',
-                'content': f'{prompt}',
+                'content': prompt,
             },
         ],
     }
-    # raise ConnectionResetError
     response = requests.post(url=config_.GPT_HOST, headers=headers, json=json_data, proxies=proxies)
+    """
+    response = requests.post(url=config_.GPT_URL, json={'content': prompt})
     # print(response.json())
     # print(response.json()['choices'][0]['message']['content'])
     # print('\n')