Fix model name instead of provider name when looking up ai config. (#…

…7919)
geldata · Oct 25, 2024 · e838e48 · e838e48
1 parent d6281c5
commit e838e48
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)