[BUG] UUID should be uuid type not str

chromadb/auth/fastapi_utils.py

@@ -1,5 +1,6 @@
 from functools import partial
 from typing import Any, Callable, Dict, Optional, Sequence, cast
+from chromadb.server.fastapi.utils import string_to_uuid
 from chromadb.api import ServerAPI
 from chromadb.auth import AuthzResourceTypes
 
@@ -46,7 +47,7 @@ def attr_from_collection_lookup(
     def _wrap(**kwargs: Any) -> Dict[str, Any]:
         _api = cast(ServerAPI, kwargs["api"])
         col = _api.get_collection(
-            id=kwargs["function_kwargs"][collection_id_arg])
+            id=string_to_uuid(kwargs["function_kwargs"][collection_id_arg]))
         return {"tenant": col.tenant, "database": col.database}
 
     return partial(_wrap, **kwargs)

chromadb/server/fastapi/__init__.py

@@ -32,7 +32,6 @@
 from chromadb.api import ServerAPI
 from chromadb.errors import (
     ChromaError,
-    InvalidUUIDError,
     InvalidDimensionException,
     InvalidHTTPVersion,
 )
@@ -51,7 +50,7 @@
 
 import logging
 
-from chromadb.server.fastapi.utils import fastapi_json_response
+from chromadb.server.fastapi.utils import fastapi_json_response, string_to_uuid as _uuid
 from chromadb.telemetry.opentelemetry.fastapi import instrument_fastapi
 from chromadb.types import Database, Tenant
 from chromadb.telemetry.product import ServerContext, ProductTelemetryClient
@@ -96,13 +95,6 @@ async def check_http_version_middleware(
     return await call_next(request)
 
 
-def _uuid(uuid_str: str) -> UUID:
-    try:
-        return UUID(uuid_str)
-    except ValueError:
-        raise InvalidUUIDError(f"Could not parse {uuid_str} as a UUID")
-
-
 class ChromaAPIRouter(fastapi.APIRouter):  # type: ignore
     # A simple subclass of fastapi's APIRouter which treats URLs with a trailing "/" the
     # same as URLs without. Docs will only contain URLs without trailing "/"s.

chromadb/server/fastapi/utils.py

@@ -1,10 +1,17 @@
+from uuid import UUID
 from starlette.responses import JSONResponse
 
-from chromadb.errors import ChromaError
+from chromadb.errors import ChromaError, InvalidUUIDError
 
 
 def fastapi_json_response(error: ChromaError) -> JSONResponse:
     return JSONResponse(
         content={"error": error.name(), "message": error.message()},
         status_code=error.code(),
     )
+
+def string_to_uuid(uuid_str: str) -> UUID:
+    try:
+        return UUID(uuid_str)
+    except ValueError:
+        raise InvalidUUIDError(f"Could not parse {uuid_str} as a UUID")