feat: add schema-linking awel example (#1081)

This commit is contained in:
junewgl
2024-01-21 09:59:59 +08:00
committed by GitHub
parent 2d905191f8
commit 4f833634df
5 changed files with 445 additions and 0 deletions

View File

@@ -0,0 +1,60 @@
from abc import ABC, abstractmethod
from typing import List
class BaseSchemaLinker(ABC):
"""Base Linker."""
def schema_linking(self, query: str) -> List:
"""
Args:
query (str): query text
Returns:
List: list of schema
"""
return self._schema_linking(query)
def schema_linking_with_vector_db(self, query: str) -> List:
"""
Args:
query (str): query text
Returns:
List: list of schema
"""
return self._schema_linking_with_vector_db(query)
async def schema_linking_with_llm(self, query: str) -> List:
""" "
Args:
query(str): query text
Returns:
List: list of schema
"""
return await self._schema_linking_with_llm(query)
@abstractmethod
def _schema_linking(self, query: str) -> List:
"""
Args:
query (str): query text
Returns:
List: list of schema
"""
@abstractmethod
def _schema_linking_with_vector_db(self, query: str) -> List:
"""
Args:
query (str): query text
Returns:
List: list of schema
"""
@abstractmethod
async def _schema_linking_with_llm(self, query: str) -> List:
"""
Args:
query (str): query text
Returns:
List: list of schema
"""