mirror of
https://github.com/agentscope-ai/ReMe.git
synced 2026-09-06 08:16:00 +00:00
22 lines
642 B
Python
22 lines
642 B
Python
from memory_scope.models import MODEL_REGISTRY
|
|
from memory_scope.models.base_model import BaseModel
|
|
from memory_scope.models.response import ModelResponse, ModelResponseGen
|
|
|
|
|
|
class BaseRankModel(BaseModel):
|
|
MODEL_REGISTRY.batch_register([
|
|
|
|
])
|
|
|
|
def before_call(self, **kwargs) -> None:
|
|
pass
|
|
|
|
def after_call(self, model_response: ModelResponse | ModelResponseGen,
|
|
**kwargs) -> ModelResponse | ModelResponseGen:
|
|
pass
|
|
|
|
def _call(self, stream: bool = False, **kwargs) -> ModelResponse | ModelResponseGen:
|
|
pass
|
|
|
|
async def _async_call(self, **kwargs) -> ModelResponse:
|
|
pass
|