github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2026-01-24 05:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

xt

Browse Source
This commit is contained in:
Eugene Yurtsev
2024-07-18 12:09:43 -04:00
parent 81db15ffa0
commit 9b7126f962
1 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
libs/core/langchain_core/indexing/in_memory.py
View File

@@ -3,7 +3,11 @@ from typing import Any, Dict, List, Optional, Sequence
from langchain_core.documents import Document
from langchain_core.indexing import UpsertResponse
from langchain_core.indexing.base import DeleteResponse, DocumentIndexer
from langchain_core.indexing.base import (
AsyncDocumentIndexer,
DeleteResponse,
DocumentIndexer,
)
class InMemoryDocumentIndexer(DocumentIndexer):
@@ -24,8 +28,9 @@ class InMemoryDocumentIndexer(DocumentIndexer):
item_.id = str(id_)
else:
item_ = item
id_ = item.id
self.store[item_.id] = item_
self.store[id_] = item_
ok_ids.append(item_.id)
return UpsertResponse(succeeded=ok_ids, failed=[])
@@ -57,7 +62,7 @@ class InMemoryDocumentIndexer(DocumentIndexer):
return found_documents
class AsyncInMemoryDocumentIndexer(DocumentIndexer):
class AsyncInMemoryDocumentIndexer(AsyncDocumentIndexer):
"""An in memory async indexer implementation."""
def __init__(self, *, store: Optional[Dict[str, Document]] = None) -> None: