Add structured outputs to Ask Mode

weaviate_agents/query/classes/__init__.py

@@ -24,6 +24,7 @@
     IsNullPropertyFilter,
     ModelUnitUsage,
     NumericMetrics,
+    ParsedAskModeResponse,
     ProgressDetails,
     ProgressMessage,
     QueryAgentResponse,
@@ -79,6 +80,7 @@
     "GeoPropertyFilter",
     "UnknownPropertyAggregation",
     "UnknownPropertyFilter",
+    "ParsedAskModeResponse",
     "ProgressDetails",
     "ProgressMessage",
     "QueryWithCollection",

weaviate_agents/query/classes/response.py

@@ -15,7 +15,7 @@
 )
 from uuid import UUID
 
-from pydantic import BaseModel, ConfigDict, Field, field_validator
+from pydantic import BaseModel, ConfigDict, Field, SerializeAsAny, field_validator
 from typing_extensions import TypedDict
 from weaviate.outputs.query import QueryReturn
 
@@ -479,6 +479,13 @@ def display(self) -> None:
         return None
 
 
+T = TypeVar("T", bound=Union[dict, BaseModel])
+
+
+class ParsedAskModeResponse(AskModeResponse, Generic[T]):
+    final_answer_parsed: SerializeAsAny[T]
+
+
 class ResearchModeResponse(BaseModel):
     output_type: Literal["final_state"] = "final_state"