From c215481531a8f36b6c196b213db8d4ec2f0c5306 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=88=98=20=E6=96=B9=E7=91=9E?= Date: Thu, 24 Aug 2023 09:26:29 +0800 Subject: [PATCH] Update default index type and metric type for MyScale vector store (#9353) We update the default index type from `IVFFLAT` to `MSTG`, a new vector type developed by MyScale. --- libs/langchain/langchain/vectorstores/myscale.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/libs/langchain/langchain/vectorstores/myscale.py b/libs/langchain/langchain/vectorstores/myscale.py index 7bbb20dcfb..ef4db34ead 100644 --- a/libs/langchain/langchain/vectorstores/myscale.py +++ b/libs/langchain/langchain/vectorstores/myscale.py @@ -46,7 +46,7 @@ class MyScaleSettings(BaseSettings): table (str) : Table name to operate on. Defaults to 'vector_table'. metric (str) : Metric to compute distance, - supported are ('l2', 'cosine', 'ip'). Defaults to 'cosine'. + supported are ('L2', 'Cosine', 'IP'). Defaults to 'Cosine'. column_map (Dict) : Column type map to project column name onto langchain semantics. Must have keys: `text`, `id`, `vector`, must be same size to number of columns. For example: @@ -69,7 +69,7 @@ class MyScaleSettings(BaseSettings): username: Optional[str] = None password: Optional[str] = None - index_type: str = "IVFFLAT" + index_type: str = "MSTG" index_param: Optional[Dict[str, str]] = None column_map: Dict[str, str] = { @@ -81,7 +81,7 @@ class MyScaleSettings(BaseSettings): database: str = "default" table: str = "langchain" - metric: str = "cosine" + metric: str = "Cosine" def __getitem__(self, item: str) -> Any: return getattr(self, item)