diff --git a/redbox-core/redbox/app.py b/redbox-core/redbox/app.py index 339de9432..3d997273a 100644 --- a/redbox-core/redbox/app.py +++ b/redbox-core/redbox/app.py @@ -1,4 +1,6 @@ +from logging import getLogger from typing import Literal + from langchain_core.embeddings import Embeddings from langchain_core.tools import StructuredTool from langchain_core.vectorstores import VectorStoreRetriever @@ -29,6 +31,9 @@ async def _default_callback(*args, **kwargs): return None +logger = getLogger(__name__) + + class Redbox: def __init__( self, @@ -91,6 +96,8 @@ async def run( activity_event_callback=_default_callback, ) -> RedboxState: final_state = None + request_dict = input["request"].model_dump() + logger.info("Request: %s", {k: request_dict[k] for k in request_dict.keys() - {"ai_settings"}}) async for event in self.graph.astream_events( input=input, version="v2",