mirror of
https://github.com/bytedance/deer-flow.git
synced 2026-05-21 07:26:50 +00:00
JeffJiang
3a99c4e81c
* Refactor API fetch calls to use a unified fetch function; enhance chat history loading with new hooks and UI components - Replaced `fetchWithAuth` with a generic `fetch` function across various API modules for consistency. - Updated `useThreadStream` and `useThreadHistory` hooks to manage chat history loading, including loading states and pagination. - Introduced `LoadMoreHistoryIndicator` component for better user experience when loading more chat history. - Enhanced message handling in `MessageList` to accommodate new loading states and history management. - Added support for run messages in the thread context, improving the overall message handling logic. - Updated translations for loading indicators in English and Chinese. * Fix test assertions for run ordering in RunManager tests - Updated assertions in `test_list_by_thread` to reflect correct ordering of runs. - Modified `test_list_by_thread_is_stable_when_timestamps_tie` to ensure stable ordering when timestamps are tied.
14 lines
675 B
Python
14 lines
675 B
Python
from typing import override
|
|
|
|
from langchain.agents.middleware import SummarizationMiddleware as BaseSummarizationMiddleware
|
|
from langchain_core.messages.human import HumanMessage
|
|
|
|
|
|
class SummarizationMiddleware(BaseSummarizationMiddleware):
|
|
@override
|
|
def _build_new_messages(self, summary: str) -> list[HumanMessage]:
|
|
"""Override the base implementation to let the human message with the special name 'summary'.
|
|
And this message will be ignored to display in the frontend, but still can be used as context for the model.
|
|
"""
|
|
return [HumanMessage(content=f"Here is a summary of the conversation to date:\n\n{summary}", name="summary")]
|