Small fixes and improvements

Author: hetzge

plugin/src/de/hetzge/eclipse/aicoder/context/PackageContextEntry.java

@@ -17,6 +17,7 @@
 import de.hetzge.eclipse.aicoder.AiCoderImageKey;
 import de.hetzge.eclipse.aicoder.util.ContextUtils;
 import de.hetzge.eclipse.aicoder.util.JavaProjectUtils;
+import de.hetzge.eclipse.aicoder.util.JdtUtils;
 import de.hetzge.eclipse.aicoder.util.LambdaExceptionUtils;
 import de.hetzge.eclipse.aicoder.util.Utils;
 
@@ -81,7 +82,8 @@ public static Optional<PackageContextEntry> create(ContextEntryKey key) throws J
 	private static PackageContextEntry create(final IPackageFragment packageFragment) throws CoreException {
 		final long before = System.currentTimeMillis();
 		final String elementName = packageFragment.getElementName();
-		final List<TypeContextEntry> entries = Arrays.stream(packageFragment.getCompilationUnits())
+		final ICompilationUnit[] compilationUnits = JdtUtils.getCompilationUnits(packageFragment);
+		final List<TypeContextEntry> entries = Arrays.stream(compilationUnits)
 				.flatMap(LambdaExceptionUtils.rethrowFunction(it -> Arrays.stream(it.getAllTypes())))
 				.filter(Utils::checkType)
 				.map(LambdaExceptionUtils.rethrowFunction(TypeContextEntry::create))

plugin/src/de/hetzge/eclipse/aicoder/inline/InlineCompletionController.java

@@ -427,6 +427,7 @@ public void accept() {
 			acceptSuggestion();
 		}
 		if (AiCoderPreferences.isCleanupCodeOnApplyEnabled()) {
+			// TODO only cleanup if syntax is complete
 			AiCoderActivator.log().info("Trigger code cleanup on apply");
 			final Optional<ICompilationUnit> compilationUnitOptional = EclipseUtils.getCompilationUnit(this.textEditor.getEditorInput());
 			if (compilationUnitOptional.isPresent()) {

plugin/src/de/hetzge/eclipse/aicoder/llm/LlmUtils.java

@@ -172,7 +172,8 @@ private static LlmResponse executeOpenAi(LlmOption llmModelOption, String system
 					Map.entry("prefix", prompt),
 					Map.entry("suffix", suffix)));
 			json.set("prompt", fimPrompt)
-					.set("max_tokens", AiCoderPreferences.getMaxTokens());
+					.set("max_tokens", AiCoderPreferences.getMaxTokens())
+					.set("stop", createStop(AiCoderPreferences.isMultilineEnabled()));
 		}
 
 		final URL url = URI.create(urlString + "/").resolve(isFillInTheMiddle ? "./v1/completions" : "./v1/chat/completions").toURL();

plugin/src/de/hetzge/eclipse/aicoder/util/JdtUtils.java

@@ -0,0 +1,21 @@
+package de.hetzge.eclipse.aicoder.util;
+
+import org.eclipse.jdt.core.ICompilationUnit;
+import org.eclipse.jdt.core.IPackageFragment;
+import org.eclipse.jdt.core.JavaModelException;
+
+import de.hetzge.eclipse.aicoder.AiCoderActivator;
+
+public final class JdtUtils {
+	private JdtUtils() {
+	}
+
+	public static ICompilationUnit[] getCompilationUnits(final IPackageFragment packageFragment) {
+		try {
+			return packageFragment.getCompilationUnits();
+		} catch (final JavaModelException exception) {
+			AiCoderActivator.log().error("Failed to get compilation units for package fragment: " + packageFragment, exception);
+			return new ICompilationUnit[0];
+		}
+	}
+}