Spaces:
Runtime error
Runtime error
| 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) |