Fix embeddings access

This commit is contained in:
Nicolas Mowen 2024-10-15 17:00:57 -06:00
parent 8ce569d365
commit 0e180cd135

View File

@ -178,7 +178,7 @@ class Embeddings:
embeddings = [] embeddings = []
for desc in event_descriptions.values(): for desc in event_descriptions.values():
embeddings.append(self.text_embedding([desc])) embeddings.append(self.text_embedding([desc])[0])
ids = list(event_descriptions.keys()) ids = list(event_descriptions.keys())