add more features to phi-2 model (#900)

kumarijy · web-flow · commit 7365f515fdb0 · 2024-04-02T08:56:27.000+01:00
diff --git a/modules/openvino_code/package-lock.json b/modules/openvino_code/package-lock.json
diff --git a/modules/openvino_code/package.json b/modules/openvino_code/package.json
@@ -1,7 +1,7 @@
 {
   "publisher": "OpenVINO",
   "name": "openvino-code-completion",
-  "version": "0.0.13",
+  "version": "0.0.14",
   "displayName": "OpenVINO Code Completion",
   "description": "VSCode extension for AI code completion with OpenVINO",
   "icon": "media/logo.png",
diff --git a/modules/openvino_code/shared/features.ts b/modules/openvino_code/shared/features.ts
@@ -2,4 +2,6 @@ export enum Features {
   CODE_COMPLETION = 'Code Completion',
   SUMMARIZATION = 'Summarization',
   FIM = 'Fill-in-the-middle',
+  QA_FORMAT = 'QA Format',
+  CHAT_FORMAT = 'Chat Format',
 }
diff --git a/modules/openvino_code/shared/model.ts b/modules/openvino_code/shared/model.ts
@@ -29,5 +29,5 @@ export const MODEL_SUPPORTED_FEATURES: Record<ModelName, Features[]> = {
   [ModelName.DECICODER_1B_OPENVINO_INT8]: [Features.CODE_COMPLETION, Features.SUMMARIZATION],
   [ModelName.STABLECODE_COMPLETION_ALPHA_3B_4K_OPENVINO_INT8]: [Features.CODE_COMPLETION, Features.SUMMARIZATION],
   [ModelName.DEEPSEEK_CODER_1_3B]: [Features.CODE_COMPLETION, Features.SUMMARIZATION, Features.FIM],
-  [ModelName.PHI_2_2_7B]: [Features.CODE_COMPLETION],
+  [ModelName.PHI_2_2_7B]: [Features.CODE_COMPLETION, Features.SUMMARIZATION, Features.QA_FORMAT, Features.CHAT_FORMAT],
 };

Original file line number	Diff line number	Diff line change
`@@ -1,7 +1,7 @@`
`1`	`1`	`{`
`2`	`2`	`"publisher": "OpenVINO",`
`3`	`3`	`"name": "openvino-code-completion",`
`4`		`- "version": "0.0.13",`
	`4`	`+ "version": "0.0.14",`
`5`	`5`	`"displayName": "OpenVINO Code Completion",`
`6`	`6`	`"description": "VSCode extension for AI code completion with OpenVINO",`
`7`	`7`	`"icon": "media/logo.png",`
Original file line number	Diff line number	Diff line change
`@@ -2,4 +2,6 @@ export enum Features {`
`2`	`2`	`CODE_COMPLETION = 'Code Completion',`
`3`	`3`	`SUMMARIZATION = 'Summarization',`
`4`	`4`	`FIM = 'Fill-in-the-middle',`
	`5`	`+ QA_FORMAT = 'QA Format',`
	`6`	`+ CHAT_FORMAT = 'Chat Format',`
`5`	`7`	`}`