update mysql conn name

This commit is contained in:
csunny 2023-05-08 00:44:58 +08:00
parent 3ce31d5ea0
commit 922d2e4f28
2 changed files with 2 additions and 2 deletions

View File

@ -89,7 +89,7 @@ class Conversation:
def gen_sqlgen_conversation(dbname):
from pilot.connections.mysql_conn import MySQLOperator
from pilot.connections.mysql import MySQLOperator
mo = MySQLOperator(
**DB_SETTINGS
)

View File

@ -12,7 +12,7 @@ import requests
from urllib.parse import urljoin
from pilot.configs.model_config import DB_SETTINGS
from pilot.server.vectordb_qa import KnownLedgeBaseQA
from pilot.connections.mysql_conn import MySQLOperator
from pilot.connections.mysql import MySQLOperator
from pilot.vector_store.extract_tovec import get_vector_storelist, load_knownledge_from_doc, knownledge_tovec_st
from pilot.configs.model_config import LOGDIR, VICUNA_MODEL_SERVER, LLM_MODEL, DATASETS_DIR