Skip to content

Commit 23a75e9

Browse files
committed
refactor(models): use Python 3.11+ stdlib features
- Import TypeAlias from typing instead of typing_extensions - Use datetime.UTC instead of datetime.timezone.utc - Remove obsolete TODO comments about Python 3.9 support
1 parent be490de commit 23a75e9

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

stackone_ai/models.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,22 +1,18 @@
1-
# TODO: Remove when Python 3.9 support is dropped
21
from __future__ import annotations
32

43
import base64
54
import json
65
import logging
76
from collections.abc import Sequence
8-
from datetime import datetime, timezone
7+
from datetime import UTC, datetime
98
from enum import Enum
10-
from typing import Annotated, Any, ClassVar, cast
9+
from typing import Annotated, Any, ClassVar, TypeAlias, cast
1110
from urllib.parse import quote
1211

1312
import httpx
1413
from langchain_core.tools import BaseTool
1514
from pydantic import BaseModel, BeforeValidator, Field, PrivateAttr
1615

17-
# TODO: Remove when Python 3.9 support is dropped
18-
from typing_extensions import TypeAlias
19-
2016
# Type aliases for common types
2117
JsonDict: TypeAlias = dict[str, Any]
2218
Headers: TypeAlias = dict[str, str]
@@ -200,7 +196,7 @@ def execute(
200196
StackOneAPIError: If the API request fails
201197
ValueError: If the arguments are invalid
202198
"""
203-
datetime.now(timezone.utc)
199+
datetime.now(UTC)
204200
feedback_options: JsonDict = {}
205201
result_payload: JsonDict | None = None
206202
response_status: int | None = None
@@ -270,7 +266,7 @@ def execute(
270266
status = "error"
271267
raise StackOneError(f"Request failed: {exc}") from exc
272268
finally:
273-
datetime.now(timezone.utc)
269+
datetime.now(UTC)
274270
metadata: JsonDict = {
275271
"http_method": self._execute_config.method,
276272
"url": url_used,

0 commit comments

Comments
 (0)