curo1305
33 lines
706 B
Python
33 lines
706 B
Python
from abc import ABC, abstractmethod
|
|
from dataclasses import dataclass, field
|
|
|
|
|
|
@dataclass
|
|
class ClassificationResult:
|
|
topics: list[str] = field(default_factory=list)
|
|
suggested_new_topics: list[str] = field(default_factory=list)
|
|
reasoning: str = ""
|
|
|
|
|
|
class AIProvider(ABC):
|
|
@abstractmethod
|
|
async def classify(
|
|
self,
|
|
document_text: str,
|
|
existing_topics: list[str],
|
|
system_prompt: str,
|
|
) -> ClassificationResult:
|
|
...
|
|
|
|
@abstractmethod
|
|
async def suggest_topics(
|
|
self,
|
|
document_text: str,
|
|
system_prompt: str,
|
|
) -> list[str]:
|
|
...
|
|
|
|
@abstractmethod
|
|
async def health_check(self) -> bool:
|
|
...
|