在本文中，我们将深入研究使用 Node.js、Express 和 OpenAI API 创建专业级股票报告生成器。我们的重点是编写高质量、可维护的代码，同时保留 OpenAI API 交互中使用的提示消息的完整性。该应用程序将获取股票数据，执行情绪和行业分析，并生成全面的投资报告。
我们的目标是构建一个 API 端点，为给定的股票代码生成详细的投资报告。该报告将包括：
我们将从外部API获取股票数据，并使用OpenAI API进行高级分析，确保提示信息被准确保留。
创建一个新目录并初始化 Node.js 项目：
mkdir stock-report-generator cd stock-report-generator npm init -y
安装必要的依赖项：
npm install express axios
设置项目结构：
mkdir routes utils data touch app.js routes/report.js utils/helpers.js
// app.js
const express = require('express');
const reportRouter = require('./routes/report');
const app = express();
app.use(express.json());
app.use('/api', reportRouter);
const PORT = process.env.PORT || 3000;
app.listen(PORT, () => {
console.log(`Server is running on port ${PORT}`);
});
在 utils/helpers.js 中，我们将定义用于数据获取和处理的实用函数。
mkdir stock-report-generator cd stock-report-generator npm init -y
在routes/report.js中，定义获取股票数据的函数。
npm install express axios
mkdir routes utils data touch app.js routes/report.js utils/helpers.js
// app.js
const express = require('express');
const reportRouter = require('./routes/report');
const app = express();
app.use(express.json());
app.use('/api', reportRouter);
const PORT = process.env.PORT || 3000;
app.listen(PORT, () => {
console.log(`Server is running on port ${PORT}`);
});
// utils/helpers.js
const axios = require('axios');
const fs = require('fs');
const path = require('path');
const BASE_URL = 'https://your-data-api.com'; // Replace with your actual data API
/**
* Get the start and end dates for the last year.
* @returns {object} - An object containing `start` and `end` dates.
*/
function getLastYearDates() {
const now = new Date();
const end = now.toISOString().split('T')[0];
now.setFullYear(now.getFullYear() - 1);
const start = now.toISOString().split('T')[0];
return { start, end };
}
/**
* Convert an object to a string, excluding specified keys.
* @param {object} obj - The object to convert.
* @param {string[]} excludeKeys - Keys to exclude.
* @returns {string} - The resulting string.
*/
function objectToString(obj, excludeKeys = []) {
return Object.entries(obj)
.filter(([key]) => !excludeKeys.includes(key))
.map(([key, value]) => `${key}: ${value}`)
.join('\n');
}
/**
* Fetch data from a specified endpoint with given parameters.
* @param {string} endpoint - API endpoint.
* @param {object} params - Query parameters.
* @param {any} defaultValue - Default value if the request fails.
* @returns {Promise<any>} - The fetched data or default value.
*/
async function fetchData(endpoint, params = {}, defaultValue = null) {
try {
const response = await axios.get(`${BASE_URL}${endpoint}`, { params });
return response.data || defaultValue;
} catch (error) {
console.error(`Error fetching data from ${endpoint}:`, error.message);
return defaultValue;
}
}
/**
* Read data from a local JSON file.
* @param {string} fileName - Name of the JSON file.
* @returns {any} - The parsed data.
*/
function readLocalJson(fileName) {
const filePath = path.join(__dirname, '../data', fileName);
const data = fs.readFileSync(filePath, 'utf-8');
return JSON.parse(data);
}
module.exports = {
fetchData,
objectToString,
getLastYearDates,
readLocalJson,
};
// routes/report.js
const express = require('express');
const {
fetchData,
objectToString,
getLastYearDates,
readLocalJson,
} = require('../utils/helpers');
const router = express.Router();
/**
* Fetches stock data including historical prices, financials, MDA, and main business info.
* @param {string} ticker - Stock ticker symbol.
* @returns {Promise<object>} - An object containing all fetched data.
*/
async function fetchStockData(ticker) {
try {
const dates = getLastYearDates();
const [historicalData, financialData, mdaData, businessData] = await Promise.all([
fetchData('/stock_zh_a_hist', {
symbol: ticker,
period: 'weekly',
start_date: dates.start,
end_date: dates.end,
}, []),
fetchData('/stock_financial_benefit_ths', {
code: ticker,
indicator: '按年度',
}, [{}]),
fetchData('/stock_mda', { code: ticker }, []),
fetchData('/stock_main_business', { code: ticker }, []),
]);
const hist = historicalData[historicalData.length - 1];
const currentPrice = (hist ? hist['开盘'] : 'N/A') + ' CNY';
const historical = historicalData
.map((item) => objectToString(item, ['股票代码']))
.join('\n----------\n');
const zsfzJson = readLocalJson('zcfz.json');
const balanceSheet = objectToString(zsfzJson.find((item) => item['股票代码'] === ticker));
const financial = objectToString(financialData[0]);
const mda = mdaData.map(item => `${item['报告期']}\n${item['内容']}`).join('\n-----------\n');
const mainBusiness = businessData.map(item =>
`主营业务: ${item['主营业务']}\n产品名称: ${item['产品名称']}\n产品类型: ${item['产品类型']}\n经营范围: ${item['经营范围']}`
).join('\n-----------\n');
return { currentPrice, historical, balanceSheet, mda, mainBusiness, financial };
} catch (error) {
console.error('Error fetching stock data:', error.message);
throw error;
}
}
在routes/report.js中添加路由处理程序：
const axios = require('axios');
const OPENAI_API_KEY = 'your-openai-api-key'; // Replace with your OpenAI API key
/**
* Interacts with the OpenAI API to get completion results.
* @param {array} messages - Array of messages, including system prompts and user messages.
* @returns {Promise<string>} - The AI's response.
*/
async function analyzeWithOpenAI(messages) {
try {
const headers = {
'Authorization': `Bearer ${OPENAI_API_KEY}`,
'Content-Type': 'application/json',
};
const requestData = {
model: 'gpt-4',
temperature: 0.3,
messages: messages,
};
const response = await axios.post(
'https://api.openai.com/v1/chat/completions',
requestData,
{ headers }
);
return response.data.choices[0].message.content.trim();
} catch (error) {
console.error('Error fetching analysis from OpenAI:', error.message);
throw error;
}
}
确保你的app.js和routes/report.js设置正确，然后启动服务器：
/**
* Performs sentiment analysis on news articles using the OpenAI API.
* @param {string} ticker - Stock ticker symbol.
* @returns {Promise<string>} - Sentiment analysis summary.
*/
async function performSentimentAnalysis(ticker) {
const systemPrompt = `You are a sentiment analysis assistant. Analyze the sentiment of the given news articles for ${ticker} and provide a summary of the overall sentiment and any notable changes over time. Be measured and discerning. You are a skeptical investor.`;
const tickerNewsResponse = await fetchData('/stock_news_specific', { code: ticker }, []);
const newsText = tickerNewsResponse
.map(item => `${item['文章来源']} Date: ${item['发布时间']}\n${item['新闻内容']}`)
.join('\n----------\n');
const messages = [
{ role: 'system', content: systemPrompt },
{
role: 'user',
content: `News articles for ${ticker}:\n${newsText || 'N/A'}\n----\nProvide a summary of the overall sentiment and any notable changes over time.`,
},
];
return await analyzeWithOpenAI(messages);
}
使用curl或Postman发送POST请求：
mkdir stock-report-generator cd stock-report-generator npm init -y
我们构建了一个高质量的库存报告生成器，具有以下功能：
在整个开发过程中，我们专注于编写专业、可维护的代码，并提供了详细的解释和注释。
免责声明：此应用程序仅用于教育目的。确保遵守所有 API 服务条款并适当处理敏感数据。
以上是使用 Node.js、Express 和 OpenAI API 构建高质量股票报告生成器的详细内容。更多信息请关注PHP中文网其他相关文章！
