Integrate MongoDB

great_ai/src/great_ai/great_ai/tracing/query_ground_truth.py

@@ -1,10 +1,8 @@
 from datetime import datetime
-from typing import List, Optional, TypeVar, Union
+from typing import List, Optional, Union
 
 from ..context import get_context
-from ..views.trace import Trace
-
-T = TypeVar("T")
+from ..views import Trace
 
 
 def query_ground_truth(
@@ -12,7 +10,7 @@ def query_ground_truth(
     *,
     since: Optional[datetime] = None,
     return_max_count: Optional[int] = None
-) -> List[Trace[T]]:
+) -> List[Trace]:
     tags = (
         conjunctive_tags if isinstance(conjunctive_tags, list) else [conjunctive_tags]
     )