diff --git a/pyproject.toml b/pyproject.toml index 9fe17ba1f..1884f1a0f 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "langflow" -version = "0.6.6" +version = "0.6.7a1" description = "A Python package with a built-in web application" authors = ["Logspace "] maintainers = [ diff --git a/src/backend/langflow/components/embeddings/OpenAIEmbeddings.py b/src/backend/langflow/components/embeddings/OpenAIEmbeddings.py index 9ff91dd0e..9a16a2088 100644 --- a/src/backend/langflow/components/embeddings/OpenAIEmbeddings.py +++ b/src/backend/langflow/components/embeddings/OpenAIEmbeddings.py @@ -95,12 +95,15 @@ class OpenAIEmbeddingsComponent(CustomComponent): tikToken_enable: bool = True, tiktoken_model_name: Optional[str] = None, ) -> Union[OpenAIEmbeddings, Callable]: + # This is to avoid errors with Vector Stores (e.g Chroma) + if disallowed_special == ["all"]: + disallowed_special = "all" return OpenAIEmbeddings( tiktoken_enabled=tikToken_enable, default_headers=default_headers, default_query=default_query, allowed_special=set(allowed_special), - disallowed_special=set(disallowed_special), + disallowed_special="all", chunk_size=chunk_size, client=client, deployment=deployment, diff --git a/src/backend/langflow/components/textsplitters/CharacterTextSplitter.py b/src/backend/langflow/components/textsplitters/CharacterTextSplitter.py index d65c28dab..1bfd2d2e3 100644 --- a/src/backend/langflow/components/textsplitters/CharacterTextSplitter.py +++ b/src/backend/langflow/components/textsplitters/CharacterTextSplitter.py @@ -1,7 +1,8 @@ -from langflow import CustomComponent +from typing import List + from langchain.text_splitter import CharacterTextSplitter from langchain_core.documents.base import Document -from typing import List +from langflow import CustomComponent class CharacterTextSplitterComponent(CustomComponent): @@ -23,8 +24,10 @@ class CharacterTextSplitterComponent(CustomComponent): chunk_size: int = 1000, separator: str = "\n", ) -> List[Document]: - return CharacterTextSplitter( + docs = CharacterTextSplitter( chunk_overlap=chunk_overlap, chunk_size=chunk_size, separator=separator, ).split_documents(documents) + self.status = docs + return docs diff --git a/src/backend/langflow/components/vectorstores/Chroma.py b/src/backend/langflow/components/vectorstores/Chroma.py index 8f25548a8..6eb8fedd6 100644 --- a/src/backend/langflow/components/vectorstores/Chroma.py +++ b/src/backend/langflow/components/vectorstores/Chroma.py @@ -29,7 +29,7 @@ class ChromaComponent(CustomComponent): "collection_name": {"display_name": "Collection Name", "value": "langflow"}, "persist": {"display_name": "Persist"}, "persist_directory": {"display_name": "Persist Directory"}, - "code": {"show": False, "display_name": "Code"}, + "code": {"advanced": True, "display_name": "Code"}, "documents": {"display_name": "Documents", "is_list": True}, "embedding": {"display_name": "Embedding"}, "chroma_server_cors_allow_origins": { diff --git a/src/frontend/src/components/headerComponent/components/menuBar/index.tsx b/src/frontend/src/components/headerComponent/components/menuBar/index.tsx index a4ff37fae..40cb7a286 100644 --- a/src/frontend/src/components/headerComponent/components/menuBar/index.tsx +++ b/src/frontend/src/components/headerComponent/components/menuBar/index.tsx @@ -7,20 +7,23 @@ import { DropdownMenuTrigger, } from "../../../ui/dropdown-menu"; -import { useNavigate } from "react-router-dom"; +import { useNavigate, useParams } from "react-router-dom"; import FlowSettingsModal from "../../../../modals/flowSettingsModal"; import useAlertStore from "../../../../stores/alertStore"; import useFlowsManagerStore from "../../../../stores/flowsManagerStore"; import IconComponent from "../../../genericIconComponent"; import { Button } from "../../../ui/button"; +import { Node } from "reactflow"; +import useFlowStore from "../../../../stores/flowStore"; -export const MenuBar = (): JSX.Element => { +export const MenuBar = ({removeFunction}: {removeFunction: (nodes: Node[]) => void}): JSX.Element => { const addFlow = useFlowsManagerStore((state) => state.addFlow); const currentFlow = useFlowsManagerStore((state) => state.currentFlow); const setErrorData = useAlertStore((state) => state.setErrorData); const undo = useFlowsManagerStore((state) => state.undo); const redo = useFlowsManagerStore((state) => state.redo); const [openSettings, setOpenSettings] = useState(false); + const n = useFlowStore((state) => state.nodes); const navigate = useNavigate(); @@ -39,6 +42,7 @@ export const MenuBar = (): JSX.Element => {