You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
from dbgpt.core.awel import DAG, InputOperator, InputSource, MapOperator, JoinOperator
196
196
from dbgpt.core.operators import PromptBuilderOperator, RequestBuilderOperator
197
-
from dbgpt.rag.operators import DBSchemaRetrieverOperator
197
+
from dbgpt_ext.rag.operators.db_schema import DBSchemaRetrieverOperator
198
198
from dbgpt.model.operators import LLMOperator
199
199
200
200
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 (
448
448
)
449
449
from dbgpt.core.operators import PromptBuilderOperator, RequestBuilderOperator
450
450
from dbgpt.datasource.operators import DatasourceOperator
451
-
from dbgpt.datasource.rdbms.conn_sqlite import SQLiteTempConnector
451
+
from dbgpt_ext.datasource.rdbms.conn_sqlite import SQLiteTempConnector
452
452
from dbgpt.model.operators import LLMOperator
453
453
from dbgpt.model.proxy import OpenAILLMClient
454
454
from dbgpt_ext.rag import ChunkParameters
455
455
from dbgpt.rag.embedding import DefaultEmbeddingFactory
456
-
from dbgpt.rag.operators import DBSchemaAssemblerOperator, DBSchemaRetrieverOperator
457
-
from dbgpt.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
456
+
from dbgpt_ext.rag.operators.db_schema import DBSchemaAssemblerOperator, DBSchemaRetrieverOperator
457
+
from dbgpt_ext.storage.vector_store.chroma_store import ChromaVectorConfig, ChromaStore
458
458
459
459
# Delete old vector store directory(/tmp/awel_with_data_vector_store)
0 commit comments