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