github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-09-13 21:47:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix flexible dimension and doc for DingoDB (#12187)

Browse Source
This commit is contained in:
Hech
2023-10-25 01:16:19 +08:00
committed by GitHub
parent 95ae40ff90
commit d76f026d72
4 changed files with 18 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
libs/langchain/langchain/vectorstores/dingo.py
View File

@@ -36,6 +36,7 @@ class Dingo(VectorStore):
*,
client: Any = None,
index_name: Optional[str] = None,
dimension: int = 1024,
host: Optional[List[str]] = None,
user: str = "root",
password: str = "123123",
@@ -67,9 +68,11 @@ class Dingo(VectorStore):
if index_name is not None and index_name not in dingo_client.get_index():
if self_id is True:
dingo_client.create_index(index_name, 1024, auto_id=False)
dingo_client.create_index(
index_name, dimension=dimension, auto_id=False
)
else:
dingo_client.create_index(index_name, 1024)
dingo_client.create_index(index_name, dimension=dimension)
self._index_name = index_name
self._embedding = embedding
@@ -268,6 +271,7 @@ class Dingo(VectorStore):
ids: Optional[List[str]] = None,
text_key: str = "text",
index_name: Optional[str] = None,
dimension: int = 1024,
client: Any = None,
host: List[str] = ["172.20.31.10:13000"],
user: str = "root",
@@ -315,11 +319,12 @@ class Dingo(VectorStore):
raise ValueError(f"Dingo failed to connect: {e}")
if kwargs is not None and kwargs.get("self_id") is True:
if index_name not in dingo_client.get_index():
dingo_client.create_index(index_name, 1024, auto_id=False)
dingo_client.create_index(
index_name, dimension=dimension, auto_id=False
)
else:
if index_name not in dingo_client.get_index():
dingo_client.create_index(index_name, 1024)
# dingo_client.create_index(index_name, 1024, index_type="hnsw")
dingo_client.create_index(index_name, dimension=dimension)
# Embed and create the documents