fix(rag): improve rag error message when docling is not enabled (#1432)

Signed-off-by: Radek Ježek <[email protected]>

Author: jezekra1

agents/rag/src/rag/helpers/vectore_store.py

@@ -55,7 +55,10 @@ async def extract_file(file: File) -> None:
             extraction = await file.get_extraction()
             final_status = extraction.status
             if final_status == "failed":
-                raise RuntimeError(f"Extraction for file {file_id} has failed: {extraction.model_dump_json()}")
+                raise RuntimeError(
+                    f"Extraction for file {file.filename} has failed. \n"
+                    f"Make sure you have docling enabled (`agentstack start --set docling.enabled=true`)"
+                )
             if final_status != "completed":
                 raise TimeoutError("Text extraction is not finished yet")
 

apps/agentstack-sdk-py/src/agentstack_sdk/server/agent.py

@@ -3,6 +3,7 @@
 
 import asyncio
 import inspect
+import json
 from asyncio import CancelledError
 from collections.abc import AsyncGenerator, AsyncIterator, Callable, Generator
 from contextlib import AbstractAsyncContextManager, AsyncExitStack, asynccontextmanager
@@ -43,6 +44,7 @@
 from agentstack_sdk.server.logging import logger
 from agentstack_sdk.server.store.context_store import ContextStore
 from agentstack_sdk.server.utils import cancel_task, close_queue
+from agentstack_sdk.util.utils import extract_messages
 
 AgentFunction: TypeAlias = Callable[[], AsyncGenerator[RunYield, RunYieldResume]]
 AgentFunctionFactory: TypeAlias = Callable[
@@ -453,7 +455,8 @@ def with_context(message: Message | None = None) -> Message | None:
             await task_updater.cancel()
         except Exception as ex:
             logger.error("Error when executing agent", exc_info=ex)
-            await task_updater.failed(task_updater.new_agent_message(parts=[Part(root=TextPart(text=str(ex)))]))
+            msg = json.dumps(extract_messages(ex), indent=2)
+            await task_updater.failed(task_updater.new_agent_message(parts=[Part(root=TextPart(text=msg))]))
         finally:  # cleanup
             await cancel_task(cancellation_task)
             is_cancelling = bool(current_task.cancelling())

apps/agentstack-sdk-py/src/agentstack_sdk/util/utils.py

@@ -29,3 +29,10 @@ async def parse_stream(response: httpx.Response) -> AsyncIterator[dict[str, Any]
     async for line in response.aiter_lines():
         if line:
             yield json.loads(re.sub("^data:", "", line).strip())
+
+
+def extract_messages(exc: BaseException) -> list[tuple[str, str]]:
+    if isinstance(exc, BaseExceptionGroup):
+        return [(exc_type, msg) for e in exc.exceptions for exc_type, msg in extract_messages(e)]
+    else:
+        return [(type(exc).__name__, str(exc))]