34 lines
767 B
Python
34 lines
767 B
Python
from abc import ABC, abstractmethod
|
|
import torch
|
|
|
|
from models.base import (BaseAnswer,
|
|
AnswerResult)
|
|
|
|
|
|
class MultimodalAnswerResult(AnswerResult):
|
|
image: str = None
|
|
|
|
|
|
class RemoteRpcModel(BaseAnswer, ABC):
|
|
|
|
@property
|
|
@abstractmethod
|
|
def _api_key(self) -> str:
|
|
"""Return _api_key of client."""
|
|
|
|
@property
|
|
@abstractmethod
|
|
def _api_base_url(self) -> str:
|
|
"""Return _api_base of client host bash url."""
|
|
|
|
@abstractmethod
|
|
def set_api_key(self, api_key: str):
|
|
"""set set_api_key"""
|
|
|
|
@abstractmethod
|
|
def set_api_base_url(self, api_base_url: str):
|
|
"""set api_base_url"""
|
|
@abstractmethod
|
|
def call_model_name(self, model_name):
|
|
"""call model name of client"""
|