Feat: 5530 feature updated at inserted at properties on retrieved records (#5540)

# Description
<!-- Please include a summary of the changes and the related issue.
Please also include relevant motivation and context. List any
dependencies that are required for this change. -->

Closes #5530 

**Type of change**
<!-- Please delete options that are not relevant. Remember to title the
PR according to the type of change -->

- Improvement (change adding some improvement to an existing
functionality)

**How Has This Been Tested**
<!-- Please add some reference about how your feature has been tested.
-->

**Checklist**
<!-- Please go over the list and make sure you've taken everything into
account -->

- I added relevant documentation
- I followed the style guidelines of this project
- I did a self-review of my code
- I made corresponding changes to the documentation
- I confirm My changes generate no new warnings
- I have added tests that prove my fix is effective or that my feature
works
- I have added relevant notes to the CHANGELOG.md file (See
https://keepachangelog.com/)

---------

Co-authored-by: Francisco Aranda <[email protected]>
Co-authored-by: Paco Aranda <[email protected]>

Author: davidberenstein1957

argilla-server/src/argilla_server/api/schemas/v1/records.py

@@ -283,8 +283,8 @@ class TermsFilter(BaseModel):
 class RangeFilter(BaseModel):
     type: Literal["range"]
     scope: FilterScope
-    ge: Optional[float]
-    le: Optional[float]
+    ge: Optional[Union[float, str]]
+    le: Optional[Union[float, str]]
 
     @root_validator(skip_on_failure=True)
     def check_ge_and_le(cls, values: dict) -> dict:

argilla/CHANGELOG.md

@@ -18,6 +18,7 @@ These are the section headers that we use:
 
 ### Added
 
+- Added `inserted_at` and `updated_at` to `Resource` model as properties. ([#5540](https://github.com/argilla-io/argilla/pull/5540))
 - Added `limit` argument when fetching records. ([#5525](https://github.com/argilla-io/argilla/pull/5525)
 
 ### Changed

argilla/src/argilla/_resource.py

@@ -13,7 +13,7 @@
 # limitations under the License.
 from abc import abstractmethod
 from datetime import datetime
-from typing import Any, TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Any, Optional
 from uuid import UUID
 
 try:
@@ -25,9 +25,9 @@
 from argilla._helpers import LoggingMixin
 
 if TYPE_CHECKING:
-    from argilla.client import Argilla
-    from argilla._models import ResourceModel
     from argilla._api._base import ResourceAPI
+    from argilla._models import ResourceModel
+    from argilla.client import Argilla
 
 
 class Resource(LoggingMixin):
@@ -63,6 +63,14 @@ def id(self) -> Optional[UUID]:
     def id(self, value: UUID) -> None:
         self._model.id = value
 
+    @property
+    def inserted_at(self) -> datetime:
+        return self._model.inserted_at
+
+    @property
+    def updated_at(self) -> datetime:
+        return self._model.updated_at
+
     @property
     def is_outdated(self) -> bool:
         """Checks if the resource is outdated based on the last API call

argilla/src/argilla/records/_resource.py

@@ -183,6 +183,7 @@ def serialize(self) -> Dict[str, Any]:
         serialized_responses = [response.serialize() for response in self.__responses]
         serialized_model["responses"] = serialized_responses
         serialized_model["suggestions"] = serialized_suggestions
+
         return serialized_model
 
     def to_dict(self) -> Dict[str, Dict]:
@@ -202,6 +203,7 @@ def to_dict(self) -> Dict[str, Dict]:
         responses = self.responses.to_dict()
         vectors = self.vectors.to_dict()
 
+        # TODO: Review model attributes when to_dict and serialize methods are unified
         return {
             "id": id,
             "fields": fields,
@@ -268,8 +270,9 @@ def from_model(cls, model: RecordModel, dataset: "Dataset") -> "Record":
         )
 
         # set private attributes
-        instance._model.id = model.id
-        instance._model.status = model.status
+        instance._dataset = dataset
+        instance._model = model
+
         # Responses and suggestions are computed separately based on the record model
         instance.responses.from_models(model.responses)
         instance.suggestions.from_models(model.suggestions)

argilla/tests/integration/test_list_records.py

@@ -193,3 +193,21 @@ def test_list_records_with_responses(client: Argilla, dataset: Dataset):
 
     assert records[1].responses["comment"][0].value == "The comment"
     assert records[1].responses["sentiment"][0].value == "negative"
+
+
+def test_list_records_with_updated_at_and_inserted_at(client: Argilla, dataset: Dataset):
+    dataset.records.log(
+        [
+            {"text": "The record text field", "id": 1},
+            {"text": "The record text field", "id": 2},
+        ]
+    )
+
+    records = list(dataset.records(with_responses=True))
+    assert len(records) == 2
+
+    assert records[0].inserted_at
+    assert records[0].updated_at
+
+    assert records[1].inserted_at
+    assert records[1].updated_at

argilla/tests/unit/test_resources/test_records.py

@@ -16,7 +16,7 @@
 
 import pytest
 
-from argilla import Record, Response, Suggestion, Dataset, Settings, TextQuestion, TextField
+from argilla import Dataset, Record, Response, Settings, Suggestion, TextField, TextQuestion
 from argilla._exceptions import ArgillaError
 from argilla._models import MetadataModel, RecordModel
 
@@ -61,6 +61,14 @@ def test_record_external_id(self):
         record = Record(fields={"name": "John", "age": "30"})
         assert record.id
 
+    def test_record_inserted_at(self):
+        record = Record(fields={"name": "John", "age": "30"})
+        assert hasattr(record, "inserted_at")
+
+    def test_record_updated_at(self):
+        record = Record(fields={"name": "John", "age": "30"})
+        assert hasattr(record, "updated_at")
+
     def test_update_record_metadata_by_key(self):
         record = Record(fields={"name": "John", "age": "30"}, metadata={"key": "value"})
 

argilla/tests/unit/test_resources/test_users.py

@@ -16,9 +16,11 @@
 import uuid
 from datetime import datetime
 
-import argilla as rg
 import httpx
 import pytest
+from pytest_httpx import HTTPXMock
+
+import argilla as rg
 from argilla._exceptions import (
     BadRequestError,
     ConflictError,
@@ -28,7 +30,6 @@
     UnprocessableEntityError,
 )
 from argilla._models import UserModel
-from pytest_httpx import HTTPXMock
 
 
 class TestUserSerialization:
@@ -113,6 +114,11 @@ def test_create_user(self, httpx_mock: HTTPXMock, status_code, expected_exceptio
                 created_user = user.create()
                 assert user.username == created_user.username
                 assert user.id == created_user.id
+                assert user.first_name == created_user.first_name
+                assert user.last_name == created_user.last_name
+                assert user.role == created_user.role
+                assert user.inserted_at == created_user.inserted_at
+                assert user.updated_at == created_user.updated_at
 
     @pytest.mark.parametrize(
         "status_code, expected_exception, expected_message",
@@ -156,6 +162,11 @@ def test_get_user(self, httpx_mock: HTTPXMock, status_code, expected_exception,
                 gotten_user = user.create()
                 assert user.username == gotten_user.username
                 assert user.id == gotten_user.id
+                assert user.first_name == gotten_user.first_name
+                assert user.last_name == gotten_user.last_name
+                assert user.role == gotten_user.role
+                assert user.inserted_at == gotten_user.inserted_at
+                assert user.updated_at == gotten_user.updated_at
 
     def test_list_users(self, httpx_mock: HTTPXMock):
         mock_return_value = {
@@ -190,6 +201,10 @@ def test_list_users(self, httpx_mock: HTTPXMock):
                 assert users[i].username == mock_return_value["items"][i]["username"]
                 assert users[i].role == mock_return_value["items"][i]["role"]
                 assert users[i].id == uuid.UUID(mock_return_value["items"][i]["id"])
+                assert users[i].first_name == mock_return_value["items"][i]["first_name"]
+                assert users[i].last_name == mock_return_value["items"][i]["last_name"]
+                assert users[i].inserted_at == datetime.fromisoformat(mock_return_value["items"][i]["inserted_at"])
+                assert users[i].updated_at == datetime.fromisoformat(mock_return_value["items"][i]["updated_at"])
 
 
 class TestUsersAPI:
@@ -212,6 +227,10 @@ def test_get_me(self, httpx_mock: HTTPXMock):
             assert user.username == mock_return_value["username"]
             assert user.id == uuid.UUID(mock_return_value["id"])
             assert user.role == mock_return_value["role"]
+            assert user.first_name == mock_return_value["first_name"]
+            assert user.last_name == mock_return_value["last_name"]
+            assert user.inserted_at == datetime.fromisoformat(mock_return_value["inserted_at"])
+            assert user.updated_at == datetime.fromisoformat(mock_return_value["updated_at"])
 
     def test_remove_user_from_workspace(self, httpx_mock: HTTPXMock):
         user_id = uuid.uuid4()
@@ -275,6 +294,11 @@ def test_list_workspace_users(self, httpx_mock: HTTPXMock):
             for i in range(len(users)):
                 assert users[i].username == mock_return_value["items"][i]["username"]
                 assert users[i].id == uuid.UUID(mock_return_value["items"][i]["id"])
+                assert users[i].first_name == mock_return_value["items"][i]["first_name"]
+                assert users[i].last_name == mock_return_value["items"][i]["last_name"]
+                assert users[i].role == mock_return_value["items"][i]["role"]
+                assert users[i].inserted_at == datetime.fromisoformat(mock_return_value["items"][i]["inserted_at"])
+                assert users[i].updated_at == datetime.fromisoformat(mock_return_value["items"][i]["updated_at"])
 
     def test_create_user(self, httpx_mock: HTTPXMock):
         user_id = uuid.uuid4()

argilla/tests/unit/test_resources/test_workspaces.py

@@ -83,6 +83,8 @@ def test_create_workspace(self, httpx_mock: HTTPXMock, status_code, expected_exc
                 created_workspace = ws.create()
                 assert created_workspace.name == mock_name
                 assert created_workspace.id == uuid.UUID(mock_return_value["id"])
+                assert created_workspace.inserted_at == datetime.fromisoformat(mock_return_value["inserted_at"])
+                assert created_workspace.updated_at == datetime.fromisoformat(mock_return_value["updated_at"])
 
     @pytest.mark.parametrize(
         "status_code, expected_exception, expected_message",
@@ -147,6 +149,8 @@ def test_list_workspaces(self, httpx_mock: HTTPXMock):
         for i in range(len(workspaces)):
             assert workspaces[i].name == mock_return_value["items"][i]["name"]
             assert workspaces[i].id == uuid.UUID(mock_return_value["items"][i]["id"])
+            assert workspaces[i].inserted_at == datetime.fromisoformat(mock_return_value["items"][i]["inserted_at"])
+            assert workspaces[i].updated_at == datetime.fromisoformat(mock_return_value["items"][i]["updated_at"])
 
 
 class TestWorkspacesAPI:
@@ -175,6 +179,8 @@ def test_get_workspace_by_name(self, httpx_mock: HTTPXMock):
             assert ws is not None
             assert ws.name == "test-workspace"
             assert ws.id == uuid.UUID(mock_return_value["items"][0]["id"])
+            assert ws.inserted_at == datetime.fromisoformat(mock_return_value["items"][0]["inserted_at"])
+            assert ws.updated_at == datetime.fromisoformat(mock_return_value["items"][0]["updated_at"])
 
     def test_multiple_clients_create_workspace(self, httpx_mock: HTTPXMock):
         mock_uuid = str(uuid.uuid4())
@@ -246,3 +252,8 @@ def test_list_workspace_datasets(self, httpx_mock: HTTPXMock):
             for i in range(len(datasets)):
                 assert datasets[i].name == mock_return_value["items"][i]["name"]
                 assert datasets[i].id == uuid.UUID(mock_return_value["items"][i]["id"])
+                assert datasets[i].guidelines == mock_return_value["items"][i]["guidelines"]
+                assert datasets[i].allow_extra_metadata == mock_return_value["items"][i]["allow_extra_metadata"]
+                assert datasets[i].workspace_id == uuid.UUID(mock_return_value["items"][i]["workspace_id"])
+                assert datasets[i].inserted_at == datetime.fromisoformat(mock_return_value["items"][i]["inserted_at"])
+                assert datasets[i].updated_at == datetime.fromisoformat(mock_return_value["items"][i]["updated_at"])