11 lines
262 B
Python
11 lines
262 B
Python
|
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
|