add embedding input type parameter (#8724)

2024-09-24 21:53:50 +08:00
parent debe5953a8
commit 4669eb24be
33 changed files with 239 additions and 35 deletions
--- a/api/core/embedding/embedding_constant.py
+++ b/api/core/embedding/embedding_constant.py
@@ -0,0 +1,10 @@
+from enum import Enum
+
+
+class EmbeddingInputType(Enum):
+    """
+    Enum for embedding input type.
+    """
+
+    DOCUMENT = "document"
+    QUERY = "query"