Skip to content

Commit

Permalink
Merge pull request #920 from parea-ai/fix-cookbook-example
Browse files Browse the repository at this point in the history
fix(cookbook): pass varible into prompt template
  • Loading branch information
jalexanderII committed Jun 4, 2024
2 parents 5cfd670 + bb22374 commit b2d628d
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions parea/cookbook/langchain/trace_class_call_method.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit b2d628d

Please sign in to comment.