47 lines
1.3 KiB
Go
47 lines
1.3 KiB
Go
package eino
|
||
|
||
import (
|
||
"context"
|
||
|
||
"github.com/cloudwego/eino-ext/components/embedding/dashscope"
|
||
"github.com/gogf/gf/v2/frame/g"
|
||
"github.com/golang/glog"
|
||
)
|
||
|
||
// 全局只初始化一次
|
||
var (
|
||
Embedder *dashscope.Embedder // 导出供其他模块使用
|
||
)
|
||
|
||
// init：程序启动时自动执行一次
|
||
func init() {
|
||
ctx := context.Background()
|
||
if !g.Cfg().MustGet(ctx, "eino.embedding").IsEmpty() {
|
||
var err error
|
||
cfg := &dashscope.EmbeddingConfig{
|
||
APIKey: g.Cfg().MustGet(ctx, "eino.embedding.apiKey").String(),
|
||
Model: g.Cfg().MustGet(ctx, "eino.embedding.model").String(),
|
||
}
|
||
// 检查是否配置了 APIType，支持 "text_api" 和 "multi_modal_api"
|
||
//if apiType := g.Cfg().MustGet(ctx, "eino.embedding.apiType").String(); apiType != "" {
|
||
// apiTypeVal := dashscope.APIType(apiType)
|
||
// cfg.APIType = &apiTypeVal
|
||
//}
|
||
Embedder, err = dashscope.NewEmbedder(ctx, cfg)
|
||
if err != nil {
|
||
glog.Fatalf("NewEmbedder of ark error: %v", err)
|
||
}
|
||
//embedding, err := embedder.EmbedStrings(ctx, []string{"hello world", "bye bye"})
|
||
//if err != nil {
|
||
// log.Printf("embedding error: %v\n", err)
|
||
// return
|
||
//}
|
||
//
|
||
//log.Printf("embedding: %v\n", embedding)
|
||
}
|
||
}
|
||
|
||
func EmbedStrings(ctx context.Context, texts []string) (embeddings [][]float64, err error) {
|
||
return Embedder.EmbedStrings(ctx, texts)
|
||
}
|