import openai
import time
import os
from multiprocessing import Pool, cpu_count
from tqdm import tqdm
openai.api_key = "123123"
PROMPT = """请从文章中抽取出所有的航空航天领域科学技术术语，以列表形式给出。\n
输出格式：\n
1. xxx \n
2. xxx \n
"""
def get_response(content, temperature=0.1, max_tokens=2048):
completion = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
temperature=0,
top_p=0,
# max_tokens=max_tokens,
messages=[
{ "role": "system", "content": f"{PROMPT}"},
{ "role": "user", "content": f"{content}" }
]
)
print(completion)
return completion.choices[0]['message']['content']
#
res = get_response("这三种在设计、任务运作方式上都有很大差别。美国航天飞机的核心级发动机安装在轨道器上起到重复使用主发动机降低成本的作用，轨道器本身是航天运载器的核心组成部分，发射时出大力。苏俄的暴风雪航天飞机则基本是完全靠能源超重型运载火箭驮上天，轨道器上没主发动机只有最后用来执行远地点轨道插入的小轨道机动发动机，轨道器发射时不出力最后入轨时小推一把。算重复使用的大型载人货运多功能飞船。欧洲被取消的赫尔墨斯航天飞机完全是放在火箭顶端如同载人飞船一样发射到太空，最终版完全取消了载荷舱专心作为可重复使用的载人飞船角色。甚至对接口、太空行走气闸舱和轨道机动发动机放在可分离的尾部的资源舱里，是一次性的在重返大气层前抛掉。上述文章涵盖的科学术语为：")
print(res)
{
"choices": [
{
"finish_reason": "stop",
"index": 0,
"message": {
"content": "1. \u822a\u5929\u98de\u673a\n2. \u6838\u5fc3\u7ea7\u53d1\u52a8\u673a\n3. \u91cd\u590d\u4f7f\u7528\n4. \u4e3b\u53d1\u52a8\u673a\n5. \u8f68\u9053\u5668\n6. \u822a\u5929\u8fd0\u8f7d\u5668\n7. \u6210\u672c\n8. \u80fd\u6e90\u8d85\u91cd\u578b\u8fd0\u8f7d\u706b\u7bad\n9. \u8fdc\u5730\u70b9\u8f68\u9053\u63d2\u5165\n10. \u5c0f\u8f68\u9053\u673a\u52a8\u53d1\u52a8\u673a\n11. \u91cd\u590d\u4f7f\u7528\u7684\u5927\u578b\u8f7d\u4eba\u8d27\u8fd0\u591a\u529f\u80fd\u98de\u8239\n12. \u8d6b\u5c14\u58a8\u65af\u822a\u5929\u98de\u673a\n13. \u706b\u7bad\u9876\u7aef\n14. \u8f7d\u4eba\u98de\u8239\n15. \u8f7d\u8377\u8231\n16. \u53ef\u91cd\u590d\u4f7f\u7528\n17. \u5bf9\u63a5\u53e3\n18. \u592a\u7a7a\u884c\u8d70\u6c14\u95f8\u8231\n19. \u8f68\u9053\u673a\u52a8\u53d1\u52a8\u673a\n20. \u8d44\u6e90\u8231\n21. \u4e00\u6b21\u6027\n22. \u91cd\u8fd4\u5927\u6c14\u5c42",
"role": "assistant"
}
}
],
"created": 1684299308,
"id": "chatcmpl-7H3ToJi1cc9aAa9c3guINfh9e5zeh",
"model": "gpt-3.5-turbo-0301",
"object": "chat.completion",
"usage": {
"completion_tokens": 234,
"prompt_tokens": 430,
"total_tokens": 664
}
}
1. 航天飞机
2. 核心级发动机
3. 重复使用
4. 主发动机
5. 轨道器
6. 航天运载器
7. 成本
8. 能源超重型运载火箭
9. 远地点轨道插入
10. 小轨道机动发动机
11. 重复使用的大型载人货运多功能飞船
12. 赫尔墨斯航天飞机
13. 火箭顶端
14. 载人飞船
15. 载荷舱
16. 可重复使用
17. 对接口
18. 太空行走气闸舱
19. 轨道机动发动机
20. 资源舱
21. 一次性
22. 重返大气层
