mirror of
https://github.com/ArvinLovegood/go-stock.git
synced 2025-07-19 00:00:09 +08:00
888a97e4d3
- 更新 SearchStockByIndicators 函数描述，使其更准确地反映功能 - 在 Resp 结构中添加 Error 字段，用于处理错误信息 - 修改 openai_api.go 和 openai_api_test.go 中的错误处理逻辑 - 优化消息发送格式，提高错误信息的可读性
57 lines
1.5 KiB
Go
57 lines
1.5 KiB
Go
package data
|
|
|
|
import (
|
|
"context"
|
|
"go-stock/backend/db"
|
|
"testing"
|
|
)
|
|
|
|
func TestNewDeepSeekOpenAiConfig(t *testing.T) {
|
|
db.Init("../../data/stock.db")
|
|
|
|
var tools []Tool
|
|
tools = append(tools, Tool{
|
|
Type: "function",
|
|
Function: ToolFunction{
|
|
Name: "SearchStockByIndicators",
|
|
Description: "根据自然语言筛选股票，返回自然语言选股条件要求的股票所有相关数据",
|
|
Parameters: FunctionParameters{
|
|
Type: "object",
|
|
Properties: map[string]any{
|
|
"words": map[string]any{
|
|
"type": "string",
|
|
"description": "选股自然语言,并且条件使用;分隔，或者条件使用,分隔。例如：创新药;PE<30;净利润增长率>50%;",
|
|
},
|
|
},
|
|
Required: []string{"words"},
|
|
},
|
|
},
|
|
})
|
|
|
|
ai := NewDeepSeekOpenAi(context.TODO())
|
|
//res := ai.NewChatStream("长电科技", "sh600584", "长电科技分析和总结", nil)
|
|
res := ai.NewSummaryStockNewsStreamWithTools("总结市场资讯，发掘潜力标的/行业/板块/概念，控制风险，最后按风险登记生成指标选股策略汇总表，每个策略中的指标分号分隔，写成一行", nil, tools)
|
|
|
|
for {
|
|
select {
|
|
case msg := <-res:
|
|
if len(msg) > 0 {
|
|
t.Log(msg)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func TestGetTopNewsList(t *testing.T) {
|
|
news := GetTopNewsList(30)
|
|
t.Log(news)
|
|
}
|
|
|
|
func TestSearchGuShiTongStockInfo(t *testing.T) {
|
|
db.Init("../../data/stock.db")
|
|
SearchGuShiTongStockInfo("hk01810", 60)
|
|
SearchGuShiTongStockInfo("sh600745", 60)
|
|
SearchGuShiTongStockInfo("gb_goog", 60)
|
|
|
|
}
|