当前位置：首页 > news >正文

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

news 2025/5/24 6:29:12

基于 OpenAI API 的博客评论生成器

在当今数字化时代，博客等社交媒体平台的影响力日益增长。为了帮助用户快速生成高质量的评论内容，本文将介绍如何利用 OpenAI API 构建一个简易的AI评论生成器。

背景知识

OpenAI 提供了强大的语言模型 API，能够根据给定的提示生成自然、连贯的文本。本文将展示如何使用 OpenAI API 来生成主题相关的评论。

实现代码

以下是使用 Python 构建AI评论生成器的核心代码：

import json
import logging
from typing import Optionalimport requests
from requests.exceptions import RequestExceptionclass OpenAIHelper:def __init__(self, api_url: str, model_name: str, api_key: str):self.api_url = api_urlself.model_name = model_nameself.api_key = api_keyself.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 Nonerequest_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 Noneelse:logging.error("请求失败，状态码: %d，响应内容: %s", response.status_code, response.text)return Noneexcept RequestException as e:logging.error("请求发生错误: %s", e)return Noneif __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("生成评论失败")

代码解析

初始化 ：OpenAIHelper 类在初始化时接收 API 地址、模型名称和 API 密钥，并设置请求头。
生成评论 ：generate_comment 方法根据给定的主题生成评论。它构造请求体，发送 POST 请求到 OpenAI API，并处理响应以提取评论内容。
使用示例 ：在主函数中，配置 API 信息和视频主题，创建 OpenAIHelper 实例并调用 generate_comment 方法。