import openai
# 设置API密钥(实际研究中应使用环境变量)
# openai.api_key = "your-api-key"
def classify_text(text, prompt_template, model="gpt-4o", temperature=0):
"""
使用OpenAI API对文本进行分类
"""
response = openai.chat.completions.create(
model=model,
messages=[
{"role": "system", "content": "你是一位专业的文本分析助手。"},
{"role": "user", "content": prompt_template.format(text=text)}
],
temperature=temperature,
max_tokens=150
)
return response.choices[0].message.content