Repository navigation
Expand file tree
/
Copy pathmain.go
More file actions
70 lines (60 loc) · 2.13 KB
/
Copy pathmain.go
File metadata and controls
70 lines (60 loc) · 2.13 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
// Copyright 2026 Google LLC
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Command gemini-search-mcp is an MCP stdio server exposing a web_search tool
// backed by Gemini + Google Search grounding (Vertex AI or AI Studio).
package main
import (
"context"
"flag"
"fmt"
"log"
"time"
"github.com/modelcontextprotocol/go-sdk/mcp"
"github.com/cwest/gemini-search-mcp/internal/config"
"github.com/cwest/gemini-search-mcp/internal/search"
"github.com/cwest/gemini-search-mcp/internal/server"
"github.com/cwest/gemini-search-mcp/internal/version"
)
func main() {
showVersion := flag.Bool("version", false, "print version and exit")
flag.Parse()
if *showVersion {
fmt.Printf("gemini-search-mcp %s (commit %s, built %s)\n", version.Version, version.Commit, version.Date)
return
}
cfg, err := config.Load()
if err != nil {
log.Fatalf("config: %v", err)
}
log.Printf("gemini-search-mcp %s: provider=%s model=%s grounding=%s", version.Version, cfg.Provider, cfg.Model, cfg.GroundingMode)
ctx := context.Background()
sc, err := search.New(ctx, cfg.Model, cfg.GroundingMode)
if err != nil {
log.Fatalf("init search: %v", err)
}
srv := server.New(timeoutSearcher{inner: sc, timeout: cfg.Timeout})
if err := srv.Run(ctx, &mcp.StdioTransport{}); err != nil {
log.Fatalf("server run: %v", err)
}
}
// timeoutSearcher wraps a Searcher with a per-call deadline.
type timeoutSearcher struct {
inner search.Searcher
timeout time.Duration
}
func (t timeoutSearcher) Search(ctx context.Context, query string) (*search.Result, error) {
ctx, cancel := context.WithTimeout(ctx, t.timeout)
defer cancel()
return t.inner.Search(ctx, query)
}