UPDATE: chat basic prompt. Limit Chatbot to conclude within three turns.
This commit is contained in:
pangkaicheng
@@ -49,10 +49,18 @@ class LCChatBot(ls.LitAPI):
|
|||||||
user_msg = Message(role=Role.USER, content=user_message)
|
user_msg = Message(role=Role.USER, content=user_message)
|
||||||
chat_history = self.redis.get_history(session_id)
|
chat_history = self.redis.get_history(session_id)
|
||||||
chat_history.append(user_msg)
|
chat_history.append(user_msg)
|
||||||
|
|
||||||
|
turn_count = sum(1 for msg in chat_history if msg.role == Role.USER)
|
||||||
|
|
||||||
if request.gender == 'male':
|
if request.gender == 'male':
|
||||||
prompt = BASIC_PROMPT.format(gender='men')
|
system_instruction = BASIC_PROMPT.format(gender='men')
|
||||||
else:
|
else:
|
||||||
prompt = BASIC_PROMPT.format(gender='women')
|
system_instruction = BASIC_PROMPT.format(gender='women')
|
||||||
|
|
||||||
|
if turn_count >= 3:
|
||||||
|
system_instruction += "\n\nCRITICAL: This is the final turn. Do not ask questions. Provide a brief summary and use the mandatory closing phrase."
|
||||||
|
else:
|
||||||
|
system_instruction += f"\n\nCURRENT STATE: This is turn {turn_count} of 3. Keep gathering info efficiently or provide a brief summary and use the mandatory closing phrase if you have sufficient information."
|
||||||
|
|
||||||
contents = []
|
contents = []
|
||||||
|
|
||||||
@@ -68,7 +76,7 @@ class LCChatBot(ls.LitAPI):
|
|||||||
model='gemini-2.5-flash',
|
model='gemini-2.5-flash',
|
||||||
contents=contents,
|
contents=contents,
|
||||||
config=types.GenerateContentConfig(
|
config=types.GenerateContentConfig(
|
||||||
system_instruction=prompt,
|
system_instruction=system_instruction,
|
||||||
# temperature=0.3,
|
# temperature=0.3,
|
||||||
)
|
)
|
||||||
)
|
)
|
||||||
@@ -99,7 +107,7 @@ class LCChatBot(ls.LitAPI):
|
|||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
sys.stdout = open('permanent.log', 'w', encoding='utf-8')
|
# sys.stdout = open('permanent.log', 'w', encoding='utf-8')
|
||||||
|
|
||||||
import asyncio
|
import asyncio
|
||||||
|
|
||||||
@@ -137,23 +145,67 @@ if __name__ == "__main__":
|
|||||||
print(chunk, end="", flush=True)
|
print(chunk, end="", flush=True)
|
||||||
|
|
||||||
|
|
||||||
text_list = [
|
# text_list = [
|
||||||
'我要去参加好朋友的婚礼，你能帮我挑一套衣服吗？',
|
# '我要去参加好朋友的婚礼，你能帮我挑一套衣服吗？',
|
||||||
'I need something to wear for a big presentation at work tomorrow. I want to look powerful but still approachable.',
|
# 'I need something to wear for a big presentation at work tomorrow. I want to look powerful but still approachable.',
|
||||||
'Who do you think is the best world leader right now?',
|
# 'Who do you think is the best world leader right now?',
|
||||||
'I’m going on a trip to Paris next week and need some outfits.',
|
# 'I’m going on a trip to Paris next week and need some outfits.',
|
||||||
'Help me find a cool outfit for a rock concert. I hate wearing dresses.',
|
# 'Help me find a cool outfit for a rock concert. I hate wearing dresses.',
|
||||||
'I want to look very cool, 或者是那种很有个性的风格 for a gallery opening.',
|
# 'I want to look very cool, 或者是那种很有个性的风格 for a gallery opening.',
|
||||||
"I'm going to a gala. Please list 5 different dress styles for me and use bold text for the names.",
|
# "I'm going to a gala. Please list 5 different dress styles for me and use bold text for the names.",
|
||||||
"I'm feeling really sad today and just want an outfit that matches my mood."
|
# "I'm feeling really sad today and just want an outfit that matches my mood."
|
||||||
]
|
# ]
|
||||||
for text in text_list:
|
# for text in text_list:
|
||||||
asyncio.run(run_simple_test(text))
|
# asyncio.run(run_simple_test(text))
|
||||||
# print("\n" + "=" * 50)
|
|
||||||
# # 启动异步事件循环
|
|
||||||
# try:
|
async def run_interactive_test():
|
||||||
# asyncio.run(run_simple_test())
|
"""
|
||||||
# except Exception as e:
|
手动输入测试：模拟真实用户与 AI 的多轮对话
|
||||||
# print(f"\n发生致命错误: {e}")
|
"""
|
||||||
#
|
# 1. 初始化
|
||||||
sys.stdout.close()
|
chatbot_api = LCChatBot()
|
||||||
|
chatbot_api.setup(device="cpu")
|
||||||
|
|
||||||
|
session_id = "manual_test_session"
|
||||||
|
user_id = "test_user"
|
||||||
|
|
||||||
|
print("--- 👗 欢迎进入 AI 造型师测试 (输入 'quit' 退出) ---")
|
||||||
|
|
||||||
|
|
||||||
|
# 清空旧的测试记录，开始新会话
|
||||||
|
chatbot_api.redis.clear_history(session_id)
|
||||||
|
print("✅ 已清空历史记录，开始新会话。")
|
||||||
|
|
||||||
|
|
||||||
|
while True:
|
||||||
|
# 2. 获取手动输入
|
||||||
|
user_input = input("\nUser (你): ")
|
||||||
|
if user_input.lower() in ['quit', 'exit', '退出']:
|
||||||
|
break
|
||||||
|
|
||||||
|
# 3. 构造请求
|
||||||
|
request_data = PredictRequest(
|
||||||
|
user_id=user_id,
|
||||||
|
session_id=session_id,
|
||||||
|
user_message=user_input,
|
||||||
|
gender="female" # 或者根据需要修改
|
||||||
|
)
|
||||||
|
|
||||||
|
print("Agent (AI): ", end="")
|
||||||
|
|
||||||
|
# 4. 调用并流式打印
|
||||||
|
# 注意：这里的 predict 内部现在会计算 turn_count
|
||||||
|
response_generator = chatbot_api.predict(request_data)
|
||||||
|
|
||||||
|
full_response = ""
|
||||||
|
async for chunk in response_generator:
|
||||||
|
print(chunk, end="", flush=True)
|
||||||
|
full_response += chunk
|
||||||
|
|
||||||
|
# 将 AI 的回复存入 Redis (如果你的 predict 函数里没存的话)
|
||||||
|
# chatbot_api.redis.add_message(session_id, Role.ASSISTANT, full_response)
|
||||||
|
|
||||||
|
print() # 换行
|
||||||
|
|
||||||
|
asyncio.run(run_interactive_test())
|
||||||
|
|||||||
@@ -7,6 +7,16 @@ CONVERSATION GOALS:
|
|||||||
3. **Vibe/Details:** Gather mood or constraints (e.g., comfort, specific colors).
|
3. **Vibe/Details:** Gather mood or constraints (e.g., comfort, specific colors).
|
||||||
4. **Item Preference:** Identify preferences for silhouettes or specific items.
|
4. **Item Preference:** Identify preferences for silhouettes or specific items.
|
||||||
|
|
||||||
|
CONVERSATION FLOW:
|
||||||
|
- You have a maximum of 3 turns (one turn = one user message and one assistant response) to gather information.
|
||||||
|
- On the 3rd turn, or *once you have sufficient info*, do NOT ask further questions.
|
||||||
|
- MANDATORY CLOSING: You must end your final response using the FINAL TURN CLOSING RULE below.
|
||||||
|
|
||||||
|
FINAL TURN CLOSING RULE:
|
||||||
|
- Once requirements are gathered, do NOT ask any more questions.
|
||||||
|
- End your final response with a creative and friendly variation of the closing phrase.
|
||||||
|
- REQUIRED COMPONENTS: Every variation MUST mention the "Generate button" and that it is located "below".
|
||||||
|
|
||||||
PERSONALITY AND COMMUNICATION RULES:
|
PERSONALITY AND COMMUNICATION RULES:
|
||||||
- Always reply in English only. If the user speaks another language, state in English that you cannot speak that language and steer back to styling.
|
- Always reply in English only. If the user speaks another language, state in English that you cannot speak that language and steer back to styling.
|
||||||
- Keep responses extremely SHORT (maximum 2 sentences).
|
- Keep responses extremely SHORT (maximum 2 sentences).
|
||||||
@@ -24,7 +34,7 @@ OUTPUT FORMAT INSTRUCTION:
|
|||||||
- **ONLY** output the plain text response spoken by the AI Assistant.
|
- **ONLY** output the plain text response spoken by the AI Assistant.
|
||||||
|
|
||||||
EXAMPLE DIALOGUES:
|
EXAMPLE DIALOGUES:
|
||||||
User: 我想找件衣服参加婚礼。 Response: I am so sorry, but I can only speak English! I would love to help you find the perfect wedding guest look if you can tell me the dress code or the venue vibe.
|
User: 我想找件衣服参加婚礼。 Response: I would love to help you find the perfect wedding guest look if you can tell me the dress code or the venue vibe.
|
||||||
|
|
||||||
User: I need a chic outfit for dinner. Response: That sounds like a fabulous evening! Is this a romantic date or a casual night out with friends, and do you prefer a sleek dress or tailored separates?
|
User: I need a chic outfit for dinner. Response: That sounds like a fabulous evening! Is this a romantic date or a casual night out with friends, and do you prefer a sleek dress or tailored separates?
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user