Skip to content

Commit 661cae1

Browse files
committed
fix(lint): ruff format integration tests and config
1 parent dc0e364 commit 661cae1

3 files changed

Lines changed: 6 additions & 10 deletions

File tree

tests/integration/test_chroma_backend.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -154,9 +154,7 @@ async def test_increment_fact_mentions(chroma_backend):
154154

155155
async def test_find_fact_by_text(chroma_backend):
156156
emb = [0.4, 0.5, 0.6, 0.7]
157-
await chroma_backend.insert_fact(
158-
text="User is vegetarian.", embedding=emb, user_id="test-user"
159-
)
157+
await chroma_backend.insert_fact(text="User is vegetarian.", embedding=emb, user_id="test-user")
160158
found = await chroma_backend.find_fact_by_text("test-user", "vegetarian")
161159
assert found is not None
162160
assert "vegetarian" in found["text"]
@@ -261,9 +259,7 @@ async def test_count_sessions(chroma_backend):
261259
async def test_delete_user(chroma_backend):
262260
emb = [0.1, 0.2, 0.3, 0.4]
263261
await chroma_backend.upsert_session("csess-del", "delete-test-user")
264-
await chroma_backend.insert_fact(
265-
text="Delete me.", embedding=emb, user_id="delete-test-user"
266-
)
262+
await chroma_backend.insert_fact(text="Delete me.", embedding=emb, user_id="delete-test-user")
267263
deleted = await chroma_backend.delete_user("delete-test-user")
268264
assert deleted >= 2
269265

tests/integration/test_lancedb_backend.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -301,9 +301,7 @@ async def test_count_sessions(lancedb_backend):
301301
async def test_delete_user(lancedb_backend):
302302
emb = [0.1, 0.2, 0.3, 0.4]
303303
await lancedb_backend.upsert_session("lsess-del", "delete-test-user")
304-
await lancedb_backend.insert_fact(
305-
text="Delete me.", embedding=emb, user_id="delete-test-user"
306-
)
304+
await lancedb_backend.insert_fact(text="Delete me.", embedding=emb, user_id="delete-test-user")
307305
deleted = await lancedb_backend.delete_user("delete-test-user")
308306
assert deleted >= 2
309307

vektori/config.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,9 @@ class VektoriConfig:
2222

2323
# Storage
2424
database_url: str | None = None # None = SQLite default (~/.vektori/vektori.db)
25-
storage_backend: str = "sqlite" # "sqlite", "postgres", "memory", "neo4j", "qdrant", "chroma", "lancedb"
25+
storage_backend: str = (
26+
"sqlite" # "sqlite", "postgres", "memory", "neo4j", "qdrant", "chroma", "lancedb"
27+
)
2628
qdrant_api_key: str | None = None # Qdrant Cloud API key (not needed for local)
2729

2830
# Embedding provider — format: "provider:model_name"

0 commit comments

Comments
 (0)