diff --git a/src/khoj/routers/research.py b/src/khoj/routers/research.py index 494a01b6..e72b25a5 100644 --- a/src/khoj/routers/research.py +++ b/src/khoj/routers/research.py @@ -198,6 +198,8 @@ async def apick_next_tool( try: response = load_complex_json(response) + if not isinstance(response, dict): + raise ValueError(f"Expected dict response, got {type(response).__name__}: {response}") selected_tool = response.get("tool", None) generated_query = response.get("query", None) scratchpad = response.get("scratchpad", None)