Skip to content

Integrate Supermemory for Long-term Agent Memory #529

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ngoiyaeric wants to merge 3 commits into
base: main
Choose a base branch
from
Open

Integrate Supermemory for Long-term Agent Memory #529

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
d5b9972
feat: integrate Supermemory for agent memory and preference learning
google-labs-jules[bot] Feb 18, 2026
b358433
feat: integrate Supermemory and fix client-side build error
google-labs-jules[bot] Feb 18, 2026
3bd50a0
feat: finalize Supermemory integration and model selection logic
google-labs-jules[bot] Feb 18, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
37 changes: 12 additions & 25 deletions app/actions.tsx
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ import { writer } from '@/lib/agents/writer'
import { saveChat, getSystemPrompt } from '@/lib/actions/chat'
import { Chat, AIMessage } from '@/lib/types'
import { UserMessage } from '@/components/user-message'
import { getCurrentUserIdOnServer } from '@/lib/auth/get-current-user'
import { BotMessage } from '@/components/message'
import { SearchSection } from '@/components/search-section'
import SearchRelated from '@/components/search-related'
Expand All @@ -41,6 +42,9 @@ async function submit(formData?: FormData, skip?: boolean) {
const isGenerating = createStreamableValue(true)
const isCollapsed = createStreamableValue(false)

const actualUserId = await getCurrentUserIdOnServer()
const userId = actualUserId || undefined

const action = formData?.get('action') as string;
const drawnFeaturesString = formData?.get('drawnFeatures') as string;
let drawnFeatures: DrawnFeature[] = [];
Expand Down Expand Up @@ -101,7 +105,7 @@ async function submit(formData?: FormData, skip?: boolean) {

async function processResolutionSearch() {
try {
const streamResult = await resolutionSearch(messages, timezone, drawnFeatures, location);
const streamResult = await resolutionSearch(messages, timezone, drawnFeatures, location, userId, aiState.get().chatId);

let fullSummary = '';
for await (const partialObject of streamResult.partialObjectStream) {
Expand Down Expand Up @@ -147,7 +151,7 @@ async function submit(formData?: FormData, skip?: boolean) {
}
return m
});
const relatedQueries = await querySuggestor(uiStream, sanitizedMessages);
const relatedQueries = await querySuggestor(uiStream, sanitizedMessages, userId, aiState.get().chatId);
uiStream.append(
<Section title="Follow-up">
<FollowupPanel />
Expand Down Expand Up @@ -397,21 +401,20 @@ async function submit(formData?: FormData, skip?: boolean) {
} as CoreMessage)
}

const userId = 'anonymous'
const currentSystemPrompt = (await getSystemPrompt(userId)) || ''
const currentSystemPrompt = (userId ? await getSystemPrompt(userId) : null) || ''
const mapProvider = formData?.get('mapProvider') as 'mapbox' | 'google'

async function processEvents() {
let action: any = { object: { next: 'proceed' } }
if (!skip) {
const taskManagerResult = await taskManager(messages)
const taskManagerResult = await taskManager(messages, userId, aiState.get().chatId)
if (taskManagerResult) {
action.object = taskManagerResult.object
}
}

if (action.object.next === 'inquire') {
const inquiry = await inquire(uiStream, messages)
const inquiry = await inquire(uiStream, messages, userId, aiState.get().chatId)
uiStream.done()
isGenerating.done()
isCollapsed.done(false)
Expand Down Expand Up @@ -441,15 +444,7 @@ async function submit(formData?: FormData, skip?: boolean) {
? answer.length === 0
: answer.length === 0 && !errorOccurred
) {
const { fullResponse, hasError, toolResponses } = await researcher(
currentSystemPrompt,
uiStream,
streamText,
messages,
mapProvider,
useSpecificAPI,
drawnFeatures
)
const { fullResponse, hasError, toolResponses } = await researcher(currentSystemPrompt, uiStream, streamText, messages, mapProvider, useSpecificAPI, drawnFeatures, userId, aiState.get().chatId)
answer = fullResponse
toolOutputs = toolResponses
errorOccurred = hasError
Expand Down Expand Up @@ -487,18 +482,13 @@ async function submit(formData?: FormData, skip?: boolean) {
: msg
) as CoreMessage[]
const latestMessages = modifiedMessages.slice(maxMessages * -1)
answer = await writer(
currentSystemPrompt,
uiStream,
streamText,
latestMessages
)
answer = await writer(currentSystemPrompt, uiStream, streamText, latestMessages, userId, aiState.get().chatId)
} else {
streamText.done()
}

if (!errorOccurred) {
const relatedQueries = await querySuggestor(uiStream, messages)
const relatedQueries = await querySuggestor(uiStream, messages, userId, aiState.get().chatId)
uiStream.append(
<Section title="Follow-up">
<FollowupPanel />
Expand Down Expand Up @@ -637,9 +627,6 @@ export const AI = createAI<AIState, UIState>({
}
]

const { getCurrentUserIdOnServer } = await import(
'@/lib/auth/get-current-user'
)
const actualUserId = await getCurrentUserIdOnServer()

if (!actualUserId) {
Expand Down
Loading