fix mypy

Author: lusu-msft

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/langgraph.py

@@ -2,6 +2,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 # pylint: disable=logging-fstring-interpolation,broad-exception-caught
+# mypy: disable-error-code="assignment,arg-type"
 import os
 import re
 from typing import Optional

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/models/langgraph_response_converter.py

@@ -2,7 +2,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 # pylint: disable=logging-fstring-interpolation,broad-exception-caught,logging-not-lazy
-# mypy: ignore-errors
+# mypy: disable-error-code="valid-type,call-overload,attr-defined"
 import copy
 from typing import List
 
@@ -23,7 +23,7 @@ def __init__(self, context: AgentRunContext, output):
         self.context = context
         self.output = output
 
-    def convert(self) -> project_models.ItemResource:
+    def convert(self) -> list[project_models.ItemResource]:
         res = []
         for step in self.output:
             for node_name, node_output in step.items():

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/models/langgraph_state_converter.py

@@ -1,7 +1,7 @@
 # ---------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
-
+# mypy: disable-error-code="call-overload,override"
 """Base interface for converting between LangGraph internal state and OpenAI-style responses.
 
 A LanggraphStateConverter implementation bridges:

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/models/langgraph_stream_response_converter.py

@@ -2,6 +2,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 # pylint: disable=logging-fstring-interpolation
+# mypy: disable-error-code="assignment,valid-type"
 from typing import List
 
 from langchain_core.messages import AnyMessage

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/models/response_event_generators/item_content_helpers.py

@@ -10,8 +10,8 @@ def __init__(self, content_type: str):
         self.has_aggregated_content = False
 
     def create_item_content(self) -> project_models.ItemContent:
-        return project_models.ItemContent(   # mypy: ignore[call-overload]
-            content_type=self.content_type,
+        return project_models.ItemContent(
+            type=self.content_type,
         )
 
 

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/models/response_event_generators/item_resource_helpers.py

@@ -92,7 +92,7 @@ class MessageItemResourceHelper(ItemResourceHelper):
     def __init__(self, item_id: str, role: project_models.ResponsesMessageRole):
         super().__init__(project_models.ItemType.MESSAGE, item_id)
         self.role = role
-        self.content = []   # mypy: ignore[var-annotated]
+        self.content: list[project_models.ItemContent] = []
 
     def create_item_resource(self, is_done: bool):
         content = {

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/models/response_event_generators/response_event_generator.py

@@ -2,6 +2,7 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 # pylint: disable=unused-argument,unnecessary-pass
+# mypy: disable-error-code="valid-type"
 from typing import List
 
 from langchain_core.messages import AnyMessage
@@ -52,24 +53,26 @@ def try_process_message(
         """
         pass
 
-    def on_start(self) -> tuple[bool, List[project_models.ResponseStreamEvent]]:   # mypy: ignore[empty-body]
+    def on_start(self) -> tuple[bool, List[project_models.ResponseStreamEvent]]:
         """
         Generate the starting events for this layer.
 
         :return: tuple of (started, events)
         :rtype: tuple[bool, List[ResponseStreamEvent]]
         """
-        pass
+        return False, []
 
-    def on_end(self) -> tuple[bool, List[project_models.ResponseStreamEvent]]:   # mypy: ignore[empty-body]
+    def on_end(
+            self, message: AnyMessage, context: AgentRunContext, stream_state: StreamEventState
+        ) -> tuple[bool, List[project_models.ResponseStreamEvent]]:
         """
         Generate the ending events for this layer.
         TODO: handle different end conditions, e.g. normal end, error end, etc.
 
         :return: tuple of (started, events)
         :rtype: tuple[bool, List[ResponseStreamEvent]]
         """
-        pass
+        return False, []
 
     def aggregate_content(self):
         """

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/models/response_event_generators/response_function_call_argument_event_generator.py

@@ -39,7 +39,7 @@ def try_process_message(
 
         if self.should_end(message):
             has_finish_reason = self.has_finish_reason(message)
-            complete_events = self.on_end(message, stream_state)
+            complete_events = self.on_end(message, context, stream_state)
             events.extend(complete_events)
             next_processor = self.parent
             is_processed = has_finish_reason  # if has finish reason, mark as processed and stop further processing
@@ -94,7 +94,7 @@ def should_end(self, event: AnyMessage) -> bool:
         return False
 
     def on_end(
-        self, message: AnyMessage, stream_state: StreamEventState
+        self, message: AnyMessage, context: AgentRunContext, stream_state: StreamEventState
     ) -> tuple[bool, List[project_models.ResponseStreamEvent]]:
         done_event = project_models.ResponseFunctionCallArgumentsDoneEvent(
             item_id=self.item_id,

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/models/response_event_generators/response_output_text_event_generator.py

@@ -2,9 +2,11 @@
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
 # pylint: disable=unused-argument
+# mypy: disable-error-code="return-value,assignment"
 from typing import List
 
 from azure.ai.agentserver.core.models import projects as project_models
+from azure.ai.agentserver.core.server.common.agent_run_context import AgentRunContext
 
 from .response_event_generator import (
     ResponseEventGenerator,
@@ -91,7 +93,7 @@ def should_end(self, message) -> bool:
         return False
 
     def on_end(   # mypy: ignore[override]
-        self, message, run_details, stream_state: StreamEventState
+        self, message, context: AgentRunContext, stream_state: StreamEventState
     ) -> tuple[bool, List[project_models.ResponseStreamEvent]]:
         if not self.started:
             return False, []

sdk/ai/azure-ai-agentserver-langgraph/azure/ai/agentserver/langgraph/models/response_event_generators/response_stream_event_generator.py

@@ -93,7 +93,7 @@ def try_process_message(
 
         if self.should_end(message):
             # received a None message, indicating end of the stream
-            done_events = self.on_end(context, stream_state)
+            done_events = self.on_end(message, context, stream_state)
             events.extend(done_events)
             is_processed = True
             next_processor = None
@@ -106,7 +106,7 @@ def should_end(self, event: langgraph_messages.AnyMessage) -> bool:
             return True
         return False
 
-    def on_end(self, context: AgentRunContext, stream_state: StreamEventState):
+    def on_end(self, message: langgraph_messages.AnyMessage, context: AgentRunContext, stream_state: StreamEventState):
         agent_id = context.get_agent_id_object()
         conversation = context.get_conversation_object()
         response_dict = {