Browse Source

Add MCP tool to get accounts

pull/7744/head
Thomas Kaul 1 day ago
parent
commit
88ea00965e
  1. 163
      apps/api/src/app/endpoints/ai/ai.service.ts
  2. 27
      apps/api/src/app/endpoints/mcp/mcp.controller.ts

163
apps/api/src/app/endpoints/ai/ai.service.ts

@ -24,6 +24,40 @@ import type { ColumnDescriptor } from 'tablemark';
@Injectable() @Injectable()
export class AiService { export class AiService {
private static readonly ACCOUNTS_TABLE_COLUMN_DEFINITIONS: ({
key:
| 'ACTIVITIES_COUNT'
| 'ALLOCATION_PERCENTAGE'
| 'BALANCE'
| 'CURRENCY'
| 'NAME'
| 'PLATFORM'
| 'VALUE';
requiresScopeToReadValues?: boolean;
} & ColumnDescriptor)[] = [
{ key: 'NAME', name: 'Name' },
{ key: 'CURRENCY', name: 'Currency' },
{ key: 'PLATFORM', name: 'Platform' },
{ align: 'right', key: 'ACTIVITIES_COUNT', name: 'Activities Count' },
{
align: 'right',
key: 'BALANCE',
name: 'Cash Balance',
requiresScopeToReadValues: true
},
{
align: 'right',
key: 'VALUE',
name: 'Value',
requiresScopeToReadValues: true
},
{
align: 'right',
key: 'ALLOCATION_PERCENTAGE',
name: 'Allocation in Percentage'
}
];
private static readonly ACTIVITIES_TABLE_COLUMN_DEFINITIONS: ({ private static readonly ACTIVITIES_TABLE_COLUMN_DEFINITIONS: ({
key: key:
| 'ACCOUNT' | 'ACCOUNT'
@ -98,16 +132,30 @@ export class AiService {
private readonly propertyService: PropertyService private readonly propertyService: PropertyService
) {} ) {}
public static getActivitiesTableColumnNames({ public static getAccountsTableColumnNames({
withValues withValues
}: { }: {
withValues: boolean; withValues: boolean;
}) { }) {
return AiService.getActivitiesTableColumnDefinitions({ withValues }).map( return AiService.getTableColumnDefinitions({
({ name }) => { withValues,
columnDefinitions: AiService.ACCOUNTS_TABLE_COLUMN_DEFINITIONS
}).map(({ name }) => {
return name; return name;
});
} }
);
public static getActivitiesTableColumnNames({
withValues
}: {
withValues: boolean;
}) {
return AiService.getTableColumnDefinitions({
withValues,
columnDefinitions: AiService.ACTIVITIES_TABLE_COLUMN_DEFINITIONS
}).map(({ name }) => {
return name;
});
} }
public static getHoldingsTableColumnNames() { public static getHoldingsTableColumnNames() {
@ -116,16 +164,18 @@ export class AiService {
}); });
} }
private static getActivitiesTableColumnDefinitions({ private static getTableColumnDefinitions<
T extends { requiresScopeToReadValues?: boolean }
>({
columnDefinitions,
withValues withValues
}: { }: {
columnDefinitions: readonly T[];
withValues: boolean; withValues: boolean;
}) { }) {
return AiService.ACTIVITIES_TABLE_COLUMN_DEFINITIONS.filter( return columnDefinitions.filter(({ requiresScopeToReadValues }) => {
({ requiresScopeToReadValues }) => {
return withValues || !requiresScopeToReadValues; return withValues || !requiresScopeToReadValues;
} });
);
} }
public async generateText({ public async generateText({
@ -154,6 +204,96 @@ export class AiService {
}); });
} }
public async getAccountsTable({
filters,
userId,
withValues
}: {
filters?: Filter[];
userId: string;
withValues: boolean;
}) {
const { accounts } =
await this.portfolioService.getAccountsWithAggregations({
filters,
userId,
withExcludedAccounts: true
});
const accountsTableColumnDefinitions = AiService.getTableColumnDefinitions({
withValues,
columnDefinitions: AiService.ACCOUNTS_TABLE_COLUMN_DEFINITIONS
});
const accountsTableRows = accounts.map(
({
activitiesCount,
allocationInPercentage,
balance,
currency,
name: label,
platform,
value
}) => {
return accountsTableColumnDefinitions.reduce(
(row, { key, name }) => {
switch (key) {
case 'ACTIVITIES_COUNT':
row[name] = activitiesCount.toString();
break;
case 'ALLOCATION_PERCENTAGE':
row[name] = `${(allocationInPercentage * 100).toFixed(3)}%`;
break;
case 'BALANCE':
row[name] = balance.toString();
break;
case 'CURRENCY':
row[name] = currency ?? '';
break;
case 'NAME':
row[name] = label ?? '';
break;
case 'PLATFORM':
row[name] = platform?.name ?? '';
break;
case 'VALUE':
row[name] = value.toString();
break;
default:
row[name] = '';
break;
}
return row;
},
{} as Record<string, string>
);
}
);
const accountsSection = ['## Accounts', ''];
if (accountsTableRows.length > 0) {
accountsSection.push(
await this.getMarkdownTable({
columnDefinitions: accountsTableColumnDefinitions,
rows: accountsTableRows
})
);
} else {
accountsSection.push('No accounts found.');
}
return accountsSection.join('\n');
}
public async getActivitiesTable({ public async getActivitiesTable({
endDate, endDate,
filters, filters,
@ -191,7 +331,10 @@ export class AiService {
}); });
const activitiesTableColumnDefinitions = const activitiesTableColumnDefinitions =
AiService.getActivitiesTableColumnDefinitions({ withValues }); AiService.getTableColumnDefinitions({
withValues,
columnDefinitions: AiService.ACTIVITIES_TABLE_COLUMN_DEFINITIONS
});
const activitiesTableRows = activities.map( const activitiesTableRows = activities.map(
({ ({

27
apps/api/src/app/endpoints/mcp/mcp.controller.ts

@ -71,6 +71,33 @@ export class GhostfolioMcpController {
private readonly apiService: ApiService private readonly apiService: ApiService
) {} ) {}
@RequiresScopeOfAccess(scopes.accountRead)
@Tool({
annotations: {
openWorldHint: false,
readOnlyHint: true,
title: 'Get accounts'
},
description: `Gives the accounts of the portfolio with these columns: ${AiService.getAccountsTableColumnNames(
{ withValues: false }
).join(
', '
)}. More columns with a monetary value are added if the access grants to read them.`,
name: 'get-accounts'
})
public async getAccounts(
@Impersonation()
{ filters, scopes: scopesOfAccess, userId }: ImpersonationContext
) {
const table = await this.aiService.getAccountsTable({
filters,
userId,
withValues: hasScope(scopesOfAccess, scopes.portfolioReadValues)
});
return { content: [{ text: table, type: 'text' as const }] };
}
@RequiresScopeOfAccess(scopes.activityRead) @RequiresScopeOfAccess(scopes.activityRead)
@Tool({ @Tool({
annotations: { annotations: {

Loading…
Cancel
Save