mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-16 06:30:02 +00:00
chore: Update dependency reference path (#2745)
This commit is contained in:
@@ -51,7 +51,7 @@ You need to prepare a vector store, here we use the `ChromaStore` as an example:
|
||||
```python
|
||||
|
||||
import shutil
|
||||
from dbgpt.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
|
||||
from dbgpt_ext.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
|
||||
|
||||
# Delete old vector store directory(/tmp/tmp_ltm_vector_stor)
|
||||
shutil.rmtree("/tmp/tmp_ltm_vector_store", ignore_errors=True)
|
||||
|
@@ -42,7 +42,7 @@ pip install chromadb
|
||||
```python
|
||||
|
||||
import shutil
|
||||
from dbgpt.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
|
||||
from dbgpt_ext.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
|
||||
|
||||
# Delete old vector store directory(/tmp/tmp_ltm_vector_stor)
|
||||
shutil.rmtree("/tmp/tmp_ltm_vector_store", ignore_errors=True)
|
||||
|
@@ -75,8 +75,9 @@ import shutil
|
||||
from dbgpt.core.awel import DAG
|
||||
from dbgpt_ext.rag import ChunkParameters
|
||||
from dbgpt.rag.knowledge import KnowledgeType
|
||||
from dbgpt.rag.operators import EmbeddingAssemblerOperator, KnowledgeOperator
|
||||
from dbgpt.storage.vector_store.chroma_store import ChromaStore, ChromaVectorConfig
|
||||
from dbgpt_ext.rag.operators import EmbeddingAssemblerOperator
|
||||
from dbgpt_ext.rag.operators.knowledge import KnowledgeOperator
|
||||
from dbgpt_ext.storage.vector_store.chroma_store import ChromaStore, ChromaVectorConfig
|
||||
|
||||
# Delete old vector store directory(/tmp/awel_rag_test_vector_store)
|
||||
shutil.rmtree("/tmp/awel_rag_test_vector_store", ignore_errors=True)
|
||||
@@ -254,10 +255,10 @@ from dbgpt.core.awel import DAG, MapOperator, InputOperator, JoinOperator, Input
|
||||
from dbgpt.core.operators import PromptBuilderOperator, RequestBuilderOperator
|
||||
from dbgpt_ext.rag import ChunkParameters
|
||||
from dbgpt.rag.knowledge import KnowledgeType
|
||||
from dbgpt.rag.operators import EmbeddingAssemblerOperator, KnowledgeOperator,
|
||||
EmbeddingRetrieverOperator
|
||||
from dbgpt_ext.rag.operators.embedding import EmbeddingAssemblerOperator, EmbeddingRetrieverOperator
|
||||
from dbgpt_ext.rag.operators import KnowledgeOperator
|
||||
from dbgpt.rag.embedding import DefaultEmbeddingFactory
|
||||
from dbgpt.storage.vector_store.chroma_store import ChromaStore, ChromaVectorConfig
|
||||
from dbgpt_ext.storage.vector_store.chroma_store import ChromaStore, ChromaVectorConfig
|
||||
from dbgpt.model.operators import LLMOperator
|
||||
from dbgpt.model.proxy import OpenAILLMClient
|
||||
|
||||
|
@@ -29,7 +29,7 @@ In this guide, we mainly focus on step 1, 2, and 3.
|
||||
First, you need to install the `dbgpt` library.
|
||||
|
||||
```bash
|
||||
pip install "dbgpt[rag]>=0.5.3rc0" -U
|
||||
pip install "dbgpt[rag]>=0.7.0" -U
|
||||
````
|
||||
|
||||
## Build Knowledge Base
|
||||
@@ -52,7 +52,7 @@ embeddings = DefaultEmbeddingFactory.openai()
|
||||
Here we create a simple SQLite database.
|
||||
|
||||
```python
|
||||
from dbgpt.datasource.rdbms.conn_sqlite import SQLiteTempConnector
|
||||
from dbgpt_ext.datasource.rdbms.conn_sqlite import SQLiteTempConnector
|
||||
|
||||
db_conn = SQLiteTempConnector.create_temporary_db()
|
||||
db_conn.create_temp_tables(
|
||||
@@ -83,8 +83,8 @@ import asyncio
|
||||
import shutil
|
||||
from dbgpt.core.awel import DAG, InputOperator
|
||||
from dbgpt_ext.rag import ChunkParameters
|
||||
from dbgpt.rag.operators import DBSchemaAssemblerOperator
|
||||
from dbgpt.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
|
||||
from dbgpt_ext.rag.operators.db_schema import DBSchemaAssemblerOperator
|
||||
from dbgpt_ext.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
|
||||
|
||||
# Delete old vector store directory(/tmp/awel_with_data_vector_store)
|
||||
shutil.rmtree("/tmp/awel_with_data_vector_store", ignore_errors=True)
|
||||
@@ -115,7 +115,7 @@ print(chunks)
|
||||
|
||||
```python
|
||||
from dbgpt.core.awel import InputSource
|
||||
from dbgpt.rag.operators import DBSchemaRetrieverOperator
|
||||
from dbgpt_ext.rag.operators.db_schema import DBSchemaRetrieverOperator
|
||||
|
||||
with DAG("retrieve_schema_dag") as retrieve_schema_dag:
|
||||
input_task = InputOperator(input_source=InputSource.from_callable())
|
||||
@@ -194,7 +194,7 @@ from dbgpt.core import (
|
||||
)
|
||||
from dbgpt.core.awel import DAG, InputOperator, InputSource, MapOperator, JoinOperator
|
||||
from dbgpt.core.operators import PromptBuilderOperator, RequestBuilderOperator
|
||||
from dbgpt.rag.operators import DBSchemaRetrieverOperator
|
||||
from dbgpt_ext.rag.operators.db_schema import DBSchemaRetrieverOperator
|
||||
from dbgpt.model.operators import LLMOperator
|
||||
|
||||
system_prompt = """You are a database expert. Please answer the user's question based on the database selected by the user and some of the available table structure definitions of the database.
|
||||
@@ -448,13 +448,13 @@ from dbgpt.core.awel import (
|
||||
)
|
||||
from dbgpt.core.operators import PromptBuilderOperator, RequestBuilderOperator
|
||||
from dbgpt.datasource.operators import DatasourceOperator
|
||||
from dbgpt.datasource.rdbms.conn_sqlite import SQLiteTempConnector
|
||||
from dbgpt_ext.datasource.rdbms.conn_sqlite import SQLiteTempConnector
|
||||
from dbgpt.model.operators import LLMOperator
|
||||
from dbgpt.model.proxy import OpenAILLMClient
|
||||
from dbgpt_ext.rag import ChunkParameters
|
||||
from dbgpt.rag.embedding import DefaultEmbeddingFactory
|
||||
from dbgpt.rag.operators import DBSchemaAssemblerOperator, DBSchemaRetrieverOperator
|
||||
from dbgpt.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
|
||||
from dbgpt_ext.rag.operators.db_schema import DBSchemaAssemblerOperator, DBSchemaRetrieverOperator
|
||||
from dbgpt_ext.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
|
||||
|
||||
# Delete old vector store directory(/tmp/awel_with_data_vector_store)
|
||||
shutil.rmtree("/tmp/awel_with_data_vector_store", ignore_errors=True)
|
||||
|
Reference in New Issue
Block a user