dep milvus

docker-compose.yml

@@ -12,6 +12,6 @@ services:
     ports:
       - "8000:8000"
     entrypoint: /app/myapp
-    # depends_on:
-    #   - standalone
-    #   - webserver
+    depends_on:
+      - milvus-standalone
+      - graph-webserver

main.go

@@ -4,7 +4,6 @@ import (
 	"context"
 	"flag"
 	"net/http"
-	"os"
 
 	"github.com/firebase/genkit/go/genkit"
 	"github.com/rs/zerolog"
@@ -68,6 +67,8 @@ func main() {
 	graphragAddr := flag.String("graphrag-addr", "graph-webserver: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")
+	tencentcloudSecretID := flag.String("tencentcloud-secret-id", "AKID64oLfmfLtESUJ6i8LPSM4gCVbiniQuBF", "Tencent Cloud Secret ID")
+    tencentcloudSecretKey := flag.String("tencentcloud-secret-key", "rX2JMBnBMJ2YqulOo37xa5OUMSN4Xnpd", "Tencent Cloud Secret Key")
 	debug := flag.Bool("debug", false, "sets log level to debug")
 	flag.Parse()
 
@@ -77,9 +78,6 @@ func main() {
 		zerolog.SetGlobalLevel(zerolog.DebugLevel)
 	}
 
-	os.Setenv("TENCENTCLOUD_SECRET_ID", "AKID64oLfmfLtESUJ6i8LPSM4gCVbiniQuBF")
-	os.Setenv("TENCENTCLOUD_SECRET_KEY", "rX2JMBnBMJ2YqulOo37xa5OUMSN4Xnpd")
-
 	ctx := context.Background()
 
 	// Initialize genkit with plugins using flag/env values
@@ -128,6 +126,8 @@ func main() {
 	kc := knowledge.NewKnowledgeClient(knowledge.ClientConfig{
 		Endpoint: "lkeap.tencentcloudapi.com",
 		Region:   "ap-guangzhou",
+		SecretID: *tencentcloudSecretID,
+		SecretKey: *tencentcloudSecretKey,
 	})
 	if err := kc.Init(context.Background()); err != nil {
 		log.Fatal().Msgf("Failed to initialize KnowledgeClient: %v", err)