@@ -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