Add support for models.search endpoint (#328)

See https://replicate.com/docs/reference/http#models.search

Signed-off-by: Mattt Zmuda <[email protected]>

Author: mattt

replicate/model.py

@@ -207,6 +207,46 @@ async def async_list(
 
         return Page[Model](**obj)
 
+    def search(self, query: str) -> Page[Model]:
+        """
+        Search for public models.
+
+        Parameters:
+            query: The search query.
+        Returns:
+            Page[Model]: A page of models matching the search query.
+        """
+        resp = self._client._request(
+            "QUERY", "/v1/models", content=query, headers={"Content-Type": "text/plain"}
+        )
+
+        obj = resp.json()
+        obj["results"] = [
+            _json_to_model(self._client, result) for result in obj["results"]
+        ]
+
+        return Page[Model](**obj)
+
+    async def async_search(self, query: str) -> Page[Model]:
+        """
+        Asynchronously search for public models.
+
+        Parameters:
+            query: The search query.
+        Returns:
+            Page[Model]: A page of models matching the search query.
+        """
+        resp = await self._client._async_request(
+            "QUERY", "/v1/models", content=query, headers={"Content-Type": "text/plain"}
+        )
+
+        obj = resp.json()
+        obj["results"] = [
+            _json_to_model(self._client, result) for result in obj["results"]
+        ]
+
+        return Page[Model](**obj)
+
     @overload
     def get(self, key: str) -> Model: ...