From f0964f4fc91f10ce7a57bb05041e067fc6ec82c2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E6=98=9F=E6=97=A5=E8=AF=AD?= <admin@weekdaycare.cn>
Date: Sat, 20 Jan 2024 02:08:08 +0800
Subject: [PATCH] fix ai postselector (#362)

* fix ai postselector

* Update tianli_gpt.ejs

---------

Co-authored-by: xaoxuu <16400144+xaoxuu@users.noreply.github.com>
---
 layout/_partial/plugins/ai/tianli_gpt.ejs | 2 +-
 layout/page.ejs                           | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/layout/_partial/plugins/ai/tianli_gpt.ejs b/layout/_partial/plugins/ai/tianli_gpt.ejs
index 31371ef..8e7ec25 100644
--- a/layout/_partial/plugins/ai/tianli_gpt.ejs
+++ b/layout/_partial/plugins/ai/tianli_gpt.ejs
@@ -1,6 +1,6 @@
 <% if (page.layout === 'post' || (page.wiki?.length > 0)) { %>
   <script>
-    let tianliGPT_postSelector = '.md-text.content.<%= page.layout === "post" ? "post" : "wiki" %>';
+    let tianliGPT_postSelector = 'article.content';
     let tianliGPT_key = '<%= theme.plugins.tianli_gpt.api %>';
     let tianliGPT_typingAnimate = '<%= theme.plugins.tianli_gpt.typingAnimate %>';
     let tianliGPT_wordLimit = <%= theme.plugins.tianli_gpt.limit %>;
diff --git a/layout/page.ejs b/layout/page.ejs
index 6403ae8..12f6b97 100755
--- a/layout/page.ejs
+++ b/layout/page.ejs
@@ -62,6 +62,9 @@ function layoutDiv() {
   if (layout === 'post' || page.wiki) {
     el += partial('_partial/main/article/read_next')
   }
+  if (isUsingTianliGPT) {
+    el += partial('_partial/plugins/ai/tianli_gpt')
+  }
   if (layout === 'post') {
     el += partial('_partial/main/article/related_posts')
   }