github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-02 17:45:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor: Refactor proxy LLM (#1064)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-01-14 21:01:37 +08:00
committed by GitHub
parent a035433170
commit 22bfd01c4b
95 changed files with 2049 additions and 1294 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
dbgpt/rag/summary/db_summary_client.py
View File

@@ -1,12 +1,9 @@
import logging
import traceback
from dbgpt.component import SystemApp
from dbgpt._private.config import Config
from dbgpt.configs.model_config import (
EMBEDDING_MODEL_CONFIG,
)
from dbgpt._private.config import Config
from dbgpt.component import SystemApp
from dbgpt.configs.model_config import EMBEDDING_MODEL_CONFIG
from dbgpt.rag.summary.rdbms_db_summary import RdbmsSummary
logger = logging.getLogger(__name__)
@@ -44,8 +41,8 @@ class DBSummaryClient:
def get_db_summary(self, dbname, query, topk):
"""get user query related tables info"""
from dbgpt.storage.vector_store.connector import VectorStoreConnector
from dbgpt.storage.vector_store.base import VectorStoreConfig
from dbgpt.storage.vector_store.connector import VectorStoreConnector
vector_store_config = VectorStoreConfig(name=dbname + "_profile")
vector_connector = VectorStoreConnector.from_default(
@@ -82,8 +79,8 @@ class DBSummaryClient:
dbname(str): dbname
"""
vector_store_name = dbname + "_profile"
from dbgpt.storage.vector_store.connector import VectorStoreConnector
from dbgpt.storage.vector_store.base import VectorStoreConfig
from dbgpt.storage.vector_store.connector import VectorStoreConnector
vector_store_config = VectorStoreConfig(name=vector_store_name)
vector_connector = VectorStoreConnector.from_default(

3
dbgpt/rag/summary/rdbms_db_summary.py
View File

@@ -1,7 +1,8 @@
from typing import List
from dbgpt._private.config import Config
from dbgpt.rag.summary.db_summary import DBSummary
from dbgpt.datasource.rdbms.base import RDBMSDatabase
from dbgpt.rag.summary.db_summary import DBSummary
CFG = Config()