Skip to content

Commit 494a60d

Browse files
committed
refactor(backend):注释掉获取股势通资讯的代码
- 在 openai_api.go 中注释掉了获取股势通资讯的相关代码 - 在 openai_api_test.go 中添加了对 SearchGuShiTongStockInfo 函数的测试用例
1 parent b3e2565 commit 494a60d

File tree

2 files changed

+26
-22
lines changed

2 files changed

+26
-22
lines changed

backend/data/openai_api.go

Lines changed: 19 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -808,25 +808,25 @@ func (o *OpenAi) NewChatStream(stock, stockCode, userQuestion string, sysPromptI
808808
return
809809
}
810810

811-
messages := SearchGuShiTongStockInfo(stockCode, o.CrawlTimeOut)
812-
if messages == nil || len(*messages) == 0 {
813-
logger.SugaredLogger.Error("获取股势通资讯失败")
814-
//ch <- "***❗获取股势通资讯失败,分析结果可能不准确***<hr>"
815-
//go runtime.EventsEmit(o.ctx, "warnMsg", "❗获取股势通资讯失败,分析结果可能不准确")
816-
return
817-
}
818-
var newsText strings.Builder
819-
for _, message := range *messages {
820-
newsText.WriteString(message + "\n")
821-
}
822-
msg = append(msg, map[string]interface{}{
823-
"role": "user",
824-
"content": stock + "相关新闻资讯",
825-
})
826-
msg = append(msg, map[string]interface{}{
827-
"role": "assistant",
828-
"content": newsText.String(),
829-
})
811+
//messages := SearchGuShiTongStockInfo(stockCode, o.CrawlTimeOut)
812+
//if messages == nil || len(*messages) == 0 {
813+
// logger.SugaredLogger.Error("获取股势通资讯失败")
814+
// //ch <- "***❗获取股势通资讯失败,分析结果可能不准确***<hr>"
815+
// //go runtime.EventsEmit(o.ctx, "warnMsg", "❗获取股势通资讯失败,分析结果可能不准确")
816+
// return
817+
//}
818+
//var newsText strings.Builder
819+
//for _, message := range *messages {
820+
// newsText.WriteString(message + "\n")
821+
//}
822+
//msg = append(msg, map[string]interface{}{
823+
// "role": "user",
824+
// "content": stock + "相关新闻资讯",
825+
//})
826+
//msg = append(msg, map[string]interface{}{
827+
// "role": "assistant",
828+
// "content": newsText.String(),
829+
//})
830830
}()
831831

832832
go func() {

backend/data/openai_api_test.go

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ package data
33
import (
44
"context"
55
"go-stock/backend/db"
6+
log "go-stock/backend/logger"
67
"testing"
78
)
89

@@ -52,9 +53,12 @@ func TestGetTopNewsList(t *testing.T) {
5253

5354
func TestSearchGuShiTongStockInfo(t *testing.T) {
5455
db.Init("../../data/stock.db")
55-
SearchGuShiTongStockInfo("hk01810", 60)
56-
SearchGuShiTongStockInfo("sh600745", 60)
57-
SearchGuShiTongStockInfo("gb_goog", 60)
56+
//SearchGuShiTongStockInfo("hk01810", 60)
57+
msgs := SearchGuShiTongStockInfo("sh600745", 60)
58+
for _, msg := range *msgs {
59+
log.SugaredLogger.Infof("%s", msg)
60+
}
61+
//SearchGuShiTongStockInfo("gb_goog", 60)
5862

5963
}
6064

0 commit comments

Comments
 (0)