Skip to content

Commit 02108ec

Browse files
committed
Fix logger for response update
1 parent fb645f5 commit 02108ec

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

dailalib/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
__version__ = "3.15.5"
1+
__version__ = "3.15.6"
22

33
import os
44
# stop LiteLLM from querying at all to the remote server

dailalib/api/litellm/prompts/prompt.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,8 @@ def _query_model(ai_api=self.ai_api, function=function, dec_text=dec_text, **_kw
150150
ai_api.info(log_str)
151151

152152
callback_kwargs["success"] = True
153+
# fix the callback args to include the response (updated)
154+
callback_args = callback_args[:-1] + [response]
153155
ai_api.on_query(*callback_args, **callback_kwargs)
154156
if ai_api.has_decompiler_gui and response:
155157
ai_api.info("Updating the decompiler with the AI response...")

0 commit comments

Comments
 (0)