Skip to content

Commit c657762

Browse files
Updating embeddings config fields from prespecified custom config file (#1298)
* Updating embeddings config fields from prespecified custom config file * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci * Update extension.py --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
1 parent 8167c7a commit c657762

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
@@ -347,7 +347,6 @@ def initialize_settings(self):
347347
self.log.info(f"Configured provider blocklist: {self.blocked_providers}")
348348
self.log.info(f"Configured model allowlist: {self.allowed_models}")
349349
self.log.info(f"Configured model blocklist: {self.blocked_models}")
350-
351350
self.settings["model_parameters"] = self.model_parameters
352351
self.log.info(f"Configured model parameters: {self.model_parameters}")
353352

@@ -356,6 +355,8 @@ def initialize_settings(self):
356355
"embeddings_provider_id": self.default_embeddings_model,
357356
"api_keys": self.default_api_keys,
358357
"fields": self.model_parameters,
358+
"embeddings_fields": self.model_parameters,
359+
"completions_fields": self.model_parameters,
359360
}
360361

361362
# Fetch LM & EM providers

0 commit comments

Comments
 (0)