feat: add poe AI (Sage, GPT4, ClaudePlus, Claude, ChatGPT, Dragonfly)

Author: josStorer

build.mjs

@@ -184,6 +184,10 @@ async function runWebpack(isWithoutKatex, isWithoutTiktoken, callback) {
           test: /\.(jpg|png|svg)$/,
           type: 'asset/inline',
         },
+        {
+          test: /\.(graphql|gql)$/,
+          loader: 'graphql-tag/loader',
+        },
         isWithoutTiktoken
           ? {
               test: /crop-text\.mjs$/,

package-lock.json

@@ -23,13 +23,15 @@
     "@picocss/pico": "^1.5.7",
     "@primer/octicons-react": "^18.2.0",
     "countries-list": "^2.6.1",
+    "diff": "^5.1.0",
     "eventsource-parser": "^0.1.0",
     "file-saver": "^2.0.5",
     "github-markdown-css": "^5.2.0",
     "gpt-3-encoder": "^1.1.4",
     "i18next": "^22.4.13",
     "katex": "^0.16.4",
     "lodash-es": "^4.17.21",
+    "md5": "^2.3.0",
     "parse5": "^6.0.1",
     "preact": "^10.13.1",
     "prop-types": "^15.8.1",
@@ -64,6 +66,7 @@
     "eslint": "^8.36.0",
     "eslint-plugin-react": "^7.32.2",
     "fs-extra": "^11.1.0",
+    "graphql-tag": "^2.12.6",
     "jsdom": "^21.1.1",
     "less-loader": "^11.1.0",
     "mini-css-extract-plugin": "^2.7.5",

package.json

@@ -1,4 +1,4 @@
-import BingAIClient from '../clients/BingAIClient'
+import BingAIClient from '../clients/bing'
 import { getUserConfig } from '../../config/index.mjs'
 import { pushRecord, setAbortController } from './shared.mjs'
 

src/background/apis/bing-web.mjs

@@ -0,0 +1,41 @@
+import { pushRecord, setAbortController } from './shared.mjs'
+import PoeAiClient from '../clients/poe'
+
+/**
+ * @param {Runtime.Port} port
+ * @param {string} question
+ * @param {Session} session
+ * @param {string} modelName
+ */
+export async function generateAnswersWithPoeWebApi(port, question, session, modelName) {
+  const bot = new PoeAiClient(session.poe_chatId)
+  const { messageListener } = setAbortController(port, () => {
+    bot.breakMsg()
+    bot.close()
+  })
+
+  let answer = ''
+  await bot
+    .ask(
+      question,
+      modelName,
+      (msg) => {
+        answer += msg
+        port.postMessage({ answer: answer, done: false, session: null })
+      },
+      () => {
+        if (bot.chatId) session.poe_chatId = bot.chatId
+
+        pushRecord(session, question, answer)
+        console.debug('conversation history', { content: session.conversationRecords })
+        port.onMessage.removeListener(messageListener)
+        port.postMessage({ answer: answer, done: true, session: session })
+        bot.close()
+      },
+    )
+    .catch((err) => {
+      port.onMessage.removeListener(messageListener)
+      bot.close()
+      throw err
+    })
+}

src/background/apis/poe-web.mjs

@@ -0,0 +1,52 @@
+mutation AddHumanMessageMutation(
+    $chatId: BigInt!
+    $bot: String!
+    $query: String!
+    $source: MessageSource
+    $withChatBreak: Boolean! = false
+) {
+    messageCreateWithStatus(
+        chatId: $chatId
+        bot: $bot
+        query: $query
+        source: $source
+        withChatBreak: $withChatBreak
+    ) {
+        message {
+            id
+            __typename
+            messageId
+            text
+            linkifiedText
+            authorNickname
+            state
+            vote
+            voteReason
+            creationTime
+            suggestedReplies
+            chat {
+                id
+                shouldShowDisclaimer
+            }
+        }
+        messageLimit{
+          canSend
+          numMessagesRemaining
+          resetTime
+          shouldShowReminder
+        }
+        chatBreak {
+            id
+            __typename
+            messageId
+            text
+            linkifiedText
+            authorNickname
+            state
+            vote
+            voteReason
+            creationTime
+            suggestedReplies
+        }
+    }
+}

src/background/clients/BingAIClient.js renamed to src/background/clients/bing/index.mjs

@@ -0,0 +1,17 @@
+mutation AddMessageBreakMutation($chatId: BigInt!) {
+    messageBreakCreate(chatId: $chatId) {
+        message {
+            id
+            __typename
+            messageId
+            text
+            linkifiedText
+            authorNickname
+            state
+            vote
+            voteReason
+            creationTime
+            suggestedReplies
+        }
+    }
+}

src/background/clients/poe/graphql/AddHumanMessageMutation.graphql

@@ -0,0 +1,7 @@
+mutation AutoSubscriptionMutation($subscriptions: [AutoSubscriptionQuery!]!) {
+    autoSubscribe(subscriptions: $subscriptions) {
+        viewer {
+            id
+        }
+    }
+}

src/background/clients/poe/graphql/AddMessageBreakMutation.graphql

@@ -0,0 +1,8 @@
+fragment BioFragment on Viewer {
+    id
+    poeUser {
+        id
+        uid
+        bio
+    }
+}