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

feat:rag graph

Browse Source
This commit is contained in:
aries_ckt
2023-10-16 14:09:04 +08:00
parent 71c31c3e2e
commit b63fa2dfe1
10 changed files with 51 additions and 51 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
pilot/graph_engine/search.py
View File

@@ -23,7 +23,7 @@ class SearchMode(str, Enum):
class BaseSearch(ABC):
"""Base Search."""
def search(self, query: str):
async def search(self, query: str):
"""Retrieve nodes given query.
Args:
@@ -32,10 +32,10 @@ class BaseSearch(ABC):
"""
# if isinstance(query, str):
return self._search(query)
return await self._search(query)
@abstractmethod
def _search(self, query: str):
async def _search(self, query: str):
"""search nodes given query.
Implemented by the user.