key_words.py

import json
import requests
import sys

def generate_key_words(key_word_model, api_key, url, outline):
    messages = [
        {"role": "system", "content": "你现在正在生成一篇论文,我将提供给你一份大纲,为了论文的准确真实性,你需要根据论文要求和大纲进行相关数据以及资料的搜索,现在,请你列出你需要进行搜索的关键句。由我来代替你上网进行搜索。注意:你的回复应当有且仅有你列出的搜索关键句,多个关键句之间用英文逗号隔开,不得输出其他任何解释。示例输出:关键句1,关键句2,关键句3"},
        {"role": "user", "content": f"大纲为:{outline}"}
    ]

    payload = json.dumps({
        "model": key_word_model,
        "messages": messages
    })
    headers = {
        'Accept': 'application/json',
        'Authorization': api_key,
        'Content-Type': 'application/json'
    }

    response = requests.post(url, headers=headers, data=payload)
    response_data = response.json()

    if 'choices' in response_data:
        search_key_word = response_data['choices'][0]['message']['content']
        print(search_key_word)
    else:
        print("Error: Unable to generate search keywords.")

if __name__ == "__main__":
    key_word_model = sys.argv[1]
    api_key = sys.argv[2]
    url = sys.argv[3]
    outline = sys.argv[4]

    generate_key_words(key_word_model, api_key, url, outline)