workflows/lib/llm_api/__init__.py
2024-05-09 22:18:51 +08:00

20 lines
541 B
Python

from .chat import chat, chat_json
from .memory.base import ChatMemory
from .memory.fixsize_memory import FixSizeChatMemory
from .openai import chat_completion_no_stream_return_json, chat_completion_stream
from .text_confirm import llm_edit_confirm
from .tools_call import chat_tools, llm_func, llm_param
__all__ = [
"chat_completion_stream",
"chat_completion_no_stream_return_json",
"chat_json",
"chat",
"llm_edit_confirm",
"llm_func",
"llm_param",
"chat_tools",
"ChatMemory",
"FixSizeChatMemory",
]