From deb1e94fdb87ab99c8362c73c784ac5437908602 Mon Sep 17 00:00:00 2001
From: "jinli.yl" <jinli.yl@alibaba-inc.com>
Date: Thu, 3 Jul 2025 18:28:40 +0800
Subject: [PATCH] format response request

---
 v1/schema/request.py  |  4 ++--
 v1/schema/response.py | 31 ++++++++++++++++---------------
 2 files changed, 18 insertions(+), 17 deletions(-)

diff --git a/v1/schema/request.py b/v1/schema/request.py
index b927628b..67e49a3d 100644
--- a/v1/schema/request.py
+++ b/v1/schema/request.py
@@ -21,10 +21,10 @@ class SummarizerRequest(BaseRequest):
     traj_list: List[Trajectory] = Field(default_factory=list)
 
 
-class VectorStoreRequest(BaseModel):
+class VectorStoreRequest(BaseRequest):
     action: str = Field(default="")
     params: dict = Field(default_factory=dict)
 
 
-class AgentRequest(BaseModel):
+class AgentRequest(BaseRequest):
     query: str = Field(default="")
diff --git a/v1/schema/response.py b/v1/schema/response.py
index 738930e7..a3d04d94 100644
--- a/v1/schema/response.py
+++ b/v1/schema/response.py
@@ -3,28 +3,29 @@ from typing import List
 
 from pydantic import BaseModel, Field
 
-from experiencemaker.schema.experience import Experience
-from experiencemaker.schema.trajectory import Trajectory, ContextMessage
+from v1.schema.experience import BaseExperienceNode
+from v1.schema.message import Message
 
 
 class BaseResponse(BaseModel, ABC):
     success: bool = Field(default=True)
     metadata: dict = Field(default_factory=dict)
 
-#
-# class AgentWrapperResponse(BaseResponse):
-#     trajectory: Trajectory = Field(default_factory=Trajectory)
-
-
-class ContextGeneratorResponse(BaseResponse):
-    experience: list[dict] = Field(default_factory=list)
-
-    merge_experience: str = Field(default="")
-
-    # when to use, experience, response
-
 
+class RetrieverResponse(BaseResponse):
+    experience_nodes: list[BaseExperienceNode] = Field(default_factory=list)
+    experience_merged: str = Field(default="")
 
 
 class SummarizerResponse(BaseResponse):
-    experiences: List[dict] = Field(default_factory=list)
+    experience_nodes: list[BaseExperienceNode] = Field(default_factory=list)
+
+
+class VectorStoreResponse(BaseResponse):
+    action: str = Field(default="")
+    params: dict = Field(default_factory=dict)
+
+
+class AgentResponse(BaseResponse):
+    answer: str = Field(default="")
+    messages: List[Message] = Field(default_factory=list)