SpannerのQueryPlanを読んで、意見を述べる機能を追加

README.md

@@ -1,6 +1,28 @@
 # tungstrix
 Spanner最適化支援Agent
 
+## Environment
+
+API Keyを利用する場合
+
+```
+GOOGLE_API_KEY={YOUR_API_KEY}
+```
+
+Vertex AIを利用する場合
+
+```
+GOOGLE_CLOUD_PROJECT={YOUR_PROJECT_ID}
+GOOGLE_CLOUD_LOCATION=us-central1
+GOOGLE_GENAI_USE_VERTEXAI=true
+```
+
+## Development
+
+```
+go run cmd/agent/main.go web api webui
+```
+
 ## 名前の由来
 
 Geminiに考えてもらいました。

agent.go

@@ -0,0 +1,41 @@
+package tungstrix
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/sinmetalcraft/tungstrix/tools/spanner"
+	"google.golang.org/adk/agent"
+	"google.golang.org/adk/agent/llmagent"
+	"google.golang.org/adk/model/gemini"
+	"google.golang.org/adk/tool"
+	"google.golang.org/adk/tool/functiontool"
+	"google.golang.org/genai"
+)
+
+func NewAgent(ctx context.Context) (agent.Agent, error) {
+	model, err := gemini.NewModel(ctx, "gemini-2.5-pro", &genai.ClientConfig{
+		HTTPOptions: genai.HTTPOptions{APIVersion: "v1"},
+	})
+	if err != nil {
+		return nil, fmt.Errorf("failed to create new model: %w", err)
+	}
+
+	analyzeQueryTool, err := functiontool.New(functiontool.Config{
+		Name:        "analyzeQuery",
+		Description: "Retrieve the Spanner QueryPlan.",
+	}, spanner.AnalyzeQuery)
+
+	a, err := llmagent.New(llmagent.Config{
+		Name:        "tungstrix",
+		Model:       model,
+		Description: "Agent that can provide advice about Spanner",
+		Instruction: prompt,
+		Tools:       []tool.Tool{analyzeQueryTool},
+	})
+	if err != nil {
+		return nil, fmt.Errorf("failed to create new agent: %w", err)
+	}
+
+	return a, nil
+}

cmd/agent/main.go

@@ -0,0 +1,32 @@
+package main
+
+import (
+	"context"
+	"log"
+	"os"
+
+	"github.com/sinmetalcraft/tungstrix"
+	"google.golang.org/adk/cmd/launcher/adk"
+	"google.golang.org/adk/cmd/launcher/full"
+	"google.golang.org/adk/server/restapi/services"
+)
+
+func main() {
+	ctx := context.Background()
+
+	root_agent, err := tungstrix.NewAgent(ctx)
+	if err != nil {
+		panic(err)
+	}
+
+	agentLoader, err := services.NewMultiAgentLoader(root_agent)
+	if err != nil {
+		log.Fatalf("failed to create agent loader: %v", err)
+	}
+	l := full.NewLauncher()
+	if err := l.Execute(ctx, &adk.Config{
+		AgentLoader: agentLoader,
+	}, os.Args[1:]); err != nil {
+		log.Fatalf("run failed: %v\n\n%s", err, l.CommandLineSyntax())
+	}
+}

go.mod

@@ -0,0 +1,81 @@
+module github.com/sinmetalcraft/tungstrix
+
+go 1.25.1
+
+require (
+	cloud.google.com/go/spanner v1.86.1
+	github.com/apstndb/spannerplan v0.1.4
+	google.golang.org/adk v0.1.0
+	google.golang.org/api v0.252.0
+	google.golang.org/genai v1.20.0
+)
+
+require (
+	cel.dev/expr v0.24.0 // indirect
+	cloud.google.com/go v0.123.0 // indirect
+	cloud.google.com/go/auth v0.17.0 // indirect
+	cloud.google.com/go/auth/oauth2adapt v0.2.8 // indirect
+	cloud.google.com/go/compute/metadata v0.9.0 // indirect
+	cloud.google.com/go/monitoring v1.24.3 // indirect
+	github.com/GoogleCloudPlatform/grpc-gcp-go/grpcgcp v1.5.3 // indirect
+	github.com/GoogleCloudPlatform/opentelemetry-operations-go/detectors/gcp v1.30.0 // indirect
+	github.com/a2aproject/a2a-go v0.3.0 // indirect
+	github.com/apstndb/lox v0.0.0-20230530141045-98c1efebcde8 // indirect
+	github.com/apstndb/treeprint v0.0.0-20250529153958-e82576b37da6 // indirect
+	github.com/awalterschulze/gographviz v2.0.3+incompatible // indirect
+	github.com/cespare/xxhash/v2 v2.3.0 // indirect
+	github.com/cncf/xds/go v0.0.0-20251014123835-2ee22ca58382 // indirect
+	github.com/envoyproxy/go-control-plane/envoy v1.35.0 // indirect
+	github.com/envoyproxy/protoc-gen-validate v1.2.1 // indirect
+	github.com/fatih/color v1.15.0 // indirect
+	github.com/felixge/httpsnoop v1.0.4 // indirect
+	github.com/go-jose/go-jose/v4 v4.1.3 // indirect
+	github.com/go-logr/logr v1.4.3 // indirect
+	github.com/go-logr/stdr v1.2.2 // indirect
+	github.com/goccy/go-yaml v1.17.1 // indirect
+	github.com/golang/groupcache v0.0.0-20210331224755-41bb18bfe9da // indirect
+	github.com/google/go-cmp v0.7.0 // indirect
+	github.com/google/jsonschema-go v0.3.0 // indirect
+	github.com/google/s2a-go v0.1.9 // indirect
+	github.com/google/uuid v1.6.0 // indirect
+	github.com/googleapis/enterprise-certificate-proxy v0.3.6 // indirect
+	github.com/googleapis/gax-go/v2 v2.15.0 // indirect
+	github.com/gorilla/mux v1.8.1 // indirect
+	github.com/gorilla/websocket v1.5.3 // indirect
+	github.com/mattn/go-colorable v0.1.13 // indirect
+	github.com/mattn/go-isatty v0.0.19 // indirect
+	github.com/mattn/go-runewidth v0.0.16 // indirect
+	github.com/mitchellh/mapstructure v1.5.0 // indirect
+	github.com/olekukonko/errors v1.1.0 // indirect
+	github.com/olekukonko/ll v0.0.9 // indirect
+	github.com/olekukonko/tablewriter v1.0.9 // indirect
+	github.com/planetscale/vtprotobuf v0.6.1-0.20240319094008-0393e58bdf10 // indirect
+	github.com/rivo/uniseg v0.2.0 // indirect
+	github.com/samber/lo v1.47.0 // indirect
+	github.com/spiffe/go-spiffe/v2 v2.6.0 // indirect
+	go.opencensus.io v0.24.0 // indirect
+	go.opentelemetry.io/auto/sdk v1.2.1 // indirect
+	go.opentelemetry.io/contrib/detectors/gcp v1.38.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/google.golang.org/grpc/otelgrpc v0.63.0 // indirect
+	go.opentelemetry.io/contrib/instrumentation/net/http/otelhttp v0.63.0 // indirect
+	go.opentelemetry.io/otel v1.38.0 // indirect
+	go.opentelemetry.io/otel/metric v1.38.0 // indirect
+	go.opentelemetry.io/otel/sdk v1.38.0 // indirect
+	go.opentelemetry.io/otel/sdk/metric v1.38.0 // indirect
+	go.opentelemetry.io/otel/trace v1.38.0 // indirect
+	golang.org/x/crypto v0.43.0 // indirect
+	golang.org/x/exp v0.0.0-20230522175609-2e198f4a06a1 // indirect
+	golang.org/x/net v0.46.0 // indirect
+	golang.org/x/oauth2 v0.32.0 // indirect
+	golang.org/x/sync v0.17.0 // indirect
+	golang.org/x/sys v0.37.0 // indirect
+	golang.org/x/text v0.30.0 // indirect
+	golang.org/x/time v0.14.0 // indirect
+	google.golang.org/genproto v0.0.0-20251014184007-4626949a642f // indirect
+	google.golang.org/genproto/googleapis/api v0.0.0-20251014184007-4626949a642f // indirect
+	google.golang.org/genproto/googleapis/rpc v0.0.0-20251014184007-4626949a642f // indirect
+	google.golang.org/grpc v1.76.0 // indirect
+	google.golang.org/protobuf v1.36.10 // indirect
+	rsc.io/omap v1.2.0 // indirect
+	rsc.io/ordered v1.1.1 // indirect
+)