llmage/llmage/init.py
2025-08-21 14:05:40 +08:00

22 lines
511 B
Python

from llmage.llmclient import (
get_llm,
inference,
get_llmcatelogs,
get_llms_by_catelog,
default_sysmessage,
default_usrmessage,
default_llmmessage
)
from ahserver.serverenv import ServerEnv
def load_llmage():
env = ServerEnv()
env.get_llm = get_llm
env.inference = inference
env.get_llms_by_catelog = get_llms_by_catelog
env.get_llmcatelogs = get_llmcatelogs
env.default_sysmessage = default_sysmessage
env.default_usrmessage = default_usrmessage
env.default_llmmessage = default_llmmessage