Skip to content

Commit a12af33

Browse files
committed
Formatting, small unit test fixes
1 parent 3608f21 commit a12af33

File tree

2 files changed

+4
-2
lines changed

2 files changed

+4
-2
lines changed

lib/idp_common_pkg/idp_common/agents/external_mcp/agent.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,7 +60,9 @@ def create_external_mcp_agent(
6060
"cognito_username",
6161
"cognito_password",
6262
]
63-
missing_fields = [field for field in required_fields if field not in mcp_server_config]
63+
missing_fields = [
64+
field for field in required_fields if field not in mcp_server_config
65+
]
6466
if missing_fields:
6567
error_msg = f"MCP server config missing required fields: {missing_fields}"
6668
logger.error(error_msg)

lib/idp_common_pkg/tests/unit/agents/common/test_monitoring.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def test_tracker_initialization_disabled(self):
126126
assert not tracker.enabled
127127
assert tracker.db_logger is None
128128

129-
@patch.dict("os.environ", {"ANALYTICS_TABLE": "test-table"})
129+
@patch.dict("os.environ", {"AGENT_TABLE": "test-table"})
130130
@patch("idp_common.agents.common.dynamodb_logger.DynamoDBMessageLogger")
131131
def test_tracker_initialization_enabled(self, mock_logger_class):
132132
"""Test tracker initialization when enabled."""

0 commit comments

Comments
 (0)