From d10881c0e441e0f37b1645c4ca890c30f2868da3 Mon Sep 17 00:00:00 2001
From: seehi <6580@pm.me>
Date: Tue, 4 Jun 2024 15:16:58 +0800
Subject: [PATCH] add trajectory schema

---
 metagpt/exp_pool/schema.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/metagpt/exp_pool/schema.py b/metagpt/exp_pool/schema.py
index b51bc3c17..e6ae4ee1d 100644
--- a/metagpt/exp_pool/schema.py
+++ b/metagpt/exp_pool/schema.py
@@ -32,6 +32,14 @@ class Metric(BaseModel):
     score: int = Field(default=1, description="Score, a value between 1 and 10.")
 
 
+class Trajectory(BaseModel):
+    """Experience Trajectory."""
+
+    plan: str = Field(default="", description="The plan.")
+    action: str = Field(default="", description="Action for the plan.")
+    observation: str = Field(default="", description="Output of the action.")
+
+
 class Experience(BaseModel):
     """Experience."""
 
@@ -41,6 +49,7 @@ class Experience(BaseModel):
     exp_type: ExperienceType = Field(default=ExperienceType.SUCCESS, description="The type of experience.")
     entry_type: EntryType = Field(default=EntryType.AUTOMATIC, description="Type of entry: Manual or Automatic.")
     tag: str = Field(default="", description="Tagging experience.")
+    traj: Optional[Trajectory] = Field(default=None, description="Trajectory.")
 
     def rag_key(self):
         return self.req