rag index connected to prompt

Author: Veikkosuhonen

src/client/components/ChatV2/ChatV2.tsx

@@ -7,7 +7,7 @@ import { enqueueSnackbar } from 'notistack'
 import { useCallback, useEffect, useRef, useState } from 'react'
 import { useTranslation } from 'react-i18next'
 import { useParams, useSearchParams } from 'react-router-dom'
-import { DEFAULT_MODEL, DEFAULT_MODEL_TEMPERATURE, FREE_MODEL, inProduction, ValidModelName, ValidModelNameSchema, validModels } from '../../../config'
+import { DEFAULT_MODEL, DEFAULT_MODEL_TEMPERATURE, FREE_MODEL, inProduction, type ValidModelName, ValidModelNameSchema, validModels } from '../../../config'
 import type { ChatMessage, MessageGenerationInfo, ToolCallResultEvent } from '../../../shared/chat'
 import { getLanguageValue } from '../../../shared/utils'
 import { useIsEmbedded } from '../../contexts/EmbeddedContext'
@@ -83,7 +83,7 @@ const ChatV2Content = () => {
     `${localStoragePrefix}-chat-model-temperature`,
     String(DEFAULT_MODEL_TEMPERATURE),
     'temperature',
-    z.number(),
+    z.coerce.number(),
   )
 
   const [messages, setMessages] = useLocalStorageState(`${localStoragePrefix}-chat-messages`, [] as ChatMessage[])
@@ -193,10 +193,6 @@ const ChatV2Content = () => {
     }
 
     try {
-      if (!streamController) {
-        throw new Error('streamController is not defined')
-      }
-
       const { tokenUsageAnalysis, stream } = await postCompletionStreamV3(
         formData,
         {

src/client/components/ChatV2/PromptState.tsx

@@ -1,13 +1,13 @@
-import React, { createContext, useContext, useState, ReactNode, useEffect } from 'react'
+import React, { createContext, useContext, ReactNode, useEffect } from 'react'
 import useLocalStorageState from '../../hooks/useLocalStorageState'
 import { useParams, useSearchParams } from 'react-router-dom'
-import { Prompt } from '../../types'
-import apiClient, { ApiError } from '../../util/apiClient'
+import type { Prompt } from '../../types'
+import apiClient, { type ApiError } from '../../util/apiClient'
 import { isAxiosError } from 'axios'
-import { UseMutateAsyncFunction, useMutation, useQuery } from '@tanstack/react-query'
+import { type UseMutateAsyncFunction, useMutation, useQuery } from '@tanstack/react-query'
 import useCourse from '../../hooks/useCourse'
 import { useAnalyticsDispatch } from '../../stores/analytics'
-import { MessageGenerationInfo } from '../../../shared/chat'
+import type { MessageGenerationInfo } from '../../../shared/chat'
 import { useTranslation } from 'react-i18next'
 
 const useUrlPromptId = () => {

src/client/components/ChatV2/api.ts

@@ -12,7 +12,7 @@ export const useToolResults = (toolCallId: string) => {
   })
 }
 
-export const postCompletionStreamV3 = async (formData: FormData, input: PostStreamSchemaV3Type, abortController: AbortController) => {
+export const postCompletionStreamV3 = async (formData: FormData, input: PostStreamSchemaV3Type, abortController?: AbortController) => {
   formData.set('data', JSON.stringify(input))
 
   return postAbortableStream('/ai/v3/stream', formData, abortController)

src/server/routes/ai/v3.ts

@@ -113,7 +113,7 @@ router.post('/stream', upload.single('file'), async (r, res) => {
         {
           model: ChatInstance,
           attributes: ['id'],
-          as: 'chatInstances',
+          as: 'chatInstance',
           where: courseId ? { courseId } : {},
         },
         {