Skip to content

Commit 241199e

Browse files
Bugfix in setting vectorizer. Uses index key_separator
1 parent 85ba8d0 commit 241199e

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

redisvl/extensions/session_manager/semantic_session.py

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -55,10 +55,9 @@ def __init__(
5555

5656
prefix = prefix or name
5757

58-
if vectorizer is None:
59-
self._vectorizer = HFTextVectorizer(
60-
model="sentence-transformers/msmarco-distilbert-cos-v5"
61-
)
58+
self._vectorizer = vectorizer or HFTextVectorizer(
59+
model="sentence-transformers/msmarco-distilbert-cos-v5"
60+
)
6261

6362
self.set_distance_threshold(distance_threshold)
6463

@@ -147,7 +146,8 @@ def drop(self, id_field: Optional[str] = None) -> None:
147146
If None then the last entry is deleted.
148147
"""
149148
if id_field:
150-
key = ":".join([self._index.schema.index.name, id_field])
149+
sep = self._index.key_separator
150+
key = sep.join([self._index.schema.index.name, id_field])
151151
else:
152152
key = self.get_recent(top_k=1, raw=True)[0]["id"] # type: ignore
153153
self._client.delete(key)
@@ -334,11 +334,12 @@ def add_messages(self, messages: List[Dict[str, str]]) -> None:
334334
Args:
335335
messages (List[Dict[str, str]]): The list of user prompts and LLM responses.
336336
"""
337+
sep = self._index.key_separator
337338
payloads = []
338339
for message in messages:
339340
vector = self._vectorizer.embed(message[self.content_field_name])
340341
timestamp = time()
341-
id_field = ":".join([self._user_tag, self._session_tag, str(timestamp)])
342+
id_field = sep.join([self._user_tag, self._session_tag, str(timestamp)])
342343
payload = {
343344
self.id_field_name: id_field,
344345
self.role_field_name: message[self.role_field_name],

0 commit comments

Comments
 (0)