ProductivitySuite UI: Add model parameter for Codegen so that it work…

…s with vLLM

Signed-off-by: sgurunat <[email protected]>

ProductivitySuite/ui/react/src/components/CodeGen/CodeGen.tsx

@@ -8,7 +8,8 @@ import { IconArrowRight } from '@tabler/icons-react'
 import { ConversationMessage } from '../Message/conversationMessage'
 import { fetchEventSource } from '@microsoft/fetch-event-source'
 import { CODE_GEN_URL } from '../../config'
-
+import { conversationSelector } from '../../redux/Conversation/ConversationSlice';
+import { useAppSelector } from "../../redux/store";
 
 
 const CodeGen = () => {
@@ -17,14 +18,16 @@ const CodeGen = () => {
     const [response,setResponse] = useState<string>("");
     const promptInputRef = useRef<HTMLTextAreaElement>(null)
     const scrollViewport = useRef<HTMLDivElement>(null)
+    const { model } = useAppSelector(conversationSelector);
 
     const toSend = "Enter"
 
     const handleSubmit = async () => {
         setResponse("")
         setSubmittedPrompt(prompt)
         const body = {
-            messages:prompt
+            messages:prompt,
+            model: model
         }
         fetchEventSource(CODE_GEN_URL, {
             method: "POST",