yumoqing/bricks
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2026-05-16 11:43:26 +08:00
parent 24ba4a0842
commit f581fc9b23
2 changed files with 66 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
bricks/llm.js
View File

@ -208,26 +208,43 @@ bricks.LlmModel = class extends bricks.JsWidget {
estimate_url:this.llmio.estimate_url
});
this.llmio.o_w.add_widget(mout);
if (this.response_mode == 'stream' || this.response_mode == 'async') {
var d = this.inputdata2uploaddata(data);
var hr = new bricks.HttpResponseStream();
var resp = await hr.post(this.opts.url, {params:d});
if (! resp) {
mout.run_stopped();
return;
}
await hr.handle_chunk(resp, this.chunk_response.bind(this, mout));
this.chunk_ended();
} else {
var d = this.inputdata2uploaddata(data);
console.log('data_inouted=', data, 'upload_data=', d);
var hj = new bricks.HttpJson()
var resp = await hj.post(this.opts.url, {params:d});
if (! resp) {
mout.run_stopped();
return;
}
mout.update_data(resp);
switch (this.response_mode) {
case 'async':
var d = this.inputdata2uploaddata(data);
console.log('data_inouted=', data, 'upload_data=', d);
var hj = new bricks.HttpJson()
var resp = await hj.post(this.opts.url, {params:d});
if (! resp) {
mout.run_stopped();
return;
}
mout.update_data(resp);
if (resp.status == 'FAILED'){
return;
}
this.query_task_status(mout, resp.taskid)
break;
case 'stream':
var d = this.inputdata2uploaddata(data);
var hr = new bricks.HttpResponseStream();
var resp = await hr.post(this.opts.url, {params:d});
if (! resp) {
mout.run_stopped();
return;
}
await hr.handle_chunk(resp, this.chunk_response.bind(this, mout));
this.chunk_ended();
break;
default:
var d = this.inputdata2uploaddata(data);
console.log('data_inouted=', data, 'upload_data=', d);
var hj = new bricks.HttpJson()
var resp = await hj.post(this.opts.url, {params:d});
if (! resp) {
mout.run_stopped();
return;
}
mout.update_data(resp);
}
mout.estimate_w.show();
}
@ -240,6 +257,22 @@ bricks.LlmModel = class extends bricks.JsWidget {
llm_msg_format(){
return this.llm_message_format || {role:'assistant', content:"${content}"}
}
query_task_status(mout, taskid){
var pt = this.opts.period_time || 30;
if (this.query_task) {
this.query_task.cancel()
}
var hj = new bricks.HttpJson()
var resp = await hj.post(this.opts.url, {params:d});
if (! resp) {
return;
}
mout.update_data(resp)
if (resp.status == 'FAILED' || resp.status == 'SUCCEEDED'){
return
}
schedule_once(this.query_task_status.bind(this, mout, taskid), pt)
}
chunk_response(mout, l){
l = l.trim();
try {
@ -249,9 +282,8 @@ bricks.LlmModel = class extends bricks.JsWidget {
return
}
if (this.opts.response_mode == 'async'){
if(d.status != 'SUCCEEDED' && d.status != 'FAILED' ){
console.log('filter all message not successed or failed', d);
return;
if (d.taskid){
this.query_task_status(mout, d.taskid)
}
}
console.log('l=', l, 'd=', d);

11
bricks/llmout.js
View File

@ -81,6 +81,7 @@ bricks.LlmOut = class extends bricks.VBox {
this.i_w = null;
this.a_w = null;
this.glb_w = null;
this.s_w = null; // 状态
this.images = [];
this.reasoning_content = '';
this.content = '';
@ -88,6 +89,13 @@ bricks.LlmOut = class extends bricks.VBox {
}
update(data){
if (data.status){
this.s_w = new bricks.Text({
width: '100%',
height: 'auto',
text: JSON.stringify(data)
});
}
if (data.audio){
var url = data.audio;
if (! data.audio.startsWith('http')){
@ -188,6 +196,9 @@ bricks.LlmOut = class extends bricks.VBox {
this.add_widget(w)
});
}
if(this.s_w){
this.add_widget(this.s_w);
}
}
}
bricks.Factory.register('LlmOut', bricks.LlmOut);