tecuts commited on
Commit
9ca0227
·
verified ·
1 Parent(s): d805a2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -65,7 +65,8 @@ When you need current information, recent events, specific facts, or when the us
65
  2. Synthesize information from multiple sources when available
66
  3. Clearly indicate when information comes from search results
67
  4. Provide comprehensive, well-structured answers
68
- 5. Cite sources appropriately
 
69
 
70
  Current date: {current_date}"""
71
 
@@ -370,13 +371,15 @@ async def chat_stream_endpoint(request: Request, _: None = Depends(verify_origin
370
  use_search = data.get("use_search", False)
371
  temperature = max(0, min(2, data.get("temperature", 0.7)))
372
  conversation_history = data.get("history", [])
 
373
 
374
  if not user_message:
375
  raise HTTPException(status_code=400, detail="No message provided")
376
 
377
  # Prepare messages
378
  current_date = datetime.now().strftime("%Y-%m-%d")
379
- system_content = (SYSTEM_PROMPT_WITH_SEARCH if use_search else SYSTEM_PROMPT_NO_SEARCH).format(current_date=current_date)
 
380
  messages = [{"role": "system", "content": system_content}] + conversation_history + [{"role": "user", "content": user_message}]
381
 
382
  logger.info(f"Stream request - search: {use_search}, temp: {temperature}, message: {user_message[:100]}...")
 
65
  2. Synthesize information from multiple sources when available
66
  3. Clearly indicate when information comes from search results
67
  4. Provide comprehensive, well-structured answers
68
+ 5. Cite sources appropriately with links.
69
+ 6. If search results conflict with my knowledge, prioritize the search results.
70
 
71
  Current date: {current_date}"""
72
 
 
371
  use_search = data.get("use_search", False)
372
  temperature = max(0, min(2, data.get("temperature", 0.7)))
373
  conversation_history = data.get("history", [])
374
+ user_prompt = data.get("system_prompt")
375
 
376
  if not user_message:
377
  raise HTTPException(status_code=400, detail="No message provided")
378
 
379
  # Prepare messages
380
  current_date = datetime.now().strftime("%Y-%m-%d")
381
+ system_content = (SYSTEM_PROMPT_WITH_SEARCH if use_search else user_prompt#SYSTEM_PROMPT_NO_SEARCH
382
+ ).format(current_date=current_date)
383
  messages = [{"role": "system", "content": system_content}] + conversation_history + [{"role": "user", "content": user_message}]
384
 
385
  logger.info(f"Stream request - search: {use_search}, temp: {temperature}, message: {user_message[:100]}...")