36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
from appPublic.base64_to_file import hex2base64
|
|
from appPublic.registerfunction import RegisterFunction
|
|
from llmage.jimeng import jimeng_auth_headers
|
|
from llmage.keling import keling_token
|
|
|
|
from llmage.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 llmage.accounting import checkCustomerBalance
|
|
from ahserver.serverenv import ServerEnv
|
|
|
|
def load_llmage():
|
|
env = ServerEnv()
|
|
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)
|