Revert logging changes
This commit is contained in:
parent
586f342a77
commit
8933f12b4d
3 changed files with 0 additions and 14 deletions
|
|
@ -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]:
|
||||
|
|
|
|||
|
|
@ -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(
|
||||
|
|
|
|||
|
|
@ -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],
|
||||
|
|
|
|||
Loading…
Reference in a new issue