apunkt/plano
1
0
Fork 0
mirror of https://github.com/katanemo/plano.git synced 2026-06-17 15:25:17 +02:00
Code Issues Projects Releases Packages Wiki Activity

fix logs

Browse source
This commit is contained in:
Adil Hafeez 2025-10-08 11:20:13 -07:00
parent 496e7d34cc
commit b3a19b1df0
No known key found for this signature in database
GPG key ID: 9B18EF7691369645
3 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
demos/use_cases/rag_agent/src/rag_agent/context_builder_agent.py
View file

@ -15,7 +15,7 @@ from .api import ChatMessage, ChatCompletionRequest, ChatCompletionResponse
# Set up logging
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - [CONTEXT_BUILDER] - %(levelname)s - %(message)s",
format="%(asctime)s - [CONTEXT_BUILDER] - %(levelname)s - %(message)s",
)
logger = logging.getLogger(__name__)
@ -252,6 +252,7 @@ if __name__ == "__main__":
def start_server(host: str = "localhost", port: int = 8000):
"""Start the REST server."""
load_knowledge_base()
# Rename the uvicorn.error logger
uvicorn.run(
app,
host=host,
@ -261,7 +262,7 @@ def start_server(host: str = "localhost", port: int = 8000):
"disable_existing_loggers": False,
"formatters": {
"default": {
"format": "%(asctime)s - [CONTEXT_BUILDER] - %(name)s - %(levelname)s - %(message)s",
"format": "%(asctime)s - [CONTEXT_BUILDER] - %(levelname)s - %(message)s",
},
},
"handlers": {

4
demos/use_cases/rag_agent/src/rag_agent/query_rewriter_agent.py
View file

@ -13,7 +13,7 @@ from .api import ChatMessage, ChatCompletionRequest, ChatCompletionResponse
# Set up logging
logging.basicConfig(
level=logging.INFO,
format="%(asctime)s - [QUERY_REWRITER] - %(levelname)s - %(message)s",
format="%(asctime)s - [QUERY_REWRITER] - %(levelname)s - %(message)s",
)
logger = logging.getLogger(__name__)
@ -170,7 +170,7 @@ def start_server(host: str = "localhost", port: int = 8000):
"disable_existing_loggers": False,
"formatters": {
"default": {
"format": "%(asctime)s - [QUERY_REWRITER] - %(name)s - %(levelname)s - %(message)s",
"format": "%(asctime)s - [QUERY_REWRITER] - %(levelname)s - %(message)s",
},
},
"handlers": {

2
demos/use_cases/rag_agent/src/rag_agent/response_generator_agent.py
View file

@ -284,7 +284,7 @@ def start_server(host: str = "localhost", port: int = 8000):
"disable_existing_loggers": False,
"formatters": {
"default": {
"format": "%(asctime)s - [RESPONSE_GENERATOR] - %(name)s - %(levelname)s - %(message)s",
"format": "%(asctime)s - [RESPONSE_GENERATOR] - %(levelname)s - %(message)s",
},
},
"handlers": {