lidp commited on
Commit
1de5cd5
·
1 Parent(s): a92172b

Remove WenCai imageurl and update investment_advisor prompt (#2606)

Browse files

### What problem does this PR solve?


### Type of change

- [x] Refactoring

agent/component/wencai.py CHANGED
@@ -64,6 +64,12 @@ class WenCai(ComponentBase, ABC):
64
  continue
65
  wencai_res.append({"content": pd.DataFrame.from_dict(item[1], orient='index').to_markdown()})
66
  continue
 
 
 
 
 
 
67
  wencai_res.append({"content": item[0] + "\n" + str(item[1])})
68
  except Exception as e:
69
  return WenCai.be_output("**ERROR**: " + str(e))
 
64
  continue
65
  wencai_res.append({"content": pd.DataFrame.from_dict(item[1], orient='index').to_markdown()})
66
  continue
67
+ if isinstance(item[1], pd.DataFrame):
68
+ if "image_url" in item[1].columns:
69
+ continue
70
+ wencai_res.append({"content": item[1].to_markdown()})
71
+ continue
72
+
73
  wencai_res.append({"content": item[0] + "\n" + str(item[1])})
74
  except Exception as e:
75
  return WenCai.be_output("**ERROR**: " + str(e))
agent/templates/investment_advisor.json CHANGED
@@ -44,7 +44,7 @@
44
  }
45
  ],
46
  "presence_penalty": 0.4,
47
- "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}",
48
  "temperature": 0.1,
49
  "top_p": 0.3
50
  }
@@ -369,7 +369,7 @@
369
  ],
370
  "presencePenaltyEnabled": true,
371
  "presence_penalty": 0.4,
372
- "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}",
373
  "temperature": 0.1,
374
  "temperatureEnabled": true,
375
  "topPEnabled": true,
 
44
  }
45
  ],
46
  "presence_penalty": 0.4,
47
+ "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}",
48
  "temperature": 0.1,
49
  "top_p": 0.3
50
  }
 
369
  ],
370
  "presencePenaltyEnabled": true,
371
  "presence_penalty": 0.4,
372
+ "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}",
373
  "temperature": 0.1,
374
  "temperatureEnabled": true,
375
  "topPEnabled": true,