fix(ui): resolve missing code source in answer engine (#4034)

* fix(ui): resolve missing code source in answer engine

* update: threadRelevantQuestions

Author: liangfung

ee/tabby-ui/app/(home)/page.tsx

@@ -4,6 +4,7 @@ import { useEffect, useLayoutEffect, useRef, useState } from 'react'
 import Image from 'next/image'
 import { useRouter } from 'next/navigation'
 import tabbyUrl from '@/assets/logo-dark.png'
+import { compact } from 'lodash-es'
 import { useQuery } from 'urql'
 import { useStore } from 'zustand'
 
@@ -95,7 +96,14 @@ function MainPanel() {
     setIsLoading(true)
     updatePendingUserMessage({
       content: question,
-      context
+      context: {
+        ...context,
+        codeSourceId: selectedRepository?.id,
+        docSourceIds: compact([
+          selectedRepository?.id,
+          ...(context?.docSourceIds ?? [])
+        ])
+      }
     })
     router.push('/search')
   }
@@ -105,8 +113,8 @@ function MainPanel() {
     updatePendingUserMessage({
       content: question,
       context: {
-        docSourceIds: [sourceId as string],
-        codeSourceIds: [sourceId as string],
+        docSourceIds: [sourceId],
+        codeSourceId: sourceId,
         modelName: selectedModel
       }
     })

ee/tabby-ui/app/search/components/search.tsx

@@ -360,7 +360,6 @@ export function Search() {
       initializing.current = true
 
       if (pendingUserMessage?.content) {
-        // setIsReady(true)
         onSubmitSearch(pendingUserMessage.content, pendingUserMessage.context)
         updatePendingUserMessage(undefined)
         return
@@ -1031,20 +1030,17 @@ function getSourceInputs(
   repositorySourceId: string | undefined,
   ctx: ThreadRunContexts | undefined
 ) {
-  let sourceIdsForDocQuery: string[] = []
-  let sourceIdForCodeQuery: string | undefined
+  let sourceIdsForDocQuery: string[] = compact([repositorySourceId])
+  let sourceIdForCodeQuery: string | undefined = repositorySourceId
   let searchPublic = false
 
   if (ctx) {
     sourceIdsForDocQuery = uniq(
       // Compatible with existing user messages
-      compact(
-        [repositorySourceId, ctx?.codeSourceIds?.[0]].concat(ctx.docSourceIds)
-      )
+      compact([repositorySourceId, ctx?.codeSourceId].concat(ctx.docSourceIds))
     )
     searchPublic = ctx.searchPublic ?? false
-    sourceIdForCodeQuery =
-      repositorySourceId || ctx.codeSourceIds?.[0] || undefined
+    sourceIdForCodeQuery = repositorySourceId || ctx.codeSourceId || undefined
   }
   return {
     sourceIdsForDocQuery,

ee/tabby-ui/lib/types/chat.ts

@@ -116,11 +116,11 @@ type MergeUnionType<T> = {
   [k in Keys<T>]?: Pick<T, k>
 }
 
-export type ThreadRunContexts = {
+export interface ThreadRunContexts {
   modelName?: string
   searchPublic?: boolean
   docSourceIds?: string[]
-  codeSourceIds?: string[]
+  codeSourceId?: string
 }
 
 export interface RelevantCodeContext extends Context {