phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

precommit

Browse source
This commit is contained in:
Xi Yan 2025-02-19 22:35:24 -08:00
parent ca687d3e86
commit ce040ad111
2 changed files with 16 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

5
llama_stack/providers/utils/kvstore/mongodb/mongodb.py
View file

@ -41,10 +41,7 @@ class MongoDBKVStoreImpl(KVStore):
return key
return f"{self.config.namespace}:{key}"
async def set(
self, key: str, value: str, expiration: Optional[datetime] = None
) -> None:
async def set(self, key: str, value: str, expiration: Optional[datetime] = None) -> None:
key = self._namespaced_key(key)
update_query = {"$set": {"value": value, "expiration": expiration}}
self.collection.update_one({"key": key}, update_query, upsert=True)