Skip to content

Commit 88e27d7

Browse files
Optimisation in passing callback manager (#94)
* Change token counter implementation & support embedding token counting (defaults to tiktoken) * Refactor code * Add a todo * Pass callback manager directly
1 parent d353d16 commit 88e27d7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/unstract/sdk/vector_db.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -136,7 +136,7 @@ def get_vector_store_index_from_storage_context(
136136
show_progress=show_progress,
137137
embed_model=self._embedding_instance,
138138
node_parser=parser,
139-
callback_manager=callback_manager,
139+
callback_manager=self._embedding_instance.callback_manager,
140140
)
141141

142142
def get_vector_store_index(self, **kwargs: Any) -> VectorStoreIndex:
@@ -145,7 +145,7 @@ def get_vector_store_index(self, **kwargs: Any) -> VectorStoreIndex:
145145
return VectorStoreIndex.from_vector_store(
146146
vector_store=self._vector_db_instance,
147147
embed_model=self._embedding_instance,
148-
callback_manager=kwargs.get("callback_manager"),
148+
callback_manager=self._embedding_instance.callback_manager,
149149
)
150150

151151
def get_storage_context(self) -> StorageContext:

0 commit comments

Comments
 (0)