create traj summarizer

This commit is contained in:
qingxu 2025-06-06 16:12:11 +08:00
parent 195d2aef4d
commit a83f83506e

View file

@ -0,0 +1,26 @@
from typing import List
from pydantic import Field
from experiencemaker.schema.trajectory import Trajectory, Sample, SummaryMessage
from experiencemaker.storage.base_vector_store import BaseVectorStore
from experiencemaker.module.summarizer.base_summarizer import BaseSummarizer
class TrajectorySummarizer(BaseSummarizer):
vector_store: BaseVectorStore | None = Field(default=None)
def extract_samples(self, trajectories: List[Trajectory], **kwargs) -> List[Sample]:
raise NotImplementedError
def insert_into_vector_store(self, samples: List[Sample], **kwargs):
raise NotImplementedError
def execute(self, trajectories: List[Trajectory], return_samples: bool = True, **kwargs) -> List[Sample]:
samples: List[Sample] = self.extract_samples(trajectories, **kwargs)
self.insert_into_vector_store(samples, **kwargs)
if return_samples:
return samples
return []