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

Update imports to use langchain_core.documents

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-01 09:37:37 -03:00
commit c0c64c99c9
15 changed files with 36 additions and 31 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.schema import Document
from langchain_core.documents 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.schema import Document
from langchain_core.documents import Document
custom_component = CustomComponent(code=code_default, function_entrypoint_name="build")
return_type = custom_component.get_function_entrypoint_return_type