Skip to content

Commit

Permalink
smoke test fixed
Browse files Browse the repository at this point in the history
  • Loading branch information
gaudyb committed Oct 31, 2024
1 parent 9fed605 commit cab502b
Showing 1 changed file with 0 additions and 8 deletions.
8 changes: 0 additions & 8 deletions tests/verbs/test_generate_text_embeddings.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,56 +50,48 @@ async def test_generate_text_embeddings():
config["community_report_full_content_embed"]["strategy"]["vector_store"] = {
"title_column": "full_content",
"collection_name": "final_community_reports_full_content_embedding",
"store_in_table": True,
"type": "lancedb",
"db_uri": "./lancedb",
}
config["community_report_summary_embed"]["strategy"]["vector_store"] = {
"title_column": "summary",
"collection_name": "final_community_reports_summary_embedding",
"store_in_table": True,
"type": "lancedb",
"db_uri": "./lancedb",
}
config["community_report_title_embed"]["strategy"]["vector_store"] = {
"title_column": "title",
"store_in_table": True,
"collection_name": "final_community_reports_title_embedding",
"type": "lancedb",
"db_uri": "./lancedb",
}
config["document_raw_content_embed"]["strategy"]["vector_store"] = {
"title_column": "raw_content",
"collection_name": "final_documents_raw_content_embedding",
"store_in_table": True,
"type": "lancedb",
"db_uri": "./lancedb",
}
config["entity_name_embed"]["strategy"]["vector_store"] = {
"title_column": "name",
"collection_name": "entity_name_embeddings",
"store_in_table": True,
"type": "lancedb",
"db_uri": "./lancedb",
}
config["entity_name_description_embed"]["strategy"]["vector_store"] = {
"title_column": "name_description",
"collection_name": "entity_description_embeddings",
"store_in_table": True,
"type": "lancedb",
"db_uri": "./lancedb",
}
config["relationship_description_embed"]["strategy"]["vector_store"] = {
"title_column": "description",
"collection_name": "relationships_description_embeddings",
"store_in_table": True,
"type": "lancedb",
"db_uri": "./lancedb",
}
config["text_unit_text_embed"]["strategy"]["vector_store"] = {
"title_column": "text",
"collection_name": "text_units_embedding",
"store_in_table": True,
"type": "lancedb",
"db_uri": "./lancedb",
}
Expand Down

0 comments on commit cab502b

Please sign in to comment.