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

refactor: The first refactored version for sdk release (#907)

Browse Source 
Co-authored-by: chengfangyin2 <chengfangyin3@jd.com>
This commit is contained in:
FangYin Cheng
2023-12-08 14:45:59 +08:00
committed by GitHub
parent e7e4aff667
commit cd725db1fb
573 changed files with 2094 additions and 3571 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/unit_tests/embedding_engine/document_test.py
View File

@@ -1,4 +1,4 @@
from pilot import EmbeddingEngine, KnowledgeType
from dbgpt import EmbeddingEngine, KnowledgeType
embedding_model = "your_embedding_model"
vector_store_type = "Chroma"

2
tests/unit_tests/embedding_engine/url_test.py
View File

@@ -1,4 +1,4 @@
from pilot import EmbeddingEngine, KnowledgeType
from dbgpt import EmbeddingEngine, KnowledgeType
url = "https://docs.dbgpt.site/docs/overview"
embedding_model = "your_embedding_model"

5
tests/unit_tests/test_plugins.py
View File

@@ -2,9 +2,8 @@ import os
import pytest
from pilot.configs.config import Config
from pilot.configs.model_config import PLUGINS_DIR
from pilot.plugins import (
from dbgpt._private.config import Config
from dbgpt.plugins import (
denylist_allowlist_check,
inspect_zip_for_modules,
scan_plugins,

6
tests/unit_tests/vector_store/test_pgvector.py
View File

@@ -1,7 +1,5 @@
import pytest
from pilot import vector_store
from pilot.vector_store.base import VectorStoreBase
from dbgpt.storage import vector_store
from dbgpt.storage.vector_store.base import VectorStoreBase
def test_vetorestore_imports() -> None: