🔧 chore(constants.py): import BaseOutputParser from langchain.schema to use it in LANGCHAIN_BASE_TYPES

This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-08-04 08:31:36 -03:00
commit 840629bef4

View file

@ -7,6 +7,7 @@ from langchain.schema import BaseRetriever, Document
from langchain.text_splitter import TextSplitter
from langchain.tools import Tool
from langchain.vectorstores.base import VectorStore
from langchain.schema import BaseOutputParser
LANGCHAIN_BASE_TYPES = {
@ -20,6 +21,7 @@ LANGCHAIN_BASE_TYPES = {
"VectorStore": VectorStore,
"Embeddings": Embeddings,
"BaseRetriever": BaseRetriever,
"BaseOutputParser": BaseOutputParser,
}
# Langchain base types plus Python base types