feat: search support images return (#35)

Co-authored-by: jinhaiyang <[email protected]>

Author: tomsun28

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "zai-sdk"
-version = "0.0.3.5"
+version = "0.0.3.6"
 description = "A SDK library for accessing big model apis from Z.ai"
 authors = ["Z.ai"]
 readme = "README.md"

src/zai/_version.py

@@ -1,2 +1,2 @@
 __title__ = 'Z.ai'
-__version__ = '0.0.3.5'
+__version__ = '0.0.3.6'

src/zai/api_resource/web_search/web_search.py

@@ -42,7 +42,8 @@ def web_search(
 		search_recency_filter: Optional[str] | NotGiven = NOT_GIVEN,
 		content_size: Optional[str] | NotGiven = NOT_GIVEN,
 		search_intent: Optional[bool] | NotGiven = NOT_GIVEN,
-		extra_headers: Headers | None = None,
+        include_image: Optional[bool] | NotGiven = NOT_GIVEN,
+        extra_headers: Headers | None = None,
 		extra_body: Body | None = None,
 		timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
 	) -> WebSearchResp:
@@ -58,6 +59,7 @@ def web_search(
 				'search_recency_filter': search_recency_filter,
 				'content_size': content_size,
 				'search_intent': search_intent,
+                'include_image': include_image
 			}
 		)
 		return self._post(

src/zai/types/web_search/web_search_resp.py

@@ -1,4 +1,4 @@
-from typing import Optional
+from typing import Optional,List
 
 from zai.core import BaseModel
 
@@ -39,6 +39,7 @@ class SearchResultResp(BaseModel):
 	media: str
 	refer: str
 	publish_date: str
+	images: List[str]
 
 
 class WebSearchResp(BaseModel):