pg connstr param

main.go

@@ -97,6 +97,13 @@ Instructions:
 
 func main() {
 
+
+    // Define command-line flags with hardcoded values as defaults
+    deepseekAPIKey := flag.String("deepseek-api-key", "sk-9f70df871a7c4b8aa566a3c7a0603706", "DeepSeek API key")
+    milvusAddr := flag.String("milvus-addr", "54.92.111.204:19530", "Milvus server address")
+    graphragAddr := flag.String("graphrag-addr", "54.92.111.204:5670", "GraphRAG server address")
+    googleAIApiKey := flag.String("googleai-api-key", "AIzaSyCoYBOmnwRWlH_-nT25lpn8pMg3T18Q0uI", "Google AI API key")
+	pgConnString := flag.String("pg-conn-string", "postgresql://postgres.awcfgdodiuqnlsobcivq:P99IU9NEoDRPsBfb@aws-0-ap-southeast-1.pooler.supabase.com:5432/postgres", "PostgreSQL connection string")
 	debug := flag.Bool("debug", false, "sets log level to debug")
     flag.Parse()
 
@@ -119,12 +126,14 @@ func main() {
 		},
 	}
 
+    // Initialize genkit with plugins using flag/env values
     g, err := genkit.Init(ctx, genkit.WithPlugins(
-		&deepseek.DeepSeek{APIKey: "sk-9f70df871a7c4b8aa566a3c7a0603706"},
-		&milvus.Milvus{Addr: "54.92.111.204:19530"},
-		&graphrag.GraphKnowledge{Addr: "54.92.111.204:5670"},
-		&googlegenai.GoogleAI{APIKey: "AIzaSyCoYBOmnwRWlH_-nT25lpn8pMg3T18Q0uI"},
-		&evaluators.GenkitEval{Metrics: metrics}))
+        &deepseek.DeepSeek{APIKey: *deepseekAPIKey},
+        &milvus.Milvus{Addr: *milvusAddr},
+        &graphrag.GraphKnowledge{Addr: *graphragAddr},
+        &googlegenai.GoogleAI{APIKey: *googleAIApiKey},
+        &evaluators.GenkitEval{Metrics: metrics},
+    ))
 
 	if err != nil {
 		log.Fatal().Msg(err.Error())
@@ -236,7 +245,7 @@ func main() {
 		log.Fatal().Msg(err.Error())
 	}
 
-	qa, err := InitQAStore()
+	qa, err := InitQAStore(*pgConnString)
 	if err != nil {
 		log.Fatal().Msgf("InitQAStore failed: %v", err)
 	}

qa.go

@@ -3,16 +3,12 @@ package main
 import (
 	"context"
 	"database/sql"
-	"flag"
 	"fmt"
 	"time"
 
 	_ "github.com/lib/pq"
 )
 
-var (
-	connString = flag.String("dbconn", "", "database connection string")
-)
 
 type QA struct {
 	ID        int64     // 主键
@@ -49,9 +45,9 @@ func NewQAStore(db *sql.DB) QAStore {
 }
 
 // 初始化数据库连接并返回 QAStore
-func InitQAStore() (QAStore, error) {
+func InitQAStore(connString string) (QAStore, error) {
 	// Supabase 提供的连接字符串
-	connString := "postgresql://postgres.awcfgdodiuqnlsobcivq:P99IU9NEoDRPsBfb@aws-0-ap-southeast-1.pooler.supabase.com:5432/postgres"
+	//connString := "postgresql://postgres.awcfgdodiuqnlsobcivq:P99IU9NEoDRPsBfb@aws-0-ap-southeast-1.pooler.supabase.com:5432/postgres"
 
 	// 打开数据库连接
 	db, err := sql.Open("postgres", connString)