LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-12-02 18:13:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: add tool calls tracking and show in frontend

Browse Source
This commit is contained in:
Siddhant Rai
2025-02-12 21:47:47 +05:30
parent 0de4241b56
commit e209699b19
13 changed files with 302 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
frontend/src/conversation/conversationSlice.ts
View File

@@ -82,6 +82,13 @@ export const fetchAnswer = createAsyncThunk<
query: { sources: data.source ?? [] },
}),
);
} else if (data.type === 'tool_calls') {
dispatch(
updateToolCalls({
index: indx ?? state.conversation.queries.length - 1,
query: { tool_calls: data.tool_calls },
}),
);
} else if (data.type === 'error') {
// set status to 'failed'
dispatch(conversationSlice.actions.setStatus('failed'));
@@ -130,7 +137,11 @@ export const fetchAnswer = createAsyncThunk<
dispatch(
updateQuery({
index: indx ?? state.conversation.queries.length - 1,
query: { response: answer.answer, sources: sourcesPrepped },
query: {
response: answer.answer,
sources: sourcesPrepped,
tool_calls: answer.toolCalls,
},
}),
);
dispatch(
@@ -156,6 +167,7 @@ export const fetchAnswer = createAsyncThunk<
query: question,
result: '',
sources: [],
tool_calls: [],
};
});
@@ -212,6 +224,15 @@ export const conversationSlice = createSlice({
state.queries[index].sources!.push(query.sources![0]);
}
},
updateToolCalls(
state,
action: PayloadAction<{ index: number; query: Partial<Query> }>,
) {
const { index, query } = action.payload;
if (!state.queries[index].tool_calls) {
state.queries[index].tool_calls = query?.tool_calls;
}
},
updateQuery(
state,
action: PayloadAction<{ index: number; query: Partial<Query> }>,
@@ -263,6 +284,7 @@ export const {
updateStreamingQuery,
updateConversationId,
updateStreamingSource,
updateToolCalls,
setConversation,
} = conversationSlice.actions;
export default conversationSlice.reducer;