apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-06-28 08:49:42 +02:00
Code Issues Projects Releases Packages Wiki Activity

fixes #100

Browse source 
Fixes multiple LLM generation when there are multiple tool calls from LLM
This commit is contained in:
Abhishek Kumar 2026-01-03 12:45:44 +05:30
parent a33fa6cffe
commit 1ef07c80b6
17 changed files with 747 additions and 454 deletions
Download patch file Download diff file Expand all files Collapse all files

6
api/services/workflow/pipecat_engine_callbacks.py
View file

@ -114,10 +114,10 @@ def create_max_duration_callback(engine: "PipecatEngine"):
def create_generation_started_callback(engine: "PipecatEngine"):
"""Return a callback that resets flags at the start of each LLM generation."""
async def handle_generation_started(): # noqa: D401
async def handle_generation_started():
logger.debug("LLM generation started in callback processor")
# Clear reference text from previous generation
engine._current_llm_reference_text = ""
engine._current_llm_generation_reference_text = ""
return handle_generation_started
@ -184,7 +184,7 @@ def create_aggregation_correction_callback(engine: "PipecatEngine"):
return "".join(out_chars)
def correct_aggregation(corrupted: str) -> str:
reference = engine._current_llm_reference_text
reference = engine._current_llm_generation_reference_text
if not reference:
logger.warning("No reference text available for aggregation correction")