Don't allow returning a Promise from a tool call (#226688)

microsoft · Aug 27, 2024 · ff7a154 · ff7a154
1 parent 422b8e9
commit ff7a154
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/src/vs/workbench/api/common/extHostLanguageModelTools.ts b/src/vs/workbench/api/common/extHostLanguageModelTools.ts
@@ -94,6 +94,13 @@ export class ExtHostLanguageModelTools implements ExtHostLanguageModelToolsShape
 			throw new CancellationError();
 		}
 
+		for (const key of Object.keys(extensionResult)) {
+			const value = extensionResult[key];
+			if (value instanceof Promise) {
+				throw new Error(`Tool result for '${key}' cannot be a Promise`);
+			}
+		}
+
 		return typeConvert.LanguageModelToolResult.from(extensionResult);
 	}