Enhance ElasticsearchClient to support sorting in search_by_match method and update LogsServer to utilize this feature for retrieving logs sorted by timestamp.

Author: Dylan Huang

eval_protocol/log_utils/elasticsearch_client.py

@@ -216,19 +216,22 @@ def search_by_term(self, field: str, value: Any, size: int = 10) -> Optional[Dic
         query = {"term": {field: value}}
         return self.search(query, size=size)
 
-    def search_by_match(self, field: str, value: str, size: int = 10) -> Optional[Dict[str, Any]]:
+    def search_by_match(
+        self, field: str, value: str, size: int = 10, sort: Optional[List[Dict[str, Any]]] = None
+    ) -> Optional[Dict[str, Any]]:
         """Search documents by text match.
 
         Args:
             field: Field name to search
             value: Text to match
             size: Number of results to return
+            sort: Sort specification (e.g., [{"@timestamp": {"order": "desc"}}])
 
         Returns:
             Dict containing search results, or None if failed
         """
         query = {"match": {field: value}}
-        return self.search(query, size=size)
+        return self.search(query, size=size, sort=sort)
 
     def search_by_match_phrase_prefix(self, field: str, value: str, size: int = 10) -> Optional[Dict[str, Any]]:
         """Search documents by phrase prefix match.

eval_protocol/utils/logs_server.py

@@ -351,8 +351,11 @@ async def get_logs(
                 raise HTTPException(status_code=503, detail="Elasticsearch is not configured for this logs server")
 
             try:
-                # Search for logs by rollout_id
-                search_results = self.elasticsearch_client.search_by_match("rollout_id", rollout_id, size=limit)
+                # Search for logs by rollout_id, sorted by timestamp in descending order (newest first)
+                sort_spec = [{"@timestamp": {"order": "desc"}}]
+                search_results = self.elasticsearch_client.search_by_match(
+                    "rollout_id", rollout_id, size=limit, sort=sort_spec
+                )
 
                 if not search_results or "hits" not in search_results:
                     # Return empty response using Pydantic model