Skip to content

Commit 57a43bc

Browse files
committed
fix: update context retrieval for message list and tokens
1 parent dedec02 commit 57a43bc

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -211,9 +211,9 @@ def get_details(self, manage, **kwargs):
211211
'run_time': self.context.get('run_time') or 0,
212212
'model_id': str(manage.context['model_id']),
213213
'message_list': self.reset_message_list(self.context['step_args'].get('message_list'),
214-
self.context['answer_text']),
215-
'message_tokens': self.context['message_tokens'],
216-
'answer_tokens': self.context['answer_tokens'],
214+
self.context.get('answer_text')),
215+
'message_tokens': self.context.get('message_tokens'),
216+
'answer_tokens': self.context.get('answer_tokens'),
217217
'cost': 0,
218218
}
219219

0 commit comments

Comments
 (0)