From 033adfb8b3ce258317f4c8c436cf64a53fdd3f0c Mon Sep 17 00:00:00 2001 From: Eugene Yurtsev Date: Thu, 18 Jul 2024 17:07:11 -0400 Subject: [PATCH] x --- libs/core/langchain_core/indexing/in_memory.py | 4 ++-- .../langchain_standard_tests/integration_tests/indexer.py | 5 +++-- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/libs/core/langchain_core/indexing/in_memory.py b/libs/core/langchain_core/indexing/in_memory.py index 0279044e5c9..63410c7a64f 100644 --- a/libs/core/langchain_core/indexing/in_memory.py +++ b/libs/core/langchain_core/indexing/in_memory.py @@ -1,5 +1,5 @@ import uuid -from typing import Any, Dict, List, Optional, Sequence +from typing import Any, Dict, List, Optional, Sequence, cast from langchain_core.documents import Document from langchain_core.indexing import UpsertResponse @@ -31,7 +31,7 @@ class InMemoryDocumentIndexer(DocumentIndexer): id_ = item.id self.store[id_] = item_ - ok_ids.append(item_.id) + ok_ids.append(cast(str, item_.id)) return UpsertResponse(succeeded=ok_ids, failed=[]) diff --git a/libs/standard-tests/langchain_standard_tests/integration_tests/indexer.py b/libs/standard-tests/langchain_standard_tests/integration_tests/indexer.py index 7fed3f69c6b..d260d4c8ba7 100644 --- a/libs/standard-tests/langchain_standard_tests/integration_tests/indexer.py +++ b/libs/standard-tests/langchain_standard_tests/integration_tests/indexer.py @@ -6,8 +6,9 @@ from abc import ABC, abstractmethod from typing import AsyncGenerator, Generator import pytest + from langchain_core.documents import Document -from langchain_core.indexing import DocumentIndexer +from langchain_core.indexing import DocumentIndexer, AsyncDocumentIndexer class DocumentIndexerTestSuite(ABC): @@ -211,7 +212,7 @@ class AsyncDocumentIndexerTestSuite(ABC): @abstractmethod @pytest.fixture - async def indexer(self) -> AsyncGenerator[DocumentIndexer, None]: + async def indexer(self) -> AsyncGenerator[AsyncDocumentIndexer, None]: """Get the indexer.""" async def test_upsert_documents_has_no_ids(self, indexer: DocumentIndexer) -> None: