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

update

Browse Source
This commit is contained in:
Eugene Yurtsev
2024-07-18 14:08:34 -04:00
parent 9b7126f962
commit ad8a4ee074
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -23,9 +23,9 @@ class InMemoryDocumentIndexer(DocumentIndexer):
for item in items:
if item.id is None:
id_ = uuid.uuid4()
id_ = str(uuid.uuid4())
item_ = item.copy()
item_.id = str(id_)
item_.id = id_
else:
item_ = item
id_ = item.id

4
libs/core/tests/unit_tests/indexing/test_in_memory_indexer.py
View File

@@ -1,6 +1,6 @@
"""Test in memory indexer"""
from typing import Generator
from typing import Generator, AsyncGenerator
import pytest
from langchain_standard_tests.integration_tests.indexer import (
@@ -23,5 +23,5 @@ class TestDocumentIndexerTestSuite(DocumentIndexerTestSuite):
class TestAsyncDocumentIndexerTestSuite(AsyncDocumentIndexerTestSuite):
@pytest.fixture()
async def indexer(self) -> Generator[AsyncDocumentIndexer, None, None]:
async def indexer(self) -> AsyncGenerator[AsyncDocumentIndexer, None]:
yield AsyncInMemoryDocumentIndexer()