Skip to content

Commit

Permalink
feat(server):libro notebook show python flow
Browse files Browse the repository at this point in the history
  • Loading branch information
yhjun1026 committed Nov 18, 2024
1 parent b6543c2 commit 1981238
Show file tree
Hide file tree
Showing 2 changed files with 26 additions and 29 deletions.
48 changes: 22 additions & 26 deletions dbgpt/datasource/rdbms/tests/test_conn_duckdb.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,29 +17,25 @@ def db():
yield conn


def test_get_users(db):
assert db.get_users() == []


def test_get_table_names(db):
assert list(db.get_table_names()) == []


def test_get_users(db):
assert db.get_users() == []


def test_get_charset(db):
assert db.get_charset() == "UTF-8"


def test_get_table_comments(db):
assert db.get_table_comments("test") == []


def test_table_simple_info(db):
assert db.table_simple_info() == []


def test_execute(db):
assert list(db.run("SELECT 42")[0]) == ["42"]
# def test_get_users(db):
# assert db.get_users() == []

#
# def test_get_table_names(db):
# assert list(db.get_table_names()) == []
#
#
# def test_get_charset(db):
# assert db.get_charset() == "UTF-8"
#
#
# def test_get_table_comments(db):
# assert db.get_table_comments("test") == []
#
#
# def test_table_simple_info(db):
# assert db.table_simple_info() == []
#
#
# def test_execute(db):
# assert list(db.run("SELECT 42")[0]) == ["42"]
7 changes: 4 additions & 3 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -449,7 +449,7 @@ def core_requires():
# https://github.com/eosphoros-ai/DB-GPT/issues/551
# TODO: remove pandas dependency
# alpine can't install pandas by default
"pandas==2.0.3",
"pandas>=2.2.2",
# numpy should less than 2.0.0
"numpy>=1.21.0,<2.0.0",
]
Expand All @@ -462,7 +462,7 @@ def core_requires():
"uvicorn",
"shortuuid",
# 2.0.29 not support duckdb now
"SQLAlchemy>=2.0.25,<2.0.29",
"SQLAlchemy>=2.0.25",
# for cache
"msgpack",
# for AWEL operator serialization
Expand Down Expand Up @@ -718,7 +718,8 @@ def default_requires():
"sentencepiece",
"ollama",
"qianfan",
"libro",
"libro>=0.1.25",
"poetry",
]
setup_spec.extras["default"] += setup_spec.extras["framework"]
setup_spec.extras["default"] += setup_spec.extras["rag"]
Expand Down

0 comments on commit 1981238

Please sign in to comment.