langchain/libs/community/tests/integration_tests/vectorstores
Ian 390ef6abe3
community[minor]: Add Initial Support for TiDB Vector Store (#15796)
This pull request introduces initial support for the TiDB vector store.
The current version is basic, laying the foundation for the vector store
integration. While this implementation provides the essential features,
we plan to expand and improve the TiDB vector store support with
additional enhancements in future updates.

Upcoming Enhancements:
* Support for Vector Index Creation: To enhance the efficiency and
performance of the vector store.
* Support for max marginal relevance search. 
* Customized Table Structure Support: Recognizing the need for
flexibility, we plan for more tailored and efficient data store
solutions.

Simple use case exmaple

```python
from typing import List, Tuple
from langchain.docstore.document import Document
from langchain_community.vectorstores import TiDBVectorStore
from langchain_openai import OpenAIEmbeddings

db = TiDBVectorStore.from_texts(
    embedding=embeddings,
    texts=['Andrew like eating oranges', 'Alexandra is from England', 'Ketanji Brown Jackson is a judge'],
    table_name="tidb_vector_langchain",
    connection_string=tidb_connection_url,
    distance_strategy="cosine",
)

query = "Can you tell me about Alexandra?"
docs_with_score: List[Tuple[Document, float]] = db.similarity_search_with_score(query)
for doc, score in docs_with_score:
    print("-" * 80)
    print("Score: ", score)
    print(doc.page_content)
    print("-" * 80)
```
2024-03-07 17:18:20 -08:00
..
cassettes community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
docarray community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
docker-compose refactor: enable connection pool usage in PGVector (#11514) 2023-12-28 15:07:16 -08:00
fixtures community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
qdrant infra: add -p to mkdir in lint steps (#17013) 2024-02-05 11:22:06 -08:00
__init__.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
conftest.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
fake_embeddings.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_alibabacloud_opensearch.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_analyticdb.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_annoy.py community[major]: breaking change in some APIs to force users to opt-in for pickling (#18696) 2024-03-06 16:43:01 -05:00
test_astradb.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_atlas.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_awadb.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_azure_cosmos_db.py community[minor]: Adding Azure Cosmos Mongo vCore Vector DB Cache (#16856) 2024-03-03 14:04:15 -08:00
test_azuresearch.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_bagel.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_baiducloud_vector_search.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_bigquery_vector_search.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_cassandra.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_chroma.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_clarifai.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_clickhouse.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_dashvector.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_deeplake.py infra: add -p to mkdir in lint steps (#17013) 2024-02-05 11:22:06 -08:00
test_documentdb.py community[minor]: Add DocumentDBVectorSearch VectorStore (#17757) 2024-03-06 15:11:34 -08:00
test_elastic_vector_search.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_elasticsearch.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_epsilla.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_hanavector.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_hippo.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_hologres.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_infinispanvs.py community[minor]: Adding support for Infinispan as VectorStore (#17861) 2024-03-06 15:11:02 -08:00
test_jaguar.py community[minor]: added jaguar vector store (#14838) 2023-12-19 10:40:18 -05:00
test_kinetica.py Langchain vectorstore integration with Kinetica (#18102) 2024-02-26 12:46:48 -08:00
test_lancedb.py community[patch]: LanceDB integration improvements/fixes (#16173) 2024-02-19 10:22:02 -08:00
test_lantern.py infra: add -p to mkdir in lint steps (#17013) 2024-02-05 11:22:06 -08:00
test_llm_rails.py infra: add print rule to ruff (#16221) 2024-02-09 16:13:30 -08:00
test_marqo.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_meilisearch.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_milvus.py community[patch]: Make some functions work with Milvus (#10695) 2024-02-09 15:21:31 -08:00
test_momento_vector_index.py community[patch]: support momento vector index filter expressions (#14978) 2023-12-20 19:11:43 -08:00
test_mongodb_atlas.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_myscale.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_neo4jvector.py Add support for parameters in neo4j retrieval query (#18310) 2024-02-29 13:00:54 -08:00
test_nucliadb.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_opensearch.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_pgvector.py refactor: enable connection pool usage in PGVector (#11514) 2023-12-28 15:07:16 -08:00
test_pinecone.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_redis.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_rocksetdb.py community[patch]: update copy of metadata in rockset vectorstore integration (#17612) 2024-02-15 23:13:40 -07:00
test_scann.py community[major]: breaking change in some APIs to force users to opt-in for pickling (#18696) 2024-03-06 16:43:01 -05:00
test_singlestoredb.py community[patch]: Added add_images method to SingleStoreDB vector store (#17871) 2024-02-21 15:16:32 -08:00
test_sqlitevss.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_tair.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_tencentvectordb.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_thirdai_neuraldb.py infra: add -p to mkdir in lint steps (#17013) 2024-02-05 11:22:06 -08:00
test_tidb_vector.py community[minor]: Add Initial Support for TiDB Vector Store (#15796) 2024-03-07 17:18:20 -08:00
test_tiledb.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_timescalevector.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_usearch.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_vald.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_vearch.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_vectara.py infra: add -p to mkdir in lint steps (#17013) 2024-02-05 11:22:06 -08:00
test_weaviate.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_xata.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_yellowbrick.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_zep.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00
test_zilliz.py community[major], core[patch], langchain[patch], experimental[patch]: Create langchain-community (#14463) 2023-12-11 13:53:30 -08:00