scriptoriumadmin/ReMe
1
0
Fork 0
mirror of https://github.com/agentscope-ai/ReMe.git synced 2026-09-21 00:22:45 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

up

Browse source
This commit is contained in:
jinli.yl 2026-05-15 09:44:44 +08:00
parent f16bdb0383
commit e713fcd804
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
reme2/component/base_step.py
View file

@ -9,6 +9,7 @@ from agentscope.token import TokenCounterBase
from .base_component import BaseComponent
from .embedding import BaseEmbeddingModel
from .file_parser import BaseFileParser
from .file_store import BaseFileStore
from .prompt_handler import PromptHandler
from .runtime_context import RuntimeContext
@ -85,6 +86,11 @@ class BaseStep(BaseComponent):
else:
return self.get_component(ComponentEnum.AS_TOKEN_COUNTER, name, "token_counter")
@property
def file_parser(self) -> BaseFileParser:
name = self.kwargs.get("file_parser", "default")
return name if isinstance(name, BaseFileParser) else self.get_component(ComponentEnum.FILE_PARSER, name)
@property
def file_store(self) -> BaseFileStore:
name = self.kwargs.get("file_store", "default")