andras/great-ai
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases 4 Packages Wiki Activity Actions

Production improvements

Browse source
This commit is contained in:
Andras Schmelczer 2022-07-09 12:34:21 +02:00
parent 7bd2e89070
commit 378047e622
No known key found for this signature in database
GPG key ID: 0EA1BC97D0AB076E
5 changed files with 21 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

11
great_ai/persistence/mongodb_driver.py
View file

@ -1,7 +1,7 @@
from datetime import datetime
from typing import Any, Dict, List, Optional, Sequence, Tuple
from pymongo import MongoClient
from pymongo import ASCENDING, DESCENDING, MongoClient
from ..utilities import chunk
from ..views import Filter, SortBy, Trace
@ -31,6 +31,11 @@ class MongodbDriver(TracingDatabaseDriver):
"Please configure the MongoDB access options by calling MongodbDriver.configure_credentials"
)
with MongoClient[Any](self.mongo_connection_string) as client:
client[self.mongo_database].traces.create_index(
[("tags", ASCENDING), ("created", DESCENDING)], background=True
)
@classmethod
def configure_credentials( # type: ignore
cls,
@ -93,11 +98,13 @@ class MongodbDriver(TracingDatabaseDriver):
"filter": {},
}
and_query: List[Dict[str, Any]] = [{}]
and_query: List[Dict[str, Any]] = []
and_query.extend({"tags": tag} for tag in conjunctive_tags)
and_query.extend(
{f.property: {self._get_operator(f): f.value}} for f in conjunctive_filters
)
if not and_query:
and_query.append({})
if since:
and_query.append({"created": {"$gte": since}})