diff --git a/generic_rag/app.py b/generic_rag/app.py index f08d014..f2e54ae 100644 --- a/generic_rag/app.py +++ b/generic_rag/app.py @@ -13,8 +13,8 @@ from generic_rag.graphs.cond_ret_gen import CondRetGenLangGraph from generic_rag.graphs.ret_gen import RetGenLangGraph from generic_rag.parsers.parser import add_pdf_files, add_urls -logging.basicConfig(level=logging.DEBUG) -logger = logging.getLogger(__name__) +logger = logging.getLogger("sogeti-rag") +logger.setLevel(logging.DEBUG) system_prompt = ( "You are an assistant for question-answering tasks. " diff --git a/generic_rag/graphs/cond_ret_gen.py b/generic_rag/graphs/cond_ret_gen.py index b2057be..2b155a0 100644 --- a/generic_rag/graphs/cond_ret_gen.py +++ b/generic_rag/graphs/cond_ret_gen.py @@ -16,7 +16,7 @@ from langgraph.graph import END, MessagesState, StateGraph from langgraph.prebuilt import InjectedStore, ToolNode, tools_condition from typing_extensions import Annotated -logger = logging.getLogger(__name__) +logger = logging.getLogger("sogeti-rag") class CondRetGenLangGraph: def __init__( diff --git a/generic_rag/graphs/ret_gen.py b/generic_rag/graphs/ret_gen.py index b293e03..6cdd4f2 100644 --- a/generic_rag/graphs/ret_gen.py +++ b/generic_rag/graphs/ret_gen.py @@ -12,7 +12,7 @@ from langgraph.checkpoint.memory import MemorySaver from langgraph.graph import END, START, StateGraph from typing_extensions import List, TypedDict -logger = logging.getLogger(__name__) +logger = logging.getLogger("sogeti-rag") class State(TypedDict):