Multi-Rag / src /pipeline /GraphRunner_pipeline.py
VashuTheGreat2's picture
Upload folder using huggingface_hub
9c90775 verified
Raw
History Blame Contribute Delete
477 Bytes
from src.utils.asyncHandler import asyncHandler
from src.components.run_graph import RunGraph
import logging
class RunGraphPipeline:
def __init__(self):
pass
@asyncHandler
async def run_graph(self, state: dict, config: dict = None) -> dict:
logging.info("Running graph pipeline...")
runner = RunGraph()
response = await runner.run(state, config=config)
logging.info("Graph pipeline completed.")
return response