tkmind/tkmind_go
1
0
Fork 0
Code Issues Pull Requests Actions 4 Packages Projects Releases Wiki Activity

Include Session ID appropriately in UI (#4901)

Browse Source
This commit is contained in:
tlongwell-block
2025-10-01 12:35:24 -04:00
committed by GitHub
parent 3b9a3b96c5
commit 5bf698cc50
6 changed files with 44 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ui/desktop/src/components/context_management/ContextManager.tsx
+12 -6
View File
@@ -12,12 +12,14 @@ interface ContextManagerActions {
handleAutoCompaction: (
messages: Message[],
setMessages: (messages: Message[]) => void,
append: (message: Message) => void
append: (message: Message) => void,
sessionId: string
) => Promise<void>;
handleManualCompaction: (
messages: Message[],
setMessages: (messages: Message[]) => void,
append?: (message: Message) => void
append?: (message: Message) => void,
sessionId?: string
) => Promise<void>;
hasCompactionMarker: (message: Message) => boolean;
}
@@ -37,6 +39,7 @@ export const ContextManagerProvider: React.FC<{ children: React.ReactNode }> = (
messages: Message[],
setMessages: (messages: Message[]) => void,
append: (message: Message) => void,
sessionId: string,
isManual: boolean = false
) => {
setIsCompacting(true);
@@ -47,6 +50,7 @@ export const ContextManagerProvider: React.FC<{ children: React.ReactNode }> = (
const summaryResponse = await manageContextFromBackend({
messages: messages,
manageAction: 'summarize',
sessionId: sessionId,
});
// Convert API messages to frontend messages
@@ -100,9 +104,10 @@ export const ContextManagerProvider: React.FC<{ children: React.ReactNode }> = (
async (
messages: Message[],
setMessages: (messages: Message[]) => void,
append: (message: Message) => void
append: (message: Message) => void,
sessionId: string
) => {
await performCompaction(messages, setMessages, append, false);
await performCompaction(messages, setMessages, append, sessionId, false);
},
[performCompaction]
);
@@ -111,9 +116,10 @@ export const ContextManagerProvider: React.FC<{ children: React.ReactNode }> = (
async (
messages: Message[],
setMessages: (messages: Message[]) => void,
append?: (message: Message) => void
append?: (message: Message) => void,
sessionId?: string
) => {
await performCompaction(messages, setMessages, append || (() => {}), true);
await performCompaction(messages, setMessages, append || (() => {}), sessionId || '', true);
},
[performCompaction]
);
ui/desktop/src/components/context_management/__tests__/ContextManager.test.tsx
+10 -8
View File
@@ -166,12 +166,13 @@ describe('ContextManager', () => {
const { result } = renderContextManager();
await act(async () => {
await result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend);
await result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend, 'test-session-id');
});
expect(mockManageContextFromBackend).toHaveBeenCalledWith({
messages: mockMessages,
manageAction: 'summarize',
sessionId: 'test-session-id',
});
// Verify conversion calls with correct parameters
@@ -225,7 +226,7 @@ describe('ContextManager', () => {
const { result } = renderContextManager();
await act(async () => {
await result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend);
await result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend, "test-session-id");
});
expect(result.current.compactionError).toBe('Backend error');
@@ -256,7 +257,7 @@ describe('ContextManager', () => {
// Start compaction
act(() => {
result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend);
result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend, "test-session-id");
});
// Should be compacting
@@ -306,7 +307,7 @@ describe('ContextManager', () => {
const { result } = renderContextManager();
await act(async () => {
await result.current.handleAutoCompaction(messages, mockSetMessages, mockAppend);
await result.current.handleAutoCompaction(messages, mockSetMessages, mockAppend, "test-session-id");
});
// No server messages -> setMessages called with empty list
@@ -369,12 +370,13 @@ describe('ContextManager', () => {
const { result } = renderContextManager();
await act(async () => {
await result.current.handleManualCompaction(mockMessages, mockSetMessages, mockAppend);
await result.current.handleManualCompaction(mockMessages, mockSetMessages, mockAppend, 'test-session-id');
});
expect(mockManageContextFromBackend).toHaveBeenCalledWith({
messages: mockMessages,
manageAction: 'summarize',
sessionId: 'test-session-id',
});
// Verify all three messages are set
@@ -481,7 +483,7 @@ describe('ContextManager', () => {
const { result } = renderContextManager();
await act(async () => {
await result.current.handleManualCompaction(mockMessages, mockSetMessages, mockAppend);
await result.current.handleManualCompaction(mockMessages, mockSetMessages, mockAppend, 'test-session-id');
});
// Verify all three messages are set
@@ -508,7 +510,7 @@ describe('ContextManager', () => {
const { result } = renderContextManager();
await act(async () => {
await result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend);
await result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend, "test-session-id");
});
expect(result.current.compactionError).toBe('Unknown error during compaction');
@@ -539,7 +541,7 @@ describe('ContextManager', () => {
const { result } = renderContextManager();
await act(async () => {
await result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend);
await result.current.handleAutoCompaction(mockMessages, mockSetMessages, mockAppend, "test-session-id");
});
// Should complete without error even if content is not text
ui/desktop/src/components/context_management/index.ts
+3 -1
View File
@@ -18,12 +18,14 @@ import { generateId } from 'ai';
export async function manageContextFromBackend({
messages,
manageAction,
sessionId,
}: {
messages: FrontendMessage[];
manageAction: 'truncation' | 'summarize';
sessionId: string;
}): Promise<ContextManageResponse> {
try {
const contextManagementRequest = { manageAction, messages };
const contextManagementRequest = { manageAction, messages, sessionId };
// Cast to the API-expected type
const result = await manageContext({