github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-30 23:28:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: FIx GraphRAG build error (#1706)

Browse Source
This commit is contained in:
Aries-ckt 2024-07-09 22:42:39 +08:00 committed by GitHub
parent bf978d2bf9
commit 0d85c7154a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 9 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
dbgpt/rag/index/base.py
View File

@ -8,7 +8,10 @@ from typing import Any, Dict, List, Optional
from dbgpt._private.pydantic import BaseModel, ConfigDict, Field, model_to_dict
from dbgpt.core import Chunk, Embeddings
from dbgpt.storage.vector_store.filters import MetadataFilters
from dbgpt.util.executor_utils import blocking_func_to_async
from dbgpt.util.executor_utils import (
blocking_func_to_async,
blocking_func_to_async_no_executor,
)
logger = logging.getLogger(__name__)
@ -195,4 +198,6 @@ class IndexStoreBase(ABC):
filters: Optional[MetadataFilters] = None,
) -> List[Chunk]:
"""Aynsc similar_search_with_score in vector database."""
return self.similar_search_with_scores(doc, topk, score_threshold, filters)
return await blocking_func_to_async_no_executor(
self.similar_search_with_scores, doc, topk, score_threshold, filters
)

8
dbgpt/rag/retriever/embedding.py
View File

@ -267,10 +267,6 @@ class EmbeddingRetriever(BaseRetriever):
"score_threshold": score_threshold,
},
):
return await blocking_func_to_async_no_executor(
self._index_store.similar_search_with_scores,
query,
self._top_k,
score_threshold,
filters,
return await self._index_store.asimilar_search_with_scores(
query, self._top_k, score_threshold, filters
)