Update app/main.py
Browse files- 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
|
506 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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
|