from beartype.claw import beartype_this_package
import importlib.metadata
from .ragflow import RAGFlow
from .modules.dataset import DataSet
from .modules.chat import Chat
from .modules.session import Session
from .modules.document import Document
from .modules.chunk import Chunk
from .modules.agent import Agent

beartype_this_package()

__version__ = importlib.metadata.version("ragflow_sdk")

__all__ = [
    "RAGFlow",
    "DataSet",
    "Chat",
    "Session",
    "Document",
    "Chunk",
    "Agent"
]