bibibi12345 commited on
Commit
f97da8b
·
verified ·
1 Parent(s): ffc6a10

Update app/main.py

Browse files
Files changed (1) hide show
  1. app/main.py +22 -3
app/main.py CHANGED
@@ -502,8 +502,27 @@ Ready for your request."""
502
  # For simplicity, we'll just pass it through as is
503
  new_messages.append(message)
504
  else:
505
- # For non-user messages, keep as is
506
- new_messages.append(message)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
507
 
508
  # Now use the standard function to convert to Gemini format
509
  return create_gemini_prompt(new_messages)
@@ -935,4 +954,4 @@ def health_check(api_key: str = Depends(get_api_key)):
935
  }
936
  }
937
 
938
- # Removed /debug/credentials endpoint
 
502
  # For simplicity, we'll just pass it through as is
503
  new_messages.append(message)
504
  else:
505
+ # For non-user messages (assistant messages)
506
+ # Check if this is the last non-user message in the list
507
+ is_last_assistant = True
508
+ for remaining_msg in messages[i+1:]:
509
+ if remaining_msg.role != "user":
510
+ is_last_assistant = False
511
+ break
512
+
513
+ if is_last_assistant:
514
+ # URL encode the last assistant message content
515
+ if isinstance(message.content, str):
516
+ new_messages.append(OpenAIMessage(
517
+ role=message.role,
518
+ content=urllib.parse.quote(message.content)
519
+ ))
520
+ else:
521
+ # For non-string content, keep as is
522
+ new_messages.append(message)
523
+ else:
524
+ # For other non-user messages, keep as is
525
+ new_messages.append(message)
526
 
527
  # Now use the standard function to convert to Gemini format
528
  return create_gemini_prompt(new_messages)
 
954
  }
955
  }
956
 
957
+ # Removed /debug/credentials endpoint