jketreno/backstory
1
0
Fork 0
Code Issues 12 Actions Releases Activity

Add assistant messages into context history.

Browse Source
This commit is contained in:
James Ketr 2025-03-31 15:43:57 -07:00
parent 2d5f6a2797
commit 254816136e
2 changed files with 28 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
src/ketr-chat/src/App.tsx
View File

@ -204,14 +204,38 @@ const App = () => {
} }
}, [conversation]); }, [conversation]);
// Set the snack pop-up and open it
const setSnack = useCallback((message: string, severity: SeverityType = "success") => {
setSnackMessage(message);
setSnackSeverity(severity);
setSnackOpen(true);
}, []);
// Set the initial chat history to "loading" or the welcome message if loaded. // Set the initial chat history to "loading" or the welcome message if loaded.
useEffect(() => { useEffect(() => {
if (sessionId === undefined) { if (sessionId === undefined) {
setConversation([loadingMessage]); setConversation([loadingMessage]);
} else { } else {
setConversation([welcomeMessage]); fetch(getConnectionBase(loc) + `/api/history/${sessionId}`, {
method: 'GET',
headers: {
'Content-Type': 'application/json',
},
})
.then(response => response.json())
.then(data => {
console.log(`Session id: ${sessionId} -- history returned from server with ${data.length} entries`)
setConversation([
welcomeMessage,
...data
]);
})
.catch(error => {
console.error('Error generating session ID:', error);
setSnack("Unable to obtain chat history.", "error");
});
} }
}, [sessionId, setConversation]); }, [sessionId, setConversation, loc, setSnack]);
// Extract the sessionId from the URL if present, otherwise // Extract the sessionId from the URL if present, otherwise
// request a sessionId from the server. // request a sessionId from the server.
@ -265,13 +289,6 @@ const App = () => {
fetchSystemPrompt(); fetchSystemPrompt();
}, [sessionId, serverSystemPrompt, setServerSystemPrompt, loc]); }, [sessionId, serverSystemPrompt, setServerSystemPrompt, loc]);
// Set the snack pop-up and open it
const setSnack = useCallback((message: string, severity: SeverityType = "success") => {
setSnackMessage(message);
setSnackSeverity(severity);
setSnackOpen(true);
}, []);
// If the tools have not been set, fetch them from the server // If the tools have not been set, fetch them from the server
useEffect(() => { useEffect(() => {
if (tools.length || sessionId === undefined) { if (tools.length || sessionId === undefined) {
@ -479,7 +496,7 @@ const App = () => {
type MessageList = Message[]; type MessageList = Message[];
const onNew = async () => { const onNew = async () => {
reset(["rags", "tools", "history", "system-prompt"], "New chat started."); reset(["history"], "New chat started.");
} }
const sendQuery = async () => { const sendQuery = async () => {

2
src/server.py
View File

@ -630,7 +630,7 @@ class WebServer:
final_message = {"role": "assistant", "content": reply, 'metadata': {"title": f"🛠️ Tool(s) used: {','.join(tools_used)}"}} final_message = {"role": "assistant", "content": reply, 'metadata': {"title": f"🛠️ Tool(s) used: {','.join(tools_used)}"}}
else: else:
final_message = {"role": "assistant", "content": reply} final_message = {"role": "assistant", "content": reply}
history.append(final_message)
yield {"status": "done", "message": final_message} yield {"status": "done", "message": final_message}
except Exception as e: except Exception as e: