From 620d672f329bb54389fb030d0a038fbe9f4785a0 Mon Sep 17 00:00:00 2001 From: Carlo Kroll Date: Thu, 29 Aug 2024 03:16:30 +0100 Subject: [PATCH] prevent dimension error when model is sentence-transformer model and not None --- src/curate_gpt/store/duckdb_adapter.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/curate_gpt/store/duckdb_adapter.py b/src/curate_gpt/store/duckdb_adapter.py index 02c44b8..f357b47 100644 --- a/src/curate_gpt/store/duckdb_adapter.py +++ b/src/curate_gpt/store/duckdb_adapter.py @@ -1020,10 +1020,9 @@ def _parse_where_clause(where: Dict[str, Any]) -> str: return " AND ".join(conditions) def _get_embedding_dimension(self, model_name: str) -> int: - if model_name is None: + if model_name is None or model_name.startswith(self.default_model): return DEFAULT_MODEL[self.default_model] if isinstance(model_name, str): - logger.info("somehow here") if model_name.startswith("openai:"): model_key = model_name.split("openai:", 1)[1] model_info = MODEL_MAP.get(model_key, DEFAULT_OPENAI_MODEL)