diff --git a/backend/modules/query_controllers/multimodal/controller.py b/backend/modules/query_controllers/multimodal/controller.py
index d61675d2..c190b37c 100644
--- a/backend/modules/query_controllers/multimodal/controller.py
+++ b/backend/modules/query_controllers/multimodal/controller.py
@@ -30,7 +30,7 @@ class MultiModalRAGQueryController(BaseQueryController):
async def _stream_vlm_answer(self, llm, message_payload, docs):
try:
async with async_timeout.timeout(GENERATION_TIMEOUT_SEC):
- yield Docs(content=self._cleanup_metadata(docs))
+ yield Docs(content=self._enrich_context_for_stream_response(docs))
async for chunk in llm.astream(message_payload):
yield Answer(content=chunk.content)
except asyncio.TimeoutError:
diff --git a/frontend/src/screens/dashboard/docsqa/main/components/ConfigSelector.tsx b/frontend/src/screens/dashboard/docsqa/main/components/ConfigSelector.tsx
index 8e524f3d..01197b0a 100644
--- a/frontend/src/screens/dashboard/docsqa/main/components/ConfigSelector.tsx
+++ b/frontend/src/screens/dashboard/docsqa/main/components/ConfigSelector.tsx
@@ -1,6 +1,6 @@
import React from 'react'
-import { MenuItem, Select } from '@mui/material'
+import { FormControl, MenuItem, Select } from '@mui/material'
interface ConfigProps {
title: string
@@ -25,31 +25,33 @@ const ConfigSelector = (props: ConfigProps) => {
return (
{title}:
-
+
+
+
)
}
diff --git a/frontend/src/screens/dashboard/docsqa/main/components/ConfigSidebar.tsx b/frontend/src/screens/dashboard/docsqa/main/components/ConfigSidebar.tsx
index 4ed5419f..33092987 100644
--- a/frontend/src/screens/dashboard/docsqa/main/components/ConfigSidebar.tsx
+++ b/frontend/src/screens/dashboard/docsqa/main/components/ConfigSidebar.tsx
@@ -88,11 +88,12 @@ const Left = (props: any) => {
{allRetrieverOptions && selectedRetriever?.key && (
{
+ console.log(allRetrieverOptions.map((retriever) => retriever.key))
const retriever = allRetrieverOptions.find(
(retriever) => retriever.key === e.target.value,
)
@@ -100,7 +101,7 @@ const Left = (props: any) => {
setPromptTemplate(retriever?.promptTemplate)
}}
renderItem={(item) => (
-