fix(lsp): Validate include parameters in query command

Author: Zhe Yu

src/vectorcode/lsp_main.py

@@ -52,6 +52,7 @@
     _prepare_formatted_result,
     get_reranked_results,
     preprocess_query_keywords,
+    verify_include,
 )
 
 DEFAULT_PROJECT_ROOT: str | None = None
@@ -144,6 +145,14 @@ async def execute_command(ls: LanguageServer, args: list[str]):
                         message=f"Querying {cleanup_path(str(final_configs.project_root))}",
                     ),
                 )
+                if not verify_include(final_configs):
+                    log_msg = "Invalid `--include` parameters!"
+                    logger.error(log_msg)
+                    ls.progress.end(
+                        progress_token,
+                        types.WorkDoneProgressEnd(message=log_msg),
+                    )
+                    return []
                 final_results = []
                 try:
                     preprocess_query_keywords(final_configs)

tests/test_lsp.py

@@ -85,6 +85,42 @@ async def test_execute_command_query(mock_language_server, mock_config):
         mock_language_server.progress.end.assert_called()
 
 
+@pytest.mark.asyncio
+async def test_execute_command_query_invalid_include(mock_language_server, mock_config):
+    with (
+        patch(
+            "vectorcode.lsp_main.parse_cli_args", new_callable=AsyncMock
+        ) as mock_parse_cli_args,
+        patch("vectorcode.lsp_main.get_database_connector"),
+        patch(
+            "vectorcode.lsp_main.get_reranked_results", new_callable=AsyncMock
+        ) as mock_get_reranked_results,
+        patch("os.path.isfile", return_value=True),
+        patch("builtins.open", MagicMock()) as mock_open,
+    ):
+        mock_parse_cli_args.return_value = mock_config
+        mock_get_reranked_results.return_value = []
+
+        # Configure the MagicMock object to return a string when read() is called
+        mock_file = MagicMock()
+        mock_file.__enter__.return_value.read.return_value = "{}"  # Return valid JSON
+        mock_open.return_value = mock_file
+
+        # Ensure parsed_args.project_root is not None
+        mock_config.project_root = "/test/project"
+        mock_config.query = ["test"]
+        mock_config.include = [QueryInclude.chunk, QueryInclude.document]
+
+        # Mock the merge_from method
+        mock_config.merge_from = AsyncMock(return_value=mock_config)
+
+        result = await execute_command(mock_language_server, ["query", "test"])
+
+        assert result == []
+        mock_language_server.progress.begin.assert_called()
+        mock_language_server.progress.end.assert_called()
+
+
 @pytest.mark.asyncio
 async def test_execute_command_query_default_proj_root(
     mock_language_server, mock_config