mirror of
https://github.com/hwchase17/langchain.git
synced 2025-08-05 11:12:47 +00:00
docs: vectorstore standard tests (#28241)
This commit is contained in:
parent
2acc83f146
commit
8738973267
@ -380,6 +380,62 @@
|
|||||||
" return {\"a\": 2, \"b\": 3}"
|
" return {\"a\": 2, \"b\": 3}"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
|
{
|
||||||
|
"cell_type": "markdown",
|
||||||
|
"metadata": {},
|
||||||
|
"source": [
|
||||||
|
"</details>\n",
|
||||||
|
"<details>\n",
|
||||||
|
" <summary>Vector Stores</summary>"
|
||||||
|
]
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"cell_type": "code",
|
||||||
|
"execution_count": null,
|
||||||
|
"metadata": {},
|
||||||
|
"outputs": [],
|
||||||
|
"source": [
|
||||||
|
"# title=\"tests/integration_tests/test_vectorstores_sync.py\"\n",
|
||||||
|
"\n",
|
||||||
|
"from typing import AsyncGenerator, Generator\n",
|
||||||
|
"\n",
|
||||||
|
"import pytest\n",
|
||||||
|
"from langchain_core.vectorstores import VectorStore\n",
|
||||||
|
"from langchain_parrot_link.vectorstores import ParrotVectorStore\n",
|
||||||
|
"from langchain_standard_tests.integration_tests.vectorstores import (\n",
|
||||||
|
" AsyncReadWriteTestSuite,\n",
|
||||||
|
" ReadWriteTestSuite,\n",
|
||||||
|
")\n",
|
||||||
|
"\n",
|
||||||
|
"\n",
|
||||||
|
"class TestSync(ReadWriteTestSuite):\n",
|
||||||
|
" @pytest.fixture()\n",
|
||||||
|
" def vectorstore(self) -> Generator[VectorStore, None, None]: # type: ignore\n",
|
||||||
|
" \"\"\"Get an empty vectorstore for unit tests.\"\"\"\n",
|
||||||
|
" store = ParrotVectorStore()\n",
|
||||||
|
" # note: store should be EMPTY at this point\n",
|
||||||
|
" # if you need to delete data, you may do so here\n",
|
||||||
|
" try:\n",
|
||||||
|
" yield store\n",
|
||||||
|
" finally:\n",
|
||||||
|
" # cleanup operations, or deleting data\n",
|
||||||
|
" pass\n",
|
||||||
|
"\n",
|
||||||
|
"\n",
|
||||||
|
"class TestAsync(AsyncReadWriteTestSuite):\n",
|
||||||
|
" @pytest.fixture()\n",
|
||||||
|
" async def vectorstore(self) -> AsyncGenerator[VectorStore, None]: # type: ignore\n",
|
||||||
|
" \"\"\"Get an empty vectorstore for unit tests.\"\"\"\n",
|
||||||
|
" store = ParrotVectorStore()\n",
|
||||||
|
" # note: store should be EMPTY at this point\n",
|
||||||
|
" # if you need to delete data, you may do so here\n",
|
||||||
|
" try:\n",
|
||||||
|
" yield store\n",
|
||||||
|
" finally:\n",
|
||||||
|
" # cleanup operations, or deleting data\n",
|
||||||
|
" pass"
|
||||||
|
]
|
||||||
|
},
|
||||||
{
|
{
|
||||||
"cell_type": "markdown",
|
"cell_type": "markdown",
|
||||||
"metadata": {},
|
"metadata": {},
|
||||||
|
Loading…
Reference in New Issue
Block a user