Compare commits
2 Commits
0e0afda45a
...
a8d69eeefe
| Author | SHA1 | Date | |
|---|---|---|---|
| a8d69eeefe | |||
| 9569e78969 |
@ -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
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user