給公眾號接入FastWiki智慧AI知識庫，讓您的公眾號加入智慧行列

2024-05-17碼農

最近由於公眾號使用者太多，我就在思考有啥方式能給微信公眾號的粉絲提供更多的更好的服務？這個時候我就想是否可以給公眾號接入一下AI？讓使用者跟微信公眾號對話，然後還能回答使用者的問題，並且我提供一些資料讓AI幫我回復使用者的資訊？

這個時候剛剛好我們的 FastWiki 計畫滿足了部份需求，然後我們就順便加入了微信公眾號，下面我們也會解析我們如何給公眾號實作接入 FastWiki 的！

給 `FastWiki` 實作接入微信公眾號

在FastWiki.Service計畫中的Service目錄建立 WeChatService 用於實作微信公眾號接入功能，具體程式碼如下，

由於微信公眾號的限制，沒有實作微信公眾號的微信認證，您的公眾號是無法主動向使用者發送資訊，並且你的介面必須在5s內回復使用者的資訊，還得是xml格式（非常想吐槽！！！），在其中，我們將使用者對話和AI回復使用Channel去分離我們的業務， AI透過讀取Channel的對話資訊，然後進行提問，並且呼叫了知識庫服務提供的介面，還可以在知識庫搜尋相關prompt資訊，然後得到大模型響應的內容，然後將響應的內容添加到記憶體緩存中，並且設定過期時間（防止使用者提問以後不在繼續造成記憶體溢位），然後當使用者發送 1 提取AI的回復的時候獲取記憶體的響應內容，然後直接返回給使用者，然後刪除記憶體緩存中的數據，這樣就避免介面超過5s導致介面響應異常!

以下程式碼是微信公眾號用於驗證介面是否可用！

if (context.Request.Method != "POST") { context.Request.Query.TryGetValue("signature", outvar signature); context.Request.Query.TryGetValue("timestamp", outvar timestamp); context.Request.Query.TryGetValue("nonce", outvar nonce); context.Request.Query.TryGetValue("echostr", outvar echostr); await context.Response.WriteAsync(echostr); return; }

WeChatService具體實作

///<summary> /// 微信服務 ///</summary> public classWeChatService { staticWeChatService() { Task.Run(AIChatAsync); } privatestaticreadonly Channel<WeChatAI> Channel = System.Threading.Channels.Channel.CreateUnbounded<WeChatAI>(); privateconststring OutputTemplate = """ 您好，歡迎關註FastWiki！由於微信限制，我們無法立即回復您的訊息，但是您的訊息已經收到，我們會盡快回復您！如果獲取訊息結果，請輸入1。如果您有其他問題，可以直接回復，我們會盡快回復您！ """; publicstaticasync Task AIChatAsync() { usingvar scope = MasaApp.RootServiceProvider.CreateScope(); var eventBus = scope.ServiceProvider.GetRequiredService<IEventBus>(); var wikiMemoryService = scope.ServiceProvider.GetRequiredService<WikiMemoryService>(); var memoryCache = scope.ServiceProvider.GetRequiredService<IMemoryCache>(); while (await Channel.Reader.WaitToReadAsync()) { var content = await Channel.Reader.ReadAsync(); await SendMessageAsync(content, eventBus, wikiMemoryService, memoryCache); } } ///<summary> /// 微信AI對話 ///</summary> ///<param name="chatAi"></param> ///<param name="eventBus"></param> ///<param name="wikiMemoryService"></param> ///<param name="memoryCache"></param> publicstaticasync Task SendMessageAsync(WeChatAI chatAi, IEventBus eventBus, WikiMemoryService wikiMemoryService, IMemoryCache memoryCache) { var chatShareInfoQuery = new ChatShareInfoQuery(chatAi.SharedId); await eventBus.PublishAsync(chatShareInfoQuery); // 如果chatShareId不存在則返回讓下面扣款 var chatShare = chatShareInfoQuery.Result; var chatApplicationQuery = new ChatApplicationInfoQuery(chatShareInfoQuery.Result.ChatApplicationId); await eventBus.PublishAsync(chatApplicationQuery); var chatApplication = chatApplicationQuery?.Result; if (chatApplication == null) { return; } int requestToken = 0; var module = new ChatCompletionDto<ChatCompletionRequestMessage>() { messages = [ new() { content = chatAi.Content, role = "user", } ] }; var chatHistory = new ChatHistory(); // 如果設定了Prompt，則添加 if (!chatApplication.Prompt.IsNullOrEmpty()) { chatHistory.AddSystemMessage(chatApplication.Prompt); } // 保存對話提問 var createChatRecordCommand = new CreateChatRecordCommand(chatApplication.Id, chatAi.Content); await eventBus.PublishAsync(createChatRecordCommand); var sourceFile = new List<FileStorage>(); var memoryServerless = wikiMemoryService.CreateMemoryServerless(chatApplication.ChatModel); // 如果為空則不使用知識庫 if (chatApplication.WikiIds.Count != 0) { var success = await OpenAIService.WikiPrompt(chatApplication, memoryServerless, chatAi.Content, eventBus, sourceFile, module); if (!success) { return; } } var output = new StringBuilder(); // 添加使用者輸入，並且計算請求token數量 module.messages.ForEach(x => { if (x.content.IsNullOrEmpty()) return; requestToken += TokenHelper.ComputeToken(x.content); chatHistory.Add(new ChatMessageContent(new AuthorRole(x.role), x.content)); }); if (chatShare != null) { // 如果token不足則返回，使用token和當前request總和大於可用token，則返回 if (chatShare.AvailableToken != -1 && (chatShare.UsedToken + requestToken) >= chatShare.AvailableToken) { output.Append("Token不足"); return; } // 如果沒有過期則繼續 if (chatShare.Expires != null && chatShare.Expires < DateTimeOffset.Now) { output.Append("Token已過期"); return; } } try { awaitforeach (var item in OpenAIService.SendChatMessageAsync(chatApplication, eventBus, wikiMemoryService, chatHistory)) { if (string.IsNullOrEmpty(item)) { continue; } output.Append(item); } //對於對話扣款 if (chatShare != null) { var updateChatShareCommand = new DeductTokenCommand(chatShare.Id, requestToken); await eventBus.PublishAsync(updateChatShareCommand); } } catch (NotModelException notModelException) { output.Clear(); output.Append(notModelException.Message); } catch (InvalidOperationException invalidOperationException) { output.Clear(); output.Append("對話異常:" + invalidOperationException.Message); } catch (ArgumentException argumentException) { output.Clear(); output.Append("對話異常:" + argumentException.Message); } catch (Exception e) { output.Clear(); output.Append("對話異常,請聯系管理員"); } finally { memoryCache.Set(chatAi.MessageId, output.ToString(), TimeSpan.FromMinutes(5)); } } ///<summary> /// 接收訊息 ///</summary> ///<param name="context"></param> publicstaticasync Task ReceiveMessageAsync(HttpContext context, string? id, IMemoryCache memoryCache) { if (context.Request.Method != "POST") { context.Request.Query.TryGetValue("signature", outvar signature); context.Request.Query.TryGetValue("timestamp", outvar timestamp); context.Request.Query.TryGetValue("nonce", outvar nonce); context.Request.Query.TryGetValue("echostr", outvar echostr); await context.Response.WriteAsync(echostr); return; } usingvar reader = new StreamReader(context.Request.Body); // xml解析 var body = await reader.ReadToEndAsync(); var doc = new XmlDocument(); doc.LoadXml(body); var root = doc.DocumentElement; var input = new WeChatMessageInput { ToUserName = root.SelectSingleNode("ToUserName")?.InnerText, FromUserName = root.SelectSingleNode("FromUserName")?.InnerText, CreateTime = long.Parse(root.SelectSingleNode("CreateTime")?.InnerText ?? "0"), MsgType = root.SelectSingleNode("MsgType")?.InnerText, Content = root.SelectSingleNode("Content")?.InnerText, MsgId = long.Parse(root.SelectSingleNode("MsgId")?.InnerText ?? "0") }; var output = new WehCahtMe { ToUserName = input.ToUserName, FromUserName = input.FromUserName, CreateTime = input.CreateTime, MsgType = input.MsgType, Content = input.Content }; if (output.Content.IsNullOrEmpty()) { return; } if (id == null) { context.Response.ContentType = "application/xml"; await context.Response.WriteAsync(GetOutputXml(output, "參數錯誤，請聯系管理員！code:id_null")); return; } var messageId = GetMessageId(output); // 從緩存中獲取,如果有則返回 memoryCache.TryGetValue(messageId, outvarvalue); // 如果value有值則，但是value為空，則返回提示,防止重復提問！ if (valueisstring str && str.IsNullOrEmpty()) { context.Response.ContentType = "application/xml"; await context.Response.WriteAsync(GetOutputXml(output, "暫無訊息，請稍後再試！code:no_message")); return; } elseif (valueisstring v && !v.IsNullOrEmpty()) { context.Response.ContentType = "application/xml"; await context.Response.WriteAsync(GetOutputXml(output, v)); return; } if (output.Content == "1") { if (valueisstring v && !v.IsNullOrEmpty()) { memoryCache.Remove(messageId); context.Response.ContentType = "application/xml"; await context.Response.WriteAsync(GetOutputXml(output, v)); return; } context.Response.ContentType = "application/xml"; await context.Response.WriteAsync(GetOutputXml(output, "暫無訊息，請稍後再試！code:no_message")); return; } // 先寫入channel，等待後續處理 Channel.Writer.TryWrite(new WeChatAI() { Content = output.Content, SharedId = id, MessageId = messageId }); // 等待4s await Task.Delay(4500); // 嘗試從緩存中獲取 memoryCache.TryGetValue(messageId, outvar outputTemplate); if (outputTemplate isstring outValue && !outValue.IsNullOrEmpty()) { context.Response.ContentType = "application/xml"; await context.Response.WriteAsync(GetOutputXml(output, outValue)); return; } context.Response.ContentType = "application/xml"; await context.Response.WriteAsync(GetOutputXml(output, OutputTemplate)); // 寫入緩存,5分鐘過期 memoryCache.Set(messageId, OutputTemplate, TimeSpan.FromMinutes(5)); } privatestaticstringGetMessageId(WehCahtMe output) { return output.FromUserName + output.ToUserName; } ///<summary> /// 獲取返回的xml ///</summary> ///<param name="output"></param> ///<param name="content"></param> ///<returns></returns> publicstaticstringGetOutputXml(WehCahtMe output, string content) { var createTime = DateTimeOffset.Now.ToUnixTimeSeconds(); var xml = $@" <xml> <ToUserName><![CDATA[{output.FromUserName}]]></ToUserName> <FromUserName><![CDATA[{output.ToUserName}]]></FromUserName> <CreateTime>{createTime}</CreateTime> <MsgType><![CDATA[text]]></MsgType> <Content><![CDATA[{content}]]></Content> </xml> "; return xml; } public classWeChatMessageInput { publicstring URL { get; set; } publicstring ToUserName { get; set; } publicstring FromUserName { get; set; } publiclong CreateTime { get; set; } publicstring MsgType { get; set; } publicstring Content { get; set; } publiclong MsgId { get; set; } } public classWehCahtMe { publicstring ToUserName { get; set; } publicstring FromUserName { get; set; } publiclong CreateTime { get; set; } publicstring MsgType { get; set; } publicstring Content { get; set; } } }

WeChat提供的API服務

上面是介面的具體實作，然後我們在 Program 中將我們的 WeChatService 對外提供API（Get是用於提供給微信公眾號驗證）， {id} 則繫結我們的介面的 string id 參數，以便動態設定。

app.MapGet("/api/v1/WeChatService/ReceiveMessage/{id}", WeChatService.ReceiveMessageAsync) .WithTags("WeChat") .WithGroupName("WeChat") .WithDescription("微信訊息驗證") .WithOpenApi(); app.MapPost("/api/v1/WeChatService/ReceiveMessage/{id}", WeChatService.ReceiveMessageAsync) .WithTags("WeChat") .WithGroupName("WeChat") .WithDescription("微信訊息接收") .WithOpenApi();