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

style:fmt

Browse Source
This commit is contained in:
aries_ckt
2023-09-22 00:21:54 +08:00
parent 25ba20925f
commit ce3b2e6213
3 changed files with 5 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
docs/conf.py
View File

@@ -27,7 +27,7 @@ extensions = [
"sphinx.ext.napoleon", "sphinx.ext.napoleon",
"sphinx.ext.viewcode", "sphinx.ext.viewcode",
"sphinxcontrib.autodoc_pydantic", "sphinxcontrib.autodoc_pydantic",
'sphinxcontrib.autodoc_pydantic_base', "sphinxcontrib.autodoc_pydantic_base",
"myst_nb", "myst_nb",
"sphinx_copybutton", "sphinx_copybutton",
"sphinx_panels", "sphinx_panels",

17
pilot/connections/conn_spark.py
View File

@@ -22,8 +22,8 @@ class SparkConnect(BaseConnect):
self, self,
file_path: str, file_path: str,
spark_session: Optional[SparkSession] = None, spark_session: Optional[SparkSession] = None,
engine_args: Optional[dict]= None, engine_args: Optional[dict] = None,
**kwargs: Any **kwargs: Any,
) -> None: ) -> None:
"""Initialize the Spark DataFrame from Datasource path """Initialize the Spark DataFrame from Datasource path
return: Spark DataFrame return: Spark DataFrame
@@ -109,15 +109,4 @@ class SparkConnect(BaseConnect):
return f"{self.table_name}{self.get_fields()}" return f"{self.table_name}{self.get_fields()}"
def get_table_comments(self, db_name): def get_table_comments(self, db_name):
session = self._db_sessions() return ""
cursor = session.execute(
text(
f"""SELECT table, comment FROM system.tables WHERE database = '{db_name}'""".format(
db_name
)
)
)
table_comments = cursor.fetchall()
return [
(table_comment[0], table_comment[1]) for table_comment in table_comments
]

1
pilot/server/llm_manage/api.py
View File

@@ -19,6 +19,7 @@ async def model_params():
print(f"/worker/model/params") print(f"/worker/model/params")
try: try:
from pilot.model.cluster import WorkerManagerFactory from pilot.model.cluster import WorkerManagerFactory
worker_manager = CFG.SYSTEM_APP.get_component( worker_manager = CFG.SYSTEM_APP.get_component(
ComponentType.WORKER_MANAGER_FACTORY, WorkerManagerFactory ComponentType.WORKER_MANAGER_FACTORY, WorkerManagerFactory
).create() ).create()