Feat: Add industry-related search keyword generation function (#9156)

### What problem does this PR solve?
Add industry-related search keyword generation function
- When generating search keywords, support for specific industries has
been added
- If the "industry" parameter is provided, industry-specific
restrictions will be added to the prompt
- This change can help users generate more precise search keywords
within specific industries

### Type of change

- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
天海蒼灆
2025-08-01 15:50:46 +08:00
committed by GitHub
parent b26088ab70
commit 6591031bad
2 changed files with 10 additions and 2 deletions

View File

@ -737,6 +737,7 @@ def related_questions(tenant_id):
if not req.get("question"):
return get_error_data_result("`question` is required.")
question = req["question"]
industry = req.get("industry", "")
chat_mdl = LLMBundle(tenant_id, LLMType.CHAT)
prompt = """
Objective: To generate search terms related to the user's search keywords, helping users find more valuable information.
@ -746,7 +747,10 @@ Instructions:
- Use common, general terms as much as possible, avoiding obscure words or technical jargon.
- Keep the term length between 2-4 words, concise and clear.
- DO NOT translate, use the language of the original keywords.
"""
if industry:
prompt += f" - Ensure all search terms are relevant to the industry: {industry}.\n"
prompt += """
### Example:
Keywords: Chinese football
Related search terms: