From 8a2cc14759744b1b9d3f9b0f0c6919379dcd5498 Mon Sep 17 00:00:00 2001 From: Reinier van der Leer Date: Mon, 16 Feb 2026 22:27:28 +0100 Subject: [PATCH] Make `get_store_listed_graphs` RPC-compatible --- .../backend/backend/copilot/tools/agent_generator/core.py | 2 +- autogpt_platform/backend/backend/data/graph.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/autogpt_platform/backend/backend/copilot/tools/agent_generator/core.py b/autogpt_platform/backend/backend/copilot/tools/agent_generator/core.py index 365bcbf244..8fbe267f74 100644 --- a/autogpt_platform/backend/backend/copilot/tools/agent_generator/core.py +++ b/autogpt_platform/backend/backend/copilot/tools/agent_generator/core.py @@ -286,7 +286,7 @@ async def search_marketplace_agents_for_generation( return [] graph_ids = [agent.agent_graph_id for agent in agents_with_graphs] - graphs = await graph_db().get_store_listed_graphs(*graph_ids) + graphs = await graph_db().get_store_listed_graphs(graph_ids) results: list[LibraryAgentSummary] = [] for agent in agents_with_graphs: diff --git a/autogpt_platform/backend/backend/data/graph.py b/autogpt_platform/backend/backend/data/graph.py index 94f99852e8..a9a92dcdd5 100644 --- a/autogpt_platform/backend/backend/data/graph.py +++ b/autogpt_platform/backend/backend/data/graph.py @@ -1147,14 +1147,14 @@ async def get_graph( return GraphModel.from_db(graph, for_export) -async def get_store_listed_graphs(*graph_ids: str) -> dict[str, GraphModel]: +async def get_store_listed_graphs(graph_ids: list[str]) -> dict[str, GraphModel]: """Batch-fetch multiple store-listed graphs by their IDs. Only returns graphs that have approved store listings (publicly available). Does not require permission checks since store-listed graphs are public. Args: - *graph_ids: Variable number of graph IDs to fetch + graph_ids: List of graph IDs to fetch Returns: Dict mapping graph_id to GraphModel for graphs with approved store listings