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
|