Function _wrapped_async

agentops/sdk/decorators/factory.py:421–450

()

            elif is_async:

                async def _wrapped_async() -> Any:
                    with _create_as_current_span(
                        operation_name,
                        entity_kind,
                        version=version,
                        attributes={CoreAttributes.TAGS: tags} if tags else None,
                    ) as span:
                        try:
                            _record_entity_input(span, args, kwargs, entity_kind=entity_kind)
                            # Set cost attribute if tool
                            if entity_kind == "tool" and cost is not None:
                                span.set_attribute(SpanAttributes.LLM_USAGE_TOOL_COST, cost)
                            # Set spec attribute if guardrail
                            if entity_kind == "guardrail" and (spec == "input" or spec == "output"):
                                span.set_attribute(
                                    SpanAttributes.AGENTOPS_DECORATOR_SPEC.format(entity_kind=entity_kind), spec
                                )
                        except Exception as e:
                            logger.warning(f"Input recording failed for '{operation_name}': {e}")
                        try:
                            result = await wrapped_func(*args, **kwargs)
                            try:
                                _record_entity_output(span, result, entity_kind=entity_kind)
                            except Exception as e:
                                logger.warning(f"Output recording failed for '{operation_name}': {e}")
                            return result
                        except Exception as e:
                            logger.error(f"Error in async function execution: {e}")
                            span.record_exception(e)
                            raise

                return _wrapped_async()
            else:  # Sync function for non-SESSION kinds