diff --git a/backend/apps/ollama/main.py b/backend/apps/ollama/main.py index 25a98686c..03a8e198e 100644 --- a/backend/apps/ollama/main.py +++ b/backend/apps/ollama/main.py @@ -21,7 +21,7 @@ import asyncio import logging import time from urllib.parse import urlparse -from typing import Optional, List, Union +from typing import Optional, Union from starlette.background import BackgroundTask diff --git a/backend/utils/misc.py b/backend/utils/misc.py index ade8d93b5..2eed58f41 100644 --- a/backend/utils/misc.py +++ b/backend/utils/misc.py @@ -2,7 +2,7 @@ from pathlib import Path import hashlib import re from datetime import timedelta -from typing import Optional, List, Tuple, Callable +from typing import Optional, Callable import uuid import time @@ -52,7 +52,7 @@ def remove_system_message(messages: list[dict]) -> list[dict]: return [message for message in messages if message["role"] != "system"] -def pop_system_message(messages: list[dict]) -> Tuple[Optional[dict], list[dict]]: +def pop_system_message(messages: list[dict]) -> tuple[Optional[dict], list[dict]]: return get_system_message(messages), remove_system_message(messages) diff --git a/backend/utils/tools.py b/backend/utils/tools.py index 76d616c59..eac36b5d9 100644 --- a/backend/utils/tools.py +++ b/backend/utils/tools.py @@ -1,5 +1,5 @@ import inspect -from typing import get_type_hints, List, Dict, Any +from typing import get_type_hints def doc_to_dict(docstring):