jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

🐛 fix(prompt_runner.py): fix issue with accessing result content when invoking chain with an empty dict input

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-04 00:20:16 -03:00
commit 929d51ea61
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/components/chains/prompt_runner.py
View file

@ -25,7 +25,9 @@ class PromptRunner(CustomComponent):
prompt: PromptTemplate, prompt: PromptTemplate,
) -> Document: ) -> Document:
chain = prompt | llm chain = prompt | llm
result = chain.invoke() # The input is an empty dict because the prompt is already filled
result = result[chain.output_key] result = chain.invoke({})
if hasattr(result, "content"):
result = result.content
self.repr_value = result self.repr_value = result
return Document(page_content=str(result)) return Document(page_content=str(result))