Python 实现基于 OpenAI API 的文章标题自动生成评论

import json
import logging
from typing import Optional

import requests
from requests.exceptions import RequestException

class OpenAIHelper:
    def __init__(self, api_url: str, model_name: str, api_key: str):
        self.api_url = api_url
        self.model_name = model_name
        self.api_key = api_key
        self.headers = {
            "Content-Type": "application/json",
            "Authorization": f"Bearer {api_key}"
        }

    def generate_comment(self, title: str) -> Optional[str]:
        if not self.api_key or not title:
            logging.error("API-KEY或标题为空")
            return None

        request_body = {
            "model": self.model_name,
            "temperature": 0.5,
            "max_tokens": 50,
            "messages": [
                {
                    "role": "user",
                    "content": f"生成一条积极的评论，主题是: {title}。评论需要自然、互动性强，使用简单语言，不包含任何特殊字符或表情符号。"
                }
            ]
        }

        try:
            response = requests.post(
                url=self.api_url,
                headers=self.headers,
                data=json.dumps(request_body)
            )

            if response.status_code == 200:
                response_data = response.json()
                if "choices" in response_data and len(response_data["choices"]) > 0 and "message" in response_data["choices"][0]:
                    return response_data["choices"][0]["message"]["content"]
                else:
                    logging.error("响应格式不正确: %s", response_data)
                    return None
            else:
                logging.error("请求失败，状态码: %d，响应内容: %s", response.status_code, response.text)
                return None

        except RequestException as e:
            logging.error("请求发生错误: %s", e)
            return None

if __name__ == "__main__":
    api_url = "https://api.siliconflow.cn/v1/chat/completions  "
    model_name = "deepseek-ai/DeepSeek-R1-Distill-Qwen-7B"
    api_key = "your-api-key"
    article_title = "your-aiticle-title"

    open_ai_helper = OpenAIHelper(
        api_url=api_url,
        model_name=model_name,
        api_key=api_key
    )

    comment = open_ai_helper.generate_comment(article_title)
    if comment:
        print(f"生成的评论: {comment}")
    else:
        print("生成评论失败")