File size: 331 Bytes
48099f7 4970725 48099f7 4970725 48099f7 4970725 | 1 2 3 4 5 6 7 8 9 10 11 12 | # adapters/base.py
from abc import ABC, abstractmethod
from typing import List
class BaseModelAdapter(ABC):
@abstractmethod
def generate(self, prompt: str) -> str: ...
@abstractmethod
def embed_text(self, text: str) -> List[float]: ...
def get_hidden_state(self, prompt: str) -> List[float]:
return []
|