|
3 | 3 |
|
4 | 4 | from autogen_core.base import AgentId, MessageContext |
5 | 5 | from autogen_core.components import RoutedAgent, message_handler |
6 | | -from autogen_core.components.models import (AssistantMessage, |
7 | | - AzureOpenAIChatCompletionClient, |
8 | | - LLMMessage, SystemMessage, |
9 | | - UserMessage) |
| 6 | +from autogen_core.components.models import ( |
| 7 | + AssistantMessage, |
| 8 | + AzureOpenAIChatCompletionClient, |
| 9 | + LLMMessage, |
| 10 | + SystemMessage, |
| 11 | + UserMessage, |
| 12 | +) |
10 | 13 | from autogen_core.components.tool_agent import tool_agent_caller_loop |
11 | 14 | from autogen_core.components.tools import Tool |
12 | 15 |
|
13 | 16 | from context.cosmos_memory import CosmosBufferedChatCompletionContext |
14 | | -from models.messages import (ActionRequest, ActionResponse, |
15 | | - AgentMessage, Step, StepStatus) |
| 17 | +from models.messages import ( |
| 18 | + ActionRequest, |
| 19 | + ActionResponse, |
| 20 | + AgentMessage, |
| 21 | + Step, |
| 22 | + StepStatus, |
| 23 | +) |
| 24 | +from azure.monitor.events.extension import track_event |
| 25 | + |
16 | 26 |
|
17 | 27 | class BaseAgent(RoutedAgent): |
18 | 28 | def __init__( |
@@ -94,15 +104,54 @@ async def handle_action_request( |
94 | 104 | step_id=message.step_id, |
95 | 105 | ) |
96 | 106 | ) |
| 107 | + |
| 108 | + track_event( |
| 109 | + "Base agent - Added into the cosmos", |
| 110 | + { |
| 111 | + "session_id": message.session_id, |
| 112 | + "user_id": self._user_id, |
| 113 | + "plan_id": message.plan_id, |
| 114 | + "content": f"{result}", |
| 115 | + "source": self._agent_name, |
| 116 | + "step_id": message.step_id, |
| 117 | + }, |
| 118 | + ) |
| 119 | + |
97 | 120 | except Exception as e: |
98 | | - print(f"Error during LLM call: {e}") |
| 121 | + logging.exception(f"Error during LLM call: {e}") |
| 122 | + track_event( |
| 123 | + "Base agent - Error during llm call, captured into the cosmos", |
| 124 | + { |
| 125 | + "session_id": message.session_id, |
| 126 | + "user_id": self._user_id, |
| 127 | + "plan_id": message.plan_id, |
| 128 | + "content": f"{e}", |
| 129 | + "source": self._agent_name, |
| 130 | + "step_id": message.step_id, |
| 131 | + }, |
| 132 | + ) |
| 133 | + |
99 | 134 | return |
100 | 135 | print(f"Task completed: {result}") |
101 | 136 |
|
102 | 137 | step.status = StepStatus.completed |
103 | 138 | step.agent_reply = result |
104 | 139 | await self._model_context.update_step(step) |
105 | 140 |
|
| 141 | + track_event( |
| 142 | + "Base agent - Updated step and updated into the cosmos", |
| 143 | + { |
| 144 | + "status": StepStatus.completed, |
| 145 | + "session_id": message.session_id, |
| 146 | + "agent_reply": f"{result}", |
| 147 | + "user_id": self._user_id, |
| 148 | + "plan_id": message.plan_id, |
| 149 | + "content": f"{result}", |
| 150 | + "source": self._agent_name, |
| 151 | + "step_id": message.step_id, |
| 152 | + }, |
| 153 | + ) |
| 154 | + |
106 | 155 | action_response = ActionResponse( |
107 | 156 | step_id=step.id, |
108 | 157 | plan_id=step.plan_id, |
|
0 commit comments