Skip to content

Commit

Permalink
ProductivitySuite UI: Add model parameter for Docsum so that it works…
Browse files Browse the repository at this point in the history
… with vLLM

Signed-off-by: sgurunat <[email protected]>
sgurunat committed Nov 11, 2024
1 parent 6f0783e commit 46736cc
Showing 1 changed file with 5 additions and 2 deletions.
7 changes: 5 additions & 2 deletions ProductivitySuite/ui/react/src/components/DocSum/DocSum.tsx
Original file line number Diff line number Diff line change
@@ -7,7 +7,8 @@ import { fetchEventSource } from '@microsoft/fetch-event-source'
import { notifications } from '@mantine/notifications'
import { DOC_SUM_URL } from '../../config'
import { FileWithPath } from '@mantine/dropzone'

import { conversationSelector } from '../../redux/Conversation/ConversationSlice';
import { useAppSelector } from "../../redux/store";

const DocSum = () => {
const [isFile, setIsFile] = useState<boolean>(false);
@@ -16,6 +17,7 @@ const DocSum = () => {
const [value, setValue] = useState<string>('');
const [fileContent, setFileContent] = useState<string>('');
const [response, setResponse] = useState<string>('');
const { model } = useAppSelector(conversationSelector);

let messagesEnd:HTMLDivElement;

@@ -59,7 +61,8 @@ const DocSum = () => {

setIsGenerating(true)
const body = {
messages: isFile ? fileContent : value
messages: isFile ? fileContent : value,
model: model
}
fetchEventSource(DOC_SUM_URL, {
method: "POST",

0 comments on commit 46736cc

Please sign in to comment.