Function main

example/main.go:24–110

()

)

func main() {
	ctx := context.Background()
	logger := slog.Make(sloghuman.Sink(os.Stderr)).Leveled(slog.LevelDebug)

	// Initialize SQLite database with WAL mode for better concurrency.
	db, err := sql.Open("sqlite", "aibridge.db?_journal_mode=WAL&_busy_timeout=5000")
	if err != nil {
		logger.Fatal(ctx, "open database", slog.Error(err))
	}
	defer db.Close()
	db.SetMaxOpenConns(1) // SQLite only supports one writer at a time.

	if err := initSchema(db); err != nil {
		logger.Fatal(ctx, "init schema", slog.Error(err))
	}

	recorder, err := NewSQLiteRecorder(db, logger)
	if err != nil {
		logger.Fatal(ctx, "create recorder", slog.Error(err))
	}
	defer recorder.Close()

	// Configure providers.
	providers := []aibridge.Provider{
		aibridge.NewAnthropicProvider(aibridge.AnthropicConfig{
			Key: os.Getenv("ANTHROPIC_API_KEY"),
		}, nil),
		aibridge.NewOpenAIProvider(aibridge.OpenAIConfig{
			Key: os.Getenv("OPENAI_API_KEY"),
		}),
	}

	// Setup metrics.
	reg := prometheus.NewRegistry()
	metrics := aibridge.NewMetrics(reg)

	// Setup tracing
	tracer := otel.GetTracerProvider().Tracer("exampleTracer")

	// Optional: Configure MCP server for centralized tool injection.
	// DeepWiki provides free access to public GitHub repo documentation.
	// See: https://mcp.deepwiki.com
	var mcpProxy mcp.ServerProxier

	deepwikiProxy, err := mcp.NewStreamableHTTPServerProxy(
		"deepwiki",                           // server name (tools prefixed as bmcp_deepwiki_*)
		"https://mcp.deepwiki.com/mcp",       // no auth required for public repos
		nil,                                  // headers
		regexp.MustCompile(`^ask_question$`), // allowlist: only ask_question tool
		nil,                                  // denylist
		logger.Named("mcp.deepwiki"),
		tracer,
	)
	if err != nil {
		logger.Fatal(ctx, "create deepwiki mcp proxy", slog.Error(err))
	}

	mcpProxy = mcp.NewServerProxyManager(map[string]mcp.ServerProxier{"deepwiki": deepwikiProxy}, tracer)