Skip to content

fix: scope agent memory files to selected agent in admin UI #1049

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

Merged
chubes4 merged 1 commit into from
Apr 14, 2026
Merged

fix: scope agent memory files to selected agent in admin UI #1049

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
48 changes: 38 additions & 10 deletions inc/Api/AgentFiles.php
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -438,9 +438,16 @@ public static function delete_agent_file( WP_REST_Request $request ) {
// =========================================================================

public static function list_daily_files( WP_REST_Request $request ) {
$result = DailyMemoryAbilities::listDaily( array(
$input = array(
'user_id' => self::resolve_scoped_user_id( $request ),
) );
);

$agent_id = $request->get_param( 'agent_id' );
if ( null !== $agent_id && '' !== $agent_id ) {
$input['agent_id'] = (int) $agent_id;
}

$result = DailyMemoryAbilities::listDaily( $input );

return rest_ensure_response( array(
'success' => true,
Expand All @@ -449,11 +456,18 @@ public static function list_daily_files( WP_REST_Request $request ) {
}

public static function get_daily_file( WP_REST_Request $request ) {
$date = sprintf( '%s-%s-%s', $request['year'], $request['month'], $request['day'] );
$result = DailyMemoryAbilities::readDaily( array(
$date = sprintf( '%s-%s-%s', $request['year'], $request['month'], $request['day'] );
$input = array(
'date' => $date,
'user_id' => self::resolve_scoped_user_id( $request ),
) );
);

$agent_id = $request->get_param( 'agent_id' );
if ( null !== $agent_id && '' !== $agent_id ) {
$input['agent_id'] = (int) $agent_id;
}

$result = DailyMemoryAbilities::readDaily( $input );

if ( ! $result['success'] ) {
return new WP_Error( 'daily_file_not_found', $result['message'], array( 'status' => 404 ) );
Expand All @@ -477,12 +491,19 @@ public static function put_daily_file( WP_REST_Request $request ) {
$content = $request->get_body();
}

$result = DailyMemoryAbilities::writeDaily( array(
$input = array(
'date' => $date,
'content' => $content,
'mode' => 'write',
'user_id' => self::resolve_scoped_user_id( $request ),
) );
);

$agent_id = $request->get_param( 'agent_id' );
if ( null !== $agent_id && '' !== $agent_id ) {
$input['agent_id'] = (int) $agent_id;
}

$result = DailyMemoryAbilities::writeDaily( $input );

if ( ! $result['success'] ) {
$status = false !== strpos( $result['message'] ?? '', 'disabled' ) ? 403 : 500;
Expand All @@ -497,11 +518,18 @@ public static function put_daily_file( WP_REST_Request $request ) {
}

public static function delete_daily_file( WP_REST_Request $request ) {
$date = sprintf( '%s-%s-%s', $request['year'], $request['month'], $request['day'] );
$result = DailyMemoryAbilities::deleteDaily( array(
$date = sprintf( '%s-%s-%s', $request['year'], $request['month'], $request['day'] );
$input = array(
'date' => $date,
'user_id' => self::resolve_scoped_user_id( $request ),
) );
);

$agent_id = $request->get_param( 'agent_id' );
if ( null !== $agent_id && '' !== $agent_id ) {
$input['agent_id'] = (int) $agent_id;
}

$result = DailyMemoryAbilities::deleteDaily( $input );

if ( ! $result['success'] ) {
$status = false !== strpos( $result['message'] ?? '', 'disabled' ) ? 403 : 404;
Expand Down
81 changes: 33 additions & 48 deletions inc/Core/Admin/Pages/Agent/assets/react/api/agentFiles.js
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -2,94 +2,79 @@
* Agent Files API
*
* REST client functions for agent memory file operations.
* Uses the shared API client for automatic agent interceptor support.
*/

/**
* WordPress dependencies
* Internal dependencies
*/
import apiFetch from '@wordpress/api-fetch';
import { client } from '@shared/utils/api';
import { useAgentStore } from '@shared/stores/agentStore';

const getConfig = () => {
const config = window.dataMachineAgentConfig || {};
return { restNamespace: config.restNamespace || 'datamachine/v1' };
/**
* Get agent_id params for mutation requests (PUT/DELETE).
* The shared client interceptor only injects into GET requests,
* so mutations must include agent_id explicitly.
*
* @return {Object} Object with agent_id if one is selected, empty otherwise.
*/
const getAgentParams = () => {
const { selectedAgentId } = useAgentStore.getState();
return selectedAgentId !== null ? { agent_id: selectedAgentId } : {};
};

export const listAgentFiles = async () => {
const config = getConfig();
return apiFetch( { path: `/${ config.restNamespace }/files/agent` } );
return client.get( '/files/agent' );
};

export const getAgentFile = async ( filename ) => {
const config = getConfig();
return apiFetch( {
path: `/${ config.restNamespace }/files/agent/${ filename }`,
} );
return client.get( `/files/agent/${ filename }` );
};

export const putAgentFile = async ( filename, content ) => {
const config = getConfig();
return apiFetch( {
path: `/${ config.restNamespace }/files/agent/${ filename }`,
method: 'PUT',
data: { content },
return client.put( `/files/agent/${ filename }`, {
content,
...getAgentParams(),
} );
};

export const deleteAgentFile = async ( filename ) => {
const config = getConfig();
return apiFetch( {
path: `/${ config.restNamespace }/files/agent/${ filename }`,
method: 'DELETE',
} );
return client.delete( `/files/agent/${ filename }`, getAgentParams() );
};

// Daily memory file operations.

export const listDailyFiles = async () => {
const config = getConfig();
return apiFetch( {
path: `/${ config.restNamespace }/files/agent/daily`,
} );
return client.get( '/files/agent/daily' );
};

export const getDailyFile = async ( year, month, day ) => {
const config = getConfig();
return apiFetch( {
path: `/${ config.restNamespace }/files/agent/daily/${ year }/${ month }/${ day }`,
} );
return client.get( `/files/agent/daily/${ year }/${ month }/${ day }` );
};

export const putDailyFile = async ( year, month, day, content ) => {
const config = getConfig();
return apiFetch( {
path: `/${ config.restNamespace }/files/agent/daily/${ year }/${ month }/${ day }`,
method: 'PUT',
data: { content },
return client.put( `/files/agent/daily/${ year }/${ month }/${ day }`, {
content,
...getAgentParams(),
} );
};

export const deleteDailyFile = async ( year, month, day ) => {
const config = getConfig();
return apiFetch( {
path: `/${ config.restNamespace }/files/agent/daily/${ year }/${ month }/${ day }`,
method: 'DELETE',
} );
return client.delete(
`/files/agent/daily/${ year }/${ month }/${ day }`,
getAgentParams()
);
};

// Context memory file operations.

export const getContextFile = async ( slug ) => {
const config = getConfig();
return apiFetch( {
path: `/${ config.restNamespace }/files/agent/contexts/${ slug }`,
} );
return client.get( `/files/agent/contexts/${ slug }` );
};

export const putContextFile = async ( slug, content ) => {
const config = getConfig();
return apiFetch( {
path: `/${ config.restNamespace }/files/agent/contexts/${ slug }`,
method: 'PUT',
data: { content },
return client.put( `/files/agent/contexts/${ slug }`, {
content,
...getAgentParams(),
} );
};
Loading
Loading