From e2331a2bff6228ed22c6083ffb6d057af38f70ac Mon Sep 17 00:00:00 2001 From: zmezei Date: Thu, 15 Feb 2024 12:46:49 +0000 Subject: [PATCH] rename var --- .../langchain-rag/src/agents/langchain_rag_user.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/integrations/langchain-rag/src/agents/langchain_rag_user.py b/integrations/langchain-rag/src/agents/langchain_rag_user.py index 56f5aeb0..efc90339 100644 --- a/integrations/langchain-rag/src/agents/langchain_rag_user.py +++ b/integrations/langchain-rag/src/agents/langchain_rag_user.py @@ -17,10 +17,10 @@ endpoint=["http://127.0.0.1:8000/submit"], ) -langchain_rag_user = Protocol("Langchain RAG user") +rag_user = Protocol("Langchain RAG user") -@langchain_rag_user.on_interval(60, messages=RagRequest) +@rag_user.on_interval(60, messages=RagRequest) async def ask_question(ctx: Context): ctx.logger.info( f"Asking RAG agent to answer {QUESTION} based on document located at {URL}, readin nested pages too: {DEEP_READ}" @@ -30,12 +30,12 @@ async def ask_question(ctx: Context): ) -@langchain_rag_user.on_message(model=UAgentResponse) +@rag_user.on_message(model=UAgentResponse) async def handle_data(ctx: Context, sender: str, data: UAgentResponse): ctx.logger.info(f"Got response from RAG agent: {data.message}") -user.include(langchain_rag_user) +user.include(rag_user) if __name__ == "__main__": - langchain_rag_user.run() + rag_user.run()