From bb22374056b975281ba11b36d9ed5d4d2ab4d8b2 Mon Sep 17 00:00:00 2001 From: Joel Alexander Date: Tue, 4 Jun 2024 11:01:31 -0400 Subject: [PATCH] fix(cookbook): pass varible into prompt template --- parea/cookbook/langchain/trace_class_call_method.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/parea/cookbook/langchain/trace_class_call_method.py b/parea/cookbook/langchain/trace_class_call_method.py index bd85bd43..fdbe50b7 100644 --- a/parea/cookbook/langchain/trace_class_call_method.py +++ b/parea/cookbook/langchain/trace_class_call_method.py @@ -27,10 +27,7 @@ def get_chain(self): @trace(name="langchain_caller_call") def __call__(self, query: str) -> str: chain = self.get_chain() - return chain.invoke( - {"input": "Write a Hello World program in Python using FastAPI."}, - config={"callbacks": [self.handler]}, - ) + return chain.invoke({"input": query}, config={"callbacks": [self.handler]}) class LLMCaller: