使用golang快速构建你的MCP Server

MCP golang sdk

目前官方并没有go版本的sdk,我们选择的是https://github.com/mark3labs/mcp-go,这个项目✨已超过3000。

MCP Server 代码

这里我们构建一个Milvus助手,让他查询有哪些数据库,具体示例代码如下:

golang 复制代码
package main

import (
    "context"
    "errors"
    "fmt"
    "github.com/milvus-io/milvus-sdk-go/v2/client"
    "log"

    "github.com/mark3labs/mcp-go/mcp"
    "github.com/mark3labs/mcp-go/server"
)

var milvusDb client.Client

func init() {
    ctx := context.Background()
    config := client.Config{
       Address: "localhost:19530",
       //DBName:  "my_database",
    }
    var err error
    milvusDb, err = client.NewClient(ctx, config)
    if err != nil {
       log.Fatal("connect err:", err)
    }
}

func main() {
    // Create MCP server
    s := server.NewMCPServer(
       "Milvus 🚀",
       "1.0.0",
    )

    // Add tool
    tool := mcp.NewTool("list_database",
       mcp.WithDescription("Get all database of milvus"))

    // Add tool handler
    s.AddTool(tool, ListDatabaseHandler)

    // Start the stdio server
    if err := server.ServeStdio(s); err != nil {
       fmt.Printf("Server error: %v\n", err)
    }
}

func ListDatabaseHandler(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
    databases, err := milvusDb.ListDatabases(ctx)
    if err != nil {
       return nil, err
    }
    cts := []mcp.Content{}
    for _, database := range databases {
       content := mcp.NewTextContent(database.Name)
       cts = append(cts, content)
    }
    return &mcp.CallToolResult{
       Content: cts,
    }, nil
}

构建完成后使用go build -o mcp_test mian.go编译出mcp_test文件,之后就可以测试这个server了。

测试Server

这里我们使用mark3labs的另外一个项目https://github.com/mark3labs/mcphost, mcphost支持本地ollama模型,我们使用Llama 3.1这个模型,需要注意的是,模型需要支持tools调用才行。

配置

ruby 复制代码
{
    "mcpServers": {
       "milvus":{
          "command": "/Users/jhonroxton/Desktop/code/golang/mcp-server-milvus/mcp_test", //路径替换成你自己的
          "args":[]
       }
  }
}

启动ollama测试

使用命令./mcphost -m ollama:llama3.1:latest启动项目

可以看到成功连接到了milvus,让它以表格形式返回milvus中有哪些数据库

可以看到调用了list_database这个tool,返回了defaultrag_test,而这两个和实际的也是对应上的:

总结

本文完整的展示了使用golang构建一个mcp server的示例,主要分四步:

  • 1.创建一个server
  • 2.向server注册tools
  • 3.为对应的tool添加相应的handler
  • 4.启动server

你也可以根据自身需求做扩展,构建出符合自己需求的servermcp在大型模型应用中展现出巨大的潜力和广阔的发展前景,希望本文能够为您的学习和了解提供有价值的参考。

相关推荐
ResponsibilityAmbiti4 小时前
AI 发展 && MCP
人工智能·llm·aigc
AI大模型5 小时前
LangGraph官方文档笔记(6)——时间旅行
程序员·langchain·llm
小雷FansUnion15 小时前
深入理解MCP架构:智能服务编排、上下文管理与动态路由实战
人工智能·架构·大模型·mcp
CoderLiu20 小时前
用这个MCP,只给大模型一个figma链接就能直接导出图片,还能自动压缩上传?
前端·llm·mcp
吴佳浩1 天前
Python入门指南-AI番外-MCP完整教程:从零开始学会Model Context Protocol
人工智能·python·mcp
智泊AI1 天前
大语言模型LLM底层技术原理到底是什么?大型语言模型如何工作?
llm
moonless02221 天前
🌈Transformer说人话版(二)位置编码 【持续更新ing】
人工智能·llm
小爷毛毛_卓寿杰1 天前
基于大模型与知识图谱的对话引导意图澄清系统技术解析
人工智能·llm
聚客AI1 天前
解构高效提示工程:分层模型、文本扩展引擎与可视化调试全链路指南
人工智能·llm·掘金·日新计划
聚客AI1 天前
🚀拒绝试错成本!企业接入MCP协议的避坑清单
人工智能·掘金·日新计划·mcp