Compare commits

...

2 Commits

Author SHA1 Message Date
a8d69eeefe bugfix 2025-09-04 17:07:11 +08:00
9569e78969 bugfix 2025-09-04 17:06:08 +08:00
2 changed files with 15 additions and 9 deletions

View File

@ -1,5 +1,5 @@
from llmage.llmclient import ( from llmage.llmclient import (
b64media, b64media2url,
get_llm, get_llm,
inference, inference,
get_llmcatelogs, get_llmcatelogs,
@ -17,7 +17,7 @@ from ahserver.serverenv import ServerEnv
def load_llmage(): def load_llmage():
env = ServerEnv() env = ServerEnv()
env.get_llm = get_llm env.get_llm = get_llm
env.b64media = b64media env.b64media2url = b64media2url
env.inference = inference env.inference = inference
env.get_llms_by_catelog = get_llms_by_catelog env.get_llms_by_catelog = get_llms_by_catelog
env.get_llmcatelogs = get_llmcatelogs env.get_llmcatelogs = get_llmcatelogs

View File

@ -4,8 +4,11 @@ from functools import partial
from traceback import format_exc from traceback import format_exc
from sqlor.dbpools import DBPools from sqlor.dbpools import DBPools
from appPublic.log import debug, exception from appPublic.log import debug, exception
from appPublic.uniqueID import getID
from appPublic.base64_to_file import base64_to_file, getFilenameFromBase64
from uapi.appapi import UAPI, sor_get_callerid, sor_get_uapi from uapi.appapi import UAPI, sor_get_callerid, sor_get_uapi
from ahserver.serverenv import get_serverenv from ahserver.serverenv import get_serverenv
from ahserver.filestorage import FileStorage
async def get_llmcatelogs(): async def get_llmcatelogs():
db = DBPools() db = DBPools()
@ -93,16 +96,19 @@ async def uapi_request(request, sor, caller_orgid, callerid, uapi, llm, params):
yield l yield l
debug(f'{d=}, {txt=}') debug(f'{d=}, {txt=}')
def b64media(mediafile): def b64media2url(request, mediafile):
env = request._run_ns
if mediafile.startswith('data:'): if mediafile.startswith('data:'):
return mediafile fs = FileStorage()
fname = getFilenameFromBase64(mediafile)
fpath = fs._name2path(fname)
base64_to_file(mediafile, fpath)
path = fs.webpath(fpath)
return env.entire_url('/idfile?path=') + env.quota(path)
if mediafile.startswith('http://') or mediafile.startswith('https://'): if mediafile.startswith('http://') or mediafile.startswith('https://'):
return mediafile return mediafile
fs = FileStorage() url = env.entire_url('/idfile?path=') + env.quota(mediafile)
fn = fs.realPath(mediafile) return url
with open(fn, 'rb') as f:
b = f.read()
return base64.b64encode(b).decode('iso-8859-1')
async def inference(request, *args, **kw): async def inference(request, *args, **kw):
env = request._run_ns env = request._run_ns