Skip to content

Commit 697bc5b

Browse files
committed
feat(context_tracker): Initialize workflow_metadata to store tool statistics
1 parent 96b5071 commit 697bc5b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ajet/context_tracker/base_tracker.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from typing import List, Tuple, Union
2+
from typing import List, Union, Tuple, Dict, Optional, Any
23

34
from ajet.schema.extended_msg import (
45
INVALID_LOG_PROB_VALUE,
@@ -135,6 +136,7 @@ def __init__(self, config, tokenizer, **kwargs):
135136
self.already_mad_flag: bool = False
136137
self.round_cnt = 0
137138
self.generation_prompt_token = None
139+
self.workflow_metadata: Optional[Dict[str, Any]] = None # 初始化 workflow_metadata 以存储工具统计信息
138140

139141
assert (
140142
self.config.ajet.data.max_prompt_length

0 commit comments

Comments
 (0)