api finalization for chat extension (#1788)

Author: justschen

src/extension/conversation/common/languageModelChatMessageHelpers.ts

@@ -4,7 +4,15 @@
  *--------------------------------------------------------------------------------------------*/
 
 
-import { ChatImageMimeType, LanguageModelDataPart } from '../../../vscodeTypes';
+import { LanguageModelDataPart } from '../../../vscodeTypes';
+
+export enum ChatImageMimeType {
+	PNG = 'image/png',
+	JPEG = 'image/jpeg',
+	GIF = 'image/gif',
+	WEBP = 'image/webp',
+	BMP = 'image/bmp',
+}
 
 export function isImageDataPart(part: unknown): part is LanguageModelDataPart {
 	if (part instanceof LanguageModelDataPart && isChatImageMimeType(part.mimeType)) {
@@ -25,4 +33,4 @@ function isChatImageMimeType(mimeType: string): mimeType is ChatImageMimeType {
 		default:
 			return false;
 	}
-}
+}

src/extension/prompt/vscode-node/requestLoggerToolResult.tsx

@@ -4,7 +4,8 @@
  *--------------------------------------------------------------------------------------------*/
 
 import { JSONTree, OutputMode, PromptElement, Raw, renderPrompt, UserMessage } from '@vscode/prompt-tsx';
-import { ChatImageMimeType, LanguageModelDataPart, LanguageModelPromptTsxPart } from '../../../vscodeTypes';
+import { LanguageModelDataPart, LanguageModelPromptTsxPart } from '../../../vscodeTypes';
+import { ChatImageMimeType } from '../../conversation/common/languageModelChatMessageHelpers';
 
 export async function renderToolResultToStringNoBudget(part: LanguageModelPromptTsxPart) {
 	const r = await renderPrompt(class extends PromptElement {

src/extension/tools/node/getNotebookCellOutputTool.tsx

@@ -11,7 +11,7 @@ import { IPromptPathRepresentationService } from '../../../platform/prompts/comm
 import { IWorkspaceService } from '../../../platform/workspace/common/workspaceService';
 import { findNotebook } from '../../../util/common/notebooks';
 import { IInstantiationService } from '../../../util/vs/platform/instantiation/common/instantiation';
-import { ChatImageMimeType, ExtendedLanguageModelToolResult, LanguageModelDataPart, LanguageModelPromptTsxPart, LanguageModelTextPart, MarkdownString } from '../../../vscodeTypes';
+import { ExtendedLanguageModelToolResult, LanguageModelDataPart, LanguageModelPromptTsxPart, LanguageModelTextPart, MarkdownString } from '../../../vscodeTypes';
 import { IBuildPromptContext } from '../../prompt/common/intents';
 import { renderPromptElementJSON } from '../../prompts/node/base/promptRenderer';
 import { ToolName } from '../common/toolNames';
@@ -21,6 +21,7 @@ import { ITelemetryService } from '../../../platform/telemetry/common/telemetry'
 import { getCellIdMap } from '../../../platform/notebook/common/helpers';
 import { INotebookService } from '../../../platform/notebook/common/notebookService';
 import { ILogService } from '../../../platform/log/common/logService';
+import { ChatImageMimeType } from '../../conversation/common/languageModelChatMessageHelpers';
 
 export class GetNotebookCellOutputTool implements ICopilotTool<IGetNotebookCellOutputToolParams> {
 	public static toolName = ToolName.ReadCellOutput;

src/extension/tools/node/runNotebookCellTool.tsx

@@ -20,7 +20,7 @@ import { findNotebook, isJupyterNotebookUri } from '../../../util/common/noteboo
 import { raceCancellationError, raceTimeout } from '../../../util/vs/base/common/async';
 import { dispose } from '../../../util/vs/base/common/lifecycle';
 import { IInstantiationService } from '../../../util/vs/platform/instantiation/common/instantiation';
-import { ChatImageMimeType, ExtendedLanguageModelToolResult, LanguageModelDataPart, LanguageModelPromptTsxPart, LanguageModelTextPart, LanguageModelToolResult, MarkdownString } from '../../../vscodeTypes';
+import { ExtendedLanguageModelToolResult, LanguageModelDataPart, LanguageModelPromptTsxPart, LanguageModelTextPart, LanguageModelToolResult, MarkdownString } from '../../../vscodeTypes';
 import { IBuildPromptContext } from '../../prompt/common/intents';
 import { renderPromptElementJSON } from '../../prompts/node/base/promptRenderer';
 import { Tag } from '../../prompts/node/base/tag';
@@ -29,6 +29,7 @@ import { ToolName } from '../common/toolNames';
 import { ICopilotTool, ToolRegistry } from '../common/toolsRegistry';
 import { IToolsService } from '../common/toolsService';
 import { IInstallExtensionToolInput } from './installExtensionTool';
+import { ChatImageMimeType } from '../../conversation/common/languageModelChatMessageHelpers';
 
 class RunNotebookTelemetryEvent {
 	public result: 'success' | 'failure' | 'skipped' = 'failure';