models.py

Code Hygiene Score: 100

Keine Issues gefunden.

Dependencies 3

Klassen 3

Code

"""Datenmodelle für Wissensextraktion."""

from dataclasses import dataclass
from enum import Enum


class KnowledgeLevel(Enum):
    """Ebene der Wissensextraktion."""

    PAGE = "page"
    SECTION = "section"
    DOCUMENT = "document"


class KnowledgeType(Enum):
    """Typ des extrahierten Wissens."""

    ENTITY = "entity"
    SEMANTIC = "semantic"
    ONTOLOGY = "ontology"
    TAXONOMY = "taxonomy"


@dataclass
class ModelConfig:
    """Konfiguration für LLM-Modell."""

    provider: str  # 'ollama' oder 'anthropic'
    model_name: str
    temperature: float = 0.3
    max_tokens: int = 2000


# Standard-Modellkonfigurationen
DEFAULT_MODELS = {
    "ollama": ModelConfig("ollama", "gemma3:27b-it-qat"),
    "anthropic": ModelConfig("anthropic", "claude-3-haiku-20240307"),
    "anthropic_opus": ModelConfig("anthropic", "claude-opus-4-5-20251101"),
}
← Übersicht Graph