refactor: use graphql instead

Author: kevinkim-ogp

packages/backend/src/routes/api/chat/readiness.ts renamed to packages/backend/src/graphql/queries/get-chat-readiness.ts

@@ -1,7 +1,5 @@
 import { startActiveObservation } from '@langfuse/tracing'
 import { generateObject } from 'ai'
-import type { Request, Response } from 'express'
-import { Router } from 'express'
 import z from 'zod/v3'
 
 import appConfig from '@/config/app'
@@ -10,21 +8,15 @@ import logger from '@/helpers/logger'
 import { model, MODEL_TYPE } from '@/helpers/pair'
 import { getPrompt } from '@/helpers/pair/get-prompt'
 
-import { getAuthenticatedContext } from '../middleware/authentication'
-
-interface ChatReadinessRequest {
-  message: string
-  sessionId: string
-}
-
-const handleChatReadiness = async (
-  req: Request,
-  res: Response,
-): Promise<void> => {
-  const context = getAuthenticatedContext(req)
+import type { QueryResolvers } from '../__generated__/types.generated'
 
+const getChatReadiness: QueryResolvers['getChatReadiness'] = async (
+  _parent,
+  params,
+  context,
+) => {
   try {
-    const { message: rawMessage } = req.body as ChatReadinessRequest
+    const { message: rawMessage, sessionId } = params
 
     const promptConfig = await getLdFlagValue(
       'ai-builder-prompt-config',
@@ -45,6 +37,7 @@ const handleChatReadiness = async (
           userId: context.currentUser.email,
           environment: appConfig.appEnv,
           tags: ['ai-builder', 'is-chat-ready'],
+          sessionId,
         })
 
         trace.update({
@@ -88,15 +81,11 @@ const handleChatReadiness = async (
       },
     )
 
-    res.json({ isReady: result.isReady })
+    return { isReady: result.isReady }
   } catch (error) {
-    logger.error('Error in chat readiness', { error })
-    res.status(500).json({ error: 'Internal server error' })
+    logger.error('Error in getChatReadiness', { error })
+    throw new Error('Error encountered. Please try again.')
   }
 }
 
-const router = Router()
-
-router.post('/', handleChatReadiness)
-
-export default router
+export default getChatReadiness

packages/backend/src/graphql/query-resolvers.ts

@@ -1,6 +1,7 @@
 import type { QueryResolvers } from './__generated__/types.generated'
 import getApp from './queries/get-app'
 import getApps from './queries/get-apps'
+import getChatReadiness from './queries/get-chat-readiness'
 import getConnectedApps from './queries/get-connected-apps'
 import getCurrentUser from './queries/get-current-user'
 import getDynamicData from './queries/get-dynamic-data'
@@ -45,6 +46,7 @@ export default {
   getPendingFlowTransfers,
   getFlowTransferDetails,
   getTemplates,
+  getChatReadiness,
   ...tilesQueryResolvers,
 
   // This is a special stub that enables us to group all our admin-related

packages/backend/src/graphql/schema.graphql

@@ -57,6 +57,8 @@ type Query {
   admin: AdminQuery!
   # Templates (including demo templates)
   getTemplates(tag: TemplateTagType): [Template!]!
+  # AI
+  getChatReadiness(message: String!, sessionId: String!): ChatReadinessResult!
 }
 
 type AdminQuery {
@@ -160,6 +162,10 @@ input CreateFlowWithStepsInput {
   aiBuilderConfig: FlowAiBuilderConfigInput!
 }
 
+type ChatReadinessResult {
+  isReady: Boolean!
+}
+
 type AdminMutation {
   pauseGroup(input: PauseGroupInput!): Boolean
   resumeGroup(input: ResumeGroupInput!): Boolean

packages/backend/src/routes/api/chat/index.ts

@@ -11,8 +11,6 @@ import { getPrompt } from '@/helpers/pair/get-prompt'
 
 import { getAuthenticatedContext } from '../middleware/authentication'
 
-import chatReadinessRouter from './readiness'
-
 interface ChatRequest {
   messages: Array<{
     role: 'user' | 'assistant' | 'system'
@@ -174,6 +172,5 @@ async function handleChatStream(req: Request, res: Response) {
 const router = Router()
 
 router.post('/', handleChatStream)
-router.use('/readiness', chatReadinessRouter)
 
 export default router