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

Revert "Refactor state management to use Zustand, migrate components to CustomComponent and other fixes"

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-05 18:29:03 -03:00 • committed by GitHub
commit f3b3d2cf68
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
239 changed files with 28527 additions and 15390 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/test_custom_component.py
View file

@ -16,7 +16,7 @@ from langflow.interface.custom.custom_component import CustomComponent
from langchain.llms.base import BaseLLM
from langchain.chains import LLMChain
from langchain.prompts import PromptTemplate
from langchain_core.documents import Document
from langchain.schema import Document
import requests
@ -224,7 +224,7 @@ def test_custom_component_get_function_entrypoint_return_type():
Test the get_function_entrypoint_return_type
property of the CustomComponent class.
"""
from langchain_core.documents import Document
from langchain.schema import Document
custom_component = CustomComponent(code=code_default, function_entrypoint_name="build")
return_type = custom_component.get_function_entrypoint_return_type