fix(cli): allow lsp to be started outside of a project root

Author: Davidyz

src/vectorcode/lsp_main.py

@@ -1,4 +1,5 @@
 import argparse
+from ast import parse
 import asyncio
 import os
 import sys
@@ -75,27 +76,30 @@ async def execute_command(ls: LanguageServer, args: list[str]):
         )
         return
     if parsed_args.project_root is None:
-        assert DEFAULT_PROJECT_ROOT is not None, (
-            "Failed to automatically resolve project root!"
-        )
-
-        parsed_args.project_root = DEFAULT_PROJECT_ROOT
+        if DEFAULT_PROJECT_ROOT is not None:
+            parsed_args.project_root = DEFAULT_PROJECT_ROOT
     elif DEFAULT_PROJECT_ROOT is None:
         DEFAULT_PROJECT_ROOT = str(parsed_args.project_root)
 
-    parsed_args.project_root = os.path.abspath(str(parsed_args.project_root))
-    await make_caches(parsed_args.project_root)
-    final_configs = await cached_project_configs[parsed_args.project_root].merge_from(
-        parsed_args
-    )
-    final_configs.pipe = True
+    if parsed_args.project_root is not None:
+        parsed_args.project_root = os.path.abspath(str(parsed_args.project_root))
+        await make_caches(parsed_args.project_root)
+        final_configs = await cached_project_configs[
+            parsed_args.project_root
+        ].merge_from(parsed_args)
+        final_configs.pipe = True
+        client = await get_client(final_configs)
+        collection = await get_collection(
+            client=client,
+            configs=final_configs,
+            make_if_missing=final_configs.action in {CliAction.vectorise},
+        )
+    else:
+        final_configs = parsed_args
+        client = await get_client(parsed_args)
+        collection = None
     progress_token = str(uuid.uuid4())
-    client = await get_client(final_configs)
-    collection = await get_collection(
-        client=client,
-        configs=final_configs,
-        make_if_missing=final_configs.action in {CliAction.vectorise},
-    )
+
     await ls.progress.create_async(progress_token)
     match final_configs.action:
         case CliAction.query:
@@ -108,9 +112,12 @@ async def execute_command(ls: LanguageServer, args: list[str]):
             )
             final_results = []
             try:
-                final_results.extend(
-                    await build_query_results(collection, final_configs)
-                )
+                if collection is None:
+                    print("Please specify a project to search in.", file=sys.stderr)
+                else:
+                    final_results.extend(
+                        await build_query_results(collection, final_configs)
+                    )
             finally:
                 ls.progress.end(
                     progress_token,

tests/test_lsp.py

@@ -265,11 +265,13 @@ async def test_execute_command_no_default_project_root(
     global DEFAULT_PROJECT_ROOT
     DEFAULT_PROJECT_ROOT = None
     mock_config.project_root = None
-    with patch(
-        "vectorcode.lsp_main.parse_cli_args", new_callable=AsyncMock
-    ) as mock_parse_cli_args:
+    with (
+        patch(
+            "vectorcode.lsp_main.parse_cli_args", new_callable=AsyncMock
+        ) as mock_parse_cli_args,
+        patch("sys.stderr.write") as stderr,
+    ):
         mock_parse_cli_args.return_value = mock_config
-        with pytest.raises(AssertionError) as excinfo:
-            await execute_command(mock_language_server, ["query", "test"])
-        assert "Failed to automatically resolve project root!" in str(excinfo.value)
+        await execute_command(mock_language_server, ["query", "test"])
+        stderr.assert_called()
     DEFAULT_PROJECT_ROOT = None  # Reset the global variable