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