9x25dillon's picture
Create adapters/base.py
48099f7 verified
# 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 []