44 lines
1.3 KiB
Python
44 lines
1.3 KiB
Python
from appPublic.base64_to_file import hex2base64
|
|
from appPublic.registerfunction import RegisterFunction
|
|
from ahserver.serverenv import ServerEnv
|
|
from .keling import keling_token
|
|
from .jimeng import jimeng_auth_headers
|
|
|
|
from .llmclient import (
|
|
b64media2url,
|
|
get_llm,
|
|
inference_generator,
|
|
inference,
|
|
llm_query_price,
|
|
get_llmproviders,
|
|
get_llms_sort_by_provider,
|
|
get_llmcatelogs,
|
|
get_llms_by_catelog
|
|
)
|
|
from .accounting import checkCustomerBalance
|
|
from .asyncinference import (
|
|
get_asynctask_status,
|
|
query_task_status,
|
|
get_today_asynctask_list
|
|
)
|
|
|
|
def load_llmage():
|
|
env = ServerEnv()
|
|
env.get_today_asynctask_list = get_today_asynctask_list
|
|
env.get_asynctask_status = get_asynctask_status
|
|
env.query_task_status = query_task_status
|
|
env.get_llm = get_llm
|
|
env.b64media2url = b64media2url
|
|
env.hex2base64 = hex2base64
|
|
env.inference = inference
|
|
env.inference_generator = inference_generator
|
|
env.get_llms_by_catelog = get_llms_by_catelog
|
|
env.get_llmcatelogs = get_llmcatelogs
|
|
env.checkCustomerBalance = checkCustomerBalance
|
|
env.get_llmproviders = get_llmproviders
|
|
env.get_llms_sort_by_provider = get_llms_sort_by_provider
|
|
env.keling_token = keling_token
|
|
env.llm_query_price = llm_query_price
|
|
rf = RegisterFunction()
|
|
rf.register('jimeng_auth_headers', jimeng_auth_headers)
|