github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-09 15:03:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

mongodb[patch]: Make ObjectId JSON-serializable on generation (#21394)

Browse Source
This commit is contained in:
Jib
2024-05-14 14:52:29 -04:00
committed by GitHub
parent 12b599c47f
commit a97473c846
3 changed files with 24 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
libs/partners/mongodb/tests/utils.py
View File

@@ -1,9 +1,9 @@
from __future__ import annotations
import uuid
from copy import deepcopy
from typing import Any, Dict, List, Mapping, Optional, cast
from bson import ObjectId
from langchain_core.callbacks.manager import (
AsyncCallbackManagerForLLMRun,
CallbackManagerForLLMRun,
@@ -162,7 +162,7 @@ class MockCollection(Collection):
def insert_many(self, to_insert: List[Any], *args, **kwargs) -> InsertManyResult: # type: ignore
mongodb_inserts = [
{"_id": str(uuid.uuid4()), "score": 1, **insert} for insert in to_insert
{"_id": ObjectId(), "score": 1, **insert} for insert in to_insert
]
self._data.extend(mongodb_inserts)
return self._insert_result or InsertManyResult(