소스 검색

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

tags/0.7.0
mochi 1 년 전
부모
커밋
ba79088ffc
No account linked to committer's email address
2개의 변경된 파일3개의 추가작업 그리고 3개의 파일을 삭제
  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 파일 보기

@@ -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 파일 보기

@@ -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

Loading…
취소
저장