fix: add forecast endpoint, bump httpx

Author: robcaulk

asknews_sdk/api/chat.py

@@ -5,6 +5,7 @@
     CreateChatCompletionRequest,
     CreateChatCompletionResponse,
     CreateChatCompletionResponseStream,
+    ForecastResponse,
     HeadlineQuestionsResponse,
     ListModelResponse,
 )
@@ -145,6 +146,36 @@ def get_headline_questions(
         )
         return HeadlineQuestionsResponse.model_validate(response.content)
 
+    def get_forecast(
+        self,
+        query: str,
+        lookback: int = 7,
+        articles_per_day: int = 1,
+        method: Literal["nl", "kw", "both"] = "kw",
+        model: Literal["gpt-4o"] = "gpt-4o",
+        *,
+        http_headers: Optional[Dict] = None,
+    ) -> ForecastResponse:
+        """
+        Get an expert forecast, complete with full sources
+        and reasoning.
+
+        https://docs.asknews.app/en/reference#get-/v1/chat/forecast
+        """
+        response = self.client.request(
+            method="GET",
+            endpoint="/v1/chat/forecast",
+            headers=http_headers,
+            query={
+                "query": query,
+                "lookback": lookback,
+                "articles_per_day": articles_per_day,
+                "method": method,
+                "model": model,
+            },
+        )
+        return ForecastResponse.model_validate(response.content)
+
 
 class AsyncChatAPI(BaseAPI):
     """
@@ -280,3 +311,33 @@ async def get_headline_questions(
             query={"queries": queries},
         )
         return HeadlineQuestionsResponse.model_validate(response.content)
+
+    async def get_forecast(
+        self,
+        query: str,
+        lookback: int = 7,
+        articles_per_day: int = 1,
+        method: Literal["nl", "kw", "both"] = "kw",
+        model: Literal["gpt-4o"] = "gpt-4o",
+        *,
+        http_headers: Optional[Dict] = None,
+    ) -> ForecastResponse:
+        """
+        Get an expert forecast, complete with full sources
+        and reasoning.
+
+        https://docs.asknews.app/en/reference#get-/v1/chat/forecast
+        """
+        response = await self.client.request(
+            method="GET",
+            endpoint="/v1/chat/forecast",
+            headers=http_headers,
+            query={
+                "query": query,
+                "lookback": lookback,
+                "articles_per_day": articles_per_day,
+                "method": method,
+                "model": model,
+            },
+        )
+        return ForecastResponse.model_validate(response.content)

asknews_sdk/api/stories.py

@@ -62,6 +62,9 @@ def search_stories(
         obj_type: Optional[List[Literal["story", "story_update"]]] = None,
         provocative: Literal["unknown", "low", "medium", "high", "all"] = "all",
         citation_method: Literal["brackets", "urls", "none"] = "brackets",
+        strategy: Literal[
+            "default", "topstories", "topstories_continents", "topstories_categories"
+        ] = "default",
         *,
         http_headers: Optional[Dict] = None,
     ) -> StoriesResponse:
@@ -129,6 +132,7 @@ def search_stories(
                 "uuids": uuids,
                 "provocative": provocative,
                 "citation_method": citation_method,
+                "strategy": strategy,
             },
             headers=http_headers,
             accept=[(StoriesResponse.__content_type__, 1.0)],
@@ -247,6 +251,9 @@ async def search_stories(
         obj_type: Optional[List[Literal["story", "story_update"]]] = None,
         provocative: Literal["unknown", "low", "medium", "high", "all"] = "all",
         citation_method: Literal["brackets", "urls", "none"] = "brackets",
+        strategy: Literal[
+            "default", "topstories", "topstories_continents", "topstories_categories"
+        ] = "default",
         *,
         http_headers: Optional[Dict] = None,
     ) -> StoriesResponse:
@@ -314,6 +321,7 @@ async def search_stories(
                 "uuids": uuids,
                 "provocative": provocative,
                 "citation_method": citation_method,
+                "strategy": strategy,
             },
             headers=http_headers,
             accept=[(StoriesResponse.__content_type__, 1.0)],

asknews_sdk/dto/chat.py

@@ -7,6 +7,7 @@
 from typing_extensions import Annotated
 
 from asknews_sdk.dto.base import BaseSchema
+from asknews_sdk.dto.news import SearchResponseDictItem
 
 
 class CreateChatCompletionRequestMessage(BaseModel):
@@ -109,5 +110,14 @@ class ListModelResponse(BaseSchema):
     data: Annotated[List[ModelItem], Field(title="Data")]
 
 
-class HeadlineQuestionsResponse(BaseSchema, RootModel[Dict[str, List[str]]]):
-    ...
+class HeadlineQuestionsResponse(BaseSchema, RootModel[Dict[str, List[str]]]): ...
+
+
+class ForecastResponse(BaseModel):
+    forecast: str
+    resolution_criteria: str
+    date: datetime
+    reasoning: str
+    sources: list[SearchResponseDictItem]
+    timeline: list[str]
+    opposite_request: str

asknews_sdk/dto/stories.py

@@ -10,7 +10,7 @@
 
 
 class RedditPerspective(BaseModel):
-    sentiment: Annotated[int, Field(title="Sentiment")]
+    sentiment: Annotated[Union[int, float], Field(title="Sentiment")]
     relevant: Annotated[bool, Field(title="Relevant")]
     summary: Annotated[str, Field(title="Summary")]
 
@@ -28,7 +28,7 @@ class RedditThread(BaseModel):
     id: Annotated[UUID, Field(title="Id")]
     key_takeaways: Annotated[List[str], Field(title="Key Takeaways")]
     keywords: Annotated[List[str], Field(title="Keywords")]
-    sentiment: Annotated[int, Field(title="Sentiment")]
+    sentiment: Annotated[Union[int, float], Field(title="Sentiment")]
     subreddit_name: Annotated[str, Field(title="Subreddit Name")]
     subreddit_url: Annotated[str, Field(title="Subreddit Url")]
     summary: Annotated[str, Field(title="Summary")]
@@ -141,4 +141,4 @@ class StoryResponse(BaseSchema):
 
 class StoriesResponse(BaseSchema):
     stories: Annotated[List[StoryResponse], Field(title="Stories")]
-    offset: Annotated[Union[int, str], Field(title="Offset")]
+    offset: Annotated[Optional[Union[int, str]], Field(title="Offset")]