Update dispatcher.py

🤦
run-llama · Feb 14, 2025 · f43e337 · f43e337
1 parent c0de5d2
commit f43e337
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llama-index-core/llama_index/core/instrumentation/dispatcher.py b/llama-index-core/llama_index/core/instrumentation/dispatcher.py
@@ -240,7 +240,7 @@ def span_exit(
             else:
                 c = c.parent
 
-    def span(self, func: Callable[..., _R]) -> _R:
+    def span(self, func: Callable[..., _R]) -> Callable[..., _R]:
         # The `span` decorator should be idempotent.
         try:
             if hasattr(func, DISPATCHER_SPAN_DECORATED_ATTR):