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