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

feat(model): Add new LLMClient and new build tools (#967)

Browse Source
This commit is contained in:
Fangyin Cheng
2023-12-23 16:33:01 +08:00
committed by GitHub
parent 12234ae258
commit 0c46c339ca
30 changed files with 1072 additions and 133 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
dbgpt/datasource/rdbms/conn_clickhouse.py
View File

@@ -1,13 +1,8 @@
import re
import sqlparse
import clickhouse_connect
from typing import List, Optional, Any, Iterable, Dict
from sqlalchemy import text
from urllib.parse import quote
from sqlalchemy.schema import CreateTable
from urllib.parse import quote_plus as urlquote
from dbgpt.datasource.rdbms.base import RDBMSDatabase
from clickhouse_connect.driver import httputil
from dbgpt.storage.schema import DBType
from sqlalchemy import (
MetaData,
@@ -56,6 +51,11 @@ class ClickhouseConnect(RDBMSDatabase):
engine_args: Optional[dict] = None,
**kwargs: Any,
) -> RDBMSDatabase:
import clickhouse_connect
from clickhouse_connect.driver import httputil
# Lazy import
big_pool_mgr = httputil.get_pool_manager(maxsize=16, num_pools=12)
client = clickhouse_connect.get_client(
host=host,