(feat:attach) send attachment ids

This commit is contained in:
ManishMadan2882
2025-03-28 18:12:38 +05:30
parent 80e55ef385
commit 5089d86095
2 changed files with 18 additions and 2 deletions

View File

@@ -13,6 +13,7 @@ export function handleFetchAnswer(
promptId: string | null,
chunks: string,
token_limit: number,
attachments?: string[],
): Promise<
| {
result: any;
@@ -48,6 +49,12 @@ export function handleFetchAnswer(
token_limit: token_limit,
isNoneDoc: selectedDocs === null,
};
// Add attachments to payload if they exist
if (attachments && attachments.length > 0) {
payload.attachments = attachments;
}
if (selectedDocs && 'id' in selectedDocs) {
payload.active_docs = selectedDocs.id as string;
}
@@ -86,6 +93,7 @@ export function handleFetchAnswerSteaming(
token_limit: number,
onEvent: (event: MessageEvent) => void,
indx?: number,
attachments?: string[],
): Promise<Answer> {
history = history.map((item) => {
return {
@@ -104,6 +112,12 @@ export function handleFetchAnswerSteaming(
isNoneDoc: selectedDocs === null,
index: indx,
};
// Add attachments to payload if they exist
if (attachments && attachments.length > 0) {
payload.attachments = attachments;
}
if (selectedDocs && 'id' in selectedDocs) {
payload.active_docs = selectedDocs.id as string;
}

View File

@@ -27,8 +27,8 @@ export function handleAbort() {
export const fetchAnswer = createAsyncThunk<
Answer,
{ question: string; indx?: number }
>('fetchAnswer', async ({ question, indx }, { dispatch, getState }) => {
{ question: string; indx?: number; attachments?: string[] }
>('fetchAnswer', async ({ question, indx, attachments }, { dispatch, getState }) => {
if (abortController) {
abortController.abort();
}
@@ -110,6 +110,7 @@ export const fetchAnswer = createAsyncThunk<
}
},
indx,
attachments
);
} else {
const answer = await handleFetchAnswer(
@@ -122,6 +123,7 @@ export const fetchAnswer = createAsyncThunk<
state.preference.prompt.id,
state.preference.chunks,
state.preference.token_limit,
attachments
);
if (answer) {
let sourcesPrepped = [];