Bläddra i källkod

Fix SQL parser Error in MyScale vdb. (#7255)

tags/0.7.0
mochi 1 år sedan
förälder
incheckning
ba79088ffc
Inget konto är kopplat till bidragsgivarens mejladress
2 ändrade filer med 3 tillägg och 3 borttagningar
  1. 2
    2
      api/core/rag/datasource/vdb/myscale/myscale_vector.py
  2. 1
    1
      docker/docker-compose.yaml

+ 2
- 2
api/core/rag/datasource/vdb/myscale/myscale_vector.py Visa fil

@@ -93,7 +93,7 @@ class MyScaleVector(BaseVector):

@staticmethod
def escape_str(value: Any) -> str:
return "".join(f"\\{c}" if c in ("\\", "'") else c for c in str(value))
return "".join(" " if c in ("\\", "'") else c for c in str(value))

def text_exists(self, id: str) -> bool:
results = self._client.query(f"SELECT id FROM {self._config.database}.{self._collection_name} WHERE id='{id}'")
@@ -118,7 +118,7 @@ class MyScaleVector(BaseVector):
return self._search(f"distance(vector, {str(query_vector)})", self._vec_order, **kwargs)

def search_by_full_text(self, query: str, **kwargs: Any) -> list[Document]:
return self._search(f"TextSearch(text, '{query}')", SortOrder.DESC, **kwargs)
return self._search(f"TextSearch('enable_nlq=false')(text, '{query}')", SortOrder.DESC, **kwargs)

def _search(self, dist: str, order: SortOrder, **kwargs: Any) -> list[Document]:
top_k = kwargs.get("top_k", 5)

+ 1
- 1
docker/docker-compose.yaml Visa fil

@@ -587,7 +587,7 @@ services:
# MyScale vector database
myscale:
container_name: myscale
image: myscale/myscaledb:1.6
image: myscale/myscaledb:1.6.4
profiles:
- myscale
restart: always

Laddar…
Avbryt
Spara