dev: remove unused callback

Author: enitrat

python/src/cairo_coder/core/rag_pipeline.py

@@ -13,7 +13,6 @@
 import langsmith as ls
 import structlog
 from dspy.adapters import XMLAdapter
-from dspy.utils.callback import BaseCallback
 from langsmith import traceable
 
 from cairo_coder.core.config import VectorStoreConfig
@@ -35,30 +34,6 @@
 SOURCE_PREVIEW_MAX_LEN = 200
 
 
-# 1. Define a custom callback class that extends BaseCallback class
-class AgentLoggingCallback(BaseCallback):
-    def on_module_start(
-        self,
-        call_id: str,
-        instance: Any,
-        inputs: dict[str, Any],
-    ) -> None:
-        logger.debug("Starting module", call_id=call_id, inputs=inputs)
-
-    # 2. Implement on_module_end handler to run a custom logging code.
-    def on_module_end(
-        self, call_id: str, outputs: dict[str, Any], exception: Exception | None
-    ) -> None:
-        step = "Reasoning" if self._is_reasoning_output(outputs) else "Acting"
-        logger.debug(f"== {step} Step ===")
-        for k, v in outputs.items():
-            logger.debug(f"  {k}: {v}")
-        logger.debug("\n")
-
-    def _is_reasoning_output(self, outputs: dict[str, Any]) -> bool:
-        return any(k.startswith("Thought") for k in outputs if isinstance(k, str))
-
-
 @dataclass
 class RagPipelineConfig:
     """Configuration for RAG Pipeline."""

python/src/cairo_coder/server/app.py

@@ -26,10 +26,7 @@
 from cairo_coder.config.manager import ConfigManager
 from cairo_coder.core.agent_factory import AgentFactory, create_agent_factory
 from cairo_coder.core.config import VectorStoreConfig
-from cairo_coder.core.rag_pipeline import (
-    AgentLoggingCallback,
-    RagPipeline,
-)
+from cairo_coder.core.rag_pipeline import RagPipeline
 from cairo_coder.core.types import Message, Role, StreamEventType
 from cairo_coder.dspy.document_retriever import SourceFilteredPgVectorRM
 from cairo_coder.dspy.suggestion_program import SuggestionGeneration
@@ -188,7 +185,6 @@ def __init__(
             lm=dspy.LM("gemini/gemini-flash-latest", max_tokens=30000, cache=False),
             adapter=ChatAdapter(),
             embedder=embedder,
-            callbacks=[AgentLoggingCallback()],
             track_usage=True,
         )