mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-10-10 03:30:56 +00:00
create traj summarizer
This commit is contained in:
parent
195d2aef4d
commit
a83f83506e
1 changed files with 26 additions and 0 deletions
26
experiencemaker/module/summarizer/traj_summarizer.py
Normal file
26
experiencemaker/module/summarizer/traj_summarizer.py
Normal 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 []
|
||||
|
||||
Loading…
Add table
Reference in a new issue