🔧 chore(process.py): replace logger import from langflow.utils.logger to loguru.logger for better logging functionality

🔧 chore(process.py): change ValueError exception to a warning log message when encountering an unknown langchain_object type
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-09-17 20:14:25 -03:00
commit d65a75c8d8

View file

@ -6,7 +6,7 @@ from langflow.interface.run import (
get_memory_key, get_memory_key,
update_memory_keys, update_memory_keys,
) )
from langflow.utils.logger import logger from loguru import logger
from langflow.graph import Graph from langflow.graph import Graph
from langchain.chains.base import Chain from langchain.chains.base import Chain
from langchain.vectorstores.base import VectorStore from langchain.vectorstores.base import VectorStore
@ -130,9 +130,8 @@ def process_graph_cached(
elif isinstance(langchain_object, Document): elif isinstance(langchain_object, Document):
result = langchain_object.dict() result = langchain_object.dict()
else: else:
raise ValueError( logger.warning(f"Unknown langchain_object type: {type(langchain_object)}")
f"Unknown langchain_object type: {type(langchain_object).__name__}"
)
return result return result