[BUG]: Use stdlib override when possible

chromadb/api/__init__.py

@@ -1,8 +1,12 @@
 from abc import ABC, abstractmethod
 from typing import Sequence, Optional, List
 from uuid import UUID
+import sys
 
-from overrides import override
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 from chromadb.api.collection_configuration import (
     CreateCollectionConfiguration,
     UpdateCollectionConfiguration,

chromadb/api/async_api.py

@@ -1,8 +1,12 @@
 from abc import ABC, abstractmethod
 from typing import Sequence, Optional, List
 from uuid import UUID
+import sys
 
-from overrides import override
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 from chromadb.api.collection_configuration import (
     CreateCollectionConfiguration,
     UpdateCollectionConfiguration,

chromadb/api/async_client.py

@@ -1,7 +1,12 @@
 import httpx
 from typing import Optional, Sequence
 from uuid import UUID
-from overrides import override
+import sys
+
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 
 from chromadb.auth import UserIdentity
 from chromadb.auth.utils import maybe_set_tenant_and_database

chromadb/api/async_fastapi.py

@@ -5,7 +5,12 @@
 from typing import Any, Optional, cast, Tuple, Sequence, Dict, List
 import logging
 import httpx
-from overrides import override
+import sys
+
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 from chromadb import __version__
 from chromadb.auth import UserIdentity
 from chromadb.api.async_api import AsyncServerAPI

chromadb/api/client.py

@@ -1,7 +1,11 @@
 from typing import Optional, Sequence
 from uuid import UUID
+import sys
 
-from overrides import override
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 import httpx
 from chromadb.api import AdminAPI, ClientAPI, ServerAPI
 from chromadb.api.collection_configuration import (

chromadb/api/configuration.py

@@ -1,6 +1,6 @@
 from abc import abstractmethod
 import json
-from overrides import override
+import sys
 from typing import (
     Any,
     ClassVar,
@@ -15,6 +15,11 @@
 from typing_extensions import Self
 from multiprocessing import cpu_count
 
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
+
 from chromadb.serde import JSONSerializable
 
 # TODO: move out of API

chromadb/api/fastapi.py

@@ -4,8 +4,13 @@
 from typing import Sequence
 from uuid import UUID
 import httpx
+import sys
 import urllib.parse
-from overrides import override
+
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 
 from chromadb.api.collection_configuration import (
     CreateCollectionConfiguration,

chromadb/api/rust.py

@@ -44,10 +44,15 @@
 
 
 from typing import Optional, Sequence, List
-from overrides import override
 from uuid import UUID
 import json
 import platform
+import sys
+
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 
 if platform.system() != "Windows":
     import resource

chromadb/api/segment.py

@@ -69,14 +69,19 @@
     Callable,
     TypeVar,
 )
-from overrides import override
 from uuid import UUID, uuid4
 from functools import wraps
 import time
 import logging
 import re
+import sys
 from chromadb.execution.expression.plan import Search
 
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
+
 T = TypeVar("T", bound=Callable[..., Any])
 
 logger = logging.getLogger(__name__)

chromadb/auth/basic_authn/__init__.py

@@ -6,8 +6,12 @@
 
 import bcrypt
 import logging
+import sys
 
-from overrides import override
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 from pydantic import SecretStr
 
 from chromadb.auth import (

chromadb/auth/simple_rbac_authz/__init__.py

@@ -1,7 +1,12 @@
 import logging
 from typing import Dict, Set
-from overrides import override
+import sys
 import yaml
+
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 from chromadb.auth import (
     AuthzAction,
     AuthzResource,

chromadb/auth/token_authn/__init__.py

@@ -2,13 +2,16 @@
 import random
 import re
 import string
+import sys
 import time
 import traceback
 from enum import Enum
 from typing import cast, Dict, List, Optional, TypedDict, TypeVar
 
-
-from overrides import override
+if sys.version_info >= (3, 12):
+    from typing import override
+else:
+    from overrides import overrides as override
 from pydantic import SecretStr
 import yaml
 

chromadb/config.py

@@ -1,14 +1,21 @@
 import importlib
 import inspect
 import logging
+import sys
 from abc import ABC
 from enum import Enum
 from graphlib import TopologicalSorter
 from typing import Optional, List, Any, Dict, Set, Iterable, Union
 from typing import Type, TypeVar, cast
 
-from overrides import EnforceOverrides
-from overrides import override
+if sys.version_info >= (3, 12):
+    from typing import override
+
+    class EnforceOverrides:
+        pass
+else:
+    from overrides import overrides as override
+    from overrides import EnforceOverrides
 from typing_extensions import Literal
 import platform
 

chromadb/db/base.py

@@ -3,7 +3,16 @@
 from typing_extensions import Protocol, Self, Literal
 from abc import ABC, abstractmethod
 from threading import local
-from overrides import override, EnforceOverrides
+import sys
+
+if sys.version_info >= (3, 12):
+    from typing import override
+
+    class EnforceOverrides:
+        pass
+else:
+    from overrides import overrides as override
+    from overrides import EnforceOverrides
 import pypika
 import pypika.queries
 from chromadb.config import System, Component