github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-08-17 08:29:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

specify comparators (#7805)

Browse Source
This commit is contained in:
Bagatur 2023-07-17 07:30:48 -07:00 committed by GitHub
parent ee40d37098
commit 111bd7ddbe
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 18 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
langchain/retrievers/self_query/pinecone.py
View File

@ -14,7 +14,15 @@ from langchain.chains.query_constructor.ir import (
class PineconeTranslator(Visitor): class PineconeTranslator(Visitor):
"""Logic for converting internal query language elements to valid filters.""" """Logic for converting internal query language elements to valid filters."""
allowed_operators = [Operator.AND, Operator.OR] allowed_comparators = (
Comparator.EQ,
Comparator.LT,
Comparator.LTE,
Comparator.GT,
Comparator.GTE,
)
"""Subset of allowed logical comparators."""
allowed_operators = (Operator.AND, Operator.OR)
"""Subset of allowed logical operators.""" """Subset of allowed logical operators."""
def _format_func(self, func: Union[Operator, Comparator]) -> str: def _format_func(self, func: Union[Operator, Comparator]) -> str:

9
langchain/retrievers/self_query/qdrant.py
View File

@ -19,6 +19,15 @@ if TYPE_CHECKING:
class QdrantTranslator(Visitor): class QdrantTranslator(Visitor):
"""Logic for converting internal query language elements to valid filters.""" """Logic for converting internal query language elements to valid filters."""
allowed_comparators = (
Comparator.EQ,
Comparator.LT,
Comparator.LTE,
Comparator.GT,
Comparator.GTE,
)
"""Subset of allowed logical comparators."""
def __init__(self, metadata_key: str): def __init__(self, metadata_key: str):
self.metadata_key = metadata_key self.metadata_key = metadata_key