init

vw-agentic-rag/service/retrieval/model.py

@@ -0,0 +1,11 @@
+from typing import Any, Optional
+
+from pydantic import BaseModel
+
+
+
+class RetrievalResponse(BaseModel):
+    """Simple response container for tool results"""
+    results: list[dict[str, Any]]
+    took_ms: Optional[int] = None
+    total_count: Optional[int] = None