33
44from autogen_core .base import AgentId , MessageContext
55from 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+ )
1013from autogen_core .components .tool_agent import tool_agent_caller_loop
1114from autogen_core .components .tools import Tool
1215
1316from src .backend .context .cosmos_memory import CosmosBufferedChatCompletionContext
14- from src .backend .models .messages import (ActionRequest , ActionResponse ,
15- AgentMessage , Step , StepStatus )
17+ from src .backend .models .messages import (
18+ ActionRequest ,
19+ ActionResponse ,
20+ AgentMessage ,
21+ Step ,
22+ StepStatus
23+ )
1624from azure .monitor .events .extension import track_event
1725
26+
1827class BaseAgent (RoutedAgent ):
1928 def __init__ (
2029 self ,
@@ -95,7 +104,7 @@ async def handle_action_request(
95104 step_id = message .step_id ,
96105 )
97106 )
98-
107+
99108 track_event (
100109 "Base agent - Added into the cosmos" ,
101110 {
@@ -107,7 +116,7 @@ async def handle_action_request(
107116 "step_id" : message .step_id ,
108117 },
109118 )
110-
119+
111120 except Exception as e :
112121 logging .exception (f"Error during LLM call: { e } " )
113122 track_event (
@@ -121,14 +130,14 @@ async def handle_action_request(
121130 "step_id" : message .step_id ,
122131 },
123132 )
124-
133+
125134 return
126135 print (f"Task completed: { result } " )
127136
128137 step .status = StepStatus .completed
129138 step .agent_reply = result
130139 await self ._model_context .update_step (step )
131-
140+
132141 track_event (
133142 "Base agent - Updated step and updated into the cosmos" ,
134143 {
0 commit comments