adapt to new base classes

This commit is contained in:
鸣山 2025-06-13 15:25:24 +08:00
parent 858e6f6e3e
commit e70ef1fd2d

View file

@ -23,7 +23,8 @@ class BaseSummarizer(BaseModel, ABC):
def execute(self, trajectories: List[Trajectory], workspace_id: str = None, **kwargs) -> List[Experience]:
experiences: List[Experience] = self._extract_experiences(trajectories, workspace_id=workspace_id, **kwargs)
nodes: List[VectorStoreNode] = [x.to_vector_store_node() for x in experiences]
self.vector_store.insert(nodes, index_name=workspace_id, **kwargs)
if experiences:
nodes: List[VectorStoreNode] = [x.to_vector_store_node() for x in experiences]
self.vector_store.insert(nodes, index_name=workspace_id, **kwargs)
return experiences