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

Update QdrantComponent parameters

Browse source
This commit is contained in:
anovazzi1 2024-01-16 16:23:30 -03:00
commit f7fc67bb9a
1 changed files with 21 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

40
src/backend/langflow/components/vectorstores/Qdrant.py
View file

@ -1,7 +1,9 @@
from langflow import CustomComponent from langflow import CustomComponent
from langchain_community.vectorstores.qdrant import Qdrant from langchain_community.vectorstores.qdrant import Qdrant
from typing import Optional, List from typing import Optional, List, Union
from langflow.field_typing import Document, Embeddings, NestedDict from langflow.field_typing import Document, Embeddings, NestedDict
from langchain.schema import BaseRetriever
from langchain.vectorstores.base import VectorStore
class QdrantComponent(CustomComponent): class QdrantComponent(CustomComponent):
@ -14,20 +16,20 @@ class QdrantComponent(CustomComponent):
"embedding": {"display_name": "Embedding"}, "embedding": {"display_name": "Embedding"},
"api_key": {"display_name": "API Key", "password": True}, "api_key": {"display_name": "API Key", "password": True},
"collection_name": {"display_name": "Collection Name"}, "collection_name": {"display_name": "Collection Name"},
"content_payload_key": {"display_name": "Content Payload Key"}, "content_payload_key": {"display_name": "Content Payload Key", "advanced": True},
"distance_func": {"display_name": "Distance Function"}, "distance_func": {"display_name": "Distance Function", "advanced": True},
"grpc_port": {"display_name": "gRPC Port"}, "grpc_port": {"display_name": "gRPC Port", "advanced": True},
"host": {"display_name": "Host"}, "host": {"display_name": "Host", "advanced": True},
"https": {"display_name": "HTTPS"}, "https": {"display_name": "HTTPS", "advanced": True},
"location": {"display_name": "Location"}, "location": {"display_name": "Location", "advanced": True},
"metadata_payload_key": {"display_name": "Metadata Payload Key"}, "metadata_payload_key": {"display_name": "Metadata Payload Key", "advanced": True},
"path": {"display_name": "Path"}, "path": {"display_name": "Path", "advanced": True},
"port": {"display_name": "Port"}, "port": {"display_name": "Port", "advanced": True},
"prefer_grpc": {"display_name": "Prefer gRPC"}, "prefer_grpc": {"display_name": "Prefer gRPC", "advanced": True},
"prefix": {"display_name": "Prefix"}, "prefix": {"display_name": "Prefix", "advanced": True},
"search_kwargs": {"display_name": "Search Kwargs"}, "search_kwargs": {"display_name": "Search Kwargs", "advanced": True},
"timeout": {"display_name": "Timeout"}, "timeout": {"display_name": "Timeout", "advanced": True},
"url": {"display_name": "URL"}, "url": {"display_name": "URL", "advanced": True},
} }
def build( def build(
@ -38,20 +40,20 @@ class QdrantComponent(CustomComponent):
collection_name: Optional[str] = None, collection_name: Optional[str] = None,
content_payload_key: str = "page_content", content_payload_key: str = "page_content",
distance_func: str = "Cosine", distance_func: str = "Cosine",
grpc_port: int = 6334, grpc_port: Optional[int] = 6334,
host: Optional[str] = None, host: Optional[str] = None,
https: bool = False, https: bool = False,
location: str = ":memory:", location: str = ":memory:",
metadata_payload_key: str = "metadata", metadata_payload_key: str = "metadata",
path: Optional[str] = None, path: Optional[str] = None,
port: int = 6333, port: Optional[int] = 6333,
prefer_grpc: bool = False, prefer_grpc: bool = False,
prefix: Optional[str] = None, prefix: Optional[str] = None,
search_kwargs: Optional[NestedDict] = None, search_kwargs: Optional[NestedDict] = None,
timeout: Optional[float] = None, timeout: Optional[float] = None,
url: Optional[str] = None, url: Optional[str] = None,
) -> Qdrant: ) -> Union[VectorStore, Qdrant, BaseRetriever]:
return Qdrant( return Qdrant.from_documents(
documents=documents, documents=documents,
embedding=embedding, embedding=embedding,
api_key=api_key, api_key=api_key,