From c457e3bc3375d4b7c2cf51e13f6fac9adf576fbe Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Tue, 31 Mar 2026 15:46:26 +0800
Subject: [PATCH] bugfix

---
 llmage/init.py           |  4 ++++
 llmage/utils.py          |  7 +++++++
 wwwroot/tasks/index.dspy | 12 ++++++++++++
 wwwroot/video/index.dspy | 25 +++++++++++++++++++++++++
 4 files changed, 48 insertions(+)
 create mode 100644 wwwroot/tasks/index.dspy
 create mode 100644 wwwroot/video/index.dspy

diff --git a/llmage/init.py b/llmage/init.py
index 11f256a..95b1d82 100644
--- a/llmage/init.py
+++ b/llmage/init.py
@@ -3,6 +3,9 @@ from appPublic.registerfunction import RegisterFunction
 from ahserver.serverenv import ServerEnv
 from .keling import keling_token
 from .jimeng import jimeng_auth_headers
+from .utils import (
+	get_llm_by_model
+)
 
 from .llmclient import (
 	b64media2url,
@@ -29,6 +32,7 @@ from .asyncinference import (
 
 def load_llmage():
 	env = ServerEnv()
+	eng.get_llm_by_model = get_llm_by_model
 	env.llm_charging = llm_charging
 	env.llm_accounting = llm_accounting
 	env.get_today_asynctask_list = get_today_asynctask_list
diff --git a/llmage/utils.py b/llmage/utils.py
index 9e9bcbc..238ecfa 100644
--- a/llmage/utils.py
+++ b/llmage/utils.py
@@ -14,6 +14,13 @@ from uapi.appapi import UAPI, sor_get_callerid, sor_get_uapi
 from ahserver.serverenv import get_serverenv, ServerEnv
 from ahserver.filestorage import FileStorage
 
+async def get_llm_by_model(id, lctype=None):
+	env = ServerEnv()
+	async with get_sor_context(env, 'llmage') as sor:
+		sql = 'select * from llm where model=${model}$'
+		recs = await sor.R('llm', {'model': model})
+		return recs
+
 def erase_apikey(e):
 	e = str(e)
 	ss = e.split('Bearer ')
diff --git a/wwwroot/tasks/index.dspy b/wwwroot/tasks/index.dspy
new file mode 100644
index 0000000..1417c0b
--- /dev/null
+++ b/wwwroot/tasks/index.dspy
@@ -0,0 +1,12 @@
+taskid = params_kw.taskid
+userid = await get_user()
+if userid is None:
+	return openai_403
+	
+s = await get_asynctask_status(taskid)
+return {
+	'status': 'ok',
+	'data': {
+		'resposne': s
+	}
+}
diff --git a/wwwroot/video/index.dspy b/wwwroot/video/index.dspy
new file mode 100644
index 0000000..6b092d2
--- /dev/null
+++ b/wwwroot/video/index.dspy
@@ -0,0 +1,25 @@
+params_kw.action = 't2v'
+lctype =
+userid = await get_user()
+if userid is None:
+	return openai_403
+
+if params_kw.model is None:
+	return openai_400()
+
+llms = await get_llm_by_model(params_kw.model)
+if llms is None:
+	debug(f'{params_kw.model=} not found in llm with error')
+	return openai_400()
+if len(llms) == 0:
+	debug(f'{params_kw.model=} not found in llm')
+	return openai_400()
+	
+f = await checkCustomerBalance(params_kw.llmid, userorgid)
+if not f:
+    return openai_429()
+
+model = params_kw.model
+env = request._run_ns
+async with get_sor.context(env, 'llmage') as sor:
+