Don't reset research mode after completing research execution

This commit is contained in:
sabaimran 2024-11-01 13:00:11 -07:00
parent c1c779a7ef
commit cead1598b9

View file

@ -757,7 +757,6 @@ async def chat(
yield research_result yield research_result
# researched_results = await extract_relevant_info(q, researched_results, agent) # researched_results = await extract_relevant_info(q, researched_results, agent)
in_research_mode = False
logger.info(f"Researched Results: {researched_results}") logger.info(f"Researched Results: {researched_results}")
for cmd in conversation_commands: for cmd in conversation_commands:
@ -1022,7 +1021,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 not in_research_mode:
async for result in text_to_image( async for result in text_to_image(
defiltered_query, defiltered_query,
user, user,