Using Request.Sessions for faster and less open connections

meilisearch/_httprequests.py

@@ -22,6 +22,7 @@ def __init__(self, config: Config) -> None:
             "Authorization": f"Bearer {self.config.api_key}",
             "User-Agent": _build_user_agent(config.client_agents),
         }
+        self.session = requests.Session()
 
     def send_request(
         self,
@@ -64,7 +65,7 @@ def send_request(
             raise MeilisearchCommunicationError(str(err)) from err
 
     def get(self, path: str) -> Any:
-        return self.send_request(requests.get, path)
+        return self.send_request(self.session.get, path)
 
     def post(
         self,
@@ -74,7 +75,7 @@ def post(
         ] = None,
         content_type: Optional[str] = "application/json",
     ) -> Any:
-        return self.send_request(requests.post, path, body, content_type)
+        return self.send_request(self.session.post, path, body, content_type)
 
     def patch(
         self,
@@ -84,7 +85,7 @@ def patch(
         ] = None,
         content_type: Optional[str] = "application/json",
     ) -> Any:
-        return self.send_request(requests.patch, path, body, content_type)
+        return self.send_request(self.session.patch, path, body, content_type)
 
     def put(
         self,
@@ -94,14 +95,14 @@ def put(
         ] = None,
         content_type: Optional[str] = "application/json",
     ) -> Any:
-        return self.send_request(requests.put, path, body, content_type)
+        return self.send_request(self.session.put, path, body, content_type)
 
     def delete(
         self,
         path: str,
         body: Optional[Union[Mapping[str, Any], Sequence[Mapping[str, Any]], List[str]]] = None,
     ) -> Any:
-        return self.send_request(requests.delete, path, body)
+        return self.send_request(self.session.delete, path, body)
 
     @staticmethod
     def __to_json(request: requests.Response) -> Any:

meilisearch/client.py

@@ -130,6 +130,7 @@ def get_indexes(self, parameters: Optional[Mapping[str, Any]] = None) -> Dict[st
                 index["primaryKey"],
                 index["createdAt"],
                 index["updatedAt"],
+                self.http,
             )
             for index in response["results"]
         ]
@@ -178,7 +179,7 @@ def get_index(self, uid: str) -> Index:
         MeilisearchApiError
             An error containing details about why Meilisearch can't process your request. Meilisearch error codes are described here: https://www.meilisearch.com/docs/reference/errors/error_codes#meilisearch-errors
         """
-        return Index(self.config, uid).fetch_info()
+        return Index(self.config, uid, http=self.http).fetch_info()
 
     def get_raw_index(self, uid: str) -> Dict[str, Any]:
         """Get the index as a dictionary.
@@ -216,7 +217,7 @@ def index(self, uid: str) -> Index:
             An Index instance.
         """
         if uid is not None:
-            return Index(self.config, uid=uid)
+            return Index(self.config, uid=uid, http=self.http)
         raise ValueError("The index UID should not be None")
 
     def multi_search(self, queries: Sequence[Mapping[str, Any]]) -> Dict[str, List[Dict[str, Any]]]:

meilisearch/index.py

@@ -31,6 +31,7 @@ def __init__(
         primary_key: Optional[str] = None,
         created_at: Optional[Union[datetime, str]] = None,
         updated_at: Optional[Union[datetime, str]] = None,
+        http: Optional[HttpRequests] = None,
     ) -> None:
         """
         Parameters
@@ -43,7 +44,7 @@ def __init__(
             Primary-key of the index.
         """
         self.config = config
-        self.http = HttpRequests(config)
+        self.http = http if http is not None else HttpRequests(config)
         self.task_handler = TaskHandler(config)
         self.uid = uid
         self.primary_key = primary_key

tests/errors/test_timeout_error_meilisearch.py

@@ -8,7 +8,7 @@
 from tests import BASE_URL, MASTER_KEY
 
 
-@patch("requests.get")
+@patch("requests.Session.get")
 def test_client_timeout_error(mock_get):
     mock_get.configure_mock(__name__="get")
     mock_get.side_effect = requests.exceptions.Timeout()