diff --git a/src/agent.py b/src/agent.py index d204b4e3..dc979e0f 100644 --- a/src/agent.py +++ b/src/agent.py @@ -221,9 +221,7 @@ async def chat( long_term_task = asyncio.create_task(fetch_long_term()) short_term_task = asyncio.create_task(run_tom_inference(chat_history, session_id)) - facts, tom_inference = await asyncio.gather( - long_term_task, short_term_task - ) + facts, tom_inference = await asyncio.gather(long_term_task, short_term_task) logger.debug(f"Retrieved {len(facts)} facts from long-term memory") logger.debug(f"TOM inference completed with {len(tom_inference)} characters") @@ -241,7 +239,9 @@ async def chat( message_id=latest_message_id, with_inference=False, ) - logger.debug(f"User representation generated: {len(user_representation)} characters") + logger.debug( + f"User representation generated: {len(user_representation)} characters" + ) # Create a Dialectic chain with the fresh user representation chain = Dialectic( diff --git a/src/deriver/tom/long_term.py b/src/deriver/tom/long_term.py index 54d791ea..a9a7f2c5 100644 --- a/src/deriver/tom/long_term.py +++ b/src/deriver/tom/long_term.py @@ -9,7 +9,6 @@ from sentry_sdk.ai.monitoring import ai_track from src.utils import parse_xml_content from src.utils.model_client import ModelClient, ModelProvider - # Configure logging logger = logging.getLogger(__name__)