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

monocle_apptrace/instrumentation/metamodel/teamsai/_helper.py

@@ -1,3 +1,4 @@
+import logging
 from monocle_apptrace.instrumentation.common.utils import MonocleSpanException
 from monocle_apptrace.instrumentation.common.utils import (
     Option,
@@ -7,6 +8,9 @@ from monocle_apptrace.instrumentation.common.utils import (
     get_exception_message,
     get_exception_status_code
 )
+
+logger = logging.getLogger(__name__)
+
 def capture_input(arguments):
     """
     Captures the input from Teams AI state.
@@ -18,6 +22,7 @@ def capture_input(arguments):
     try:
         # Get the memory object from kwargs
         kwargs = arguments.get("kwargs", {})
+        messages = []
 
         # If memory exists, try to get the input from temp
         if "memory" in kwargs:
@@ -29,14 +34,20 @@ def capture_input(arguments):
                 if temp and hasattr(temp, "get"):
                     input_value = temp.get("input")
                     if input_value:
-                        return str(input_value)
-
+                        messages.append({'user': str(input_value)})
+        system_prompt = ""
+        try:
+            system_prompt = kwargs.get("template").prompt.sections[0].sections[0].template
+            messages.append({'system': system_prompt})
+        except Exception as e:
+            print(f"Debug - Error accessing system prompt: {str(e)}")
+            
         # Try alternative path through context if memory path fails
         context = kwargs.get("context")
         if hasattr(context, "activity") and hasattr(context.activity, "text"):
-            return str(context.activity.text)
+            messages.append({'user': str(context.activity.text)})
 
-        return "No input found in memory or context"
+        return [str(message) for message in messages]
     except Exception as e:
         print(f"Debug - Arguments structure: {str(arguments)}")
         print(f"Debug - kwargs: {str(kwargs)}")
@@ -59,6 +70,41 @@ def capture_prompt_info(arguments):
     except Exception as e:
         return f"Error capturing prompt: {str(e)}"
 
+def capture_prompt_template_info(arguments):
+    """Captures prompt information from ActionPlanner state"""
+    try:
+        kwargs = arguments.get("kwargs", {})
+        prompt = kwargs.get("prompt")
+
+        if hasattr(prompt,"prompt") and prompt.prompt is not None:
+            if "_text" in prompt.prompt.__dict__:
+                prompt_template = prompt.prompt.__dict__.get("_text", None)
+                return prompt_template
+            elif "_sections" in prompt.prompt.__dict__ and prompt.prompt._sections is not None:
+                sections = prompt.prompt._sections[0].__dict__.get("_sections", None)
+                if sections is not None and "_template" in sections[0].__dict__:
+                    return sections[0].__dict__.get("_template", None)
+
+
+        return "No prompt information found"
+    except Exception as e:
+        return f"Error capturing prompt: {str(e)}"
+
+def status_check(arguments):
+    if hasattr(arguments["result"], "error") and arguments["result"].error is not None:
+        error_msg:str = arguments["result"].error
+        error_code:str = arguments["result"].status if hasattr(arguments["result"], "status") else "unknown"
+        raise MonocleSpanException(f"Error: {error_code} - {error_msg}")
+
+def get_prompt_template(arguments):
+    pass
+    return {
+        "prompt_template_name": capture_prompt_info(arguments),
+        "prompt_template": capture_prompt_template_info(arguments),
+        "prompt_template_description": get_nested_value(arguments.get("kwargs", {}), ["prompt", "config", "description"]),
+        "prompt_template_type": get_nested_value(arguments.get("kwargs", {}), ["prompt", "config", "type"])
+    }
+
 def get_status_code(arguments):
     if arguments["exception"] is not None:
         return get_exception_status_code(arguments)

monocle_apptrace/instrumentation/metamodel/teamsai/entities/inference/actionplanner_output_processor.py

@@ -28,22 +28,42 @@ ACTIONPLANNER_OUTPUT_PROCESSOR = {
             {
                 "attribute": "tokenizer",
                 "accessor": lambda arguments: arguments["instance"]._options.tokenizer.__class__.__name__ if hasattr(arguments["instance"], "_options") else "GPTTokenizer"
-            },
-            {
-                "attribute": "prompt_name",
-                "accessor": _helper.capture_prompt_info
-            },
-            {
-                "attribute": "validator",
-                "accessor": lambda arguments: arguments["kwargs"].get("validator").__class__.__name__ if arguments.get("kwargs", {}).get("validator") else "DefaultResponseValidator"
-            },
-            {
-                "attribute": "memory_type",
-                "accessor": lambda arguments: arguments["kwargs"].get("memory").__class__.__name__ if arguments.get("kwargs", {}).get("memory") else "unknown"
             }
         ]
     ],
     "events": [
+        {
+            "name": "data.input",
+            "_comment": "input configuration to ActionPlanner",
+            "attributes": [
+               {
+                    "_comment": "execution metadata",
+                    "accessor": lambda arguments: _helper.get_prompt_template(arguments)
+                },
+                {
+                    "attribute": "validator",
+                    "accessor": lambda arguments: arguments["kwargs"].get("validator").__class__.__name__ if arguments.get("kwargs", {}).get("validator") else "DefaultResponseValidator"
+                },
+                {
+                    "attribute": "memory_type",
+                    "accessor": lambda arguments: arguments["kwargs"].get("memory").__class__.__name__ if arguments.get("kwargs", {}).get("memory") else "unknown"
+                }
+            ]
+        },
+        {
+            "name": "data.output",
+            "_comment": "output from ActionPlanner",
+            "attributes": [
+                {
+                    "attribute": "status",
+                    "accessor": lambda arguments: _helper.status_check(arguments)
+                },
+                {
+                    "attribute": "response",
+                    "accessor": lambda arguments: arguments["result"].message.content if hasattr(arguments["result"], "message") else str(arguments["result"])
+                }
+            ]
+        },
         {
             "name": "metadata",
             "attributes": [

monocle_apptrace/instrumentation/metamodel/teamsai/methods.py

@@ -9,31 +9,44 @@ from monocle_apptrace.instrumentation.metamodel.teamsai.entities.inference.teams
 )
 from monocle_apptrace.instrumentation.metamodel.teamsai.entities.inference.actionplanner_output_processor import (
     ACTIONPLANNER_OUTPUT_PROCESSOR,
-)
-
-
+) 
 def get_id(args, kwargs):
     """
     Extracts the ID from the context.
     """
     scopes: dict[str, dict[str:str]] = {}
     context = kwargs.get("context")
-    if context and context.activity and context.activity.conversation.id:
-        conversation_id = context.activity.conversation.id or ""
-        user_aad_object_id = context.activity.from_property.aad_object_id or ""
-        user_teams_id = context.activity.from_property.id or ""
+    if context and context.activity and context.activity.channel_id:
         channel_id = context.activity.channel_id or ""
-        recipient_id = context.activity.recipient.id or ""
-        recipient_aad_object_id = context.activity.recipient.aad_object_id or ""
-        scopes[f"teams.conversation.conversation.id"] = conversation_id
-        scopes[f"teams.user.from_property.aad_object_id"] = user_aad_object_id
-        scopes[f"teams.user.from_property.id"] = user_teams_id
         scopes[f"teams.channel.channel_id"] = channel_id
-        scopes[f"teams.channel.recipient.id"] = recipient_id
-        scopes[f"teams.channel.recipient.aad_object_id"] = recipient_aad_object_id
+        if channel_id == "msteams":
+            scopes[f"msteams.activity.type"] = context.activity.type or ""
 
-    return scopes
+            if hasattr(context.activity,"conversation"):
+                scopes[f"msteams.conversation.id"] = context.activity.conversation.id or ""
+                scopes[f"msteams.conversation.type"] = context.activity.conversation.conversation_type or ""
+                scopes[f"msteams.conversation.name"] = context.activity.conversation.name or ""
 
+            if hasattr(context.activity,"from_property"):
+                scopes[f"msteams.user.from_property.id"] = context.activity.from_property.id or ""
+                scopes[f"msteams.user.from_property.name"] = context.activity.from_property.name or ""
+                scopes[f"msteams.user.from_property.role"] = context.activity.from_property.role or ""
+
+            if hasattr(context.activity,"recipient"):
+                scopes[f"msteams.recipient.id"] = context.activity.recipient.id or ""
+
+            if hasattr(context.activity,"channel_data"):
+                if "tenant" in context.activity.channel_data:
+                    scopes[f"msteams.channel_data.tenant.id"] = context.activity.channel_data['tenant']['id'] or ""
+                if "team" in context.activity.channel_data:
+                    scopes[f"msteams.channel_data.team.id"] = context.activity.channel_data['team']['id'] or ""
+                    if "name" in context.activity.channel_data['team']:
+                        scopes[f"msteams.channel_data.team.name"] = context.activity.channel_data['team']['name'] or ""
+                if "channel" in context.activity.channel_data:
+                    scopes[f"msteams.channel_data.channel.id"] = context.activity.channel_data['channel']['id'] or ""
+                    if "name" in context.activity.channel_data['channel']:
+                        scopes[f"msteams.channel_data.channel.name"] = context.activity.channel_data['channel']['name'] or ""
+    return scopes
 
 TEAMAI_METHODS = [
     {
@@ -55,6 +68,6 @@ TEAMAI_METHODS = [
         "object": "ActionPlanner",
         "method": "complete_prompt",
         "scope_values": get_id,
-        "wrapper_method": ascopes_wrapper,
-    },
+        "wrapper_method": ascopes_wrapper,  
+    }
 ]