fix(Article): AI transcript uses attachment hash as doc id

cofacts · Aug 30, 2023 · eea35ab · eea35ab
1 parent e9e4614
commit eea35ab
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 2 deletions.
diff --git a/src/graphql/models/AIResponse.js b/src/graphql/models/AIResponse.js
@@ -85,7 +85,8 @@ export const AIReply = new GraphQLObjectType({
 
 export const AITranscript = new GraphQLObjectType({
   name: 'AITranscript',
-  description: 'Transcript from OCR or speech-to-text AI models.',
+  description:
+    'Transcript from OCR or speech-to-text AI models for the specified MediaEntry ID as docId.',
   interfaces: [Node, AIResponse],
   fields: {
     ...commonAiResponseFields,

diff --git a/src/graphql/models/Article.js b/src/graphql/models/Article.js
@@ -36,7 +36,7 @@ import mediaManager, {
 } from 'util/mediaManager';
 import ArticleReplyStatusEnum from './ArticleReplyStatusEnum';
 import ArticleReply from './ArticleReply';
-import { AIReply } from './AIResponse';
+import { AIReply, AITranscript } from './AIResponse';
 import ArticleCategoryStatusEnum from './ArticleCategoryStatusEnum';
 import ReplyRequestStatusEnum from './ReplyRequestStatusEnum';
 import ArticleCategory from './ArticleCategory';
@@ -174,6 +174,34 @@ const Article = new GraphQLObjectType({
       },
     },
 
+    aiTranscripts: {
+      type: new GraphQLNonNull(
+        new GraphQLList(new GraphQLNonNull(AITranscript))
+      ),
+      description: 'Automated transcript',
+      async resolve({ attachmentHash }, _, { loaders }) {
+        return loaders.searchResultLoader.load({
+          index: 'airesponses',
+          type: 'doc',
+          body: {
+            query: {
+              bool: {
+                must: [
+                  { term: { type: 'TRANSCRIPT' } },
+                  { term: { docId: attachmentHash } },
+                  { term: { status: 'SUCCESS' } },
+                ],
+              },
+            },
+            sort: {
+              createdAt: 'desc',
+            },
+            size: 10,
+          },
+        });
+      },
+    },
+
     articleCategories: {
       type: new GraphQLList(ArticleCategory),
       args: {