monocle-apptrace 0.4.0b2__py3-none-any.whl → 0.4.1__py3-none-any.whl

monocle_apptrace/instrumentation/__init__.py

@@ -1 +1,2 @@
-from .common import *
+from .common import *
+from .metamodel.azfunc.wrapper import monocle_trace_azure_function_route

monocle_apptrace/instrumentation/common/constants.py

@@ -54,6 +54,9 @@ llm_type_map = {
     "anthropic": "anthropic",
     "chatanthropic":"anthropic",
     "anthropicchatgenerator":"anthropic",
+    "chatcompletionsclient": "azure_ai_inference",
+    "embeddingsclient": "azure_ai_inference",
+    "imageembeddingsclient": "azure_ai_inference",
 }
 
 MONOCLE_INSTRUMENTOR = "monocle_apptrace"

monocle_apptrace/instrumentation/common/instrumentor.py

@@ -362,7 +362,7 @@ def monocle_trace_http_route(func):
     if inspect.iscoroutinefunction(func):
         @wraps(func)
         async def wrapper(*args, **kwargs):
-            return http_async_route_handler(func, *args, **kwargs)
+            return await http_async_route_handler(func, *args, **kwargs)
         return wrapper
     else:
         @wraps(func)

monocle_apptrace/instrumentation/common/span_handler.py

@@ -83,10 +83,10 @@ class SpanHandler:
     def post_task_processing(self, to_wrap, wrapped, instance, args, kwargs, result, span:Span):
         pass
 
-    def hydrate_span(self, to_wrap, wrapped, instance, args, kwargs, result, span, ex:Exception = None) -> bool:
+    def hydrate_span(self, to_wrap, wrapped, instance, args, kwargs, result, span, parent_span = None, ex:Exception = None) -> bool:
         try:
             detected_error_in_attribute = self.hydrate_attributes(to_wrap, wrapped, instance, args, kwargs, result, span)
-            detected_error_in_event = self.hydrate_events(to_wrap, wrapped, instance, args, kwargs, result, span, ex)
+            detected_error_in_event = self.hydrate_events(to_wrap, wrapped, instance, args, kwargs, result, span, parent_span, ex)
             if detected_error_in_attribute or detected_error_in_event:
                 span.set_attribute(MONOCLE_DETECTED_SPAN_ERROR, True)
         finally:
@@ -134,18 +134,20 @@ class SpanHandler:
             span.set_attribute("entity.count", span_index)
         return detected_error
 
-    def hydrate_events(self, to_wrap, wrapped, instance, args, kwargs, ret_result, span, ex:Exception=None) -> bool:
+    def hydrate_events(self, to_wrap, wrapped, instance, args, kwargs, ret_result, span, parent_span=None, ex:Exception=None) -> bool:
         detected_error:bool = False
         if 'output_processor' in to_wrap and to_wrap["output_processor"] is not None:
             output_processor=to_wrap['output_processor']
             skip_processors:list[str] = self.skip_processor(to_wrap, wrapped, instance, span, args, kwargs) or []
 
             arguments = {"instance": instance, "args": args, "kwargs": kwargs, "result": ret_result, "exception":ex}
-            if 'events' in output_processor and 'events' not in skip_processors:
+            # Process events if they are defined in the output_processor.
+            # In case of inference.modelapi skip the event processing unless the span has an exception
+            if 'events' in output_processor and ('events' not in skip_processors or ex is not None):
                 events = output_processor['events']
                 for event in events:
                     event_name = event.get("name")
-                    if 'events.'+event_name in skip_processors:
+                    if 'events.'+event_name in skip_processors and ex is None:
                         continue
                     event_attributes = {}
                     attributes = event.get("attributes", [])
@@ -157,7 +159,7 @@ class SpanHandler:
                                 result = accessor(arguments)
                                 if result and isinstance(result, dict):
                                     result = dict((key, value) for key, value in result.items() if value is not None)
-                                if result and isinstance(result, (str, list, dict)):
+                                if result and isinstance(result, (int, str, list, dict)):
                                     if attribute_key is not None:
                                         event_attributes[attribute_key] = result
                                     else:

monocle_apptrace/instrumentation/common/utils.py

@@ -236,6 +236,13 @@ def set_scopes_from_baggage(baggage_context:Context):
             scope_name = scope_key[len(MONOCLE_SCOPE_NAME_PREFIX):]
             set_scope(scope_name, scope_value)
 
+def get_parent_span() -> Span:
+    parent_span: Span = None
+    _parent_span_context = get_current()
+    if _parent_span_context is not None and _parent_span_context.get(_SPAN_KEY, None):
+        parent_span = _parent_span_context.get(_SPAN_KEY, None)    
+    return parent_span
+
 def extract_http_headers(headers) -> object:
     global http_scopes
     trace_context:Context = extract(headers, context=get_current())
@@ -362,11 +369,19 @@ def get_llm_type(instance):
     except:
         pass
 
+def get_status(arguments):
+    if arguments['exception'] is not None:
+        return 'error'
+    else:
+        return 'success'
+
 def get_exception_status_code(arguments):
-    if arguments['exception'] is not None and hasattr(arguments['exception'], 'code'):
+    if arguments['exception'] is not None and hasattr(arguments['exception'], 'code') and arguments['exception'].code is not None:
         return arguments['exception'].code
-    else:
+    elif arguments['exception'] is not None:
         return 'error'
+    else:
+        return 'success'
 
 def get_exception_message(arguments):
     if arguments['exception'] is not None:
@@ -377,6 +392,22 @@ def get_exception_message(arguments):
     else:
         return ''
 
+def get_status_code(arguments):
+    if arguments["exception"] is not None:
+        return get_exception_status_code(arguments)
+    elif hasattr(arguments["result"], "status"):
+        return arguments["result"].status
+    else:
+        return 'success'
+
+def get_status(arguments):
+    if arguments["exception"] is not None:
+        return 'error'
+    elif get_status_code(arguments) == 'success':
+        return 'success'
+    else:
+        return 'error'
+
 def patch_instance_method(obj, method_name, func):
     """
     Patch a special method (like __iter__) for a single instance.

monocle_apptrace/instrumentation/common/wrapper.py

@@ -9,7 +9,8 @@ from monocle_apptrace.instrumentation.common.utils import (
     set_scopes,
     with_tracer_wrapper,
     set_scope,
-    remove_scope
+    remove_scope,
+    get_parent_span
 )
 from monocle_apptrace.instrumentation.common.constants import WORKFLOW_TYPE_KEY, ADD_NEW_WORKFLOW
 logger = logging.getLogger(__name__)
@@ -35,10 +36,10 @@ def pre_process_span(name, tracer, handler, add_workflow_span, to_wrap, wrapped,
         except Exception as e:
             logger.info(f"Warning: Error occurred in pre_task_processing: {e}")
 
-def post_process_span(handler, to_wrap, wrapped, instance, args, kwargs, return_value, span, ex = None):
+def post_process_span(handler, to_wrap, wrapped, instance, args, kwargs, return_value, span, parent_span=None, ex = None):
     if not (SpanHandler.is_root_span(span) or get_value(ADD_NEW_WORKFLOW) == True):
         try:
-            handler.hydrate_span(to_wrap, wrapped, instance, args, kwargs, return_value, span, ex)
+            handler.hydrate_span(to_wrap, wrapped, instance, args, kwargs, return_value, span, parent_span, ex)
         except Exception as e:
             logger.info(f"Warning: Error occurred in hydrate_span: {e}")
         
@@ -59,45 +60,35 @@ def monocle_wrapper_span_processor(tracer: Tracer, handler: SpanHandler, to_wrap
     name = get_span_name(to_wrap, instance)
     return_value = None
     span_status = None
-    if(get_auto_close_span(to_wrap, kwargs)):
-        with tracer.start_as_current_span(name) as span:
-            pre_process_span(name, tracer, handler, add_workflow_span, to_wrap, wrapped, instance, args, kwargs, span, source_path)
-            
-            if SpanHandler.is_root_span(span) or add_workflow_span:
-                # Recursive call for the actual span
-                return_value, span_status = monocle_wrapper_span_processor(tracer, handler, to_wrap, wrapped, instance, source_path, False, args, kwargs)
-                span.set_status(span_status)
-            else:
-                ex:Exception = None
-                try:
-                    with SpanHandler.workflow_type(to_wrap, span):
-                        return_value = wrapped(*args, **kwargs)
-                except Exception as e:
-                    ex = e
-                    raise
-                finally:
-                    post_process_span(handler, to_wrap, wrapped, instance, args, kwargs, return_value, span, ex)
-                span_status = span.status
-    else:
-        span = tracer.start_span(name)
-        
+    auto_close_span = get_auto_close_span(to_wrap, kwargs)
+    parent_span = get_parent_span()
+    with tracer.start_as_current_span(name, end_on_exit=auto_close_span) as span:
         pre_process_span(name, tracer, handler, add_workflow_span, to_wrap, wrapped, instance, args, kwargs, span, source_path)
         
-        def post_process_span_internal(ret_val):
-            nonlocal handler, to_wrap, wrapped, instance, args, kwargs, span
-            post_process_span(handler, to_wrap, wrapped, instance, args, kwargs, ret_val, span)
-            span.end()
-                    
-        try:
-            with SpanHandler.workflow_type(to_wrap, span):
-                return_value = wrapped(*args, **kwargs)
-        finally:
-            if to_wrap.get("output_processor") and to_wrap.get("output_processor").get("response_processor"):
-                # Process the stream
-                to_wrap.get("output_processor").get("response_processor")(to_wrap, return_value, post_process_span_internal)
-            else: 
+        if SpanHandler.is_root_span(span) or add_workflow_span:
+            # Recursive call for the actual span
+            return_value, span_status = monocle_wrapper_span_processor(tracer, handler, to_wrap, wrapped, instance, source_path, False, args, kwargs)
+            span.set_status(span_status)
+            if not auto_close_span:
                 span.end()
-        span_status = span.status
+        else:
+            ex:Exception = None
+            try:
+                with SpanHandler.workflow_type(to_wrap, span):
+                    return_value = wrapped(*args, **kwargs)
+            except Exception as e:
+                ex = e
+                raise
+            finally:
+                def post_process_span_internal(ret_val):
+                    post_process_span(handler, to_wrap, wrapped, instance, args, kwargs, ret_val, span, parent_span ,ex)
+                    if not auto_close_span:
+                        span.end()
+                if ex is None and not auto_close_span and to_wrap.get("output_processor") and to_wrap.get("output_processor").get("response_processor"):
+                    to_wrap.get("output_processor").get("response_processor")(to_wrap, return_value, post_process_span_internal)
+                else:
+                    post_process_span_internal(return_value)
+            span_status = span.status
     return return_value, span_status
 
 def monocle_wrapper(tracer: Tracer, handler: SpanHandler, to_wrap, wrapped, instance, source_path, args, kwargs):
@@ -129,44 +120,34 @@ async def amonocle_wrapper_span_processor(tracer: Tracer, handler: SpanHandler,
     name = get_span_name(to_wrap, instance)
     return_value = None
     span_status = None
-    if(get_auto_close_span(to_wrap, kwargs)):
-        with tracer.start_as_current_span(name) as span:
-            pre_process_span(name, tracer, handler, add_workflow_span, to_wrap, wrapped, instance, args, kwargs, span, source_path)
-            
-            if SpanHandler.is_root_span(span) or add_workflow_span:
-                # Recursive call for the actual span
-                return_value, span_status = await amonocle_wrapper_span_processor(tracer, handler, to_wrap, wrapped, instance, source_path, False, args, kwargs)
-                span.set_status(span_status)
-            else:
-                ex:Exception = None
-                try:
-                    with SpanHandler.workflow_type(to_wrap, span):
-                        return_value = await wrapped(*args, **kwargs)
-                except Exception as e:
-                    ex = e
-                    raise
-                finally:
-                    post_process_span(handler, to_wrap, wrapped, instance, args, kwargs, return_value, span, ex)
-                span_status = span.status
-    else:
-        span = tracer.start_span(name)
-        
+    auto_close_span = get_auto_close_span(to_wrap, kwargs)
+    parent_span = get_parent_span()
+    with tracer.start_as_current_span(name, end_on_exit=auto_close_span) as span:
         pre_process_span(name, tracer, handler, add_workflow_span, to_wrap, wrapped, instance, args, kwargs, span, source_path)
         
-        def post_process_span_internal(ret_val):
-            nonlocal handler, to_wrap, wrapped, instance, args, kwargs, span
-            post_process_span(handler, to_wrap, wrapped, instance, args, kwargs, ret_val, span)
-            span.end()
-
-        try:
-            with SpanHandler.workflow_type(to_wrap, span):
-                return_value = await wrapped(*args, **kwargs)
-        finally:
-            if to_wrap.get("output_processor") and to_wrap.get("output_processor").get("response_processor"):
-                # Process the stream
-                to_wrap.get("output_processor").get("response_processor")(to_wrap, return_value, post_process_span_internal)
-            else: 
+        if SpanHandler.is_root_span(span) or add_workflow_span:
+            # Recursive call for the actual span
+            return_value, span_status = await amonocle_wrapper_span_processor(tracer, handler, to_wrap, wrapped, instance, source_path, False, args, kwargs)
+            span.set_status(span_status)
+            if not auto_close_span:
                 span.end()
+        else:
+            ex:Exception = None
+            try:
+                with SpanHandler.workflow_type(to_wrap, span):
+                    return_value = await wrapped(*args, **kwargs)
+            except Exception as e:
+                ex = e
+                raise
+            finally:
+                def post_process_span_internal(ret_val):
+                    post_process_span(handler, to_wrap, wrapped, instance, args, kwargs, ret_val, span, parent_span,ex)
+                    if not auto_close_span:
+                        span.end()
+                if ex is None and not auto_close_span and to_wrap.get("output_processor") and to_wrap.get("output_processor").get("response_processor"):
+                    to_wrap.get("output_processor").get("response_processor")(to_wrap, return_value, post_process_span_internal)
+                else:
+                    post_process_span_internal(return_value)
         span_status = span.status
     return return_value, span.status
 
@@ -229,7 +210,7 @@ async def ascope_wrapper(tracer: Tracer, handler: SpanHandler, to_wrap, wrapped,
 @with_tracer_wrapper
 def scopes_wrapper(tracer: Tracer, handler: SpanHandler, to_wrap, wrapped, instance, source_path, args, kwargs):
     scope_values = to_wrap.get('scope_values', None)
-    scope_values = evaluate_scope_values(args, kwargs, scope_values)
+    scope_values = evaluate_scope_values(args, kwargs, to_wrap, scope_values)
     token = None
     try:
         if scope_values:
@@ -243,7 +224,7 @@ def scopes_wrapper(tracer: Tracer, handler: SpanHandler, to_wrap, wrapped, insta
 @with_tracer_wrapper
 async def ascopes_wrapper(tracer: Tracer, handler: SpanHandler, to_wrap, wrapped, instance, source_path, args, kwargs):
     scope_values = to_wrap.get('scope_values', None)
-    scope_values = evaluate_scope_values(args, kwargs, scope_values)
+    scope_values = evaluate_scope_values(args, kwargs, to_wrap, scope_values)
     token = None
     try:
         if scope_values:
@@ -254,7 +235,7 @@ async def ascopes_wrapper(tracer: Tracer, handler: SpanHandler, to_wrap, wrapped
         if token:
             remove_scope(token)
 
-def evaluate_scope_values(args, kwargs, scope_values):
+def evaluate_scope_values(args, kwargs, to_wrap, scope_values):
     if callable(scope_values):
         try:
             scope_values = scope_values(args, kwargs)
@@ -263,5 +244,4 @@ def evaluate_scope_values(args, kwargs, scope_values):
             scope_values = None
     if isinstance(scope_values, dict):
         return scope_values
-    return None
-    
+    return None

monocle_apptrace/instrumentation/common/wrapper_method.py

@@ -2,6 +2,7 @@
 from typing import Any, Dict
 from monocle_apptrace.instrumentation.common.wrapper import task_wrapper, scope_wrapper
 from monocle_apptrace.instrumentation.common.span_handler import SpanHandler, NonFrameworkSpanHandler
+from monocle_apptrace.instrumentation.metamodel.azureaiinference.methods import AZURE_AI_INFERENCE_METHODS
 from monocle_apptrace.instrumentation.metamodel.botocore.methods import BOTOCORE_METHODS
 from monocle_apptrace.instrumentation.metamodel.botocore.handlers.botocore_span_handler import BotoCoreSpanHandler
 from monocle_apptrace.instrumentation.metamodel.langchain.methods import (
@@ -20,6 +21,8 @@ from monocle_apptrace.instrumentation.metamodel.teamsai.methods import (TEAMAI_M
 from monocle_apptrace.instrumentation.metamodel.anthropic.methods import (ANTHROPIC_METHODS, )
 from monocle_apptrace.instrumentation.metamodel.aiohttp.methods import (AIOHTTP_METHODS, )
 from monocle_apptrace.instrumentation.metamodel.aiohttp._helper import aiohttpSpanHandler
+from monocle_apptrace.instrumentation.metamodel.azfunc._helper import (azureSpanHandler)
+from monocle_apptrace.instrumentation.metamodel.azfunc.methods import AZFUNC_HTTP_METHODS
 class WrapperMethod:
     def __init__(
             self,
@@ -68,7 +71,7 @@ class WrapperMethod:
     def get_span_handler(self) -> SpanHandler:
         return self.span_handler()
 
-DEFAULT_METHODS_LIST = LANGCHAIN_METHODS + LLAMAINDEX_METHODS + HAYSTACK_METHODS + BOTOCORE_METHODS + FLASK_METHODS + REQUESTS_METHODS + LANGGRAPH_METHODS + OPENAI_METHODS + TEAMAI_METHODS + ANTHROPIC_METHODS + AIOHTTP_METHODS
+DEFAULT_METHODS_LIST = LANGCHAIN_METHODS + LLAMAINDEX_METHODS + HAYSTACK_METHODS + BOTOCORE_METHODS + FLASK_METHODS + REQUESTS_METHODS + LANGGRAPH_METHODS + OPENAI_METHODS + TEAMAI_METHODS + ANTHROPIC_METHODS + AIOHTTP_METHODS + AZURE_AI_INFERENCE_METHODS + AZFUNC_HTTP_METHODS
 
 MONOCLE_SPAN_HANDLERS: Dict[str, SpanHandler] = {
     "default": SpanHandler(),
@@ -79,4 +82,5 @@ MONOCLE_SPAN_HANDLERS: Dict[str, SpanHandler] = {
     "request_handler": RequestSpanHandler(),
     "non_framework_handler": NonFrameworkSpanHandler(),
     "openai_handler": OpenAISpanHandler(),
+    "azure_func_handler": azureSpanHandler(),
 }

monocle_apptrace/instrumentation/metamodel/anthropic/_helper.py

@@ -9,6 +9,7 @@ from monocle_apptrace.instrumentation.common.utils import (
     get_keys_as_tuple,
     get_nested_value,
     try_option,
+    get_exception_message,
 )
 
 
@@ -39,12 +40,36 @@ def extract_messages(kwargs):
         logger.warning("Warning: Error occurred in extract_messages: %s", str(e))
         return []
 
+def get_exception_status_code(arguments):
+    if arguments['exception'] is not None and hasattr(arguments['exception'], 'status_code') and arguments['exception'].status_code is not None:
+        return arguments['exception'].status_code
+    elif arguments['exception'] is not None:
+        return 'error'
+    else:
+        return 'success'
 
-def extract_assistant_message(response):
+def get_status_code(arguments):
+    if arguments["exception"] is not None:
+        return get_exception_status_code(arguments)
+    elif hasattr(arguments["result"], "status"):
+        return arguments["result"].status
+    else:
+        return 'success'
+
+def extract_assistant_message(arguments):
     try:
-        if response is not None and hasattr(response,"content") and len(response.content) >0:
-            if hasattr(response.content[0],"text"):
-                return response.content[0].text
+        status = get_status_code(arguments)
+        response: str = ""
+        if status == 'success':
+            if arguments['result'] is not None and hasattr(arguments['result'],"content") and len(arguments['result'].content) >0:
+                if hasattr(arguments['result'].content[0],"text"):
+                    response = arguments['result'].content[0].text
+        else:
+            if arguments["exception"] is not None:
+                response = get_exception_message(arguments)
+            elif hasattr(arguments["result"], "error"):
+                response = arguments["result"].error
+        return response
     except (IndexError, AttributeError) as e:
         logger.warning("Warning: Error occurred in extract_assistant_message: %s", str(e))
         return None

monocle_apptrace/instrumentation/metamodel/anthropic/entities/inference.py

@@ -1,7 +1,9 @@
 from monocle_apptrace.instrumentation.metamodel.anthropic import (
     _helper,
 )
-from monocle_apptrace.instrumentation.common.utils import resolve_from_alias, get_llm_type
+from monocle_apptrace.instrumentation.common.utils import (resolve_from_alias, get_llm_type,
+    get_status, get_status_code
+)
 
 INFERENCE = {
     "type": "inference",
@@ -52,10 +54,18 @@ INFERENCE = {
         {
             "name": "data.output",
             "attributes": [
+                {
+                    "attribute": "status",
+                    "accessor": lambda arguments: get_status(arguments)
+                },
+                {
+                    "attribute": "status_code",
+                    "accessor": lambda arguments: _helper.get_status_code(arguments)
+                },
                 {
                     "_comment": "this is result from LLM",
                     "attribute": "response",
-                    "accessor": lambda arguments: _helper.extract_assistant_message(arguments['result'])
+                    "accessor": lambda arguments: _helper.extract_assistant_message(arguments)
                 }
             ]
         },

monocle_apptrace/instrumentation/metamodel/azfunc/_helper.py

@@ -0,0 +1,78 @@
+import logging
+from threading import local
+from monocle_apptrace.instrumentation.common.utils import extract_http_headers, clear_http_scopes, try_option, Option, MonocleSpanException
+from monocle_apptrace.instrumentation.common.span_handler import SpanHandler
+from monocle_apptrace.instrumentation.common.constants import HTTP_SUCCESS_CODES
+from urllib.parse import unquote, urlparse, ParseResult
+
+
+logger = logging.getLogger(__name__)
+MAX_DATA_LENGTH = 1000
+token_data = local()
+token_data.current_token = None
+
+def get_url(kwargs) -> ParseResult:
+    url_str = try_option(getattr, kwargs['req'], 'url')
+    url = url_str.unwrap_or(None)
+    if url is not None:
+        return urlparse(url)
+    else:
+        return None
+
+def get_route(kwargs) -> str:
+    url:ParseResult = get_url(kwargs)
+    if url is not None:
+        return url.path
+
+def get_method(kwargs) -> str:
+#    return args[0]['method'] if 'method' in args[0] else ""
+    http_method: Option[str] = try_option(getattr, kwargs['req'], 'method')
+    return http_method.unwrap_or("")
+
+def get_params(kwargs) -> dict:
+    url:ParseResult = get_url(kwargs)
+    if url is not None:
+        return unquote(url.query)
+
+def get_body(kwargs) -> dict:
+    if hasattr(kwargs['req'], 'get_body'):
+        response = kwargs.get_body()
+        if isinstance(response, bytes):
+            response = response.decode('utf-8', errors='ignore')
+    else:
+        response = ""
+    return response
+
+def extract_response(result) -> str:
+    if hasattr(result, 'get_body'):
+        response = result.get_body() #  text[0:max(result.text.__len__(), MAX_DATA_LENGTH)]
+        if isinstance(response, bytes):
+            response = response.decode('utf-8', errors='ignore')
+    else:
+        response = ""
+    return response
+
+def extract_status(result) -> str:
+    status = f"{result.status_code}" if hasattr(result, 'status_code') else ""
+    if status not in HTTP_SUCCESS_CODES:
+        error_message = extract_response(result)
+        raise MonocleSpanException(f"error: {status} - {error_message}")
+    return status
+
+def azure_func_pre_tracing(kwargs):
+    headers = kwargs['req'].headers if hasattr(kwargs['req'], 'headers') else {}
+    token_data.current_token = extract_http_headers(headers)
+
+def azure_func_post_tracing():
+    clear_http_scopes(token_data.current_token)
+    token_data.current_token = None
+
+class azureSpanHandler(SpanHandler):
+
+    def pre_tracing(self, to_wrap, wrapped, instance, args, kwargs):
+        azure_func_pre_tracing(kwargs)
+        return super().pre_tracing(to_wrap, wrapped, instance, args, kwargs)
+    
+    def post_tracing(self, to_wrap, wrapped, instance, args, kwargs, return_value):
+        azure_func_post_tracing()
+        return super().post_tracing(to_wrap, wrapped, instance, args, kwargs, return_value)

monocle_apptrace/instrumentation/metamodel/azfunc/entities/http.py

@@ -0,0 +1,51 @@
+from monocle_apptrace.instrumentation.metamodel.azfunc import _helper
+AZFUNC_HTTP_PROCESSOR = {
+    "type": "http.process",
+    "attributes": [
+        [
+            {
+                "_comment": "request method, request URI",
+                "attribute": "method",
+                "accessor": lambda arguments: _helper.get_method(arguments['kwargs'])
+            },
+            {
+                "_comment": "request method, request URI",
+                "attribute": "route",
+                "accessor": lambda arguments: _helper.get_route(arguments['kwargs'])
+            },
+            {
+                "_comment": "request method, request URI",
+                "attribute": "body",
+                "accessor": lambda arguments: _helper.get_body(arguments['kwargs'])
+            },
+        ]
+    ],
+    "events": [
+        {
+            "name": "data.input",
+            "attributes": [
+                {
+                    "_comment": "route params",
+                    "attribute": "params",
+                    "accessor": lambda arguments: _helper.get_params(arguments['kwargs'])
+                }
+            ]
+        },
+        {
+            "name": "data.output",
+            "attributes": [
+                {
+                    "_comment": "status from HTTP response",
+                    "attribute": "status",
+                    "accessor": lambda arguments: _helper.extract_status(arguments['result'])
+                },
+                {
+                    "_comment": "this is result from LLM",
+                    "attribute": "response",
+                    "accessor": lambda arguments: _helper.extract_response(arguments['result'])
+                }
+            ]
+        }
+
+    ]
+}

monocle_apptrace/instrumentation/metamodel/azfunc/methods.py

@@ -0,0 +1,23 @@
+from monocle_apptrace.instrumentation.common.wrapper import atask_wrapper, task_wrapper
+from monocle_apptrace.instrumentation.metamodel.azfunc.entities.http import AZFUNC_HTTP_PROCESSOR
+
+AZFUNC_HTTP_METHODS = [
+    {
+        "package": "monocle_apptrace.instrumentation.metamodel.azfunc.wrapper",
+        "object": "AzureFunctionRouteWrapper",
+        "method": "run_async",
+        "span_name": "azure_function_route",
+        "wrapper_method": atask_wrapper,
+        "span_handler": "azure_func_handler",
+        "output_processor": AZFUNC_HTTP_PROCESSOR
+    },
+    {
+        "package": "monocle_apptrace.instrumentation.metamodel.azfunc.wrapper",
+        "object": "AzureFunctionRouteWrapper",
+        "method": "run_sync",
+        "span_name": "azure_function_route",
+        "wrapper_method": task_wrapper,
+        "span_handler": "azure_func_handler",
+        "output_processor": AZFUNC_HTTP_PROCESSOR
+    }
+]

monocle_apptrace/instrumentation/metamodel/azfunc/wrapper.py

@@ -0,0 +1,23 @@
+from functools import wraps
+import inspect
+
+def monocle_trace_azure_function_route(func):
+    if inspect.iscoroutinefunction(func):
+        @wraps(func)
+        async def wrapper(*args, **kwargs):
+            return await AzureFunctionRouteWrapper.run_async(func, *args, **kwargs)
+        return wrapper
+    else:
+        @wraps(func)
+        def wrapper(*args, **kwargs):
+            return AzureFunctionRouteWrapper.run_sync(func ,*args, **kwargs)
+        return wrapper
+
+class AzureFunctionRouteWrapper:
+    @staticmethod
+    async def run_async(func, *args, **kwargs):
+        return await func(*args, **kwargs)
+
+    @staticmethod
+    def run_sync(func, *args, **kwargs):
+        return func(*args, **kwargs)

monocle_apptrace/instrumentation/metamodel/azureaiinference/__init__.py

@@ -0,0 +1 @@
+# Azure AI Inference instrumentation module