fix: embedding model name should be span attribute (#95)

Arize-ai · Jan 11, 2024 · bf6579b · bf6579b
1 parent 82c5d83
commit bf6579b
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/python/openinference-semantic-conventions/src/openinference/semconv/trace/__init__.py b/python/openinference-semantic-conventions/src/openinference/semconv/trace/__init__.py
@@ -19,6 +19,10 @@ class SpanAttributes:
     """
     A list of objects containing embedding data, including the vector and represented piece of text.
     """
+    EMBEDDING_MODEL_NAME = "embedding.model_name"
+    """
+    The name of the embedding model.
+    """
 
     LLM_FUNCTION_CALL = "llm.function_call"
     """
@@ -178,14 +182,6 @@ class EmbeddingAttributes:
     Attributes for an embedding
     """
 
-    EMBEDDING_EMBEDDINGS = "embedding.embeddings"
-    """
-    A list of objects containing embedding data, including the vector and represented piece of text.
-    """
-    EMBEDDING_MODEL_NAME = "embedding.model_name"
-    """
-    The name of the embedding model.
-    """
     EMBEDDING_TEXT = "embedding.text"
     """
     The text represented by the embedding.