fix test

Author: nimarb

langfuse/__init__.py

@@ -4,7 +4,7 @@
 from ._client.attributes import LangfuseOtelSpanAttributes
 from ._client.get_client import get_client
 from ._client.observe import observe
-from ._client.span import LangfuseEvent, LangfuseGeneration, LangfuseSpan
+from ._client.span import LangfuseEvent, LangfuseGeneration, LangfuseSpan, LangfuseAgent, LangfuseTool, LangfuseChain, LangfuseEmbedding, LangfuseRetriever
 
 Langfuse = _client_module.Langfuse
 
@@ -16,4 +16,9 @@
     "LangfuseGeneration",
     "LangfuseEvent",
     "LangfuseOtelSpanAttributes",
+    "LangfuseAgent",
+    "LangfuseTool",
+    "LangfuseChain",
+    "LangfuseEmbedding",
+    "LangfuseRetriever",
 ]

langfuse/_client/client.py

@@ -432,7 +432,7 @@ def start_as_current_span(
         return cast(
             _AgnosticContextManager[LangfuseSpan],
             self._start_as_current_otel_span_with_processed_media(
-                as_type=as_type,
+                as_type="span",
                 name=name,
                 end_on_exit=end_on_exit,
                 input=input,