feat: Add charts 🎉

Author: robcaulk

asknews_sdk/api/chat.py

@@ -7,6 +7,8 @@
 from asknews_sdk.client import APIClient, AsyncAPIClient
 from asknews_sdk.dto.alert import AlertLog, AlertResponse, CreateAlertRequest, UpdateAlertRequest
 from asknews_sdk.dto.chat import (
+    ChartResponse,
+    CreateChartRequest,
     CreateChatCompletionRequest,
     CreateChatCompletionResponse,
     CreateChatCompletionResponseStream,
@@ -643,6 +645,32 @@ def _stream():
         else:
             return CreateDeepNewsResponse.model_validate(response.content)
 
+    def create_chart(
+        self,
+        payload: CreateChartRequest,
+        *,
+        http_headers: Optional[Dict] = None,
+    ) -> ChartResponse:
+        """
+        Create a chart based on news data using a natural language query
+        """
+
+        response = self.client.request(
+            method="POST",
+            endpoint="/v1/chat/charts",
+            body=payload.model_dump(mode="json")
+            if isinstance(payload, CreateChartRequest)
+            else payload,
+            headers={
+                **(http_headers or {}),
+                "Content-Type": CreateChartRequest.__content_type__,
+            },
+            accept=[
+                (ChartResponse.__content_type__, 1.0),
+            ],
+        )
+        return ChartResponse.model_validate(response.content)
+
 
 class AsyncChatAPI(BaseAPI[AsyncAPIClient]):
     """
@@ -1231,3 +1259,29 @@ async def _stream():
             return _stream()
         else:
             return CreateDeepNewsResponse.model_validate(response.content)
+
+    async def create_chart(
+        self,
+        payload: CreateChartRequest,
+        *,
+        http_headers: Optional[Dict] = None,
+    ) -> ChartResponse:
+        """
+        Create a chart based on news data using a natural language query
+        """
+
+        response = await self.client.request(
+            method="POST",
+            endpoint="/v1/chat/charts",
+            body=payload.model_dump(mode="json")
+            if isinstance(payload, CreateChartRequest)
+            else payload,
+            headers={
+                **(http_headers or {}),
+                "Content-Type": CreateChartRequest.__content_type__,
+            },
+            accept=[
+                (ChartResponse.__content_type__, 1.0),
+            ],
+        )
+        return ChartResponse.model_validate(response.content)

asknews_sdk/api/news.py

@@ -313,7 +313,6 @@ def get_index_counts(
         :return: The index counts response.
         :rtype: IndexCountsResponse
         """
-        print(f" Datetimes {start_datetime.isoformat()}, {end_datetime.isoformat()}")
         response = self.client.request(
             method="GET",
             endpoint="/v1/index_counts",

asknews_sdk/dto/alert.py

@@ -31,6 +31,7 @@
     "o3-mini",
     "claude-3-5-sonnet-latest",
     "claude-sonnet-4-20250514",
+    "claude-sonnet-4-5-20250929",
     "meta-llama/Meta-Llama-3.1-405B-Instruct",
     "meta-llama/Meta-Llama-3.3-70B-Instruct",
 ]

asknews_sdk/dto/chat.py

@@ -66,12 +66,10 @@ class CreateChatCompletionRequest(BaseSchema):
     asknews_watermark: Annotated[
         Optional[bool], Field(title='Append "Generated by AskNews AI" watermark')
     ] = True
-    append_references: Annotated[
-        Optional[bool], Field(title="Append References or not")
-    ] = True
-    conversational_awareness: Annotated[
-        Optional[bool], Field(title="Conversational Awareness")
-    ] = False
+    append_references: Annotated[Optional[bool], Field(title="Append References or not")] = True
+    conversational_awareness: Annotated[Optional[bool], Field(title="Conversational Awareness")] = (
+        False
+    )
     filter_params: Annotated[
         Optional[Dict[str, Any]],
         Field(title="Any filter param available on the /news endpoint."),
@@ -327,3 +325,53 @@ class FilterParamsMetadata(BaseModel):
 class FilterParamsResponse(BaseModel):
     metadata: FilterParamsMetadata
     filter_params: FilterParams
+
+
+class SeriesConfig(BaseModel):
+    name: str
+    start_datetime: datetime
+    end_datetime: datetime
+    sampling: Optional[str] = "1d"
+    time_filter: Optional[str] = "pub_date"
+    categories: Optional[List[str]] = None
+    domains: Optional[List[str]] = None
+    bad_domain_url: Optional[List[str]] = None
+    page_rank: Optional[int] = None
+    string_guarantee: Optional[List[str]] = None
+    string_guarantee_op: Optional[str] = "OR"
+    reverse_string_guarantee: Optional[List[str]] = None
+    entity_guarantee: Optional[List[str]] = None
+    entity_guarantee_op: Optional[str] = "OR"
+    countries: Optional[List[str]] = None
+    countries_blacklist: Optional[List[str]] = None
+    continents: Optional[List[str]] = None
+    languages: Optional[List[str]] = None
+    sentiment: Optional[str] = None
+    reporting_voice: Optional[List[str]] = None
+    provocative: Optional[str] = "all"
+
+
+class ChartParameters(BaseModel):
+    chart_type: Literal["line", "bar", "bar_pct"] = Field(
+        ..., description="The type of chart to create"
+    )
+    title: Optional[str] = Field(None, description="The title of the chart")
+    series: List[SeriesConfig] = Field(..., description="The series configurations for the chart")
+
+
+class CreateChartRequest(BaseSchema):
+    query: str = Field(..., description="The chart query to create")
+
+
+class ChartResponse(BaseSchema):
+    success: bool
+    chart_json: Annotated[
+        dict, Field(description="The plotly fig.to_json() JSON representation of the chart")
+    ]
+    query: str = None
+    reasoning: Optional[str] = None
+    parameters: Optional[ChartParameters] = None
+    context: Optional[dict] = None
+    iterations_used: Optional[int] = None
+    last_reasoning: Optional[str] = None
+    last_parameters: Optional[ChartParameters] = None