correct syntax

Unstructured-IO · Oct 3, 2023 · d6eea67 · d6eea67
1 parent 2eb7fe4
commit d6eea67
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/unstructured/embed/openai.py b/unstructured/embed/openai.py
@@ -27,7 +27,7 @@ def is_unit_vector(self):
         return np.isclose(np.linalg.norm(self.examplary_embedding), 1.0)
 
     def embed_query(self, query):
-        return self.openai_client.embed_documents([str(query)])
+        return self.openai_client.embed_documents(str(query))
 
     def embed_documents(self, elements: List[Element]) -> List[Element]:
         embeddings = self.openai_client.embed_documents([str(e) for e in elements])