Merge d39aa46 into 7886f6d

.env.sample

@@ -100,5 +100,9 @@ GCS_CREDENTIALS=
 GCS_BUCKET_NAME=
 GCS_MEDIA_FOLDER=media/
 
+# Open AI API key for AI replies, etc.
+#
+OPENAI_API_KEY=
+
 # When LOG_REQUESTS exists, it also shows incoming GraphQL request, variables, and resolved user info
 LOG_REQUESTS=

package.json

@@ -44,6 +44,7 @@
     "koa-session2": "^2.2.9",
     "koa-static": "^5.0.0",
     "node-fetch": "^2.2.0",
+    "openai": "^3.2.1",
     "passport-facebook": "^3.0.0",
     "passport-github2": "^0.1.10",
     "passport-google-oauth20": "^2.0.0",

src/graphql/models/AIResponse.js

@@ -0,0 +1,88 @@
+import {
+  GraphQLObjectType,
+  GraphQLInterfaceType,
+  GraphQLString,
+  GraphQLNonNull,
+  GraphQLID,
+  GraphQLInt,
+} from 'graphql';
+
+import { createConnectionType } from 'graphql/util';
+import Node from '../interfaces/Node';
+import AIResponseTypeEnum from './AIResponseTypeEnum';
+import AIResponseStatusEnum from './AIResponseStatusEnum';
+import User, { userFieldResolver } from './User';
+
+const commonAiResponseFields = {
+  id: { type: new GraphQLNonNull(GraphQLID) },
+
+  docId: {
+    description: 'The id for the document that this AI response is for.',
+    type: GraphQLID,
+  },
+
+  type: {
+    description: 'AI response type',
+    type: new GraphQLNonNull(AIResponseTypeEnum),
+  },
+
+  user: {
+    description: 'The user triggered this AI response',
+    type: User,
+    resolve: userFieldResolver,
+  },
+
+  status: {
+    description: 'Processing status of AI',
+    type: new GraphQLNonNull(AIResponseStatusEnum),
+  },
+
+  text: {
+    description: 'AI response text. Populated after status becomes SUCCESS.',
+    type: GraphQLString,
+  },
+
+  createdAt: { type: new GraphQLNonNull(GraphQLString) },
+  updatedAt: { type: GraphQLString },
+};
+
+const AIResponse = new GraphQLInterfaceType({
+  name: 'AIResponse',
+  description: 'Denotes an AI processed response and its processing status.',
+  interfaces: [Node],
+  fields: commonAiResponseFields,
+  resolveType(doc) {
+    switch (doc.type) {
+      case 'AI_REPLY':
+        return AIReply;
+    }
+  },
+});
+
+export const AIReply = new GraphQLObjectType({
+  name: 'AIReply',
+  description: 'A ChatGPT reply for an article with no human fact-checks yet',
+  interfaces: [Node, AIResponse],
+  fields: {
+    ...commonAiResponseFields,
+    usage: {
+      description:
+        'The usage returned from OpenAI. Populated after status becomes SUCCESS.',
+      type: new GraphQLObjectType({
+        name: 'OpenAICompletionUsage',
+        fields: {
+          promptTokens: { type: new GraphQLNonNull(GraphQLInt) },
+          completionTokens: { type: new GraphQLNonNull(GraphQLInt) },
+          totalTokens: { type: new GraphQLNonNull(GraphQLInt) },
+        },
+      }),
+    },
+  },
+});
+
+export const AIResponseConnection = createConnectionType(
+  'AIResponseConnection',
+  AIResponse
+);
+
+export default AIResponse;

src/graphql/models/AIResponseStatusEnum.js

@@ -0,0 +1,10 @@
+import { GraphQLEnumType } from 'graphql';
+
+export default new GraphQLEnumType({
+  name: 'AIResponseStatusEnum',
+  values: {
+    LOADING: { value: 'LOADING' },
+    SUCCESS: { value: 'SUCCESS' },
+    ERROR: { value: 'ERROR' },
+  },
+});

src/graphql/models/AIResponseTypeEnum.js

@@ -0,0 +1,13 @@
+import { GraphQLEnumType } from 'graphql';
+
+export default new GraphQLEnumType({
+  name: 'AIResponseTypeEnum',
+  values: {
+    AI_REPLY: {
+      description:
+        'The AI Response is an automated analysis / reply of an article.',
+      value: 'AI_REPLY',
+    },
+    // TBA: speach-to-text result, OCR result, reply review response, etc
+  },
+});

src/graphql/models/Article.js

@@ -35,6 +35,7 @@ import mediaManager, {
 } from 'util/mediaManager';
 import ArticleReplyStatusEnum from './ArticleReplyStatusEnum';
 import ArticleReply from './ArticleReply';
+import { AIReply } from './AIResponse';
 import ArticleCategoryStatusEnum from './ArticleCategoryStatusEnum';
 import ReplyRequestStatusEnum from './ReplyRequestStatusEnum';
 import ArticleCategory from './ArticleCategory';
@@ -144,6 +145,33 @@ const Article = new GraphQLObjectType({
       },
     },
 
+    aiReplies: {
+      type: new GraphQLNonNull(new GraphQLList(new GraphQLNonNull(AIReply))),
+      description:
+        'Automated reply from AI before human fact checkers compose an fact check',
+      async resolve({ id }, _, { loaders }) {
+        return loaders.searchResultLoader.load({
+          index: 'airesponses',
+          type: 'doc',
+          body: {
+            query: {
+              bool: {
+                must: [
+                  { term: { type: 'AI_REPLY' } },
+                  { term: { docId: id } },
+                  { term: { status: 'SUCCESS' } },
+                ],
+              },
+            },
+            sort: {
+              createdAt: 'desc',
+            },
+            size: 10,
+          },
+        });
+      },
+    },
+
     articleCategories: {
       type: new GraphQLList(ArticleCategory),
       args: {