Skip to content

Commit 8ac52cf

Browse files
committed
fix: Add pagination to live web search
1 parent ec36fd1 commit 8ac52cf

File tree

2 files changed

+19
-2
lines changed

2 files changed

+19
-2
lines changed

asknews_sdk/api/chat.py

Lines changed: 18 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -242,6 +242,8 @@ def live_web_search(
242242
queries: List[str],
243243
lookback: Optional[int] = None,
244244
domains: Optional[List[str]] = None,
245+
strict: Optional[bool] = False,
246+
offset: Optional[int] = None,
245247
*,
246248
http_headers: Optional[Dict] = None,
247249
) -> WebSearchResponse:
@@ -255,7 +257,13 @@ def live_web_search(
255257
method="GET",
256258
endpoint="/v1/chat/websearch",
257259
headers=http_headers,
258-
query={"queries": queries, "lookback": lookback, "domains": domains},
260+
query={
261+
"queries": queries,
262+
"lookback": lookback,
263+
"domains": domains,
264+
"strict": strict,
265+
"offset": offset,
266+
},
259267
)
260268
return WebSearchResponse.model_validate(response.content)
261269

@@ -770,6 +778,8 @@ async def live_web_search(
770778
queries: List[str],
771779
lookback: Optional[int] = None,
772780
domains: Optional[List[str]] = None,
781+
strict: Optional[bool] = False,
782+
offset: Optional[int] = None,
773783
*,
774784
http_headers: Optional[Dict] = None,
775785
) -> WebSearchResponse:
@@ -783,7 +793,13 @@ async def live_web_search(
783793
method="GET",
784794
endpoint="/v1/chat/websearch",
785795
headers=http_headers,
786-
query={"queries": queries, "lookback": lookback, "domains": domains},
796+
query={
797+
"queries": queries,
798+
"lookback": lookback,
799+
"domains": domains,
800+
"strict": strict,
801+
"offset": offset,
802+
},
787803
)
788804
return WebSearchResponse.model_validate(response.content)
789805

asknews_sdk/dto/chat.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,7 @@ class WebSearchResult(BaseModel):
152152
class WebSearchResponse(BaseModel):
153153
as_string: str
154154
as_dicts: List[WebSearchResult]
155+
offset: Optional[int] = None
155156

156157

157158
class ForecastResponse(BaseModel):

0 commit comments

Comments
 (0)