LLM/DocsGPT
1
0
Fork 0
mirror of https://github.com/arc53/DocsGPT.git synced 2025-11-30 09:03:15 +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

35
frontend/src/conversation/conversationHandlers.ts
View File

@@ -1,6 +1,7 @@
import conversationService from '../api/services/conversationService';
import { Doc } from '../models/misc';
import { Answer, FEEDBACK, RetrievalPayload } from './conversationModels';
import { ToolCallsType } from './types';
export function handleFetchAnswer(
question: string,
@@ -16,6 +17,7 @@ export function handleFetchAnswer(
result: any;
answer: any;
sources: any;
toolCalls: ToolCallsType[];
conversationId: any;
query: string;
}
@@ -23,13 +25,18 @@ export function handleFetchAnswer(
result: any;
answer: any;
sources: any;
toolCalls: ToolCallsType[];
query: string;
conversationId: any;
title: any;
}
> {
history = history.map((item) => {
return { prompt: item.prompt, response: item.response };
return {
prompt: item.prompt,
response: item.response,
tool_calls: item.tool_calls,
};
});
const payload: RetrievalPayload = {
question: question,
@@ -60,6 +67,7 @@ export function handleFetchAnswer(
query: question,
result,
sources: data.sources,
toolCalls: data.tool_calls,
conversationId: data.conversation_id,
};
});
@@ -78,7 +86,11 @@ export function handleFetchAnswerSteaming(
indx?: number,
): Promise<Answer> {
history = history.map((item) => {
return { prompt: item.prompt, response: item.response };
return {
prompt: item.prompt,
response: item.response,
tool_calls: item.tool_calls,
};
});
const payload: RetrievalPayload = {
question: question,
@@ -155,7 +167,11 @@ export function handleSearch(
token_limit: number,
) {
history = history.map((item) => {
return { prompt: item.prompt, response: item.response };
return {
prompt: item.prompt,
response: item.response,
tool_calls: item.tool_calls,
};
});
const payload: RetrievalPayload = {
question: question,
@@ -183,7 +199,11 @@ export function handleSearchViaApiKey(
history: Array<any> = [],
) {
history = history.map((item) => {
return { prompt: item.prompt, response: item.response };
return {
prompt: item.prompt,
response: item.response,
tool_calls: item.tool_calls,
};
});
return conversationService
.search({
@@ -230,7 +250,11 @@ export function handleFetchSharedAnswerStreaming( //for shared conversations
onEvent: (event: MessageEvent) => void,
): Promise<Answer> {
history = history.map((item) => {
return { prompt: item.prompt, response: item.response };
return {
prompt: item.prompt,
response: item.response,
tool_calls: item.tool_calls,
};
});
return new Promise<Answer>((resolve, reject) => {
@@ -330,6 +354,7 @@ export function handleFetchSharedAnswer(
query: question,
result,
sources: data.sources,
toolCalls: data.tool_calls,
};
});
}