mirror of
https://github.com/ArvinLovegood/go-stock.git
synced 2025-07-19 00:00:09 +08:00
ebeaf104bb
- 在 OpenAi 结构中添加了新的方法 NewSummaryStockNewsStreamWithTools，支持使用工具进行股票分析 - 在 app.go 中调用了新方法，集成了股票搜索工具- 修改了 SearchStockApi 的 SearchStock 方法，增加了 pageSize 参数 - 更新了相关测试文件以适应新的功能
26 lines
619 B
Go
26 lines
619 B
Go
package data
|
|
|
|
import (
|
|
"go-stock/backend/db"
|
|
"go-stock/backend/logger"
|
|
"testing"
|
|
)
|
|
|
|
func TestSearchStock(t *testing.T) {
|
|
db.Init("../../data/stock.db")
|
|
|
|
res := NewSearchStockApi("算力股;净利润连续3年增长").SearchStock(10)
|
|
data := res["data"].(map[string]any)
|
|
result := data["result"].(map[string]any)
|
|
dataList := result["dataList"].([]any)
|
|
for _, v := range dataList {
|
|
d := v.(map[string]any)
|
|
logger.SugaredLogger.Infof("%s:%s", d["INDUSTRY"], d["SECURITY_SHORT_NAME"])
|
|
}
|
|
//columns := result["columns"].([]any)
|
|
//for _, v := range columns {
|
|
// logger.SugaredLogger.Infof("v:%+v", v)
|
|
//}
|
|
|
|
}
|