Skip to content

Commit 24aa51d

Browse files
authored
feat: 支持gpu stack (#71)
1 parent ad1d031 commit 24aa51d

File tree

6 files changed

+140
-1
lines changed

6 files changed

+140
-1
lines changed

domain/gpustack.go

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,16 @@
1+
package domain
2+
3+
type GPUStackListModelResp struct {
4+
Items []*struct {
5+
Name string `json:"name"`
6+
} `json:"items"`
7+
}
8+
9+
// ParseModels 实现ModelResponseParser接口
10+
func (o *GPUStackListModelResp) ParseModels() []ModelListItem {
11+
var models []ModelListItem
12+
for _, item := range o.Items {
13+
models = append(models, ModelListItem{Model: item.Name})
14+
}
15+
return models
16+
}

test/ui_example/src/App.tsx

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -112,6 +112,8 @@ function App() {
112112
<MenuItem value={'audio'}>音频模型</MenuItem>
113113
<MenuItem value={'reranker'}>重排序模型</MenuItem>
114114
<MenuItem value={'analysis'}>分析模型</MenuItem>
115+
<MenuItem value={'monkeycode-chat'}>对话模型</MenuItem>
116+
<MenuItem value={'monkeycode-code'}>代码补全模型</MenuItem>
115117
</Select>
116118
</FormControl>
117119
</Box>

ui/ModelModal/src/ModelModal.tsx

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,8 @@ const titleMap: Record<string, string> = {
4343
["rerank"]: '重排序模型',
4444
["reranker"]: '重排序模型',
4545
["analysis"]: '分析模型',
46+
["monkeycode-chat"]: '对话模型',
47+
["monkeycode-code"]: '代码补全模型',
4648
};
4749

4850
export const ModelModal: React.FC<ModelModalProps> = ({
@@ -449,6 +451,10 @@ export const ModelModal: React.FC<ModelModalProps> = ({
449451
return it.rerank;
450452
case 'analysis':
451453
return it.analysis;
454+
case 'monkeycode-chat':
455+
return it.monkeycode_chat;
456+
case 'monkeycode-code':
457+
return it.monkeycode_code;
452458
default:
453459
return it.label === 'BaiZhiCloud' || it.label === 'Other';
454460
}

0 commit comments

Comments
 (0)