Skip to content

Commit 796a203

Browse files
committed
Linting fixes
Signed-off-by: Madhav Kandukuri <[email protected]>
1 parent b25911e commit 796a203

File tree

1 file changed

+9
-15
lines changed

1 file changed

+9
-15
lines changed

mcpgateway/services/gateway_service.py

Lines changed: 9 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -1156,13 +1156,13 @@ async def update_gateway(
11561156
new_prompt_names = [prompt.name for prompt in prompts]
11571157

11581158
# Update tools using helper method
1159-
tools_to_add = self._update_or_create_tools(db, tools, gateway, "update", append_to_gateway=True)
1159+
tools_to_add = self._update_or_create_tools(db, tools, gateway, "update")
11601160

11611161
# Update resources using helper method
1162-
resources_to_add = self._update_or_create_resources(db, resources, gateway, "update", append_to_gateway=True)
1162+
resources_to_add = self._update_or_create_resources(db, resources, gateway, "update")
11631163

11641164
# Update prompts using helper method
1165-
prompts_to_add = self._update_or_create_prompts(db, prompts, gateway, "update", append_to_gateway=True)
1165+
prompts_to_add = self._update_or_create_prompts(db, prompts, gateway, "update")
11661166

11671167
# Log newly added items
11681168
items_added = len(tools_to_add) + len(resources_to_add) + len(prompts_to_add)
@@ -1375,9 +1375,9 @@ async def toggle_gateway_status(self, db: Session, gateway_id: str, activate: bo
13751375
new_prompt_names = [prompt.name for prompt in prompts]
13761376

13771377
# Update tools, resources, and prompts using helper methods
1378-
tools_to_add = self._update_or_create_tools(db, tools, gateway, "rediscovery", append_to_gateway=True)
1379-
resources_to_add = self._update_or_create_resources(db, resources, gateway, "rediscovery", append_to_gateway=True)
1380-
prompts_to_add = self._update_or_create_prompts(db, prompts, gateway, "rediscovery", append_to_gateway=True)
1378+
tools_to_add = self._update_or_create_tools(db, tools, gateway, "rediscovery")
1379+
resources_to_add = self._update_or_create_resources(db, resources, gateway, "rediscovery")
1380+
prompts_to_add = self._update_or_create_prompts(db, prompts, gateway, "rediscovery")
13811381

13821382
# Log newly added items
13831383
items_added = len(tools_to_add) + len(resources_to_add) + len(prompts_to_add)
@@ -2519,7 +2519,7 @@ def _create_db_tool(
25192519
visibility="public", # Federated tools should be public for discovery
25202520
)
25212521

2522-
def _update_or_create_tools(self, db: Session, tools: List[Any], gateway: DbGateway, created_via: str, append_to_gateway: bool = False) -> List[DbTool]:
2522+
def _update_or_create_tools(self, db: Session, tools: List[Any], gateway: DbGateway, created_via: str) -> List[DbTool]:
25232523
"""Helper to handle update-or-create logic for tools from MCP server.
25242524
25252525
Args:
@@ -2583,16 +2583,14 @@ def _update_or_create_tools(self, db: Session, tools: List[Any], gateway: DbGate
25832583
# Attach relationship to avoid NoneType during flush
25842584
db_tool.gateway = gateway
25852585
tools_to_add.append(db_tool)
2586-
if append_to_gateway:
2587-
gateway.tools.append(db_tool)
25882586
logger.debug(f"Created new tool: {tool.name}")
25892587
except Exception as e:
25902588
logger.warning(f"Failed to process tool {getattr(tool, 'name', 'unknown')}: {e}")
25912589
continue
25922590

25932591
return tools_to_add
25942592

2595-
def _update_or_create_resources(self, db: Session, resources: List[Any], gateway: DbGateway, created_via: str, append_to_gateway: bool = False) -> List[DbResource]:
2593+
def _update_or_create_resources(self, db: Session, resources: List[Any], gateway: DbGateway, created_via: str) -> List[DbResource]:
25962594
"""Helper to handle update-or-create logic for resources from MCP server.
25972595
25982596
Args:
@@ -2649,16 +2647,14 @@ def _update_or_create_resources(self, db: Session, resources: List[Any], gateway
26492647
visibility=gateway.visibility,
26502648
)
26512649
resources_to_add.append(db_resource)
2652-
if append_to_gateway:
2653-
gateway.resources.append(db_resource)
26542650
logger.debug(f"Created new resource: {resource.uri}")
26552651
except Exception as e:
26562652
logger.warning(f"Failed to process resource {getattr(resource, 'uri', 'unknown')}: {e}")
26572653
continue
26582654

26592655
return resources_to_add
26602656

2661-
def _update_or_create_prompts(self, db: Session, prompts: List[Any], gateway: DbGateway, created_via: str, append_to_gateway: bool = False) -> List[DbPrompt]:
2657+
def _update_or_create_prompts(self, db: Session, prompts: List[Any], gateway: DbGateway, created_via: str) -> List[DbPrompt]:
26622658
"""Helper to handle update-or-create logic for prompts from MCP server.
26632659
26642660
Args:
@@ -2710,8 +2706,6 @@ def _update_or_create_prompts(self, db: Session, prompts: List[Any], gateway: Db
27102706
visibility=gateway.visibility,
27112707
)
27122708
prompts_to_add.append(db_prompt)
2713-
if append_to_gateway:
2714-
gateway.prompts.append(db_prompt)
27152709
logger.debug(f"Created new prompt: {prompt.name}")
27162710
except Exception as e:
27172711
logger.warning(f"Failed to process prompt {getattr(prompt, 'name', 'unknown')}: {e}")

0 commit comments

Comments
 (0)