qa chat

main.go

@@ -2,6 +2,7 @@ package main
 
 import (
 	"context"
+	"encoding/json"
 	"fmt"
 	"log"
 	"net/http"
@@ -27,6 +28,11 @@ type Input struct {
 
 func main() {
 
+
+
+	mainQA()
+
+	
 	ctx := context.Background()
 
 	ds := deepseek.DeepSeek{
@@ -48,7 +54,13 @@ func main() {
 	// Define a simple flow that generates jokes about a given topic
 	genkit.DefineFlow(g, "chat", func(ctx context.Context, input *Input) (string, error) {
 
-		fmt.Println("input-------------------------------",input.Content)
+		inputAsJson, err := json.Marshal(input)
+
+		if err != nil{
+			return "",err 
+		}
+		
+		fmt.Println("input-------------------------------",string(inputAsJson))
 
 		resp, err := genkit.Generate(ctx, g,
 			ai.WithModel(m),
@@ -80,12 +92,6 @@ func main() {
 		mux.Handle("POST /"+a.Name(), handler)
 	}
 
-	//mux.HandleFunc("/swagger/", httpSwagger.WrapHandler)
-
-	// 暴露 Swagger UI，使用 swagger.yaml
-	// mux.HandleFunc("/swagger/", httpSwagger.Handler(
-	// 	httpSwagger.URL("/swagger/doc.yaml"), // 指定 YAML 文件路径
-	// ))
 
 	    // 暴露 Swagger UI，使用 swagger.yaml
 		mux.HandleFunc("/swagger/", httpSwagger.Handler(

qa.go

@@ -15,7 +15,6 @@ var (
 	connString = flag.String("dbconn", "", "database connection string")
 )
 
-// QA 结构体表示 qa 表的记录
 type QA struct {
     ID        int64     // 主键
     CreatedAt time.Time // 创建时间
@@ -23,6 +22,7 @@ type QA struct {
     Username  *string   // 可空的用户名
     Question  *string   // 可空的问题
     Answer    *string   // 可空的答案
+    Summary   *string   // 可空的摘要
 }
 
 // QAStore 定义 DAO 接口
@@ -43,10 +43,9 @@ func NewQAStore(db *sql.DB) QAStore {
 	return &qaStore{db: db}
 }
 
-// GetLatestQA 从 latest_qa 视图读取数据
 func (s *qaStore) GetLatestQA(ctx context.Context, userID *int64) ([]QA, error) {
     query := `
-        SELECT id, created_at, user_id, username, question, answer
+        SELECT id, created_at, user_id, username, question, answer, summary
         FROM latest_qa
         WHERE user_id = $1 OR (user_id IS NULL AND $1 IS NULL)`
     args := []interface{}{userID}
@@ -64,8 +63,8 @@ func (s *qaStore) GetLatestQA(ctx context.Context, userID *int64) ([]QA, error)
     for rows.Next() {
         var qa QA
         var userIDVal sql.NullInt64
-		var username, question, answer sql.NullString
-		if err := rows.Scan(&qa.ID, &qa.CreatedAt, &userIDVal, &username, &question, &answer); err != nil {
+        var username, question, answer, summary sql.NullString
+        if err := rows.Scan(&qa.ID, &qa.CreatedAt, &userIDVal, &username, &question, &answer, &summary); err != nil {
             return nil, fmt.Errorf("scan row: %w", err)
         }
         if userIDVal.Valid {
@@ -80,6 +79,9 @@ func (s *qaStore) GetLatestQA(ctx context.Context, userID *int64) ([]QA, error)
         if answer.Valid {
             qa.Answer = &answer.String
         }
+        if summary.Valid {
+            qa.Summary = &summary.String
+        }
         results = append(results, qa)
     }
     if err := rows.Err(); err != nil {
@@ -88,17 +90,16 @@ func (s *qaStore) GetLatestQA(ctx context.Context, userID *int64) ([]QA, error)
     return results, nil
 }
 
-// WriteQA 插入或更新 qa 表记录
 func (s *qaStore) WriteQA(ctx context.Context, qa QA) (int64, error) {
     if qa.ID != 0 {
         // 更新记录
         query := `
             UPDATE qa
-            SET user_id = $1, username = $2, question = $3, answer = $4
-            WHERE id = $5
+            SET user_id = $1, username = $2, question = $3, answer = $4, summary = $5
+            WHERE id = $6
             RETURNING id`
         var updatedID int64
-		err := s.db.QueryRowContext(ctx, query, qa.UserID, qa.Username, qa.Question, qa.Answer, qa.ID).Scan(&updatedID)
+        err := s.db.QueryRowContext(ctx, query, qa.UserID, qa.Username, qa.Question, qa.Answer, qa.Summary, qa.ID).Scan(&updatedID)
         if err == sql.ErrNoRows {
             return 0, fmt.Errorf("no record found with id %d", qa.ID)
         }
@@ -110,11 +111,11 @@ func (s *qaStore) WriteQA(ctx context.Context, qa QA) (int64, error) {
 
     // 插入新记录
     query := `
-        INSERT INTO qa (user_id, username, question, answer)
-        VALUES ($1, $2, $3, $4)
+        INSERT INTO qa (user_id, username, question, answer, summary)
+        VALUES ($1, $2, $3, $4, $5)
         RETURNING id`
     var newID int64
-	err := s.db.QueryRowContext(ctx, query, qa.UserID, qa.Username, qa.Question, qa.Answer).Scan(&newID)
+    err := s.db.QueryRowContext(ctx, query, qa.UserID, qa.Username, qa.Question, qa.Answer, qa.Summary).Scan(&newID)
     if err != nil {
         return 0, fmt.Errorf("insert qa: %w", err)
     }
@@ -122,14 +123,15 @@ func (s *qaStore) WriteQA(ctx context.Context, qa QA) (int64, error) {
 }
 
 func mainQA() {
-	flag.Parse()
-	ctx := context.Background()
 
-	if *connString == "" {
-		log.Fatal("need -dbconn")
-	}
+	//  postgres://postgres.awcfgdodiuqnlsobcivq:[YOUR-PASSWORD]@aws-0-ap-southeast-1.pooler.supabase.com:6543/postgres
+	// postgres://postgres.awcfgdodiuqnlsobcivq:rNVrqxaapXypEGg8@aws-0-ap-southeast-1.pooler.supabase.com:6543/postgres
+	//connString := "postgres://postgres.awcfgdodiuqnlsobcivq:rNVrqxaapXypEGg8@aws-0-ap-southeast-1.pooler.supabase.com:6543/postgres"
+	// var connString = "postgres://postgres.awcfgdodiuqnlsobcivq:rNVrqxaapXypEGg8@aws-0-ap-southeast-1.pooler.supabase.com:6543/postgres"
+
+	var connString = "postgresql://postgres.awcfgdodiuqnlsobcivq:P8L00j7k3wHRwUJw@aws-0-ap-southeast-1.pooler.supabase.com:6543/postgres"
 
-	db, err := sql.Open("postgres", *connString)
+    db, err := sql.Open("postgres", connString)
     if err != nil {
         log.Fatalf("open database: %v", err)
     }
@@ -137,14 +139,17 @@ func mainQA() {
 
     store := NewQAStore(db)
 
+	ctx := context.Background()
+
     // 示例：读取 user_id=101 的最新 QA
     results, err := store.GetLatestQA(ctx, int64Ptr(101))
     if err != nil {
         log.Fatalf("get latest QA: %v", err)
     }
     for _, qa := range results {
-		fmt.Printf("ID: %d, CreatedAt: %v, UserID: %v, Username: %v, Question: %v, Answer: %v\n",
-			qa.ID, qa.CreatedAt, derefInt64(qa.UserID), derefString(qa.Username), derefString(qa.Question), derefString(qa.Answer))
+        fmt.Printf("ID: %d, CreatedAt: %v, UserID: %v, Username: %v, Question: %v, Answer: %v, Summary: %v\n",
+            qa.ID, qa.CreatedAt, derefInt64(qa.UserID), derefString(qa.Username), 
+            derefString(qa.Question), derefString(qa.Answer), derefString(qa.Summary))
     }
 
     // 示例：插入新 QA
@@ -153,6 +158,7 @@ func mainQA() {
         Username: stringPtr("alice"),
         Question: stringPtr("What is AI?"),
         Answer:   stringPtr("AI is..."),
+        Summary:  stringPtr("AI overview"),
     }
     newID, err := store.WriteQA(ctx, newQA)
     if err != nil {
@@ -167,6 +173,7 @@ func mainQA() {
         Username: stringPtr("alice_updated"),
         Question: stringPtr("What is NLP?"),
         Answer:   stringPtr("NLP is..."),
+        Summary:  stringPtr("NLP overview"),
     }
     updatedID, err := store.WriteQA(ctx, updateQA)
     if err != nil {
@@ -175,6 +182,7 @@ func mainQA() {
     fmt.Printf("Updated QA with ID: %d\n", updatedID)
 }
 
+
 // 辅助函数：处理指针类型的空值
 func int64Ptr(i int64) *int64 {
 	return &i