fix build

Author: lusu-msft

sdk/agentserver/azure-ai-agentserver-agentframework/azure/ai/agentserver/agentframework/models/agent_framework_output_streaming_converter.py

@@ -7,7 +7,7 @@
 
 import datetime
 import json
-from typing import Any, AsyncIterable, List, Optional
+from typing import Any, AsyncIterable, List
 
 from agent_framework import AgentRunResponseUpdate, BaseContent, FunctionApprovalRequestContent, FunctionResultContent
 from agent_framework._types import (
@@ -48,7 +48,7 @@
 class _BaseStreamingState:
     """Base interface for streaming state handlers."""
 
-    def convert_contents(self, contents: AsyncIterable[BaseContent]) -> AsyncIterable[ResponseStreamEvent]:  # pylint: disable=unused-argument
+    async def convert_contents(self, contents: AsyncIterable[BaseContent]) -> AsyncIterable[ResponseStreamEvent]:  # pylint: disable=unused-argument
         raise NotImplementedError
 
 
@@ -126,7 +126,9 @@ class _FunctionCallStreamingState(_BaseStreamingState):
     def __init__(self, parent: AgentFrameworkOutputStreamingConverter):
         self._parent = parent
 
-    async def convert_contents(self, contents: AsyncIterable[FunctionCallContent]) -> AsyncIterable[ResponseStreamEvent]:
+    async def convert_contents(
+            self, contents: AsyncIterable[FunctionCallContent]
+        ) -> AsyncIterable[ResponseStreamEvent]:
         content_by_call_id = {}
         ids_by_call_id = {}
 
@@ -149,18 +151,17 @@ async def convert_contents(self, contents: AsyncIterable[FunctionCallContent]) -
                         arguments="",
                     ),
                 )
-                continue
             else:
                 content_by_call_id[content.call_id] = content_by_call_id[content.call_id] + content
                 item_id, output_index = ids_by_call_id[content.call_id]
 
-            args_delta = content.arguments if isinstance(content.arguments, str) else ""
-            yield ResponseFunctionCallArgumentsDeltaEvent(
-                sequence_number=self._parent.next_sequence(),
-                item_id=item_id,
-                output_index=output_index,
-                delta=args_delta,
-            )
+                args_delta = content.arguments if isinstance(content.arguments, str) else ""
+                yield ResponseFunctionCallArgumentsDeltaEvent(
+                    sequence_number=self._parent.next_sequence(),
+                    item_id=item_id,
+                    output_index=output_index,
+                    delta=args_delta,
+                )
 
         for call_id, content in content_by_call_id.items():
             item_id, output_index = ids_by_call_id[call_id]
@@ -194,7 +195,9 @@ class _FunctionCallOutputStreamingState(_BaseStreamingState):
     def __init__(self, parent: AgentFrameworkOutputStreamingConverter):
         self._parent = parent
 
-    async def convert_contents(self, contents: AsyncIterable[FunctionResultContent]) -> AsyncIterable[ResponseStreamEvent]:
+    async def convert_contents(
+            self, contents: AsyncIterable[FunctionResultContent]
+        ) -> AsyncIterable[ResponseStreamEvent]:
         async for content in contents:
             item_id = self._parent.context.id_generator.generate_function_output_id()
             output_index = self._parent.next_output_index()
@@ -281,7 +284,7 @@ async def convert(self, updates: AsyncIterable[AgentRunResponseUpdate]) -> Async
         )
 
         is_changed = (
-            lambda a, b: a is not None and b is not None and a.message_id != b.message_id
+            lambda a, b: a is not None and b is not None and a.message_id != b.message_id  # pylint: disable=unnecessary-lambda-assignment
         )
         async for group in chunk_on_change(updates, is_changed):
             has_value, first, contents = await peek(self._read_updates(group))

sdk/agentserver/azure-ai-agentserver-agentframework/azure/ai/agentserver/agentframework/models/utils/async_iter.py

@@ -4,7 +4,7 @@
 from __future__ import annotations
 
 from collections.abc import AsyncIterable, AsyncIterator, Callable
-from typing import TypeVar, Optional, Tuple, Awaitable
+from typing import TypeVar, Optional, Tuple
 
 TSource = TypeVar("TSource")
 TKey = TypeVar("TKey")
@@ -19,9 +19,12 @@ async def chunk_on_change(
     Chunks an async iterable into groups based on when consecutive elements change.
 
     :param source: Async iterable of items.
+    :type source: AsyncIterable[TSource]
     :param is_changed: Function(prev, current) -> bool indicating if value changed.
                        If None, uses != by default.
+    :type is_changed: Optional[Callable[[Optional[TSource], Optional[TSource]], bool]]
     :return: An async iterator of async iterables (chunks).
+    :rtype: AsyncIterator[AsyncIterable[TSource]]
     """
 
     if is_changed is None:
@@ -46,9 +49,13 @@ async def chunk_by_key(
     Chunks the async iterable into groups based on a key selector.
 
     :param source: Async iterable of items.
+    :type source: AsyncIterable[TSource]
     :param key_selector: Function mapping item -> key.
+    :type key_selector: Callable[[TSource], TKey]
     :param key_equal: Optional equality function for keys. Defaults to '=='.
+    :type key_equal: Optional[Callable[[TKey, TKey], bool]]
     :return: An async iterator of async iterables (chunks).
+    :rtype: AsyncIterator[AsyncIterable[TSource]]
     """
 
     if key_equal is None:
@@ -104,7 +111,9 @@ async def peek(
     Peeks at the first element of an async iterable without consuming it.
 
     :param source: Async iterable.
+    :type source: AsyncIterable[T]
     :return: (has_value, first, full_sequence_including_first)
+    :rtype: Tuple[bool, Optional[T], AsyncIterable[T]]
     """
 
     it = source.__aiter__()
@@ -131,6 +140,6 @@ async def sequence() -> AsyncIterator[T]:
 
 
 async def _empty_async() -> AsyncIterator[T]:
-    if False:
+    if False:  # pylint: disable=using-constant-test
         # This is just to make this an async generator for typing
         yield None  # type: ignore[misc]

sdk/agentserver/azure-ai-agentserver-core/azure/ai/agentserver/core/client/tools/_client.py

@@ -1,7 +1,7 @@
 # ---------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
-
+# pylint: disable=protected-access
 from typing import Any, List, Mapping, Union
 from azure.core import PipelineClient
 from azure.core.pipeline import policies

sdk/agentserver/azure-ai-agentserver-core/azure/ai/agentserver/core/client/tools/aio/_client.py

@@ -1,7 +1,7 @@
 # ---------------------------------------------------------
 # Copyright (c) Microsoft Corporation. All rights reserved.
 # ---------------------------------------------------------
-
+# pylint: disable=protected-access,do-not-import-asyncio
 from typing import Any, List, Mapping, Union, TYPE_CHECKING
 from asyncio import gather
 from azure.core import AsyncPipelineClient

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

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