zhuchaowe/maxkb
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
liqiang-fit2cloud 2024-11-29 15:27:30 +08:00
commit a480ab0b6b
2 changed files with 17 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
apps/application/flow/step_node/image_understand_step_node/impl/base_image_understand_node.py
View File

@ -92,15 +92,24 @@ class BaseImageUnderstandNode(IImageUnderstandNode):
'history_message': history_message, 'question': question.content}, {}, 'history_message': history_message, 'question': question.content}, {},
_write_context=write_context) _write_context=write_context)
@staticmethod
def get_history_message_for_details(history_chat_record, dialogue_number): def get_history_message_for_details(self, history_chat_record, dialogue_number):
start_index = len(history_chat_record) - dialogue_number start_index = len(history_chat_record) - dialogue_number
history_message = reduce(lambda x, y: [*x, *y], [ history_message = reduce(lambda x, y: [*x, *y], [
[history_chat_record[index].get_human_message(), history_chat_record[index].get_ai_message()] [self.generate_history_human_message_for_details(history_chat_record[index]), history_chat_record[index].get_ai_message()]
for index in for index in
range(start_index if start_index > 0 else 0, len(history_chat_record))], []) range(start_index if start_index > 0 else 0, len(history_chat_record))], [])
return history_message return history_message
def generate_history_human_message_for_details(self, chat_record):
for data in chat_record.details.values():
if self.node.id == data['node_id'] and 'image_list' in data:
image_list = data['image_list']
if len(image_list) == 0 or data['dialogue_type'] == 'WORKFLOW':
return HumanMessage(content=chat_record.problem_text)
return HumanMessage(content=data['question'])
return HumanMessage(content=chat_record.problem_text)
def get_history_message(self, history_chat_record, dialogue_number): def get_history_message(self, history_chat_record, dialogue_number):
start_index = len(history_chat_record) - dialogue_number start_index = len(history_chat_record) - dialogue_number
history_message = reduce(lambda x, y: [*x, *y], [ history_message = reduce(lambda x, y: [*x, *y], [

7
ui/src/views/application-workflow/index.vue
View File

@ -241,9 +241,12 @@ function onmousedown(item: any) {
function clickoutside() { function clickoutside() {
showPopover.value = false showPopover.value = false
} }
function publicHandle() { async function publicHandle() {
// 先执行保存 // 先执行保存
saveApplication() const obj = {
work_flow: getGraphData()
}
await application.asyncPutApplication(id, obj)
// 后执行发布 // 后执行发布
workflowRef.value workflowRef.value
?.validate() ?.validate()