feat(rag): Support RAG SDK (#1322)

dbgpt/rag/operators/schema_linking.py

@@ -7,7 +7,7 @@ from typing import Any, Optional
 
 from dbgpt.core import LLMClient
 from dbgpt.core.awel import MapOperator
-from dbgpt.datasource.rdbms.base import RDBMSConnector
+from dbgpt.datasource.base import BaseConnector
 from dbgpt.rag.schemalinker.schema_linking import SchemaLinking
 from dbgpt.storage.vector_store.connector import VectorStoreConnector
 
@@ -17,7 +17,7 @@ class SchemaLinkingOperator(MapOperator[Any, Any]):
 
     def __init__(
         self,
-        connection: RDBMSConnector,
+        connector: BaseConnector,
         model_name: str,
         llm: LLMClient,
         top_k: int = 5,
@@ -27,14 +27,14 @@ class SchemaLinkingOperator(MapOperator[Any, Any]):
         """Create the schema linking operator.
 
         Args:
-            connection (RDBMSConnector): The connection.
+            connector (BaseConnector): The connection.
             llm (Optional[LLMClient]): base llm
         """
         super().__init__(**kwargs)
 
         self._schema_linking = SchemaLinking(
             top_k=top_k,
-            connection=connection,
+            connector=connector,
             llm=llm,
             model_name=model_name,
             vector_store_connector=vector_store_connector,