rag ok

README.md

@@ -132,35 +132,44 @@ curl -X POST http://localhost:8000/indexGraph \
 
 
 
+curl -X POST http://localhost:8000/indexDocuments \
+-H "Content-Type: application/json" \
+-d '{"content": "What is the capital of UK?", "metadata": {"user_id": "user456", "username": "Bob"}}'
+{"result": "Document indexed successfully"}
 
 
 
 
 
 
+		d1 := ai.DocumentFromText("Paris is the capital of France", nil)
+		d2 := ai.DocumentFromText("USA is the largest importer of coffee", nil)
+		d3 := ai.DocumentFromText("Water exists in 3 states - solid, liquid and gas", nil)
 
 
 
 
+curl -X POST http://localhost:8000/indexDocuments \
+-H "Content-Type: application/json" \
+-d '{"content": "What is the capital of UK?", "metadata": {"user_id": "user456", "username": "Bob"}}'
+{"result": "Paris is the capital of France"}
 
 
+curl -X POST http://localhost:8000/indexDocuments \
+-H "Content-Type: application/json" \
+-d '{"content": "What is the capital of UK?", "metadata": {"user_id": "user456", "username": "Bob"}}'
+{"result": "USA is the largest importer of coffee"}
 
 
 
 
+curl -X POST http://localhost:8000/indexDocuments \
+-H "Content-Type: application/json" \
+-d '{"content": "Water exists in 3 states - solid, liquid and gas", "metadata": {"user_id": "user456", "username": "Bob"}}'
+{"result": "USA is the largest importer of coffee"}
 
 
 
-
-
-
-
-
-
-curl -X POST http://localhost:8000/indexDocuments \
--H "Content-Type: application/json" \
--d '{"content": "What is the capital of UK?", "metadata": {"user_id": "user456", "username": "Bob"}}'
-{"result": "Document indexed successfully"}
 
 
 

main.go

@@ -6,6 +6,7 @@ import (
 	"fmt"
 	"log"
 	"net/http"
+	"strings"
 
 	"github.com/firebase/genkit/go/ai"
 	"github.com/firebase/genkit/go/genkit"
@@ -47,6 +48,16 @@ type GraphInput struct {
 	Metadata map[string]interface{} `json:"metadata,omitempty"`
 }
 
+const simpleQaPromptTemplate = `
+You're a helpful agent that answers the user's common questions based on the context provided.
+
+Here is the user's query: {{query}}
+
+Here is the context you should use: {{context}}
+
+Please provide the best answer you can.
+`
+
 func main() {
 
 	ctx := context.Background()
@@ -79,12 +90,12 @@ func main() {
 		log.Fatal(err)
 	}
 
-	m := ds.DefineModel(g,
-		deepseek.ModelDefinition{
-			Name: "deepseek-chat", // Choose an appropriate model
-			Type: "chat",          // Must be chat for tool support
-		},
-		nil)
+	// m := ds.DefineModel(g,
+	// 	deepseek.ModelDefinition{
+	// 		Name: "deepseek-chat", // Choose an appropriate model
+	// 		Type: "chat",          // Must be chat for tool support
+	// 	},
+	// 	nil)
 
 	embedder := googlegenai.GoogleAIEmbedder(g, "embedding-001")
 	if embedder == nil {
@@ -157,6 +168,16 @@ func main() {
 		return fmt.Sprintf("Document indexed successfully, docname %s", resDocName), nil
 	})
 
+	simpleQaPrompt, err := genkit.DefinePrompt(g, "simpleQaPrompt",
+		ai.WithModelName("googleai/gemini-2.0-flash"),
+		ai.WithPrompt(simpleQaPromptTemplate),
+		ai.WithInputType(simpleQaPromptInput{}),
+		ai.WithOutputFormat(ai.OutputFormatText),
+	)
+	if err != nil {
+		log.Fatal(err)
+	}
+
 	// Define a simple flow that generates jokes about a given topic
 	genkit.DefineFlow(g, "chat", func(ctx context.Context, input *Input) (string, error) {
 
@@ -175,29 +196,28 @@ func main() {
 		}
 
 		for _, d := range response.Documents {
-			fmt.Println("d.Content[0].Text",d.Content[0].Text)
+			fmt.Println("d.Content[0].Text", d.Content[0].Text)
 		}
 
-		return "",nil 
-
-
-		resp, err := genkit.Generate(ctx, g,
-			ai.WithModel(m),
-			ai.WithPrompt(`Tell silly short jokes about apple`))
-
-		if err != nil {
-			fmt.Println(err.Error())
-			return "", err
+		var sb strings.Builder
+		for _, d := range response.Documents {
+			sb.WriteString(d.Content[0].Text)
+			sb.WriteByte('\n')
 		}
 
-		fmt.Println("resp.Text()", resp.Text())
+		promptInput := &simpleQaPromptInput{
+			Query:   input.Content,
+			Context: sb.String(),
+		}
 
+		resp, err := simpleQaPrompt.Execute(ctx, ai.WithInput(promptInput))
 		if err != nil {
 			return "", err
 		}
+		return resp.Text(), nil
 
-		text := resp.Text()
-		return text, nil
+		//ai.WithPrompt(promptInput))
+		//ai.WithPrompt(`Tell silly short jokes about apple`)
 	})
 
 	// 配置限速器：每秒 10 次请求，突发容量 20，最大并发 5
@@ -225,3 +245,8 @@ func main() {
 		log.Fatalf("Server failed: %v", err)
 	}
 }
+
+type simpleQaPromptInput struct {
+	Query   string `json:"query"`
+	Context string `json:"context"`
+}

main_test.go

@@ -13,7 +13,7 @@ import (
 
 //  func TestGenerateEmbedding(t *testing.T)
 
-func TestGenerateEmbedding(t *testing.T){
+func TestGenerateEmbedding(t *testing.T) {
 	ctx := context.Background()
 
 	// Initialize Genkit with Google AI plugin

plugins/milvus/milvus.go

@@ -265,10 +265,6 @@ type docStore struct {
 // 	}, nil
 // }
 
-
-
-
-
 // newDocStore creates a docStore.
 func (m *Milvus) newDocStore(ctx context.Context, cfg *CollectionConfig) (*docStore, error) {
 	if m.client == nil {
@@ -389,15 +385,6 @@ func (m *Milvus) newDocStore(ctx context.Context, cfg *CollectionConfig) (*docSt
 	}, nil
 }
 
-
-
-
-
-
-
-
-
-
 // Indexer returns the indexer for a collection.
 func Indexer(g *genkit.Genkit, collection string) ai.Indexer {
 	return genkit.LookupIndexer(g, provider, collection)
@@ -472,10 +459,6 @@ func (ds *docStore) Index(ctx context.Context, req *ai.IndexerRequest) error {
 	return nil
 }
 
-
-
-
-
 // // RetrieverOptions for Milvus retrieval.
 // type RetrieverOptions struct {
 //     Count      int    `json:"count,omitempty"`       // Max documents to retrieve.
@@ -653,14 +636,6 @@ func (ds *docStore) Index(ctx context.Context, req *ai.IndexerRequest) error {
 //     return strings.Join(strs, sep)
 // }
 
-
-
-
-
-
-
-
-
 // RetrieverOptions for Milvus retrieval.
 type RetrieverOptions struct {
 	Count      int    `json:"count,omitempty"`       // Max documents to retrieve.
@@ -787,4 +762,3 @@ func (ds *docStore) Retrieve(ctx context.Context, req *ai.RetrieverRequest) (*ai
 		Documents: docs,
 	}, nil
 }
-