Skip to content

Commit 2f247da

Browse files
Gaudy BlancoGaudy Blanco
authored andcommitted
fixes for cosmosdb test
1 parent a810c1c commit 2f247da

File tree

2 files changed

+15
-5
lines changed

2 files changed

+15
-5
lines changed

.vscode/launch.json

Lines changed: 13 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,18 @@
3737
"--config",
3838
"<path_to_ragtest_root_demo>/settings.yaml",
3939
]
40-
}
40+
},
41+
{
42+
"name": "Debug Integration Pytest",
43+
"type": "debugpy",
44+
"request": "launch",
45+
"module": "pytest",
46+
"args": [
47+
"./tests/integration/vector_stores",
48+
"-k", "test_lancedb"
49+
],
50+
"console": "integratedTerminal",
51+
"justMyCode": false
52+
},
4153
]
4254
}

tests/integration/vector_stores/test_cosmosdb.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -25,8 +25,7 @@
2525
def test_vector_store_operations():
2626
"""Test basic vector store operations with CosmosDB."""
2727
vector_store = CosmosDBVectorStore(
28-
collection_name="testvector",
29-
vector_store_schema_config=VectorStoreSchemaConfig(),
28+
vector_store_schema_config=VectorStoreSchemaConfig(index_name="testvector"),
3029
)
3130

3231
try:
@@ -80,8 +79,7 @@ def mock_embedder(text: str) -> list[float]:
8079
def test_clear():
8180
"""Test clearing the vector store."""
8281
vector_store = CosmosDBVectorStore(
83-
collection_name="testclear",
84-
vector_store_schema_config=VectorStoreSchemaConfig(),
82+
vector_store_schema_config=VectorStoreSchemaConfig(index_name="testclear"),
8583
)
8684
try:
8785
vector_store.connect(

0 commit comments

Comments
 (0)