lower case

Author: nimarb

langfuse/_client/client.py

@@ -359,14 +359,12 @@ def start_as_current_span(
             Literal[
                 "generation",
                 "span",
-                "GENERATION",
-                "EVENT",
-                "SPAN",
-                "AGENT",
-                "TOOL",
-                "CHAIN",
-                "RETRIEVER",
-                "EMBEDDING",
+                "event",
+                "agent",
+                "tool",
+                "chain",
+                "retriever",
+                "embedding",
             ]
         ],
     ) -> _AgnosticContextManager[LangfuseSpan]:
@@ -691,30 +689,30 @@ def _get_span_class(
             "span",
             "generation",
             "event",
-            "AGENT",
-            "TOOL",
-            "CHAIN",
-            "RETRIEVER",
-            "EMBEDDING",
+            "agent",
+            "tool",
+            "chain",
+            "retriever",
+            "embedding",
         ],
     ):
         """Get the appropriate span class based on as_type."""
         # TODO: make it case insensitive
-        if as_type == "AGENT":
+        if as_type == "agent":
             return LangfuseAgent
-        elif as_type == "TOOL":
+        elif as_type == "tool":
             return LangfuseTool
-        elif as_type == "CHAIN":
+        elif as_type == "chain":
             return LangfuseChain
-        elif as_type == "RETRIEVER":
+        elif as_type == "retriever":
             return LangfuseRetriever
-        elif as_type == "EMBEDDING":
+        elif as_type == "embedding":
             return LangfuseEmbedding
-        elif as_type in ("generation", "GENERATION"):
+        elif as_type in ("generation"):
             return LangfuseGeneration
-        elif as_type in ("event", "EVENT"):
+        elif as_type in ("event"):
             return LangfuseEvent
-        elif as_type in ("span", "SPAN"):
+        elif as_type in ("span"):
             return LangfuseSpan
         else:
             return LangfuseSpan
@@ -775,7 +773,14 @@ def _start_as_current_otel_span_with_processed_media(
         name: str,
         as_type: Optional[
             Literal[
-                "generation", "span", "event", "AGENT", "TOOL", "CHAIN", "RETRIEVER"
+                "generation",
+                "span",
+                "event",
+                "agent",
+                "tool",
+                "chain",
+                "retriever",
+                "embedding",
             ]
         ] = None,
         end_on_exit: Optional[bool] = None,

langfuse/_client/constants.py

@@ -7,12 +7,12 @@
 
 # Valid observation types for the @observe decorator
 VALID_OBSERVATION_TYPES = {
-    "SPAN",
-    "EVENT",
-    "GENERATION",
-    "AGENT",
-    "TOOL",
-    "CHAIN",
-    "RETRIEVER",
-    "EMBEDDING",
+    "span",
+    "event",
+    "generation",
+    "agent",
+    "tool",
+    "chain",
+    "retriever",
+    "embedding",
 }

langfuse/_client/observe.py

@@ -70,14 +70,14 @@ def observe(
         as_type: Optional[Literal["generation"]] = None,
         type: Optional[
             Literal[
-                "SPAN",
-                "EVENT",
-                "GENERATION",
-                "AGENT",
-                "TOOL",
-                "CHAIN",
-                "RETRIEVER",
-                "EMBEDDING",
+                "span",
+                "event",
+                "generation",
+                "agent",
+                "tool",
+                "chain",
+                "retriever",
+                "embedding",
             ]
         ] = None,
         capture_input: Optional[bool] = None,
@@ -93,14 +93,14 @@ def observe(
         as_type: Optional[Literal["generation"]] = None,
         type: Optional[
             Literal[
-                "SPAN",
-                "EVENT",
-                "GENERATION",
-                "AGENT",
-                "TOOL",
-                "CHAIN",
-                "RETRIEVER",
-                "EMBEDDING",
+                "span",
+                "event",
+                "generation",
+                "agent",
+                "tool",
+                "chain",
+                "retriever",
+                "embedding",
             ]
         ] = None,
         capture_input: Optional[bool] = None,
@@ -122,7 +122,7 @@ def observe(
             as_type (Optional[Literal["generation"]]): Set to "generation" to create a specialized LLM generation span
                     with model metrics support, suitable for tracking language model outputs.
             type (Optional[Literal]): Set the observation type for agentic workflows. Supported values:
-                    "GENERATION", "AGENT", "TOOL", "CHAIN", "RETRIEVER", "EMBEDDING". When specified, creates spans with
+                    "generation", "agent", "tool", "chain", "retriever", "embedding". When specified, creates spans with
                     the specified type for graph visualization and filtering in the Langfuse UI.
 
         Returns:
@@ -151,7 +151,7 @@ async def generate_answer(query):
 
             For automatic graph instrumentation with agent workflows:
             ```python
-            @observe(type="AGENT")
+            @observe(type="agent")
             def planning_agent():
                 return create_plan()
             ```
@@ -259,15 +259,14 @@ def _async_observe(
         name: Optional[str],
         as_type: Optional[
             Literal[
+                "span",
                 "generation",
-                "GENERATION",
-                "EVENT",
-                "SPAN",
-                "AGENT",
-                "TOOL",
-                "CHAIN",
-                "RETRIEVER",
-                "EMBEDDING",
+                "event",
+                "agent",
+                "tool",
+                "chain",
+                "retriever",
+                "embedding",
             ]
         ],
         capture_input: bool,

langfuse/_client/span.py

@@ -1224,7 +1224,7 @@ class LangfuseGraphObservation(LangfuseSpanWrapper):
     """Base class for agentic workflow observations.
 
     This class provides a common foundation for specialized observation types
-    used in agentic workflows: AGENT, TOOL, CHAIN, RETRIEVER, and EMBEDDING.
+    used in agentic workflows: agent, tool, chain, retriever, and embedding.
     All graph observations use span-type attributes but with specific observation types.
     """
 
@@ -1245,7 +1245,7 @@ def __init__(
         """Initialize a new graph observation span.
 
         Args:
-            observation_type: The specific observation type (e.g., "AGENT", "TOOL")
+            observation_type: The specific observation type (e.g., "agent", "tool", ...)
             otel_span: The OpenTelemetry span to wrap
             langfuse_client: Reference to the parent Langfuse client
             input: Input data for the observation
@@ -1331,45 +1331,45 @@ def update(
 
 
 class LangfuseAgent(LangfuseGraphObservation):
-    """Specialized span for AGENT observations in agentic workflows."""
+    """Specialized span for agent observations in agentic workflows."""
 
     def __init__(self, **kwargs):
         """Initialize a new LangfuseAgent span."""
-        kwargs["observation_type"] = "AGENT"
+        kwargs["observation_type"] = "agent"
         super().__init__(**kwargs)
 
 
 class LangfuseTool(LangfuseGraphObservation):
-    """Specialized span for TOOL observations in agentic workflows."""
+    """Specialized span for tool observations in agentic workflows."""
 
     def __init__(self, **kwargs):
         """Initialize a new LangfuseTool span."""
-        kwargs["observation_type"] = "TOOL"
+        kwargs["observation_type"] = "tool"
         super().__init__(**kwargs)
 
 
 class LangfuseChain(LangfuseGraphObservation):
-    """Specialized span for CHAIN observations in agentic workflows."""
+    """Specialized span for chain observations in agentic workflows."""
 
     def __init__(self, **kwargs):
         """Initialize a new LangfuseChain span."""
-        kwargs["observation_type"] = "CHAIN"
+        kwargs["observation_type"] = "chain"
         super().__init__(**kwargs)
 
 
 class LangfuseRetriever(LangfuseGraphObservation):
-    """Specialized span for RETRIEVER observations in agentic workflows."""
+    """Specialized span for retriever observations in agentic workflows."""
 
     def __init__(self, **kwargs):
         """Initialize a new LangfuseRetriever span."""
-        kwargs["observation_type"] = "RETRIEVER"
+        kwargs["observation_type"] = "retriever"
         super().__init__(**kwargs)
 
 
 class LangfuseEmbedding(LangfuseGraphObservation):
-    """Specialized span for EMBEDDING observations in agentic workflows."""
+    """Specialized span for embedding observations in agentic workflows."""
 
     def __init__(self, **kwargs):
         """Initialize a new LangfuseEmbedding span."""
-        kwargs["observation_type"] = "EMBEDDING"
+        kwargs["observation_type"] = "embedding"
         super().__init__(**kwargs)