33from __future__ import annotations
44
55import os
6- from typing import TYPE_CHECKING , Any , Dict , Union , Mapping
6+ from typing import TYPE_CHECKING , Any , Dict , List , Union , Mapping
77from typing_extensions import Self , Literal , override
88
99import httpx
@@ -334,7 +334,9 @@ def profile(
334334 self ,
335335 * ,
336336 container_tag : str ,
337+ buckets : SequenceNotStr [str ] | Omit = omit ,
337338 filters : client_profile_params .Filters | Omit = omit ,
339+ include : List [Literal ["static" , "dynamic" , "buckets" ]] | Omit = omit ,
338340 q : str | Omit = omit ,
339341 threshold : float | Omit = omit ,
340342 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -351,9 +353,15 @@ def profile(
351353 container_tag: Tag to filter the profile by. This can be an ID for your user, a project ID, or
352354 any other identifier you wish to use to filter memories.
353355
356+ buckets: Specific bucket keys to return. Omit to return all configured buckets. Only
357+ relevant when "buckets" is included.
358+
354359 filters: Optional metadata filters to apply to profile results and search results.
355360 Supports complex AND/OR queries with multiple conditions.
356361
362+ include: Profile sections to return. Omit to return all sections. Pass a subset to reduce
363+ payload — e.g. ["buckets"] skips static and dynamic entirely.
364+
357365 q: Optional search query to include search results in the response
358366
359367 threshold: Threshold for search results. Only results with a score above this threshold
@@ -372,7 +380,9 @@ def profile(
372380 body = maybe_transform (
373381 {
374382 "container_tag" : container_tag ,
383+ "buckets" : buckets ,
375384 "filters" : filters ,
385+ "include" : include ,
376386 "q" : q ,
377387 "threshold" : threshold ,
378388 },
@@ -681,7 +691,9 @@ async def profile(
681691 self ,
682692 * ,
683693 container_tag : str ,
694+ buckets : SequenceNotStr [str ] | Omit = omit ,
684695 filters : client_profile_params .Filters | Omit = omit ,
696+ include : List [Literal ["static" , "dynamic" , "buckets" ]] | Omit = omit ,
685697 q : str | Omit = omit ,
686698 threshold : float | Omit = omit ,
687699 # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -698,9 +710,15 @@ async def profile(
698710 container_tag: Tag to filter the profile by. This can be an ID for your user, a project ID, or
699711 any other identifier you wish to use to filter memories.
700712
713+ buckets: Specific bucket keys to return. Omit to return all configured buckets. Only
714+ relevant when "buckets" is included.
715+
701716 filters: Optional metadata filters to apply to profile results and search results.
702717 Supports complex AND/OR queries with multiple conditions.
703718
719+ include: Profile sections to return. Omit to return all sections. Pass a subset to reduce
720+ payload — e.g. ["buckets"] skips static and dynamic entirely.
721+
704722 q: Optional search query to include search results in the response
705723
706724 threshold: Threshold for search results. Only results with a score above this threshold
@@ -719,7 +737,9 @@ async def profile(
719737 body = await async_maybe_transform (
720738 {
721739 "container_tag" : container_tag ,
740+ "buckets" : buckets ,
722741 "filters" : filters ,
742+ "include" : include ,
723743 "q" : q ,
724744 "threshold" : threshold ,
725745 },
0 commit comments