Skip image, code generation if in research mode

This commit is contained in:
sabaimran 2024-10-10 18:06:29 -07:00
parent 1b13d069f5
commit 01a58b71a5

View file

@ -952,7 +952,7 @@ async def chat(
) )
## Gather Code Results ## Gather Code Results
if ConversationCommand.Code in conversation_commands: if ConversationCommand.Code in conversation_commands and pending_research:
try: try:
previous_iteration_history = ( previous_iteration_history = (
f"# Iteration 1:\n#---\nNotes:\n{compiled_references}\n\nOnline Results:{online_results}" f"# Iteration 1:\n#---\nNotes:\n{compiled_references}\n\nOnline Results:{online_results}"
@ -993,7 +993,7 @@ async def chat(
# Generate Output # Generate Output
## Generate Image Output ## Generate Image Output
if ConversationCommand.Image in conversation_commands: if ConversationCommand.Image in conversation_commands and pending_research:
async for result in text_to_image( async for result in text_to_image(
q, q,
user, user,