feat: Add orderBy to ListMemories

PiperOrigin-RevId: 804536990

Author: vertex-sdk-bot

tests/unit/vertexai/genai/replays/test_list_agent_engine_memories.py

@@ -21,20 +21,37 @@
 def test_list_memories(client):
     agent_engine = client.agent_engines.create()
     assert not list(
-        client.agent_engines.list_memories(
+        client.agent_engines.memories.list(
             name=agent_engine.api_resource.name,
         )
     )
-    client.agent_engines.create_memory(
+    client.agent_engines.memories.create(
         name=agent_engine.api_resource.name,
         fact="memory_fact",
         scope={"user_id": "123"},
+        config={
+            "wait_for_completion": True,
+        },
     )
-    memory_list = client.agent_engines.list_memories(
+    client.agent_engines.memories.create(
         name=agent_engine.api_resource.name,
+        fact="memory_fact_2",
+        scope={"user_id": "456"},
+        config={
+            "wait_for_completion": True,
+        },
+    )
+    memory_list = client.agent_engines.memories.list(
+        name=agent_engine.api_resource.name,
+        config=types.ListAgentEngineMemoryConfig(
+            page_size=1,
+            order_by="create_time asc",
+        ),
     )
     assert len(memory_list) == 1
     assert isinstance(memory_list[0], types.Memory)
+    assert memory_list[0].fact == "memory_fact"
+    assert memory_list[0].scope["user_id"] == "123"
     # Clean up resources.
     agent_engine.delete(force=True)
 

vertexai/_genai/memories.py

@@ -249,6 +249,9 @@ def _ListAgentEngineMemoryConfig_to_vertex(
     if getv(from_object, ["filter"]) is not None:
         setv(parent_object, ["_query", "filter"], getv(from_object, ["filter"]))
 
+    if getv(from_object, ["order_by"]) is not None:
+        setv(parent_object, ["_query", "orderBy"], getv(from_object, ["order_by"]))
+
     return to_object
 
 

vertexai/_genai/types.py

@@ -5214,6 +5214,17 @@ class ListAgentEngineMemoryConfig(_common.BaseModel):
         description="""An expression for filtering the results of the request.
       For field names both snake_case and camelCase are supported.""",
     )
+    order_by: Optional[str] = Field(
+        default=None,
+        description="""The standard list order by string. If not specified, the default
+      order is `create_time desc`. If specified, the default sorting order of
+      provided fields is ascending. More detail in
+      [AIP-132](https://google.aip.dev/132).
+
+      Supported fields:
+      * `create_time`
+      * `update_time`""",
+    )
 
 
 class ListAgentEngineMemoryConfigDict(TypedDict, total=False):
@@ -5232,6 +5243,16 @@ class ListAgentEngineMemoryConfigDict(TypedDict, total=False):
     """An expression for filtering the results of the request.
       For field names both snake_case and camelCase are supported."""
 
+    order_by: Optional[str]
+    """The standard list order by string. If not specified, the default
+      order is `create_time desc`. If specified, the default sorting order of
+      provided fields is ascending. More detail in
+      [AIP-132](https://google.aip.dev/132).
+
+      Supported fields:
+      * `create_time`
+      * `update_time`"""
+
 
 ListAgentEngineMemoryConfigOrDict = Union[
     ListAgentEngineMemoryConfig, ListAgentEngineMemoryConfigDict