Fix tool tests

This commit is contained in:
Ashwin Bharambe 2025-01-22 20:31:18 -08:00
parent 0bff6e1658
commit 6c205e1d5a
2 changed files with 4 additions and 5 deletions

View file

@ -88,7 +88,7 @@ class TestTools:
tools_impl = tools_stack.impls[Api.tool_runtime]
# Register memory bank
await vector_dbs_impl.register(
await vector_dbs_impl.register_vector_db(
vector_db_id="test_bank",
embedding_model="all-MiniLM-L6-v2",
embedding_dimension=384,