Skip to content

Commit cec400a

Browse files
seanzhougooglecopybara-github
authored andcommitted
chore: Autoformat history_management agent codes
PiperOrigin-RevId: 785003632
1 parent d2461ec commit cec400a

1 file changed

Lines changed: 8 additions & 2 deletions

File tree

  • contributing/samples/history_management

contributing/samples/history_management/agent.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -66,11 +66,17 @@ async def check_prime(nums: list[int]) -> str:
6666

6767

6868
def create_slice_history_callback(n_recent_turns):
69-
async def before_model_callback(callback_context: CallbackContext, llm_request: LlmRequest):
69+
async def before_model_callback(
70+
callback_context: CallbackContext, llm_request: LlmRequest
71+
):
7072
if n_recent_turns < 1:
7173
return
7274

73-
user_indexes = [i for i, content in enumerate(llm_request.contents) if content.role == "user"]
75+
user_indexes = [
76+
i
77+
for i, content in enumerate(llm_request.contents)
78+
if content.role == 'user'
79+
]
7480

7581
if n_recent_turns > len(user_indexes):
7682
return

0 commit comments

Comments
 (0)