clean up

src/common/constants.ts

@@ -132,15 +132,6 @@ export const DEFAULT_PROVIDER_FORM_VALUES = {
   type: 'chat'
 }
 
-export const FIM_TEMPLATE_TYPE = [
-  'automatic',
-  'stable-code',
-  'codellama',
-  'deepseek',
-  'starcoder',
-  'custom-template'
-]
-
 export const WASM_LANGAUAGES: { [key: string]: string } = {
   cpp: 'cpp',
   hpp: 'cpp',

src/extension/fim-templates.ts

@@ -3,7 +3,7 @@ import {
   STOP_DEEPSEEK,
   STOP_LLAMA,
   STOP_STARCODER,
-  STOP_CODEGEMMA,
+  STOP_CODEGEMMA
 } from '../common/constants'
 import { supportedLanguages } from '../common/languages'
 import { FimPromptTemplate } from '../common/types'
@@ -74,19 +74,9 @@ export const getFimPromptTemplateStableCode = ({
 }
 
 export const getFimPromptTemplateCodegemma = ({
-  context,
-  header,
-  useFileContext,
-  prefixSuffix,
-  language
+  prefixSuffix
 }: FimPromptTemplate) => {
   const { prefix, suffix } = prefixSuffix
-  const languageId =
-    supportedLanguages[language as keyof typeof supportedLanguages]
-  const fileContext = useFileContext
-    ? `${languageId?.syntaxComments?.start}${context}${languageId?.syntaxComments?.end}`
-    : ''
-  const heading = header ? header : ''
   return `<|fim_prefix|>${prefix}<|fim_suffix|>${suffix}<|fim_middle|>`
 }
 

src/webview/providers.tsx

@@ -14,7 +14,7 @@ import styles from './providers.module.css'
 import { TwinnyProvider } from '../extension/provider-manager'
 import {
   DEFAULT_PROVIDER_FORM_VALUES,
-  FIM_TEMPLATE_TYPE
+  FIM_TEMPLATE_FORMAT,
 } from '../common/constants'
 
 export const Providers = () => {
@@ -223,7 +223,7 @@ function ProviderForm({ onClose, provider }: ProviderFormProps) {
               onChange={handleChangeDropdown}
               value={formState.fimTemplate}
             >
-              {FIM_TEMPLATE_TYPE.map((type, index) => (
+              {Object.values(FIM_TEMPLATE_FORMAT).map((type, index) => (
                 <VSCodeOption key={index} value={type || 'automatic'}>
                   {type}
                 </VSCodeOption>