52 lines
958 B
Python
52 lines
958 B
Python
"""
|
|
SimpleChat core module.
|
|
"""
|
|
from .providers import (
|
|
BaseProvider,
|
|
ChatResponse,
|
|
ChatConfig,
|
|
OpenAIProvider,
|
|
ClaudeProvider,
|
|
GeminiProvider,
|
|
PROVIDERS,
|
|
get_provider,
|
|
)
|
|
from .image_utils import (
|
|
tensor_to_pil,
|
|
pil_to_tensor,
|
|
tensor_to_base64,
|
|
base64_to_tensor,
|
|
create_data_uri,
|
|
)
|
|
from .noass import (
|
|
format_noass_prompt,
|
|
build_noass_messages,
|
|
extract_noass_response,
|
|
build_full_history,
|
|
get_stop_sequences,
|
|
)
|
|
|
|
__all__ = [
|
|
# Providers
|
|
"BaseProvider",
|
|
"ChatResponse",
|
|
"ChatConfig",
|
|
"OpenAIProvider",
|
|
"ClaudeProvider",
|
|
"GeminiProvider",
|
|
"PROVIDERS",
|
|
"get_provider",
|
|
# Image utils
|
|
"tensor_to_pil",
|
|
"pil_to_tensor",
|
|
"tensor_to_base64",
|
|
"base64_to_tensor",
|
|
"create_data_uri",
|
|
# NoASS
|
|
"format_noass_prompt",
|
|
"build_noass_messages",
|
|
"extract_noass_response",
|
|
"build_full_history",
|
|
"get_stop_sequences",
|
|
]
|