Skip to content

Commit bb69b89

Browse files
Backport PR #1298: Updating embeddings config fields from prespecified custom config file (#1302)
Co-authored-by: Sanjiv Das <[email protected]>
1 parent ce17532 commit bb69b89

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

packages/jupyter-ai/jupyter_ai/extension.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -208,7 +208,6 @@ def initialize_settings(self):
208208
self.log.info(f"Configured provider blocklist: {self.blocked_providers}")
209209
self.log.info(f"Configured model allowlist: {self.allowed_models}")
210210
self.log.info(f"Configured model blocklist: {self.blocked_models}")
211-
212211
self.settings["model_parameters"] = self.model_parameters
213212
self.log.info(f"Configured model parameters: {self.model_parameters}")
214213

@@ -217,6 +216,8 @@ def initialize_settings(self):
217216
"embeddings_provider_id": self.default_embeddings_model,
218217
"api_keys": self.default_api_keys,
219218
"fields": self.model_parameters,
219+
"embeddings_fields": self.model_parameters,
220+
"completions_fields": self.model_parameters,
220221
}
221222

222223
# Fetch LM & EM providers

0 commit comments

Comments
 (0)