AI/ragflow
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/infiniflow/ragflow.git synced 2025-08-12 20:29:06 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove WenCai imageurl and update investment_advisor prompt (#2606)

Browse Source 
### What problem does this PR solve?


### Type of change

- [x] Refactoring
This commit is contained in:
lidp 2024-09-26 17:27:53 +08:00 committed by GitHub
parent 1de3032650
commit 240450ea52
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
agent/component/wencai.py
View File

@ -64,6 +64,12 @@ class WenCai(ComponentBase, ABC):
continue continue
wencai_res.append({"content": pd.DataFrame.from_dict(item[1], orient='index').to_markdown()}) wencai_res.append({"content": pd.DataFrame.from_dict(item[1], orient='index').to_markdown()})
continue continue
if isinstance(item[1], pd.DataFrame):
if "image_url" in item[1].columns:
continue
wencai_res.append({"content": item[1].to_markdown()})
continue
wencai_res.append({"content": item[0] + "\n" + str(item[1])}) wencai_res.append({"content": item[0] + "\n" + str(item[1])})
except Exception as e: except Exception as e:
return WenCai.be_output("**ERROR**: " + str(e)) return WenCai.be_output("**ERROR**: " + str(e))

4
agent/templates/investment_advisor.json
View File

@ -44,7 +44,7 @@
} }
], ],
"presence_penalty": 0.4, "presence_penalty": 0.4,
"prompt": "Role: You are a professional financial counseling assistant.\n\nTasks\nStep 1: Organize and output the content provided by Man Finance in the form of a table based on the user's questions.\nStep 2: Organize and output the news information provided by AkShare, making sure that each news item has a corresponding URL link.\n\nNotice:\n- All data provided by WenCai needs to be collated and output, you cannot replace the output data with ellipses!\n- Output no more than 5 news items from AkShare.\n\nContent provided by Wencai: \n{wencai_input}\n\nContent provided by AkShare: \n{ak_input}", "prompt": "Role: You are a professional financial counseling assistant.\n\nTask: Answer user's question based on content provided by Wencai and AkShare.\n\nNotice:\n- Output no more than 5 news items from AkShare if there's content provided by Wencai.\n- Items from AkShare MUST have a corresponding URL link.\n\n############\nContent provided by Wencai: \n{wencai_input}\n\n################\nContent provided by AkShare: \n{ak_input}",
"temperature": 0.1, "temperature": 0.1,
"top_p": 0.3 "top_p": 0.3
} }
@ -369,7 +369,7 @@
], ],
"presencePenaltyEnabled": true, "presencePenaltyEnabled": true,
"presence_penalty": 0.4, "presence_penalty": 0.4,
"prompt": "Role: You are a professional financial counseling assistant.\n\nTasks\nStep 1: Organize and output the content provided by Man Finance in the form of a table based on the user's questions.\nStep 2: Organize and output the news information provided by AkShare, making sure that each news item has a corresponding URL link.\n\nNotice:\n- All data provided by WenCai needs to be collated and output, you cannot replace the output data with ellipses!\n- Output no more than 5 news items from AkShare.\n\nContent provided by Wencai: \n{wencai_input}\n\nContent provided by AkShare: \n{ak_input}", "prompt": "Role: You are a professional financial counseling assistant.\n\nTask: Answer user's question based on content provided by Wencai and AkShare.\n\nNotice:\n- Output no more than 5 news items from AkShare if there's content provided by Wencai.\n- Items from AkShare MUST have a corresponding URL link.\n\n############\nContent provided by Wencai: \n{wencai_input}\n\n################\nContent provided by AkShare: \n{ak_input}",
"temperature": 0.1, "temperature": 0.1,
"temperatureEnabled": true, "temperatureEnabled": true,
"topPEnabled": true, "topPEnabled": true,