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

chore: Add pylint for DB-GPT rag lib (#1267)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-03-07 23:27:43 +08:00
committed by GitHub
parent aaaf34db17
commit 7446817340
70 changed files with 1135 additions and 587 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
dbgpt/rag/schemalinker/base_linker.py
View File

@@ -1,3 +1,5 @@
"""Base Linker."""
from abc import ABC, abstractmethod
from typing import List
@@ -6,7 +8,8 @@ class BaseSchemaLinker(ABC):
"""Base Linker."""
def schema_linking(self, query: str) -> List:
"""
"""Query schema info.
Args:
query (str): query text
Returns:
@@ -15,7 +18,8 @@ class BaseSchemaLinker(ABC):
return self._schema_linking(query)
def schema_linking_with_vector_db(self, query: str) -> List:
"""
"""Query schema info with vector db.
Args:
query (str): query text
Returns:
@@ -24,7 +28,8 @@ class BaseSchemaLinker(ABC):
return self._schema_linking_with_vector_db(query)
async def schema_linking_with_llm(self, query: str) -> List:
""" "
"""Query schema info with llm.
Args:
query(str): query text
Returns:
@@ -34,7 +39,8 @@ class BaseSchemaLinker(ABC):
@abstractmethod
def _schema_linking(self, query: str) -> List:
"""
"""Get DB schema info.
Args:
query (str): query text
Returns:
@@ -43,7 +49,8 @@ class BaseSchemaLinker(ABC):
@abstractmethod
def _schema_linking_with_vector_db(self, query: str) -> List:
"""
"""Query schema info with vector db.
Args:
query (str): query text
Returns:
@@ -52,7 +59,8 @@ class BaseSchemaLinker(ABC):
@abstractmethod
async def _schema_linking_with_llm(self, query: str) -> List:
"""
"""Query schema info with llm.
Args:
query (str): query text
Returns: