chore: update create chroma store param (#2798)

This commit is contained in:
alanchen
2025-06-27 18:01:32 +08:00
committed by GitHub
parent bf6f38906d
commit f423b1fb2c
7 changed files with 38 additions and 42 deletions

View File

@@ -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."""

View File

@@ -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"},