github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-28 09:28:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

add variables for field names (#9563)

Browse Source
This commit is contained in:
Harrison Chase 2023-08-22 07:43:21 -07:00 committed by GitHub
parent dfb9ff1079
commit 1720e99397
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -52,6 +52,9 @@ class Milvus(VectorStore):
default of index. default of index.
drop_old (Optional[bool]): Whether to drop the current collection. Defaults drop_old (Optional[bool]): Whether to drop the current collection. Defaults
to False. to False.
primary_field (str): Name of the primary key field. Defaults to "pk".
text_field (str): Name of the text field. Defaults to "text".
vector_field (str): Name of the vector field. Defaults to "vector".
The connection args used for this class comes in the form of a dict, The connection args used for this class comes in the form of a dict,
here are a few of the options: here are a few of the options:
@ -107,6 +110,10 @@ class Milvus(VectorStore):
index_params: Optional[dict] = None, index_params: Optional[dict] = None,
search_params: Optional[dict] = None, search_params: Optional[dict] = None,
drop_old: Optional[bool] = False, drop_old: Optional[bool] = False,
*,
primary_field: str = "pk",
text_field: str = "text",
vector_field: str = "vector",
): ):
"""Initialize the Milvus vector store.""" """Initialize the Milvus vector store."""
try: try:
@ -138,11 +145,11 @@ class Milvus(VectorStore):
self.consistency_level = consistency_level self.consistency_level = consistency_level
# In order for a collection to be compatible, pk needs to be auto'id and int # In order for a collection to be compatible, pk needs to be auto'id and int
self._primary_field = "pk" self._primary_field = primary_field
# In order for compatiblility, the text field will need to be called "text" # In order for compatibility, the text field will need to be called "text"
self._text_field = "text" self._text_field = text_field
# In order for compatibility, the vector field needs to be called "vector" # In order for compatibility, the vector field needs to be called "vector"
self._vector_field = "vector" self._vector_field = vector_field
self.fields: list[str] = [] self.fields: list[str] = []
# Create the connection to the server # Create the connection to the server
if connection_args is None: if connection_args is None: