{"id":1578,"date":"2024-05-23T17:35:47","date_gmt":"2024-05-23T17:35:47","guid":{"rendered":"https:\/\/www.nicekj.com\/?p=1578"},"modified":"2024-05-23T17:35:47","modified_gmt":"2024-05-23T17:35:47","slug":"jiaonishiyong-go-yuyanfangwenzhipu-ai-damoxing-2","status":"publish","type":"post","link":"https:\/\/www.nicekj.com\/jiaonishiyong-go-yuyanfangwenzhipu-ai-damoxing-2.html","title":{"rendered":"\u6559\u4f60\u4f7f\u7528 Go \u8bed\u8a00\u8bbf\u95ee\u667a\u8c31 AI \u5927\u6a21\u578b\uff01"},"content":{"rendered":"

\u6700\u8fd1\u6587\u7ae0\u5feb\u51c9\u51c9\u4e86\uff0c\u5927\u5bb6\u8bb0\u5f97\u7ed9\u6587\u7ae0\u70b9\u8d5e\u652f\u6301\uff5e\u8c22\u8c22\u5927\u5bb6\u7684\u559c\u6b22\uff01\u5173\u6ce8\u6211\uff0c\u5206\u4eab\u6709\u8da3\u7684\u77e5\u8bc6\u5171\u540c\u5b66\u4e60\uff01<\/strong><\/p>\n

AI \u5927\u6a21\u578b\u592a\u706b\u7206\u4e86\uff01\u6211\u5728\u5de5\u4f5c\u4e2d\u7ecf\u5e38\u4f7f\u7528\u5b83\u4eec\uff0c\u611f\u89c9\u5df2\u7ecf\u79bb\u4e0d\u5f00\u4e86\uff01<\/p>\n

\u6700\u8fd1\u6d4b\u8bd5\u4e86\u667a\u8c31 AI \u5927\u6a21\u578b\uff0c\u5b9e\u6d4b\u4e0b\u6765\u611f\u89c9\u8fd8\u633a\u4e0d\u9519\u7684\u3002<\/p>\n

\u5b98\u65b9\u63d0\u4f9b\u4e86\u4f53\u9a8c\u5305\uff0c\u6ce8\u518c\u5e76\u5b9e\u540d\u8ba4\u8bc1\u4e4b\u540e\u6709 300 \u4e07 token\uff0c\u975e\u5e38\u58d5\uff01<\/p>\n

\u4eca\u5929\u7684\u80a1\u5e02\u592a\u7eff\u4e86\uff0c\u6211\u8ba9\u5b83\u7ed9\u6211\u5199\u4e2a\u6587\u6848\uff0c\u6ca1\u60f3\u5230\u8fd8\u7ed9\u80a1\u6c11\u6309\u6469\u4e86\u54c8\u54c8\uff01<\/p>\n

\n

\u4f60\u662f\u4e00\u540d\u6587\u5b66\u5bb6\uff0c\u8bf7\u4f60\u4e3a\u6211\u521b\u4f5c\u6587\u6848\uff0c\u4e3b\u9898\u662f 2024 \u5e74\uff0cA \u80a1\u8d70\u52bf\u7eff\u5f97\u53ef\u6015\u3002\u8981\u6c42\uff1a\u8bed\u8a00\u6d41\u7545\u901a\u987a\uff0c\u89c2\u70b9\u72ec\u7279\uff0c\u884c\u6587\u98ce\u9a9a\u3002<\/p>\n

2024\u5e74\uff0cA\u80a1\u8d70\u52bf\u7eff\u610f\u76ce\u7136\uff0c\u604d\u82e5\u6625\u65e5\u91cc\u7684\u7fe0\u7af9\uff0c\u8282\u8282\u8d25\u9000\u3002\u90a3\u62b9\u7eff\uff0c\u5982\u540c\u4e00\u5e45\u6cfc\u58a8\u5c71\u6c34\uff0c\u6325\u6beb\u4e4b\u95f4\uff0c\u60ca\u5fc3\u52a8\u9b44\u3002
\n\u8fd9\u4e00\u5e74\uff0c\u80a1\u5e02\u7684\u7eff\u610f\u72b9\u5982\u6c5f\u5357\u7684\u6885\u96e8\uff0c\u7ef5\u5ef6\u4e0d\u7edd\uff0c\u6295\u8d44\u8005\u4eec\u7684\u5fc3\u60c5\u4e5f\u968f\u7740\u8fd9\u7eff\u610f\uff0c\u8d77\u4f0f\u4e0d\u5b9a\u3002\u6709\u4eba\u8c03\u4f83\uff1a\u201c\u7eff\u201d\u6210\u4e86\u5fc3\u5934\u5927\u60a3\uff0c\u4e5f\u6709\u4eba\u620f\u8a00\uff1a\u201c\u8981\u60f3\u751f\u6d3b\u8fc7\u5f97\u53bb\uff0c\u5934\u4e0a\u603b\u5f97\u5e26\u70b9\u7eff\u3002\u201d
\n\u7136\u800c\uff0c\u5728\u8fd9\u7247\u7eff\u6d77\u4e4b\u4e2d\uff0c\u667a\u8005\u770b\u5230\u4e86\u673a\u9047\uff0c\u52c7\u8005\u6355\u6349\u5230\u4e86\u9006\u88ad\u7684\u66d9\u5149\u3002\u7eff\u5f97\u53ef\u6015\uff0c\u6070\u4f3c\u98ce\u96e8\u524d\u7684\u4e4c\u4e91\uff0c\u9884\u793a\u7740\u4e00\u573a\u7ffb\u5929\u8986\u5730\u7684\u53d8\u9769\u5373\u5c06\u6765\u4e34\u3002
\n\u98ce\u9a9a\u72ec\u6b65\u7684\u6295\u8d44\u8005\uff0c\u628a\u8fd9\u7247\u7eff\u8272\u89c6\u4e3a\u4e00\u79cd\u6311\u6218\uff0c\u4e00\u79cd\u78e8\u783a\u3002\u4ed6\u4eec\u6df1\u77e5\uff0c\u5728\u8fd9\u7247\u7eff\u610f\u76ce\u7136\u7684A\u80a1\u5e02\u573a\uff0c\u552f\u6709\u72ec\u5177\u6167\u773c\uff0c\u65b9\u80fd\u6d1e\u5bdf\u5148\u673a\uff0c\u628a\u63e1\u90a3\u7a0d\u7eb5\u5373\u901d\u7684\u76c8\u5229\u673a\u4f1a\u3002
\n\u7eff\u5f97\u53ef\u6015\uff0c\u662f\u56e0\u4e3a\u5e02\u573a\u5728\u6d17\u724c\uff0c\u5728\u91cd\u5851\u3002\u90a3\u4e9b\u7ad9\u5728\u98ce\u53e3\u6d6a\u5c16\u7684\u4f01\u4e1a\uff0c\u6216\u5c06\u6d85\u69c3\u91cd\u751f\uff0c\u6216\u5c06\u9eef\u7136\u79bb\u573a\u3002\u800c\u4f5c\u4e3a\u6295\u8d44\u8005\u7684\u6211\u4eec\uff0c\u552f\u6709\u7d27\u8ddf\u65f6\u4ee3\u6b65\u4f10\uff0c\u62e5\u62b1\u53d8\u9769\uff0c\u65b9\u80fd\u5728\u8fd9\u7247\u7eff\u8272\u7684\u80a1\u5e02\u4e2d\uff0c\u7834\u8327\u6210\u8776\uff0c\u7fe9\u7fe9\u8d77\u821e\u3002
\n2024\u5e74\uff0cA\u80a1\u8d70\u52bf\u7eff\u5f97\u53ef\u6015\uff0c\u4f46\u8fd9\u7eff\u8272\u80cc\u540e\uff0c\u5374\u8574\u85cf\u7740\u65e0\u5c3d\u7684\u673a\u9047\u4e0e\u6311\u6218\u3002\u8ba9\u6211\u4eec\u4ee5\u72ec\u7279\u7684\u89c6\u89d2\uff0c\u5ba1\u89c6\u8fd9\u7247\u7eff\u8272\u7684\u80a1\u5e02\uff0c\u643a\u624b\u5171\u8fdb\uff0c\u4e66\u5199\u5c5e\u4e8e\u6211\u4eec\u7684\u8f89\u714c\u7bc7\u7ae0\u3002<\/p>\n<\/blockquote>\n

\u6211\u5e73\u65f6\u4f7f\u7528 Go \u8bed\u8a00\uff0c\u4f46\u6211\u53d1\u73b0\u5b98\u65b9\u53ea\u7ed9\u4e86 Python \u548c Java \u7684\u63a5\u53e3\uff01<\/p>\n

Python\uff1a github.com\/zhipuai\/zhi\u2026<\/a><\/p>\n

Java\uff1a github.com\/zhipuai\/zhi\u2026<\/a><\/p>\n

\u8fd9\u5bf9\u4e8e\u6211\u4eec Go \u8bed\u8a00\u5f00\u53d1\u8005\u6765\u8bf4\uff0c\u786e\u5b9e\u6709\u70b9\u4e0d\u65b9\u4fbf\u3002<\/p>\n

\u7ffb\u904d\u4e86\u5168\u7403\u6700\u5927\u7684\u7537\u6027\u4ea4\u53cb\u5e73\u53f0\u2014\u2014GitHub\uff01Go \u8bed\u8a00\u4e00\u4e2a SDK \u90fd\u6ca1\u6709\uff01\u96be\u9053\u8981\u81ea\u5df1\u624b\u5199\uff1f<\/p>\n

\u5bf9\u6211\u8fd9\u79cd\u559c\u6b22\u5077\u61d2\u7684\u4eba\u6765\u8bf4\uff0c\u662f\u4e0d\u53ef\u80fd\u81ea\u5df1\u5199\u7684\u3002<\/p>\n

\u6211\u574f\u4e86\u5341\u4e2a\u952e\u76d8\uff0c\u6bcf\u4e00\u4e2a\u90fd\u662f\u574f\u4e86 Ctrl C V \u8fd9\u4e09\u4e2a\u952e\u3002<\/strong><\/p>\n

\u7814\u7a76\u4e86\u4e0b\u63a5\u53e3\u6587\u6863\uff08\u667a\u8c31AI\u5f00\u653e\u5e73\u53f0<\/a>\uff09\uff0c\u53d1\u73b0\u667a\u8c31 AI \u7684 v4 \u7248\u672c\u63a5\u53e3\u548c OpenAI \u7684\u63a5\u53e3\u957f\u5f97\u633a\u50cf\u7684\uff01<\/p>\n

\u90a3\u5c31\u7528 go-openai \u8fd9\u4e2a\u5305\u8bd5\u8bd5\u6c34\uff01GitHub – sashabaranov\/go-openai: OpenAI ChatGPT, GPT-3, GPT-4, DALL\u00b7E, Whisper API wrapper for Go<\/a><\/p>\n

\u6298\u817e\u4e0b\u6765\uff0c\u53d1\u73b0\u4e24\u4e2a\u95ee\u9898\uff1a<\/p>\n

\u7b2c\u4e00\u4e2a\u95ee\u9898\u662f\uff0c\u667a\u8c31 AI \u7684\u63a5\u53e3\u8ba4\u8bc1\u8981\u7528 jwt \u751f\u6210 token\uff0c\u8fd9\u4e2a\u5728\u7f51\u4e0a\u4e00\u641c\uff0c\u6ca1\u4eba\u505a\u8fc7\u3002<\/p>\n

\u6839\u636e\u6587\u6863\uff0c\u6211\u81ea\u5df1\u5199\u4e86\u4e2a\u51fd\u6570\uff0c\u5f97\u5230 Authorization \u5934\u90e8\u540e\uff0c\u4f20\u7ed9 go-openai \u4f5c\u4e3a\u8bf7\u6c42\u7684 Header \u5934\u53d1\u51fa\u53bb\uff01<\/p>\n

<\/div>
go<\/span>
\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div>import<\/span> \"github.com\/golang-jwt\/jwt\/v5\"<\/span><\/span>\n<\/span>\n\/\/ apikey \u4e3a\u667a\u8c31\u63d0\u4f9b\u7684 API Key<\/span><\/span>\nfunc<\/span> splitAPIKey<\/span>(apikey string<\/span>)<\/span><\/span> (string<\/span>, string<\/span>) {<\/span>\n\tparts := strings.Split(apikey, \".\"<\/span>)<\/span>\n\tif<\/span> len<\/span>(parts) != 2<\/span> {<\/span>\n\t\treturn<\/span> \"\"<\/span>, \"\"<\/span><\/span>\n\t}<\/span>\n\treturn<\/span> parts[0<\/span>], parts[1<\/span>]<\/span>\n}<\/span>\n<\/span>\nfunc<\/span> getAuthorization<\/span>(authToken string<\/span>)<\/span><\/span> string<\/span> {<\/span>\n\tid, secret := splitAPIKey(authToken)<\/span>\n\tpayload := jwt.MapClaims{<\/span>\n\t\t\"api_key\"<\/span>:   id,<\/span>\n\t\t\"exp\"<\/span>:       time.Now().Add(3<\/span> * time.Minute).UnixMilli(),<\/span>\n\t\t\"timestamp\"<\/span>: time.Now().UnixMilli(),<\/span>\n\t}<\/span>\n\ttoken := jwt.NewWithClaims(jwt.SigningMethodHS256, payload)<\/span>\n\ttoken.Header[\"sign_type\"<\/span>] = \"SIGN\"<\/span><\/span>\n\tres, _ := token.SignedString([]byte<\/span>(secret))<\/span>\n\treturn<\/span> res<\/span>\n}<\/span>\n<\/code><\/pre>\n
\u7b2c\u4e8c\u4e2a\u95ee\u9898\u662f\uff0c\u4f60\u80fd\u901a\u8fc7\u8ba4\u8bc1\uff0c\u4f46\u8c03\u7528\u5931\u8d25\uff01\uff01\uff01<\/p>\n
\u6211\u6363\u9f13\u4e86\u4e00\u4e0b\u5348\uff0c\u603b\u7b97\u627e\u5230\u4e86\u7f6a\u9b41\u7978\u9996\uff01<\/p>\n
\u6392\u67e5\u7684\u65b9\u5f0f\u662f\uff0c\u5c06\u6211\u7684\u8bf7\u6c42\u590d\u5236\u6210 cURL \u5230\u7ec8\u7aef\uff0c\u53d1\u73b0\u662f\u80fd\u6b63\u5e38\u54cd\u5e94\u7684\uff01\uff01\uff01<\/p>\n
\u800c go-openai \u5305\u5728\u5e95\u5c42\u53d1\u9001\u8bf7\u6c42\u7684\u65f6\u5019\uff0c\u4f1a\u81ea\u5df1\u52a0\u4e0a\u4ee5\u4e0b\u5934\u90e8\uff1a<\/p>\n
<\/div>
go<\/span>
\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div>\treq.Header.Set(\"Content-Type\"<\/span>, \"application\/json\"<\/span>)<\/span>\n\treq.Header.Set(\"Accept\"<\/span>, \"text\/event-stream\"<\/span>)<\/span>\n\treq.Header.Set(\"Cache-Control\"<\/span>, \"no-cache\"<\/span>)<\/span>\n\treq.Header.Set(\"Connection\"<\/span>, \"keep-alive\"<\/span>)<\/span>\n<\/span>\n\tresp, err := client.config.HTTPClient.Do(req) \/\/nolint:bodyclose \/\/ body is closed in stream.Close()<\/span><\/span>\n\tif<\/span> err != nil<\/span> {<\/span>\n\t\treturn<\/span> new<\/span>(streamReader[T]), err<\/span>\n\t}<\/span>\n<\/code><\/pre>\n
\u667a\u8c31 AI \u670d\u52a1\u5668\u7aef\uff0c\u662f\u4e0d\u652f\u6301 Accept: text\/event-stream<\/code> \u7684\uff01\uff01\uff01<\/p>\n
\u611f\u5174\u8da3\u7684\u540c\u5b66\u53ef\u4ee5\u81ea\u5df1\u6d4b\u8bd5\u4e0b\uff0c\u53ea\u8981\u5e26\u4e0a Accept \u5934\u5c31\u4f1a\u62a5\u9519\uff0c\u5373\u4fbf\u8fd4\u56de\u63a5\u53e3\u5c31\u662f text\/event-stream<\/code>\uff0c\u539f\u56e0\u4e0d\u660e\u3002<\/p>\n
\u90a3\u8be5\u5982\u4f55\u89e3\u51b3\u8fd9\u4e2a\u95ee\u9898\u5462\uff1f\u6211\u53d1\u73b0 go-openai \u5305\u91cc\u7684 requestBuilder \u662f\u4e2a interface\uff0c\u4f46\u53ef\u60dc\u6ca1\u6709\u66b4\u9732\u51fa\u6765\uff01<\/p>\n
\u5982\u679c\u5b83\u66b4\u9732\u51fa\u6765\u4e86\uff0c\u6211\u53ef\u4ee5\u5957\u5a03\u540e\u66ff\u6362\u5b83\uff01\u4e0d\u8fc7\u8bf7\u6c42\u5934\u662f\u5728 requestBuilder \u4e4b\u540e\uff0c\u6240\u4ee5\u8fd9\u4e2a\u65b9\u6848\u53ea\u80fd\u653e\u5f03\u3002<\/p>\n
\u4f46\u5b83\u63d0\u4f9b\u4e86\u81ea\u5b9a\u4e49\u7684 HTTPClient\uff01<\/p>\n
\u800c HTTPClient \u6700\u7ec8\u4f1a\u8bf7\u6c42\u5230 client.config.HTTPClient.Do(req)<\/code> \u65b9\u6cd5\uff01<\/p>\n
\u6240\u4ee5\u7528 RoundTrip(*Request) (*Response, error)<\/code> \u63a5\u53e3\u622a\u83b7\u8bf7\u6c42\u5c31\u5b8c\u4e8b\u4e86\uff01<\/p>\n
\u600e\u4e48\u641e\u5462\uff1f\u7070\u5e38\u7b80\u5355\uff1a<\/p>\n
<\/div>
go<\/span>
\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div>type<\/span> removeAcceptHeaderRoundTrip struct<\/span>{}<\/span>\n<\/span>\nfunc<\/span> (s *removeAcceptHeaderRoundTrip)<\/span><\/span> RoundTrip(req *http.Request) (resp *http.Response, err error<\/span>) {<\/span>\n\treq.Header.Del(\"Accept\"<\/span>)<\/span>\n\tresp, err = http.DefaultTransport.RoundTrip(req)<\/span>\n\treturn<\/span><\/span>\n}<\/span>\n<\/code><\/pre>\n
\u7136\u540e\u521d\u59cb\u5316\u7684\u65f6\u5019\uff0c\u6362\u6210\u6211\u4eec\u7684 RoundTrip \u5c31\u597d\u4e86\uff01<\/p>\n
<\/div>
go<\/span>
\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div>\tcfg := openai.DefaultConfig(authorization)<\/span>\n\tcfg.BaseURL = \"https:\/\/open.bigmodel.cn\/api\/paas\/v4\"<\/span><\/span>\n\tcfg.HTTPClient.Transport = &removeAcceptHeaderRoundTrip{} \/\/ \u6362\u6210\u6211\u4eec\u7684 RoundTrip<\/span><\/span>\n\tcli := openai.NewClientWithConfig(cfg)<\/span>\n\tstream, err := cli.CreateChatCompletionStream(ctx, openai.ChatCompletionRequest{<\/span>\n\t\t\/\/ Model: \"glm-3-turbo\",<\/span><\/span>\n\t\tModel: \"glm-4\"<\/span>,<\/span>\n\t\tMessages: []openai.ChatCompletionMessage{<\/span>\n\t\t\t{Role: openai.ChatMessageRoleUser, Content: input},<\/span>\n\t\t},<\/span>\n\t})<\/span>\n<\/code><\/pre>\n
\u662f\u4e0d\u662f\u5f88\u60ca\u8273\uff01\uff01\uff01interface \u5927\u6cd5\u597d\uff01\u62bd\u8c61\u5f97\u5f53\uff0c\u6709\u65f6\u5019\u5199\u4ee3\u7801\u771f\u7684\u5c31\u662f\u51e0\u5206\u949f\u7684\u4e8b\u60c5\uff01\uff01\uff01<\/p>\n
\u4e0d\u5f97\u4e0d\u8bf4 Go \u7684\u6807\u51c6\u5e93\u662f\u771f\u7684\u8212\u670d\uff0c\u7b80\u5355\u7ffb\u4e00\u4e0b\u4ee3\u7801\uff0c\u5c31\u77e5\u9053\u600e\u4e48\u5199\u4e86\uff0c1 \u5206\u949f\u7684\u4e8b\u60c5\u5e72\u5b8c\u4e0b\u73ed\uff01\uff01\uff01<\/p>\n
\uff08\u5982\u679c\u89c9\u5f97\u6587\u7ae0\u6709\u5e2e\u52a9\uff0c\u5efa\u8bae\u6536\u85cf\u8d77\u6765\uff0c\u65b9\u4fbf\u4e0b\u6b21\u9605\u8bfb\uff09<\/strong><\/p>\n
\u4e4b\u524d\u5199\u6587\u7ae0\u7684\u65f6\u5019\uff0c\u6709\u4eba\u55b7\u6211\u7528 interface \u7528 wire \u628a\u4ee3\u7801\u641e\u5f97\u592a\u590d\u6742\u4e86\u2026\u2026<\/p>\n
\u4ec1\u626c\uff1a\u6211\u662f\u5982\u4f55\u7ec4\u7ec7 Go \u4ee3\u7801\u7684\uff08\u76ee\u5f55\u7ed3\u6784 \u4f9d\u8d56\u6ce8\u5165 wire\uff09<\/a><\/p>\n
\u73b0\u5728\u8fd9\u4e2a\u5c31\u662f\u6d3b\u751f\u751f\u7684\u4f8b\u5b50\u2026\u2026<\/strong><\/p>\n
\u5047\u5982\u6ca1\u6709\u66b4\u9732 RoundTrip \u63a5\u53e3\uff0c\u90a3\u4f60\u4ec0\u4e48\u90fd\u505a\u4e0d\u4e86\uff0c\u4f60\u53ea\u80fd\u786c\u8d28\u5934\u76ae\u81ea\u5df1\u5199 SDK \u5199\u53cd\u5c04\u5199\u5404\u79cd\u6076\u5fc3\u7684\u9ed1\u79d1\u6280\u2026\u2026<\/p>\n
\u603b\u4e4b\uff0c\u5199\u4ee3\u7801\u62bd\u8c61\u601d\u7ef4\u5f88\u91cd\u8981\uff0c\u9700\u8981\u65f6\u523b\u6ce8\u610f\u4ee3\u7801\u7684\u9886\u57df\u548c\u8fb9\u754c\uff01\uff01\uff01<\/strong><\/p>\n
\u591c\u5df2\u6df1\uff0c\u6211\u628a\u5b8c\u6574\u7684\u4ee3\u7801\u8d34\u4e0a\u6765\u5427\uff0c\u5927\u5bb6\u81ea\u884c\u98df\u7528\uff1a<\/p>\n
<\/div>
go<\/span>
\u590d\u5236\u4ee3\u7801<\/div><\/div><\/div>package<\/span> main<\/span>\n<\/span>\nimport<\/span> (<\/span>\n\t\"context\"<\/span><\/span>\n\t\"errors\"<\/span><\/span>\n\t\"fmt\"<\/span><\/span>\n\t\"io\"<\/span><\/span>\n\t\"net\/http\"<\/span><\/span>\n\t\"strings\"<\/span><\/span>\n\t\"time\"<\/span><\/span>\n<\/span>\n\t\"github.com\/golang-jwt\/jwt\/v5\"<\/span><\/span>\n\t\"github.com\/sashabaranov\/go-openai\"<\/span><\/span>\n)<\/span>\n<\/span>\nfunc<\/span> main<\/span>()<\/span><\/span> {<\/span>\n\tinput := `\u4f60\u662f\u4e00\u540d\u6587\u5b66\u5bb6\uff0c\u8bf7\u4f60\u4e3a\u6211\u521b\u4f5c\u6587\u6848\uff0c\u4e3b\u9898\u662f2024\u5e74\uff0cA\u80a1\u8d70\u52bf\u7eff\u5f97\u53ef\u6015\u3002\u8981\u6c42\uff1a\u8bed\u8a00\u6d41\u7545\u901a\u987a\uff0c\u89c2\u70b9\u72ec\u7279\uff0c\u884c\u6587\u98ce\u9a9a\u3002<\/span><\/span>\n`<\/span>\n\tctx := context.Background()<\/span>\n\tauthToken := \"......\"<\/span><\/span>\n\tauthorization := getAuthorization(authToken)<\/span>\n\tcfg := openai.DefaultConfig(authorization)<\/span>\n\tcfg.BaseURL = \"https:\/\/open.bigmodel.cn\/api\/paas\/v4\"<\/span><\/span>\n\tcfg.HTTPClient.Transport = &removeAcceptHeaderRoundTrip{} \/\/ \u66ff\u6362\u6210\u6211\u4eec\u81ea\u5df1\u7684<\/span><\/span>\n\tcli := openai.NewClientWithConfig(cfg)<\/span>\n\tstream, err := cli.CreateChatCompletionStream(ctx, openai.ChatCompletionRequest{<\/span>\n\t\t\/\/ Model: \"glm-3-turbo\",<\/span><\/span>\n\t\tModel: \"glm-4\"<\/span>,<\/span>\n\t\tMessages: []openai.ChatCompletionMessage{<\/span>\n\t\t\t{Role: openai.ChatMessageRoleUser, Content: input},<\/span>\n\t\t},<\/span>\n\t})<\/span>\n\tif<\/span> err != nil<\/span> {<\/span>\n\t\tfmt.Println(err.Error())<\/span>\n\t\treturn<\/span><\/span>\n\t}<\/span>\n\tdefer<\/span> stream.Close()<\/span>\n<\/span>\n\tfmt.Print(input)<\/span>\n\tfmt.Println(\"\"<\/span>)<\/span>\n<\/span>\n\tfor<\/span> {<\/span>\n\t\tresponse, e := stream.Recv()<\/span>\n\t\tif<\/span> errors.Is(e, io.EOF) {<\/span>\n\t\t\tbreak<\/span><\/span>\n\t\t}<\/span>\n\t\tif<\/span> e != nil<\/span> {<\/span>\n\t\t\tbreak<\/span><\/span>\n\t\t}<\/span>\n\t\tcontent := response.Choices[0<\/span>].Delta.Content<\/span>\n\t\tfmt.Print(content)<\/span>\n\t}<\/span>\n<\/span>\n\tfmt.Println(\"\"<\/span>)<\/span>\n<\/span>\n\treturn<\/span><\/span>\n}<\/span>\n<\/span>\nfunc<\/span> getAuthorization<\/span>(authToken string<\/span>)<\/span><\/span> string<\/span> {<\/span>\n\tid, secret := splitAPIKey(authToken)<\/span>\n\tpayload := jwt.MapClaims{<\/span>\n\t\t\"api_key\"<\/span>:   id,<\/span>\n\t\t\"exp\"<\/span>:       time.Now().Add(3<\/span> * time.Minute).UnixMilli(),<\/span>\n\t\t\"timestamp\"<\/span>: time.Now().UnixMilli(),<\/span>\n\t}<\/span>\n\ttoken := jwt.NewWithClaims(jwt.SigningMethodHS256, payload)<\/span>\n\ttoken.Header[\"sign_type\"<\/span>] = \"SIGN\"<\/span><\/span>\n\tres, _ := token.SignedString([]byte<\/span>(secret))<\/span>\n\treturn<\/span> res<\/span>\n}<\/span>\n<\/span>\nfunc<\/span> splitAPIKey<\/span>(apikey string<\/span>)<\/span><\/span> (string<\/span>, string<\/span>) {<\/span>\n\tparts := strings.Split(apikey, \".\"<\/span>)<\/span>\n\tif<\/span> len<\/span>(parts) != 2<\/span> {<\/span>\n\t\treturn<\/span> \"\"<\/span>, \"\"<\/span><\/span>\n\t}<\/span>\n\treturn<\/span> parts[0<\/span>], parts[1<\/span>]<\/span>\n}<\/span>\n<\/span>\ntype<\/span> removeAcceptHeaderRoundTrip struct<\/span>{}<\/span>\n<\/span>\nfunc<\/span> (s *removeAcceptHeaderRoundTrip)<\/span><\/span> RoundTrip(req *http.Request) (resp *http.Response, err error<\/span>) {<\/span>\n\treq.Header.Del(\"Accept\"<\/span>)<\/span>\n\tresp, err = http.DefaultTransport.RoundTrip(req)<\/span>\n\treturn<\/span><\/span>\n}<\/span>\n<\/code><\/pre>\n
\u5982\u679c\u89c9\u5f97\u6587\u7ae0\u6709\u5e2e\u52a9\uff0c\u5efa\u8bae\u6536\u85cf\u8d77\u6765\uff0c\u65b9\u4fbf\u4e0b\u6b21\u9605\u8bfb\u3002<\/strong><\/p>\n
\u559c\u6b22\u6211\u7684\u6587\u7ae0\u53ef\u4ee5\u5173\u6ce8\u6211\uff0c\u5206\u4eab\u6709\u8da3\u7684\u77e5\u8bc6\u5171\u540c\u5b66\u4e60\uff01<\/strong><\/p>\n","protected":false},"excerpt":{"rendered":"
AI \u5927\u6a21\u578b\u592a\u706b\u7206\u4e86\uff01\u6211\u5728\u5de5\u4f5c\u4e2d\u7ecf\u5e38\u4f7f\u7528\u5b83\u4eec\uff0c\u611f\u89c9\u5df2\u7ecf\u79bb\u4e0d\u5f00\u4e86\uff01\u6700\u8fd1\u6d4b\u8bd5\u4e86\u667a\u8c31 AI \u5927\u6a21\u578b\uff0c\u5b9e\u6d4b\u4e0b\u6765\u611f\u89c9\u8fd8\u633a\u4e0d\u9519\u7684\u3002\u5b98\u65b9\u63d0\u4f9b\u4e86\u4f53\u9a8c\u5305\uff0c\u6ce8\u518c\u5e76\u5b9e\u540d\u8ba4\u8bc1\u4e4b\u540e\u6709 300 \u4e07 token\uff0c\u975e\u5e38\u58d5\uff01<\/p>\n","protected":false},"author":1,"featured_media":0,"comment_status":"open","ping_status":"open","sticky":false,"template":"","format":"standard","meta":{"_acf_changed":false,"rank_math_title":"","rank_math_description":"","rank_math_focus_keyword":"","views":"1","footnotes":""},"categories":[3],"tags":[126,127,128,129,136],"collection":[],"class_list":["post-1578","post","type-post","status-publish","format-standard","hentry","category-fenlei2","tag-gpt","tag-ai","tag-128","tag-129","tag-136"],"acf":[],"_links":{"self":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/posts\/1578","targetHints":{"allow":["GET"]}}],"collection":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/posts"}],"about":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/types\/post"}],"author":[{"embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/users\/1"}],"replies":[{"embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/comments?post=1578"}],"version-history":[{"count":0,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/posts\/1578\/revisions"}],"wp:attachment":[{"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/media?parent=1578"}],"wp:term":[{"taxonomy":"category","embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/categories?post=1578"},{"taxonomy":"post_tag","embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/tags?post=1578"},{"taxonomy":"collection","embeddable":true,"href":"https:\/\/www.nicekj.com\/nicekj2024\/wp\/v2\/collection?post=1578"}],"curies":[{"name":"wp","href":"https:\/\/api.w.org\/{rel}","templated":true}]}}