Skip to content

Commit 9468b95

Browse files
committed
feat: remove unnecessary logging in aggregator and schema retriever nodes
1 parent 4238067 commit 9468b95

3 files changed

Lines changed: 0 additions & 7 deletions

File tree

packages/core/src/nl2sql/pipeline/nodes/aggregator/node.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,9 +26,7 @@ def __init__(self, ctx: NL2SQLContext):
2626
def __call__(self, state: GraphState) -> Dict[str, Any]:
2727
try:
2828
planner_response = state.global_planner_response
29-
logger.info(f"Planner response: {planner_response.model_dump_json(indent=2)}")
3029
artifact_refs = state.artifact_refs
31-
logger.info(f"Artifact references: {artifact_refs}")
3230

3331
dag = planner_response.execution_dag
3432

packages/core/src/nl2sql/pipeline/nodes/ast_planner/node.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -74,8 +74,6 @@ def __call__(self, state: SubgraphExecutionState) -> Dict[str, Any]:
7474
}
7575
)
7676

77-
logger.info(f"Generated Plan: {plan.model_dump_json(indent=2)}")
78-
7977
return {
8078
"ast_planner_response": ASTPlannerResponse(plan=plan),
8179
"reasoning": [

packages/core/src/nl2sql/pipeline/nodes/schema_retriever/node.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,6 @@ def _build_tables_from_snapshot(
7070
resolved_tables: Optional[Dict[str, Set[str]]] = None,
7171
schema_version: Optional[str] = None,
7272
) -> List[Table]:
73-
logger.info(f"Building tables from snapshot: {snapshot.model_dump_json(indent=2)}")
7473
if not snapshot:
7574
return []
7675

@@ -201,7 +200,6 @@ def __call__(self, state: SubgraphExecutionState) -> Dict[str, Any]:
201200
resolved_tables=None,
202201
schema_version=schema_version,
203202
)
204-
logger.info(f"Length of relevant tables for planning: {len(relevant_tables)}")
205203
return {
206204
"relevant_tables": relevant_tables,
207205
"reasoning": [
@@ -227,7 +225,6 @@ def __call__(self, state: SubgraphExecutionState) -> Dict[str, Any]:
227225
)
228226

229227

230-
logger.info(f"Length of relevant tables for planning: {len(relevant_tables)}")
231228

232229
return {
233230
"relevant_tables": relevant_tables,

0 commit comments

Comments
 (0)