mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-09-15 22:19:28 +00:00
chore: update create chroma store param (#2798)
This commit is contained in:
@@ -51,16 +51,16 @@ INPUT_PROMPT = "\n###Input:\n{}\n###Response:"
|
||||
|
||||
def _create_vector_connector():
|
||||
"""Create vector connector."""
|
||||
config = ChromaVectorConfig(
|
||||
persist_path=PILOT_PATH,
|
||||
config = ChromaVectorConfig(persist_path=PILOT_PATH)
|
||||
|
||||
return ChromaStore(
|
||||
config,
|
||||
name="embedding_rag_test",
|
||||
embedding_fn=DefaultEmbeddingFactory(
|
||||
default_model_name=os.path.join(MODEL_PATH, "text2vec-large-chinese"),
|
||||
).create(),
|
||||
)
|
||||
|
||||
return ChromaStore(config)
|
||||
|
||||
|
||||
def _create_temporary_connection():
|
||||
"""Create a temporary database connection for testing."""
|
||||
|
@@ -60,12 +60,12 @@ db_conn.create_temp_tables(
|
||||
}
|
||||
)
|
||||
|
||||
config = ChromaVectorConfig(
|
||||
persist_path=PILOT_PATH,
|
||||
config = ChromaVectorConfig(persist_path=PILOT_PATH)
|
||||
vector_store = ChromaStore(
|
||||
config,
|
||||
name="db_schema_vector_store",
|
||||
embedding_fn=embeddings,
|
||||
)
|
||||
vector_store = ChromaStore(config)
|
||||
|
||||
antv_charts = [
|
||||
{"response_line_chart": "used to display comparative trend analysis data"},
|
||||
|
Reference in New Issue
Block a user