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

update rag agent - rename content->context

Browse source
This commit is contained in:
Adil Hafeez 2025-10-08 10:49:50 -07:00
parent 39a97d4a89
commit 496e7d34cc
5 changed files with 81 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

6
demos/use_cases/rag_agent/src/rag_agent/__init__.py
View file

@ -25,8 +25,10 @@ def main(host, port, agent, transport, rest_server, rest_port):
start_server(host=host, port=rest_port)
return
elif agent == "content_builder":
from rag_agent.content_builder_agent import start_server
elif agent == "context_builder":
from rag_agent.context_builder_agent import (
start_server,
)
start_server(host=host, port=rest_port)
return

26
demos/use_cases/rag_agent/src/rag_agent/content_builder_agent.py → demos/use_cases/rag_agent/src/rag_agent/context_builder_agent.py
View file

@ -252,4 +252,28 @@ if __name__ == "__main__":
def start_server(host: str = "localhost", port: int = 8000):
"""Start the REST server."""
load_knowledge_base()
uvicorn.run(app, host=host, port=port)
uvicorn.run(
app,
host=host,
port=port,
log_config={
"version": 1,
"disable_existing_loggers": False,
"formatters": {
"default": {
"format": "%(asctime)s - [CONTEXT_BUILDER] - %(name)s - %(levelname)s - %(message)s",
},
},
"handlers": {
"default": {
"formatter": "default",
"class": "logging.StreamHandler",
"stream": "ext://sys.stdout",
},
},
"root": {
"level": "INFO",
"handlers": ["default"],
},
},
)

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

@ -161,4 +161,28 @@ def parse_query(query):
def start_server(host: str = "localhost", port: int = 8000):
"""Start the REST server."""
uvicorn.run(app, host=host, port=port)
uvicorn.run(
app,
host=host,
port=port,
log_config={
"version": 1,
"disable_existing_loggers": False,
"formatters": {
"default": {
"format": "%(asctime)s - [QUERY_REWRITER] - %(name)s - %(levelname)s - %(message)s",
},
},
"handlers": {
"default": {
"formatter": "default",
"class": "logging.StreamHandler",
"stream": "ext://sys.stdout",
},
},
"root": {
"level": "INFO",
"handlers": ["default"],
},
},
)

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

@ -275,4 +275,28 @@ async def health_check():
def start_server(host: str = "localhost", port: int = 8000):
"""Start the REST server."""
uvicorn.run(app, host=host, port=port)
uvicorn.run(
app,
host=host,
port=port,
log_config={
"version": 1,
"disable_existing_loggers": False,
"formatters": {
"default": {
"format": "%(asctime)s - [RESPONSE_GENERATOR] - %(name)s - %(levelname)s - %(message)s",
},
},
"handlers": {
"default": {
"formatter": "default",
"class": "logging.StreamHandler",
"stream": "ext://sys.stdout",
},
},
"root": {
"level": "INFO",
"handlers": ["default"],
},
},
)

4
demos/use_cases/rag_agent/start_agents.sh
View file

@ -25,8 +25,8 @@ log "Starting query_parser agent on port 10500..."
uv run python -m rag_agent --rest-server --host 0.0.0.0 --rest-port 10500 --agent query_parser &
WAIT_FOR_PIDS+=($!)
log "Starting content_builder agent on port 10501..."
uv run python -m rag_agent --rest-server --host 0.0.0.0 --rest-port 10501 --agent content_builder &
log "Starting context_builder agent on port 10501..."
uv run python -m rag_agent --rest-server --host 0.0.0.0 --rest-port 10501 --agent context_builder &
WAIT_FOR_PIDS+=($!)
log "Starting response_generator agent on port 10502..."