fix: use sync function for single query, avoid sync open() in async context

Author: vitali87

codebase_rag/cli.py

@@ -18,8 +18,8 @@
     connect_memgraph,
     export_graph_to_file,
     main_async,
-    main_async_single_query,
     main_optimize_async,
+    main_single_query,
     prompt_for_unignored_directories,
     style,
     update_model_settings,
@@ -247,11 +247,7 @@ def start(
 
     try:
         if ask_agent:
-            asyncio.run(
-                main_async_single_query(
-                    target_repo_path, effective_batch_size, ask_agent
-                )
-            )
+            main_single_query(target_repo_path, effective_batch_size, ask_agent)
         else:
             asyncio.run(main_async(target_repo_path, effective_batch_size))
     except KeyboardInterrupt:

codebase_rag/main.py

@@ -1023,14 +1023,12 @@ def _initialize_services_and_agent(
     return rag_agent, confirmation_tool_names
 
 
-async def main_async_single_query(
-    repo_path: str, batch_size: int, question: str
-) -> None:
+def main_single_query(repo_path: str, batch_size: int, question: str) -> None:
     _setup_common_initialization(repo_path)
 
     with connect_memgraph(batch_size) as ingestor:
         rag_agent, _ = _initialize_services_and_agent(repo_path, ingestor)
-        response = await rag_agent.run(question, message_history=[])
+        response = asyncio.run(rag_agent.run(question, message_history=[]))
         print(response.output)  # noqa: T201
 
 

codebase_rag/mcp/client.py

@@ -13,7 +13,8 @@
 
 
 async def query_mcp_server(question: str) -> dict[str, str]:
-    with open(os.devnull, "w") as devnull:
+    devnull = open(os.devnull, "w")  # noqa: ASYNC230, SIM115
+    try:
         server_params = StdioServerParameters(
             command=sys.executable,
             args=["-m", "codebase_rag.cli", "mcp-server"],
@@ -37,6 +38,8 @@ async def query_mcp_server(question: str) -> dict[str, str]:
                     except json.JSONDecodeError:
                         return {"output": response_text}
                 return {"output": "No response from server"}
+    finally:
+        devnull.close()
 
 
 @app.command()