fix: Add concrete return type for chat headline questions method

Author: wizrds

asknews_sdk/api/chat.py

@@ -5,6 +5,7 @@
     CreateChatCompletionRequest,
     CreateChatCompletionResponse,
     CreateChatCompletionResponseStream,
+    HeadlineQuestionsResponse,
     ListModelResponse,
 )
 from asknews_sdk.response import EventSource
@@ -106,7 +107,9 @@ def list_chat_models(self) -> ListModelResponse:
         )
         return ListModelResponse.model_validate(response.content)
 
-    def get_headline_questions(self, queries: Optional[List[str]] = None) -> Dict[str, List[str]]:
+    def get_headline_questions(
+        self, queries: Optional[List[str]] = None
+    ) -> HeadlineQuestionsResponse:
         """
         Get headline questions for a given query.
 
@@ -115,14 +118,14 @@ def get_headline_questions(self, queries: Optional[List[str]] = None) -> Dict[st
         :param queries: List of queries to get headline questions for
         :type queries: Optional[List[str]]
         :return: Headline questions
-        :rtype: Dict[str, List[str]]
+        :rtype: HeadlineQuestionsResponse
         """
         response = self.client.request(
             method="GET",
             endpoint="/v1/chat/questions",
             query={"queries": queries}
         )
-        return response.content
+        return HeadlineQuestionsResponse.model_validate(response.content)
 
 
 class AsyncChatAPI(BaseAPI):
@@ -224,7 +227,7 @@ async def list_chat_models(self) -> ListModelResponse:
 
     async def get_headline_questions(
         self, queries: Optional[List[str]] = None
-    ) -> Dict[str, List[str]]:
+    ) -> HeadlineQuestionsResponse:
         """
         Get headline questions for a given query.
 
@@ -233,11 +236,11 @@ async def get_headline_questions(
         :param queries: List of queries to get headline questions for
         :type queries: Optional[List[str]]
         :return: Headline questions
-        :rtype: Dict[str, List[str]]
+        :rtype: HeadlineQuestionsResponse
         """
         response = await self.client.request(
             method="GET",
             endpoint="/v1/chat/questions",
             query={"queries": queries}
         )
-        return response.content
+        return HeadlineQuestionsResponse.model_validate(response.content)

asknews_sdk/dto/chat.py

@@ -3,7 +3,7 @@
 from datetime import datetime, timezone
 from typing import Any, Dict, List, Literal, Optional, Union
 
-from pydantic import BaseModel, ConfigDict, Field
+from pydantic import BaseModel, ConfigDict, Field, RootModel
 from typing_extensions import Annotated
 
 from asknews_sdk.dto.base import BaseSchema
@@ -107,3 +107,7 @@ class ListModelResponse(BaseSchema):
 
     object: Annotated[Optional[str], Field("list", title="Object")]
     data: Annotated[List[ModelItem], Field(title="Data")]
+
+
+class HeadlineQuestionsResponse(BaseSchema, RootModel[Dict[str, List[str]]]):
+    ...