From 3d269ce9d1dee9fc578f328246cea55f70e6d6de Mon Sep 17 00:00:00 2001
From: dmy <dmy@dmy.com>
Date: Mon, 12 Jan 2026 18:59:17 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E9=87=8D=E6=9E=84AI=E6=8E=A8=E8=8D=90?=
 =?UTF-8?q?=E5=8A=9F=E8=83=BD=E5=B9=B6=E4=BC=98=E5=8C=96=E7=88=AC=E8=99=AB?=
 =?UTF-8?q?=E5=9F=BA=E7=A1=80URL?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

重构前端AI推荐组件，移除本地过滤逻辑，改为从后端获取日期范围内的数据
新增AI服务模块，包含Prompt和推荐逻辑
为投标服务添加按日期范围查询接口
统一各爬虫服务的baseURL格式
---
 frontend/src/components/Dashboard-AI.vue    | 52 ++++---------
 src/ai/Prompt.ts                            |  1 +
 src/ai/ai.controller.ts                     |  3 -
 src/ai/ai.service.ts                        | 83 +++++++++++++++++++++
 src/bids/controllers/bid.controller.ts      |  5 ++
 src/bids/services/bid.service.ts            | 18 +++++
 src/crawler/services/cdt_target.ts          |  2 +-
 src/crawler/services/ceic_target.ts         |  2 +-
 src/crawler/services/cgnpc_target.ts        |  2 +-
 src/crawler/services/cnncecp_target.ts      |  2 +-
 src/crawler/services/cnooc_target.ts        |  2 +-
 src/crawler/services/eps_target.ts          |  2 +-
 src/crawler/services/espic_target.ts        |  2 +-
 src/crawler/services/powerbeijing_target.ts |  2 +-
 src/crawler/services/sdicc_target.ts        |  2 +-
 src/crawler/services/szecp_target.ts        |  2 +-
 16 files changed, 131 insertions(+), 51 deletions(-)
 create mode 100644 src/ai/Prompt.ts
 create mode 100644 src/ai/ai.service.ts

diff --git a/frontend/src/components/Dashboard-AI.vue b/frontend/src/components/Dashboard-AI.vue
index b47bf30..a3fa65b 100644
--- a/frontend/src/components/Dashboard-AI.vue
+++ b/frontend/src/components/Dashboard-AI.vue
@@ -106,7 +106,7 @@
 </template>
 
 <script setup lang="ts">
-import { ref, computed, watch } from 'vue'
+import { ref } from 'vue'
 import axios from 'axios'
 import { ElMessage } from 'element-plus'
 import { MagicStick, Loading, InfoFilled, List } from '@element-plus/icons-vue'
@@ -132,36 +132,6 @@ const showAllBids = ref(false)
 const bidsLoading = ref(false)
 const bidsByDateRange = ref<any[]>([])
 
-// 根据日期范围过滤 bids
-const filteredBids = computed(() => {
-  let result = props.bids
-
-  // 按日期范围筛选（只限制开始时间，不限制结束时间）
-  if (dateRange.value && dateRange.value.length === 2) {
-    const [startDate] = dateRange.value
-    result = result.filter(bid => {
-      if (!bid.publishDate) return false
-      const bidDate = new Date(bid.publishDate)
-      const start = new Date(startDate)
-      // 设置时间为当天的开始
-      start.setHours(0, 0, 0, 0)
-      return bidDate >= start
-    })
-  }
-
-  return result
-})
-
-// 监听日期范围变化并显示提示
-watch(dateRange, () => {
-  const totalBids = props.bids.length
-  const filteredCount = filteredBids.value.length
-
-  if (totalBids > 0 && filteredCount < totalBids) {
-    ElMessage.info(`筛选结果：共 ${filteredCount} 条数据（总共 ${totalBids} 条）`)
-  }
-})
-
 // 设置日期范围为最近3天
 const setLast3Days = () => {
   const endDate = new Date()
@@ -200,12 +170,9 @@ const setLast7Days = () => {
 const fetchAIRecommendations = async () => {
   loading.value = true
   try {
-    // 准备发送给后端的数据（使用过滤后的 bids）
-    const bidsData = filteredBids.value.map(bid => ({
-      title: bid.title,
-      url: bid.url,
-      source: bid.source,
-      publishDate: bid.publishDate
+    // 准备发送给后端的数据（只包含 title）
+    const bidsData = bidsByDateRange.value.map(bid => ({
+      title: bid.title
     }))
 
     // 调用后端 API
@@ -213,7 +180,16 @@ const fetchAIRecommendations = async () => {
       bids: bidsData
     })
 
-    aiRecommendations.value = response.data
+    // 根据 title 从 bidsByDateRange 中更新 url 和 source
+    aiRecommendations.value = response.data.map((rec: any) => {
+      const bid = bidsByDateRange.value.find(b => b.title === rec.title)
+      return {
+        title: rec.title,
+        url: bid?.url || '',
+        source: bid?.source || '',
+        confidence: rec.confidence
+      }
+    })
 
     ElMessage.success('AI 推荐获取成功')
   } catch (error: any) {
diff --git a/src/ai/Prompt.ts b/src/ai/Prompt.ts
new file mode 100644
index 0000000..b64ea12
--- /dev/null
+++ b/src/ai/Prompt.ts
@@ -0,0 +1 @@
+export const PromptString: string = `先给我说你统计了多少个项目。我只对辽宁、山东、江苏、浙江、福建、广东、广西、河北这些地方的海上风电、海上光伏、漂浮式光伏、滩涂光伏、滩涂风电、渔光互补项目感兴趣。从我提供的这些工程里面找到我感兴趣的工程。如果没有推荐的，也要给出思考过程。`;
diff --git a/src/ai/ai.controller.ts b/src/ai/ai.controller.ts
index df38c61..5810d69 100644
--- a/src/ai/ai.controller.ts
+++ b/src/ai/ai.controller.ts
@@ -3,9 +3,6 @@ import { AiService } from './ai.service';
 
 export class BidDataDto {
   title: string;
-  url: string;
-  source: string;
-  publishDate: string;
 }
 
 export class BidsRequestDto {
diff --git a/src/ai/ai.service.ts b/src/ai/ai.service.ts
new file mode 100644
index 0000000..8e8d564
--- /dev/null
+++ b/src/ai/ai.service.ts
@@ -0,0 +1,83 @@
+import { Injectable, Logger } from '@nestjs/common';
+import { ConfigService } from '@nestjs/config';
+import OpenAI from 'openai';
+import { PromptString } from './Prompt';
+
+export interface BidDataDto {
+  title: string;
+}
+
+export interface AIRecommendation {
+  title: string;
+  url: string;
+  source: string;
+  confidence: number;
+}
+
+@Injectable()
+export class AiService {
+  private readonly logger = new Logger(AiService.name);
+  private openai: OpenAI;
+
+  constructor(private readonly configService: ConfigService) {
+    const apiKey = this.configService.get<string>('ARK_API_KEY');
+    this.openai = new OpenAI({
+      apiKey: apiKey || '',
+      baseURL: 'https://ark.cn-beijing.volces.com/api/v3',
+      timeout: 120000, // 120秒超时
+    });
+  }
+
+  async getRecommendations(bids: BidDataDto[]): Promise<AIRecommendation[]> {
+    this.logger.log('开始获取 AI 推荐');
+    this.logger.log(`发送给 AI 的数据数量: ${bids.length}`);
+
+    try {
+      const prompt =PromptString+ `请根据以下投标项目标题列表，推荐最值得关注的 5-10 个项目。请以 JSON 格式返回，格式如下：
+[
+  {
+    "title": "项目标题",
+    "confidence": 推荐度(0-100的数字)
+  }
+]
+
+投标项目标题列表：
+${JSON.stringify(bids.map(b => b.title), null, 2)}`;
+      this.logger.log('发给AI的内容',prompt);
+      const completion = await this.openai.chat.completions.create({
+        model: 'doubao-seed-1-6-lite-251015',
+        max_tokens: 32768,
+        reasoning_effort: 'medium',
+        messages: [
+          {
+            role: 'user',
+            content: prompt,
+          },
+        ],
+      });
+
+      this.logger.log('AI API 响应成功');
+
+      const aiContent = completion.choices[0].message.content;
+      this.logger.log('AI 返回的内容:', aiContent);
+
+      if (!aiContent) {
+        throw new Error('AI 返回内容为空');
+      }
+
+      const jsonMatch = aiContent.match(/\[[\s\S]*\]/);
+
+      if (!jsonMatch) {
+        throw new Error('AI 返回格式不正确');
+      }
+
+      const recommendations = JSON.parse(jsonMatch[0]) as AIRecommendation[];
+      this.logger.log(`解析后的推荐结果: ${recommendations.length} 个`);
+
+      return recommendations;
+    } catch (error) {
+      this.logger.error('获取 AI 推荐失败:', error);
+      throw error;
+    }
+  }
+}
diff --git a/src/bids/controllers/bid.controller.ts b/src/bids/controllers/bid.controller.ts
index 0415236..e7ca920 100644
--- a/src/bids/controllers/bid.controller.ts
+++ b/src/bids/controllers/bid.controller.ts
@@ -24,4 +24,9 @@ export class BidsController {
   getSources() {
     return this.bidsService.getSources();
   }
+
+  @Get('by-date-range')
+  getByDateRange(@Query('startDate') startDate: string, @Query('endDate') endDate: string) {
+    return this.bidsService.getBidsByDateRange(startDate, endDate);
+  }
 }
diff --git a/src/bids/services/bid.service.ts b/src/bids/services/bid.service.ts
index f16271c..7bf8262 100644
--- a/src/bids/services/bid.service.ts
+++ b/src/bids/services/bid.service.ts
@@ -87,4 +87,22 @@ export class BidsService {
       .orderBy('bid.publishDate', 'DESC')
       .getMany();
   }
+
+  async getBidsByDateRange(startDate?: string, endDate?: string) {
+    const qb = this.bidRepository.createQueryBuilder('bid');
+
+    if (startDate) {
+      const start = new Date(startDate);
+      start.setHours(0, 0, 0, 0);
+      qb.andWhere('bid.publishDate >= :startDate', { startDate: start });
+    }
+
+    if (endDate) {
+      const end = new Date(endDate);
+      end.setHours(23, 59, 59, 999);
+      qb.andWhere('bid.publishDate <= :endDate', { endDate: end });
+    }
+
+    return qb.orderBy('bid.publishDate', 'DESC').getMany();
+  }
 }
diff --git a/src/crawler/services/cdt_target.ts b/src/crawler/services/cdt_target.ts
index 6ae6697..df3c436 100644
--- a/src/crawler/services/cdt_target.ts
+++ b/src/crawler/services/cdt_target.ts
@@ -55,7 +55,7 @@ export interface CdtResult {
 export const CdtCrawler = {
   name: '中国大唐集团电子商务平台',
   url: 'https://tang.cdt-ec.com/home/index.html',
-  baseUrl: 'https://tang.cdt-ec.com',
+  baseUrl: 'https://tang.cdt-ec.com/',
 
   async crawl(browser: puppeteer.Browser): Promise<CdtResult[]> {
     const logger = new Logger('CdtCrawler');
diff --git a/src/crawler/services/ceic_target.ts b/src/crawler/services/ceic_target.ts
index fcf6417..5d85bbd 100644
--- a/src/crawler/services/ceic_target.ts
+++ b/src/crawler/services/ceic_target.ts
@@ -50,7 +50,7 @@ async function simulateHumanScrolling(page: puppeteer.Page) {
 export const CeicCrawler = {
   name: '国家能源集团生态协作平台',
   url: 'https://ceic.dlnyzb.com/3001',
-  baseUrl: 'https://ceic.dlnyzb.com',
+  baseUrl: 'https://ceic.dlnyzb.com/',
 
   async crawl(browser: puppeteer.Browser): Promise<ChdtpResult[]> {
     const logger = new Logger('CeicCrawler');
diff --git a/src/crawler/services/cgnpc_target.ts b/src/crawler/services/cgnpc_target.ts
index 1b6bc1e..3b833bc 100644
--- a/src/crawler/services/cgnpc_target.ts
+++ b/src/crawler/services/cgnpc_target.ts
@@ -55,7 +55,7 @@ export interface CgnpcResult {
 export const CgnpcCrawler = {
   name: '中广核电子商务平台',
   url: 'https://ecp.cgnpc.com.cn/zbgg.html',
-  baseUrl: 'https://ecp.cgnpc.com.cn',
+  baseUrl: 'https://ecp.cgnpc.com.cn/',
 
   async crawl(browser: puppeteer.Browser): Promise<CgnpcResult[]> {
     const logger = new Logger('CgnpcCrawler');
diff --git a/src/crawler/services/cnncecp_target.ts b/src/crawler/services/cnncecp_target.ts
index fed1b07..725ce26 100644
--- a/src/crawler/services/cnncecp_target.ts
+++ b/src/crawler/services/cnncecp_target.ts
@@ -55,7 +55,7 @@ export interface CnncecpResult {
 export const CnncecpCrawler = {
   name: '中核集团电子采购平台',
   url: 'https://www.cnncecp.com/xzbgg/index.jhtml',
-  baseUrl: 'https://www.cnncecp.com',
+  baseUrl: 'https://www.cnncecp.com/',
 
   async crawl(browser: puppeteer.Browser): Promise<CnncecpResult[]> {
     const logger = new Logger('CnncecpCrawler');
diff --git a/src/crawler/services/cnooc_target.ts b/src/crawler/services/cnooc_target.ts
index 14220b7..e85ab01 100644
--- a/src/crawler/services/cnooc_target.ts
+++ b/src/crawler/services/cnooc_target.ts
@@ -55,7 +55,7 @@ export interface CnoocResult {
 export const CnoocCrawler = {
   name: '中海油招标平台',
   url: 'https://buy.cnooc.com.cn/cbjyweb/001/001001/moreinfo.html',
-  baseUrl: 'https://buy.cnooc.com.cn',
+  baseUrl: 'https://buy.cnooc.com.cn/',
 
   async crawl(browser: puppeteer.Browser): Promise<CnoocResult[]> {
     const logger = new Logger('CnoocCrawler');
diff --git a/src/crawler/services/eps_target.ts b/src/crawler/services/eps_target.ts
index 351426f..0344322 100644
--- a/src/crawler/services/eps_target.ts
+++ b/src/crawler/services/eps_target.ts
@@ -55,7 +55,7 @@ export interface EpsResult {
 export const EpsCrawler = {
   name: '中国三峡集团电子商务平台',
   url: 'https://eps.ctg.com.cn/cms/channel/1ywgg1/index.htm',
-  baseUrl: 'https://eps.ctg.com.cn',
+  baseUrl: 'https://eps.ctg.com.cn/',
 
   async crawl(browser: puppeteer.Browser): Promise<EpsResult[]> {
     const logger = new Logger('EpsCrawler');
diff --git a/src/crawler/services/espic_target.ts b/src/crawler/services/espic_target.ts
index 8aff546..341f9ec 100644
--- a/src/crawler/services/espic_target.ts
+++ b/src/crawler/services/espic_target.ts
@@ -54,7 +54,7 @@ export interface EspicResult {
 
 export const EspicCrawler = {
   name: '电能e招采平台（国电投）',
-  baseUrl: 'https://ebid.espic.com.cn',
+  baseUrl: 'https://ebid.espic.com.cn/',
 
   // 生成动态 URL，使用当前日期
   getUrl(page: number = 1): string {
diff --git a/src/crawler/services/powerbeijing_target.ts b/src/crawler/services/powerbeijing_target.ts
index 631b499..842b0e2 100644
--- a/src/crawler/services/powerbeijing_target.ts
+++ b/src/crawler/services/powerbeijing_target.ts
@@ -55,7 +55,7 @@ export interface PowerbeijingResult {
 export const PowerbeijingCrawler = {
   name: '北京京能电子商务平台',
   url: 'https://www.powerbeijing-ec.com/jncms/search/bulletin.html?dates=300&categoryId=2&tabName=%E6%8B%9B%E6%A0%87%E5%85%AC%E5%91%8A&page=1',
-  baseUrl: 'https://www.powerbeijing-ec.com',
+  baseUrl: 'https://www.powerbeijing-ec.com/',
 
   async crawl(browser: puppeteer.Browser): Promise<PowerbeijingResult[]> {
     const logger = new Logger('PowerbeijingCrawler');
diff --git a/src/crawler/services/sdicc_target.ts b/src/crawler/services/sdicc_target.ts
index 7345776..e3ff710 100644
--- a/src/crawler/services/sdicc_target.ts
+++ b/src/crawler/services/sdicc_target.ts
@@ -55,7 +55,7 @@ export interface SdiccResult {
 export const SdiccCrawler = {
   name: '国投集团电子采购平台',
   url: 'https://www.sdicc.com.cn/cgxx/ggList',
-  baseUrl: 'https://www.sdicc.com.cn',
+  baseUrl: 'https://www.sdicc.com.cn/',
 
   async crawl(browser: puppeteer.Browser): Promise<SdiccResult[]> {
     const logger = new Logger('SdiccCrawler');
diff --git a/src/crawler/services/szecp_target.ts b/src/crawler/services/szecp_target.ts
index 19f2a5b..81007f1 100644
--- a/src/crawler/services/szecp_target.ts
+++ b/src/crawler/services/szecp_target.ts
@@ -50,7 +50,7 @@ async function simulateHumanScrolling(page: puppeteer.Page) {
 export const SzecpCrawler = {
   name: '华润守正采购交易平台',
   url: 'https://www.szecp.com.cn/first_zbgg/index.html',
-  baseUrl: 'https://www.szecp.com.cn',
+  baseUrl: 'https://www.szecp.com.cn/',
 
   async crawl(browser: puppeteer.Browser): Promise<ChdtpResult[]> {
     const logger = new Logger('SzecpCrawler');