Fix: Add python client QueryResult model field definitions

streampipes-client-python/streampipes/endpoint/api/data_lake_measure.py

@@ -316,12 +316,21 @@ def _resource_cls(self) -> Type[QueryResult]:
         that the resource class of the resource container is
         the return type of the get endpoint.
         Therefore, this is only a temporary implementation and will be removed soon.
+
+        Returns
+        -------
+        [QueryResult][streampipes.model.resource.QueryResult]
         """
         return QueryResult
 
     @property
     def _container_cls(self) -> Type[ResourceContainer]:
-        """Defines the model container class the endpoint refers to."""
+        """Defines the model container class the endpoint refers to.
+
+        Returns
+        -------
+        [DataLakeMeasures][streampipes.model.container.DataLakeMeasures]
+        """
         return DataLakeMeasures
 
     @property
@@ -350,7 +359,7 @@ def get(self, identifier: str, **kwargs: Optional[Dict[str, Any]]) -> QueryResul
 
         Returns
         -------
-        measurement: DataLakeMeasures
+        measurement: QueryResult
             The specified data lake measure
 
         Examples

streampipes-client-python/streampipes/model/resource/query_result.py

@@ -16,7 +16,7 @@
 #
 
 from itertools import chain
-from typing import Any, Dict, List, Literal, Union
+from typing import Any, Dict, List, Literal, Optional, Union
 
 import pandas as pd
 from pydantic.v1 import Field, StrictInt, StrictStr
@@ -73,6 +73,9 @@ def convert_to_pandas_representation(self) -> Dict[str, Union[List[str], List[Li
     headers: List[StrictStr]
     all_data_series: List[DataSeries]
     query_status: Literal["OK", "TOO_MUCH_DATA"] = Field(alias="spQueryStatus")
+    source_index: StrictInt
+    for_id: Optional[str]
+    last_timestamp: StrictInt
 
     def to_pandas(self) -> pd.DataFrame:
         """Returns the data lake series in representation of a Pandas Dataframe.

streampipes-client-python/tests/client/test_data_lake_series.py

@@ -103,6 +103,9 @@ def test_to_pandas(self, server_version: MagicMock, http_session: MagicMock):
             "headers": self.headers,
             "spQueryStatus": "OK",
             "allDataSeries": [self.data_series],
+            "sourceIndex": 0,
+            "forId": None,
+            "lastTimestamp": 1717936808802,
         }
 
         result_pd = self.get_result_as_panda(http_session, query_result)
@@ -125,6 +128,9 @@ def test_group_by_to_pandas(self, server_version: MagicMock, http_session: Magic
             "headers": self.headers,
             "spQueryStatus": "OK",
             "allDataSeries": [self.data_series, self.data_series],
+            "sourceIndex": 0,
+            "forId": None,
+            "lastTimestamp": 1717936808802,
         }
 
         result_pd = self.get_result_as_panda(http_session, query_result)
@@ -147,6 +153,9 @@ def test_different_headers_exception(self, server_version: MagicMock, http_sessi
             "headers": ["one"],
             "spQueryStatus": "OK",
             "allDataSeries": [self.data_series],
+            "sourceIndex": 0,
+            "forId": None,
+            "lastTimestamp": 1717936808802,
         }
 
         with self.assertRaises(StreamPipesUnsupportedDataSeries):