langchain-postgres 0.0.7__tar.gz → 0.0.9__tar.gz

{langchain_postgres-0.0.7 → langchain_postgres-0.0.9}/PKG-INFO

@@ -1,10 +1,10 @@
 Metadata-Version: 2.1
 Name: langchain-postgres
-Version: 0.0.7
+Version: 0.0.9
 Summary: An integration package connecting Postgres and LangChain
 Home-page: https://github.com/langchain-ai/langchain-postgres
 License: MIT
-Requires-Python: >=3.9,<4.0
+Requires-Python: >=3.8.1,<4.0.0
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.9

{langchain_postgres-0.0.7 → langchain_postgres-0.0.9}/langchain_postgres/__init__.py

@@ -1,6 +1,7 @@
 from importlib import metadata
 
 from langchain_postgres.chat_message_histories import PostgresChatMessageHistory
+from langchain_postgres.translator import PGVectorTranslator
 from langchain_postgres.vectorstores import PGVector
 
 try:
@@ -13,4 +14,5 @@ __all__ = [
     "__version__",
     "PostgresChatMessageHistory",
     "PGVector",
+    "PGVectorTranslator",
 ]

langchain_postgres-0.0.9/langchain_postgres/translator.py

@@ -0,0 +1,52 @@
+from typing import Dict, Tuple, Union
+
+from langchain_core.structured_query import (
+    Comparator,
+    Comparison,
+    Operation,
+    Operator,
+    StructuredQuery,
+    Visitor,
+)
+
+
+class PGVectorTranslator(Visitor):
+    """Translate `PGVector` internal query language elements to valid filters."""
+
+    allowed_operators = [Operator.AND, Operator.OR]
+    """Subset of allowed logical operators."""
+    allowed_comparators = [
+        Comparator.EQ,
+        Comparator.NE,
+        Comparator.GT,
+        Comparator.LT,
+        Comparator.IN,
+        Comparator.NIN,
+        Comparator.CONTAIN,
+        Comparator.LIKE,
+    ]
+    """Subset of allowed logical comparators."""
+
+    def _format_func(self, func: Union[Operator, Comparator]) -> str:
+        self._validate_func(func)
+        return f"${func.value}"
+
+    def visit_operation(self, operation: Operation) -> Dict:
+        args = [arg.accept(self) for arg in operation.arguments]
+        return {self._format_func(operation.operator): args}
+
+    def visit_comparison(self, comparison: Comparison) -> Dict:
+        return {
+            comparison.attribute: {
+                self._format_func(comparison.comparator): comparison.value
+            }
+        }
+
+    def visit_structured_query(
+        self, structured_query: StructuredQuery
+    ) -> Tuple[str, dict]:
+        if structured_query.filter is None:
+            kwargs = {}
+        else:
+            kwargs = {"filter": structured_query.filter.accept(self)}
+        return structured_query.query, kwargs

{langchain_postgres-0.0.7 → langchain_postgres-0.0.9}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "langchain-postgres"
-version = "0.0.7"
+version = "0.0.9"
 description = "An integration package connecting Postgres and LangChain"
 authors = []
 readme = "README.md"
@@ -11,7 +11,7 @@ license = "MIT"
 "Source Code" = "https://github.com/langchain-ai/langchain-postgres/tree/master/langchain_postgres"
 
 [tool.poetry.dependencies]
-python = "^3.9"
+python = "^3.8.1"
 langchain-core = ">=0.1.50,<0.3"
 psycopg = "^3"
 psycopg-pool = "^3.2.1"