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

🔥 refactor(run.py): remove NotEnoughElementsException import and exception handling

Browse source 
Chroma implemented a fix for this so this exception won't be thrown anymore.
This commit is contained in:
Gabriel Almeida 2023-05-23 12:02:55 -03:00
commit ddf75d7eb9
1 changed files with 0 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/backend/langflow/interface/run.py
View file

@ -2,7 +2,6 @@ import contextlib
import io import io
from typing import Any, Dict, List, Tuple from typing import Any, Dict, List, Tuple
from chromadb.errors import NotEnoughElementsException # type: ignore
from langchain.schema import AgentAction from langchain.schema import AgentAction
from langflow.api.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler # type: ignore from langflow.api.callback import AsyncStreamingLLMCallbackHandler, StreamingLLMCallbackHandler # type: ignore
@ -195,10 +194,6 @@ async def get_result_and_steps(langchain_object, message: str, **kwargs):
else output else output
) )
thought = format_actions(intermediate_steps) if intermediate_steps else "" thought = format_actions(intermediate_steps) if intermediate_steps else ""
except NotEnoughElementsException as exc:
raise ValueError(
"Error: Not enough documents for ChromaDB to index. Try reducing chunk size in TextSplitter."
) from exc
except Exception as exc: except Exception as exc:
raise ValueError(f"Error: {str(exc)}") from exc raise ValueError(f"Error: {str(exc)}") from exc
return result, thought return result, thought