From e838e486217d95dab9eaaeb3f8ceab8d3aac53e6 Mon Sep 17 00:00:00 2001 From: dnwpark Date: Fri, 25 Oct 2024 13:43:46 -0400 Subject: [PATCH] Fix model name instead of provider name when looking up ai config. (#7919) --- edb/server/protocol/ai_ext.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/edb/server/protocol/ai_ext.py b/edb/server/protocol/ai_ext.py index 1e259364480..707f458bec5 100644 --- a/edb/server/protocol/ai_ext.py +++ b/edb/server/protocol/ai_ext.py @@ -1801,7 +1801,7 @@ async def _get_model_provider( model_name: str, ) -> str: provider = await _get_model_annotation_as_json( - db, base_model_type, model_name, "ext::ai::model_name") + db, base_model_type, model_name, "ext::ai::model_provider") return cast(str, provider)