| """ | |
| MediGenius — agents/memory.py | |
| MemoryAgent: trims conversation history to the last 20 turns. | |
| """ | |
| from app.core.state import AgentState | |
| def MemoryAgent(state: AgentState) -> AgentState: | |
| """Trim conversation history to the last 20 turns to avoid context overflow.""" | |
| history = state.get("conversation_history", []) | |
| if len(history) > 20: | |
| history = history[-20:] | |
| state["conversation_history"] = history | |
| return state | |