Source code for langchain_community.docstore.in_memory
"""简单的内存文档存储,以字典的形式。"""
from typing import Dict, List, Optional, Union
from langchain_core.documents import Document
from langchain_community.docstore.base import AddableMixin, Docstore
[docs]class InMemoryDocstore(Docstore, AddableMixin):
"""简单的内存文档存储,以字典的形式。"""
[docs] def __init__(self, _dict: Optional[Dict[str, Document]] = None):
"""使用字典进行初始化。"""
self._dict = _dict if _dict is not None else {}
[docs] def add(self, texts: Dict[str, Document]) -> None:
"""将文本添加到内存字典中。
参数:
texts:id -> 文档的字典。
返回:
无。
"""
overlapping = set(texts).intersection(self._dict)
if overlapping:
raise ValueError(f"Tried to add ids that already exist: {overlapping}")
self._dict = {**self._dict, **texts}
[docs] def delete(self, ids: List) -> None:
"""从内存中的字典中删除ID。"""
overlapping = set(ids).intersection(self._dict)
if not overlapping:
raise ValueError(f"Tried to delete ids that does not exist: {ids}")
for _id in ids:
self._dict.pop(_id)
[docs] def search(self, search: str) -> Union[str, Document]:
"""通过直接查找进行搜索。
参数:
search:要搜索的文档的id。
返回:
如果找到文档,则返回文档,否则返回错误消息。
"""
if search not in self._dict:
return f"ID {search} not found."
else:
return self._dict[search]