15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Merge branch 'main' of https://github.com/maxkb-dev/maxkb

This commit is contained in:
liqiang-fit2cloud 2024-11-20 13:58:51 +08:00
parent 7923be3c68 aba33eb00a
commit c06d5cc8b4
4 changed files with 18 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
apps/setting/models_provider/impl/openai_model_provider/model/image.py
View File

@ -17,9 +17,10 @@ class OpenAIImage(MaxKBBaseModel, ChatOpenAI):
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)
return OpenAIImage(
model=model_name,
model_name=model_name,
openai_api_base=model_credential.get('api_base'),
openai_api_key=model_credential.get('api_key'),
stream_options={"include_usage": True},
# stream_options={"include_usage": True},
streaming=True,
**optional_params,
)

5
apps/setting/models_provider/impl/qwen_model_provider/model/image.py
View File

@ -13,10 +13,11 @@ class QwenVLChatModel(MaxKBBaseModel, ChatOpenAI):
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)
chat_tong_yi = QwenVLChatModel(
model=model_name,
model_name=model_name,
openai_api_key=model_credential.get('api_key'),
openai_api_base='https://dashscope.aliyuncs.com/compatible-mode/v1',
stream_options={"include_usage": True},
# stream_options={"include_usage": True},
streaming=True,
model_kwargs=optional_params,
)
return chat_tong_yi

5
apps/setting/models_provider/impl/tencent_model_provider/model/image.py
View File

@ -17,9 +17,10 @@ class TencentVision(MaxKBBaseModel, ChatOpenAI):
def new_instance(model_type, model_name, model_credential: Dict[str, object], **model_kwargs):
optional_params = MaxKBBaseModel.filter_optional_params(model_kwargs)
return TencentVision(
model=model_name,
model_name=model_name,
openai_api_base='https://api.hunyuan.cloud.tencent.com/v1',
openai_api_key=model_credential.get('api_key'),
stream_options={"include_usage": True},
# stream_options={"include_usage": True},
streaming=True,
**optional_params,
)

9
ui/src/workflow/nodes/image-understand/index.vue
View File

@ -193,6 +193,7 @@ import applicationApi from '@/api/application'
import { app } from '@/main'
import useStore from '@/stores'
import NodeCascader from '@/workflow/common/NodeCascader.vue'
import type { FormInstance } from 'element-plus'
const { model } = useStore()
@ -204,6 +205,12 @@ const props = defineProps<{ nodeModel: any }>()
const modelOptions = ref<any>(null)
const providerOptions = ref<Array<Provider>>([])
const aiChatNodeFormRef = ref<FormInstance>()
const validate = () => {
return aiChatNodeFormRef.value?.validate().catch((err) => {
return Promise.reject({ node: props.nodeModel, errMessage: err })
})
}
const wheel = (e: any) => {
if (e.ctrlKey === true) {
@ -277,6 +284,8 @@ function submitDialog(val: string) {
onMounted(() => {
getModel()
getProvider()
set(props.nodeModel, 'validate', validate)
})
</script>