Remove inline from "inline completion" as per review request

jupyterlab · Jan 24, 2024 · b235e0f · b235e0f
1 parent be4345f
commit b235e0f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
@@ -340,9 +340,9 @@ def get_chat_prompt_template(self) -> PromptTemplate:
                 + CHAT_DEFAULT_TEMPLATE,
             )
 
-    def get_inline_completion_prompt_template(self) -> PromptTemplate:
+    def get_completion_prompt_template(self) -> PromptTemplate:
         """
-        Produce a prompt template optimised for code or text completion.
+        Produce a prompt template optimised for inline code or text completion.
         The template should take variables: prefix, suffix, language, filename.
         """
         if self.is_chat_provider:

diff --git a/packages/jupyter-ai/jupyter_ai/completions/handlers/default.py b/packages/jupyter-ai/jupyter_ai/completions/handlers/default.py
@@ -31,7 +31,7 @@ def create_llm_chain(
         model_parameters = self.get_model_parameters(provider, provider_params)
         llm = provider(**provider_params, **model_parameters)
 
-        prompt_template = llm.get_inline_completion_prompt_template()
+        prompt_template = llm.get_completion_prompt_template()
 
         self.llm = llm
         self.llm_chain = prompt_template | llm | StrOutputParser()