great-ai/great_ai/tracing/query_ground_truth.py
2022-07-04 19:31:15 +02:00

22 lines
599 B
Python

from datetime import datetime
from typing import List, Optional, Union
from ..context import get_context
from ..views import Trace
def query_ground_truth(
conjunctive_tags: Union[List[str], str] = [],
*,
since: Optional[datetime] = None,
return_max_count: Optional[int] = None
) -> List[Trace]:
tags = (
conjunctive_tags if isinstance(conjunctive_tags, list) else [conjunctive_tags]
)
db = get_context().tracing_database
items, length = db.query(
conjunctive_tags=tags, since=since, take=return_max_count, has_feedback=True
)
return items