remove unknown context types

codota · Dec 12, 2023 · da43de1 · da43de1
1 parent 1c2490a
commit da43de1
Showing 1 changed file with 13 additions and 11 deletions.
diff --git a/...rc/main/java/com/tabnineCommon/chat/commandHandlers/context/GetEnrichingContextHandler.kt b/...rc/main/java/com/tabnineCommon/chat/commandHandlers/context/GetEnrichingContextHandler.kt
@@ -30,18 +30,20 @@ class GetEnrichingContextHandler(gson: Gson) :
         val contextTypesSet = payload?.contextTypes?.toSet() ?: return EnrichingContextResponsePayload()
         val editor = getEditorFromProject(project) ?: return EnrichingContextResponsePayload()
 
-        val enrichingContextData = contextTypesSet.map {
-            when (it) {
-                EnrichingContextType.Editor -> EditorContext.createFuture(editor)
-                EnrichingContextType.Workspace -> WorkspaceContext.createFuture(
-                    editor,
-                    project,
-                    payload.workspaceCommands ?: emptyList()
-                )
-
-                EnrichingContextType.Diagnostics -> DiagnosticsContext.createFuture(editor, project)
+        val enrichingContextData = contextTypesSet
+            // remove unknown context types
+            .filter { it != null }.map {
+                when (it) {
+                    EnrichingContextType.Editor -> EditorContext.createFuture(editor)
+                    EnrichingContextType.Workspace -> WorkspaceContext.createFuture(
+                        editor,
+                        project,
+                        payload.workspaceCommands ?: emptyList()
+                    )
+
+                    EnrichingContextType.Diagnostics -> DiagnosticsContext.createFuture(editor, project)
+                }
             }
-        }
 
         CompletableFuture.allOf(*enrichingContextData.toTypedArray()).get(3, TimeUnit.SECONDS)