|
2 | 2 | from abc import ABC |
3 | 3 | from collections.abc import AsyncGenerator, Awaitable |
4 | 4 | from dataclasses import dataclass, field |
5 | | -from typing import Any, Optional, TypedDict, Union, cast |
| 5 | +from typing import Any, Optional, TypedDict, cast |
6 | 6 |
|
7 | 7 | from azure.search.documents.agent.aio import KnowledgeAgentRetrievalClient |
8 | 8 | from azure.search.documents.agent.models import ( |
@@ -190,7 +190,7 @@ def build_filter(self, overrides: dict[str, Any]) -> Optional[str]: |
190 | 190 | filters.append("category eq '{}'".format(include_category.replace("'", "''"))) |
191 | 191 | if exclude_category: |
192 | 192 | filters.append("category ne '{}'".format(exclude_category.replace("'", "''"))) |
193 | | - return None if len(filters) == 0 else " and ".join(filters) |
| 193 | + return None if not filters else " and ".join(filters) |
194 | 194 |
|
195 | 195 | async def search( |
196 | 196 | self, |
@@ -520,7 +520,7 @@ def create_chat_completion( |
520 | 520 | temperature: Optional[float] = None, |
521 | 521 | n: Optional[int] = None, |
522 | 522 | reasoning_effort: Optional[ChatCompletionReasoningEffort] = None, |
523 | | - ) -> Union[Awaitable[ChatCompletion], Awaitable[AsyncStream[ChatCompletionChunk]]]: |
| 523 | + ) -> Awaitable[ChatCompletion] | Awaitable[AsyncStream[ChatCompletionChunk]]: |
524 | 524 | if chatgpt_model in self.GPT_REASONING_MODELS: |
525 | 525 | params: dict[str, Any] = { |
526 | 526 | # max_tokens is not supported |
|
0 commit comments