Skip to content

Commit 977454a

Browse files
committed
Yield original message type
1 parent ee41cbe commit 977454a

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

text_2_sql/autogen/src/autogen_text_2_sql/custom_agents/parallel_query_solving_agent.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -75,10 +75,7 @@ async def consume_inner_messages_from_agentic_flow(
7575
complete_inner_messages[identifier] = []
7676
complete_inner_messages[identifier].append(inner_message)
7777

78-
yield TextMessage(
79-
content=json.dumps(inner_message),
80-
source=f"{self.name}-{identifier}",
81-
)
78+
yield inner_message
8279

8380
inner_solving_generators = []
8481
complete_inner_messages = {}
@@ -102,6 +99,7 @@ async def consume_inner_messages_from_agentic_flow(
10299
)
103100
)
104101

102+
logging.info("Created %i Inner Solving Generators", inner_solving_generators)
105103
logging.info("Starting Inner Solving Generators")
106104
combined_message_streams = stream.merge(*inner_solving_generators)
107105

0 commit comments

Comments
 (0)