0基础Go语言Eino框架智能体实战-chatModel
0基础Go语言Eino框架智能体实战-chatModel摘要本文详细介绍如何使用Go语言、Eino框架和Gin框架构建一个完整的智能聊天服务。涵盖环境搭建、大模型调用、API创建、日志输出、异常处理等核心知识点适合零基础入门人工智能应用开发源码在此。一、项目概述1.1 技术栈技术说明版本Go服务端开发语言1.21Eino字节跳动开源的AI应用框架最新Gin高性能HTTP Web框架最新DeepSeek大模型提供商-1.2 功能特性基于DeepSeek大模型的智能对话RESTful API接口设计Swagger UI在线调试日志输出到文件或标准输出完整的异常捕获机制1.3 最终效果┌─────────────────────────────────────────────────────┐ │ 智能聊天服务 │ ├─────────────────────────────────────────────────────┤ │ POST /chat 聊天接口 │ │ GET /health 健康检查 │ │ GET /swagger/*any API文档 │ └─────────────────────────────────────────────────────┘二、环境准备2.1 安装Go环境首先确保已安装Go语言环境# 检查Go版本go version# 输出示例go version go1.21.6 darwin/arm642.2 安装swag工具swag用于生成Swagger API文档# 安装swag命令行工具goinstallgithub.com/swaggo/swag/cmd/swaglatest# 验证安装~/go/bin/swag--version# v1.16.32.3 创建项目结构mkdir-p~/projects/chat-servicecd~/projects/chat-servicemkdir-pdocs# Swagger文档目录三、配置文件3.1 创建config.yml在项目根目录创建配置文件model:base_url:https://api.minimaxi.com/v1# API基础地址api_key:your-api-key-here# API密钥model_name:MiniMax-M2.7# 模型名称app:host:0.0.0.0# 监听地址port:8080# 监听端口3.2 Go配置结构体定义// Config 模型配置typeConfigstruct{Model ModelConfigyaml:modelApp AppConfigyaml:app}// ModelConfig 大模型配置typeModelConfigstruct{BaseURLstringyaml:base_urlAPIKeystringyaml:api_keyModelNamestringyaml:model_name}// AppConfig 应用配置typeAppConfigstruct{Hoststringyaml:hostPortintyaml:port}3.3 读取配置文件import(fmtosgopkg.in/yaml.v2)funcloadConfig(configPathstring)(*Config,error){// 读取配置文件内容data,err:os.ReadFile(configPath)iferr!nil{returnnil,fmt.Errorf(读取配置文件失败: %w,err)}// 解析YAML格式的配置varconfig Configiferr:yaml.Unmarshal(data,config);err!nil{returnnil,fmt.Errorf(解析配置文件失败: %w,err)}returnconfig,nil}关键点说明yaml.Unmarshal()自动将YAML字段映射到Go结构体yaml:base_url标签指定YAML中的字段名Go字段名可以不同驼峰但标签必须匹配YAML四、Eino框架创建大模型聊天服务4.1 初始化聊天模型Eino框架是字节跳动开源的AI应用框架提供了统一的大模型调用接口import(contextgithub.com/cloudwego/eino-ext/components/model/deepseekgithub.com/cloudwego/eino/schema)varchatModel*deepseek.ChatModel// 全局聊天模型实例funcinitChatModel(cfg*Config)error{ctx:context.Background()// 创建DeepSeek聊天模型实例model,err:deepseek.NewChatModel(ctx,deepseek.ChatModelConfig{APIKey:cfg.Model.APIKey,// API密钥Model:cfg.Model.ModelName,// 模型名称BaseURL:cfg.Model.BaseURL,// API基础地址})iferr!nil{returnfmt.Errorf(创建ChatModel实例失败: %w,err)}chatModelmodelreturnnil}4.2 聊天请求与响应结构体// ChatRequest 聊天请求typeChatRequeststruct{// SystemPrompt 系统提示词用于设定AI角色和行为SystemPromptstringjson:system_prompt example:你是一个有帮助的助手 description:系统提示词// UserMessage 用户消息即用户的输入UserMessagestringjson:user_message example:你好 description:用户消息}// ChatResponse 聊天响应typeChatResponsestruct{// Content 模型回复内容Contentstringjson:content example:你好有什么可以帮助你的吗 description:模型回复内容// PromptTokens 输入token数量PromptTokensintjson:prompt_tokens example:100 description:输入token数量// OutputTokens 输出token数量OutputTokensintjson:output_tokens example:50 description:输出token数量// TotalTokens 总token数量TotalTokensintjson:total_tokens example:150 description:总token数量}4.3 聊天处理函数import(net/httptime)funcchatHandler(c*gin.Context){varreq ChatRequest// 绑定并验证JSON请求体iferr:c.ShouldBindJSON(req);err!nil{c.JSON(http.StatusBadRequest,gin.H{error:无效的请求参数})return}// 构建消息列表系统提示 用户消息messages:[]*schema.Message{schema.SystemMessage(req.SystemPrompt),schema.UserMessage(req.UserMessage),}// 创建带超时的context防止模型调用耗时过长ctx,cancel:context.WithTimeout(c.Request.Context(),30*time.Second)defercancel()// 调用聊天模型生成响应response,err:chatModel.Generate(ctx,messages)iferr!nil{c.JSON(http.StatusInternalServerError,gin.H{error:生成响应失败请稍后重试})return}// 构建响应结构resp:ChatResponse{Content:response.Content,}// 提取token使用统计信息ifresponse.ResponseMeta!nilresponse.ResponseMeta.Usage!nil{resp.PromptTokensresponse.ResponseMeta.Usage.PromptTokens resp.OutputTokensresponse.ResponseMeta.Usage.CompletionTokens resp.TotalTokensresponse.ResponseMeta.Usage.TotalTokens}c.JSON(http.StatusOK,resp)}4.4 消息类型说明函数说明使用场景schema.SystemMessage(content)系统消息设置AI角色和行为schema.UserMessage(content)用户消息用户输入schema.AssistantMessage(content)助手消息对话历史schema.ToolMessage(content, toolCallID)工具消息工具调用结果五、Gin框架创建API服务5.1 Gin基本使用Gin是一个用Go语言编写的高性能HTTP Web框架importgithub.com/gin-gonic/ginfuncmain(){// 创建Gin实例r:gin.Default()// 注册路由r.GET(/ping,func(c*gin.Context){c.JSON(200,gin.H{message:pong})})// 启动服务r.Run(:8080)}5.2 健康检查接口funchealthHandler(c*gin.Context){c.JSON(http.StatusOK,gin.H{status:ok})}5.3 注册完整路由funcmain(){// 使用ReleaseMode减少日志输出gin.SetMode(gin.ReleaseMode)r:gin.New()// 注册HTTP处理函数r.GET(/health,healthHandler)// 健康检查r.POST(/chat,chatHandler)// 聊天接口// 启动服务addr:fmt.Sprintf(%s:%d,cfg.App.Host,cfg.App.Port)log.Printf(服务启动中监听地址: %s,addr)r.Run(addr)}5.4 Swagger文档集成第一步添加Swagger注释// chatHandler 处理聊天请求//// Summary 聊天接口// Description 与DeepSeek大模型对话// Tags chat// Accept json// Produce json// Param request body ChatRequest true 聊天请求// Success 200 {object} ChatResponse// Failure 400 {object} map[string]string// Failure 500 {object} map[string]string// Router /chat [post]funcchatHandler(c*gin.Context){// ...}第二步生成Swagger文档swag init-gchat_quickstart.go第三步注册Swagger路由import(swaggerFilesgithub.com/swaggo/filesginSwaggergithub.com/swaggo/gin-swagger_your_module/lab01/docs// 导入自动生成的docs包)funcmain(){r:gin.Default()r.GET(/swagger/*any,ginSwagger.WrapHandler(swaggerFiles.Handler))}常用Swagger注释说明注释说明Summary接口概要Description接口详细描述Tags接口分组标签Param请求参数Success成功响应Failure失败响应Router路由路径六、Go语言日志输出6.1 标准日志库使用Go语言内置了log包提供基本的日志功能importlogfuncmain(){// 设置日志输出格式log.SetFlags(log.LstdFlags|log.Lshortfile)// 输出日志log.Println(这是一条普通日志)log.Printf(用户%s的操作失败错误码: %d,张三,500)log.Fatalf(严重错误: %v,err)// 输出后程序退出}6.2 日志输出到文件通过命令行参数指定日志文件import(flagos)funcmain(){// 解析命令行参数logFile:flag.String(log,,日志输出文件路径)flag.Parse()// 设置日志输出if*logFile!{f,err:os.OpenFile(*logFile,os.O_CREATE|os.O_WRONLY|os.O_APPEND,0666)iferr!nil{log.Fatalf(打开日志文件失败: %v,err)}deferf.Close()log.SetOutput(f)log.SetFlags(log.LstdFlags|log.Lshortfile)log.Printf(日志将输出到文件: %s,*logFile)}else{log.SetFlags(log.LstdFlags|log.Lshortfile)log.Println(日志将输出到标准输出)}}6.3 日志使用示例# 输出到标准输出默认go run chat_quickstart.go# 输出到指定日志文件go run chat_quickstart.go-logapp.log# 输出到指定日志文件绝对路径go run chat_quickstart.go-log/var/log/chat_service.log6.4 日志格式说明2026/04/07 10:30:15 main.go:45: 配置加载成功2026/04/07 10:30:15- 日期和时间main.go:45- 代码位置文件名:行号日志内容6.5 日志Flag说明Flag说明Ldate日期2009/01/23Ltime时间01:23:23Lshortfile完整路径和行号Llongfile完整路径和行号LstdFlags标准格式日期时间七、Go语言异常捕获与抛出7.1 Go异常机制概述Go语言使用panic和recover机制处理异常不同于传统语言的try-catch// panic: 触发异常中断程序执行panic(这是一个严重错误)// recover: 捕获panic防止程序崩溃deferfunc(){ifr:recover();r!nil{fmt.Println(捕获到异常:,r)}}()7.2 Panic Recovery中间件为了防止服务器崩溃我们需要一个中间件来捕获所有未处理的panicimportruntimefuncrecoveryMiddleware()gin.HandlerFunc{returnfunc(c*gin.Context){deferfunc(){ifr:recover();r!nil{// 记录panic到日志log.Printf([PANIC RECOVERED] 异常信息: %v\n堆栈跟踪:\n%s,r,getStackTrace())// 返回内部错误响应c.JSON(http.StatusInternalServerError,gin.H{error:服务器内部错误请稍后重试,})c.Abort()}}()c.Next()}}// getStackTrace 获取当前goroutine的堆栈跟踪信息funcgetStackTrace()string{varbuf[4096]byten:runtime.Stack(buf[:],false)returnstring(buf[:n])}7.3 异常抛出函数提供一个辅助函数方便在错误发生时主动抛出异常funcpanicIfErr(errerror,msgstring){iferr!nil{log.Printf([PANIC THROW] %s: %v,msg,err)panic(fmt.Sprintf(%s: %v,msg,err))}}7.4 使用示例示例一在初始化阶段使用funcmain(){// 初始化聊天模型iferr:initChatModel(cfg);err!nil{panicIfErr(err,初始化聊天模型失败)}}示例二在业务逻辑中主动抛出funcchatHandler(c*gin.Context){response,err:chatModel.Generate(ctx,messages)iferr!nil{log.Printf([ERROR] 模型调用失败: %v,err)panic(chat model generation failed)}// ...}7.5 中间件注册funcmain(){gin.SetMode(gin.ReleaseMode)r:gin.New()// 注册panic恢复中间件r.Use(recoveryMiddleware())// 注册路由r.GET(/health,healthHandler)r.POST(/chat,chatHandler)}7.6 日志中的异常标记标记含义[PANIC RECOVERED]panic被recoveryMiddleware捕获[PANIC THROW]panic被panicIfErr函数主动抛出[ERROR]普通错误日志7.7 异常日志输出示例[PANIC RECOVERED] 异常信息: chat model generation failed 堆栈跟踪: goroutine 8 [running]: main.chatHandler(0xc0000a2000) /path/to/chat_quickstart.go:143 ...八、完整代码整合8.1 main函数完整实现funcmain(){// -------------------- 0. 解析命令行参数 --------------------logFile:flag.String(log,,日志输出文件路径)flag.Parse()if*logFile!{f,err:os.OpenFile(*logFile,os.O_CREATE|os.O_WRONLY|os.O_APPEND,0666)iferr!nil{log.Fatalf(打开日志文件失败: %v,err)}deferf.Close()log.SetOutput(f)log.SetFlags(log.LstdFlags|log.Lshortfile)}else{log.SetFlags(log.LstdFlags|log.Lshortfile)}// -------------------- 1. 加载配置 --------------------cfg,err:loadConfig(config.yml)iferr!nil{log.Fatalf(加载配置失败: %v,err)}log.Printf(配置加载成功: base_url%s, model%s,cfg.Model.BaseURL,cfg.Model.ModelName)// -------------------- 2. 初始化聊天模型 --------------------iferr:initChatModel(cfg);err!nil{log.Fatalf(初始化聊天模型失败: %v,err)}log.Println(聊天模型初始化成功)// -------------------- 3. 设置Gin路由 --------------------gin.SetMode(gin.ReleaseMode)r:gin.New()r.Use(recoveryMiddleware())r.GET(/health,healthHandler)r.POST(/chat,chatHandler)r.GET(/swagger/*any,ginSwagger.WrapHandler(swaggerFiles.Handler))// -------------------- 4. 启动服务 --------------------addr:fmt.Sprintf(%s:%d,cfg.App.Host,cfg.App.Port)log.Printf(服务启动中监听地址: %s,addr)iferr:r.Run(addr);err!nil{log.Fatalf(服务启动失败: %v,err)}}8.2 项目目录结构lab01/ ├── chat_quickstart.go # 主程序 ├── config.yml # 配置文件 ├── go.mod # Go模块文件 ├── go.sum # 依赖校验文件 └── docs/ # Swagger文档 ├── docs.go ├── swagger.json └── swagger.yaml九、启动与测试9.1 启动服务# 生成Swagger文档~/go/bin/swag init-gchat_quickstart.go# 启动服务输出到标准输出go run chat_quickstart.go# 启动服务输出到日志文件go run chat_quickstart.go-logapp.log9.2 测试接口# 健康检查curlhttp://localhost:8080/health# 聊天接口curl-XPOST http://localhost:8080/chat\-HContent-Type: application/json\-d{system_prompt: 你是一个有帮助的助手, user_message: 你好}9.3 访问Swagger UI打开浏览器访问http://localhost:8080/swagger/index.html十、总结本文详细介绍了如何使用Go语言、Eino框架和Gin框架构建一个完整的智能聊天服务主要包括Eino框架使用通过deepseek.NewChatModel创建大模型实例调用Generate方法生成对话Gin框架使用创建路由、注册中间件、绑定请求参数、返回JSON响应日志输出使用flag解析命令行参数通过os.OpenFile将日志输出到文件异常处理使用recover捕获panic实现recoveryMiddleware中间件防止服务器崩溃通过本文的学习你应该能够掌握Go语言Web服务开发AI大模型调用日志与异常处理API文档生成参考资料Eino框架官方文档Gin框架官方文档Swag官方文档作者[Your Name]首发于CSDN转载请注明出处
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2499400.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!