Function main

examples/rag-basic/main.go:15–63

()

)

func main() {
	pipe, err := core.NewPipeline(core.PipelineConfig{
		Store:       vector.NewMemoryStore(),
		Embedder:    vector.NewMockEmbedder(32),
		Namespace:   "demo",
		DefaultTopK: 3,
	})
	if err != nil {
		log.Fatalf("init pipeline: %v", err)
	}

	ctx := context.Background()

	// 入库
	text := `Aster 是面向生产的多 Agent 框架。
它内置工作记忆、语义记忆、Workflow 编排以及丰富的安全护栏。`

	chunks, err := pipe.Ingest(ctx, core.IngestRequest{
		ID:        "intro",
		Text:      text,
		Namespace: "demo",
		Metadata: map[string]any{
			"source": "docs/intro",
		},
	})
	if err != nil {
		log.Fatalf("ingest: %v", err)
	}
	if _, err := fmt.Fprintf(os.Stdout, "Inserted %d chunks\n", len(chunks)); err != nil {
		log.Printf("Failed to write output: %v", err)
	}

	// 检索
	hits, err := pipe.Search(ctx, "什么是 Aster？", 3, map[string]any{
		"namespace": "demo",
	})
	if err != nil {
		log.Fatalf("search: %v", err)
	}

	if _, err := fmt.Fprintf(os.Stdout, "Top hits:\n"); err != nil {
		log.Printf("Failed to write output: %v", err)
	}
	for i, h := range hits {
		if _, err := fmt.Fprintf(os.Stdout, "%d) score=%.4f text=%.50s...\n", i+1, h.Score, h.Text); err != nil {
			log.Printf("Failed to write output: %v", err)
		}
	}
}