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

Add typing imports and update return type in LLMMathChainComponent

Browse source
This commit is contained in:
anovazzi1 2024-01-16 18:13:16 -03:00
commit 4f5f8258f3
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/components/chains/LLMMathChain.py
View file

@ -1,10 +1,11 @@
from typing import Callable, Optional, Union
from langflow import CustomComponent from langflow import CustomComponent
from langchain.chains import LLMChain from langchain.chains import LLMChain
from typing import Optional from typing import Optional
from langflow.field_typing import ( from langflow.field_typing import (
BaseLanguageModel, BaseLanguageModel,
BaseMemory, BaseMemory,
Chain
) )
class LLMMathChainComponent(CustomComponent): class LLMMathChainComponent(CustomComponent):
@ -28,5 +29,5 @@ class LLMMathChainComponent(CustomComponent):
input_key: str, input_key: str,
output_key: str, output_key: str,
memory: Optional[BaseMemory] = None, memory: Optional[BaseMemory] = None,
) -> LLMChain: ) -> Union[Chain, Callable]:
return LLMChain(llm=llm, prompt=llm_chain, input_key=input_key, output_key=output_key, memory=memory) return LLMChain(llm=llm, prompt=llm_chain, input_key=input_key, output_key=output_key, memory=memory)