火车头采集器python CHATGPT/AI openai改写创作/批量原创插件代码使用教程！

原创

于 2024-01-09 10:57:17 发布 · 1k 阅读

8 ·

CC 4.0 BY-SA版权

文章标签：

#python #chatgpt #人工智能

# -*- coding: utf-8 -*-
import requests
from urllib import parse
import json
import sys

def chat_with_api(api_key, model, messages, temperature=1, top_p=1, n=1, stop=None):
    base_url = "https://api地址"  # 请替换为你实际的 OpenAI API URL

    headers = {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"}

    data = {
        "model": model,
        "messages": messages,
        "temperature": temperature,
        "top_p": top_p,
        "n": n,
    }

    if stop is not None:
        data["stop"] = stop

    response = requests.post(base_url, headers=headers, data=json.dumps(data))

    if response.status_code == 200:
        return response.json()
    else:
        raise Exception(f"Error {response.status_code}: {response.text}")

def main():
    if len(sys.argv) != 5:
        print("命令行参数长度不为5")
        sys.exit()

    LabelCookie = parse.unquote(sys.argv[1])
    LabelUrl = parse.unquote(sys.argv[2])
    PageType = sys.argv[3]
    S