605219f8c7
* feat: Add real world dates extraction * fix: Linter * fix: 💄 mypy errors * chore: handle invalid dates returned by the llm * chore: Polish prompt * reformat * style: 💄 reformat
19 lines
417 B
Python
19 lines
417 B
Python
import typing
|
|
from abc import ABC, abstractmethod
|
|
|
|
from ..prompts.models import Message
|
|
from .config import LLMConfig
|
|
|
|
|
|
class LLMClient(ABC):
|
|
@abstractmethod
|
|
def __init__(self, config: LLMConfig):
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_embedder(self) -> typing.Any:
|
|
pass
|
|
|
|
@abstractmethod
|
|
async def generate_response(self, messages: list[Message]) -> dict[str, typing.Any]:
|
|
pass
|