From 2222cf33793577f222e1316b14803d5433cfce22 Mon Sep 17 00:00:00 2001
From: seehi <6580@pm.me>
Date: Mon, 12 Aug 2024 18:24:10 +0800
Subject: [PATCH] use exclude

---
 metagpt/roles/di/data_analyst.py | 4 +++-
 metagpt/roles/di/team_leader.py  | 6 +++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/metagpt/roles/di/data_analyst.py b/metagpt/roles/di/data_analyst.py
index 3a43f72e0..f65042217 100644
--- a/metagpt/roles/di/data_analyst.py
+++ b/metagpt/roles/di/data_analyst.py
@@ -1,5 +1,7 @@
 from __future__ import annotations
 
+from typing import Annotated
+
 from pydantic import Field, model_validator
 
 from metagpt.actions.di.execute_nb_code import ExecuteNbCode
@@ -31,7 +33,7 @@ class DataAnalyst(RoleZero):
     tools: list[str] = ["Plan", "DataAnalyst", "RoleZero", "Browser"]
     custom_tools: list[str] = ["web scraping", "Terminal"]
     custom_tool_recommender: ToolRecommender = None
-    experience_retriever: ExpRetriever = KeywordExpRetriever()
+    experience_retriever: Annotated[ExpRetriever, Field(exclude=True)] = KeywordExpRetriever()
 
     use_reflection: bool = True
     write_code: WriteAnalysisCode = Field(default_factory=WriteAnalysisCode, exclude=True)
diff --git a/metagpt/roles/di/team_leader.py b/metagpt/roles/di/team_leader.py
index 12b4b3a18..353e00620 100644
--- a/metagpt/roles/di/team_leader.py
+++ b/metagpt/roles/di/team_leader.py
@@ -1,5 +1,9 @@
 from __future__ import annotations
 
+from typing import Annotated
+
+from pydantic import Field
+
 from metagpt.actions.di.run_command import RunCommand
 from metagpt.prompts.di.team_leader import (
     FINISH_CURRENT_TASK_CMD,
@@ -24,7 +28,7 @@ class TeamLeader(RoleZero):
 
     tools: list[str] = ["Plan", "RoleZero", "TeamLeader"]
 
-    experience_retriever: ExpRetriever = SimpleExpRetriever()
+    experience_retriever: Annotated[ExpRetriever, Field(exclude=True)] = SimpleExpRetriever()
 
     def _update_tool_execution(self):
         self.tool_execution_map.update(