From 8933f12b4daa323648b191a39ff2594d790f1c45 Mon Sep 17 00:00:00 2001 From: Yiorgis Gozadinos Date: Mon, 20 Oct 2025 19:22:42 +0300 Subject: [PATCH] Revert logging changes --- src/haiku/rag/client.py | 1 - src/haiku/rag/logging.py | 1 - src/haiku/rag/qa/agent.py | 12 ------------ 3 files changed, 14 deletions(-) diff --git a/src/haiku/rag/client.py b/src/haiku/rag/client.py index 47b6408a..686d7414 100644 --- a/src/haiku/rag/client.py +++ b/src/haiku/rag/client.py @@ -582,7 +582,6 @@ class HaikuRAG: from haiku.rag.qa import get_qa_agent qa_agent = get_qa_agent(self, use_citations=cite, system_prompt=system_prompt) - logger.debug("Using QA agent: %s", qa_agent.__class__.__name__) return await qa_agent.answer(question) async def rebuild_database(self) -> AsyncGenerator[str, None]: diff --git a/src/haiku/rag/logging.py b/src/haiku/rag/logging.py index 24bb90fc..d727f84e 100644 --- a/src/haiku/rag/logging.py +++ b/src/haiku/rag/logging.py @@ -7,7 +7,6 @@ from rich.logging import RichHandler def get_logger() -> logging.Logger: """Return the library logger configured with a Rich handler.""" - logger = logging.getLogger("haiku.rag") handler = RichHandler( diff --git a/src/haiku/rag/qa/agent.py b/src/haiku/rag/qa/agent.py index 9da9b036..f1abe7ab 100644 --- a/src/haiku/rag/qa/agent.py +++ b/src/haiku/rag/qa/agent.py @@ -1,5 +1,3 @@ -import logging - from pydantic import BaseModel, Field from pydantic_ai import Agent, RunContext from pydantic_ai.models.openai import OpenAIChatModel @@ -10,8 +8,6 @@ from haiku.rag.client import HaikuRAG from haiku.rag.config import Config from haiku.rag.qa.prompts import QA_SYSTEM_PROMPT, QA_SYSTEM_PROMPT_WITH_CITATIONS -logger = logging.getLogger(__name__) - class SearchResult(BaseModel): content: str = Field(description="The document text content") @@ -51,14 +47,6 @@ class QuestionAnswerAgent: retries=3, ) - logger.info( - "Initialized QuestionAnswerAgent %s, with agent: %s, model: %s, client: %s", - self, - self._agent, - model_obj, - client, - ) - @self._agent.tool async def search_documents( ctx: RunContext[Dependencies],