From 5ea362c96da92eff13e1c2252e8030515d71c085 Mon Sep 17 00:00:00 2001
From: "Jonathan C. McKinney" <pseudotensor@gmail.com>
Date: Thu, 24 Oct 2024 14:09:58 -0700
Subject: [PATCH] Only count executable code blocks for tool usage

---
 openai_server/autogen_utils.py | 3 ++-
 src/version.py                 | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/openai_server/autogen_utils.py b/openai_server/autogen_utils.py
index cf3c2a6a4..166aea732 100644
--- a/openai_server/autogen_utils.py
+++ b/openai_server/autogen_utils.py
@@ -363,6 +363,7 @@ def is_in_container() -> bool:
 
     def _execute_code_dont_check_setup(self, code_blocks: List[CodeBlock]) -> CommandLineCodeResult:
         multiple_executable_code_detected = False
+        code_blocks_exec = code_blocks
         try:
             # skip code blocks with # execution: false
             code_blocks_len0 = len(code_blocks)
@@ -433,7 +434,7 @@ def _execute_code_dont_check_setup(self, code_blocks: List[CodeBlock]) -> Comman
                 raise
         
         # Update agent tool usage if there is any
-        self.update_agent_tool_usages(code_blocks)
+        self.update_agent_tool_usages(code_blocks_exec)
         # Truncate output if it is too long
         ret = self.truncate_output(ret)
         # Add executed code note if needed
diff --git a/src/version.py b/src/version.py
index 5e70b8f0a..343f32d11 100644
--- a/src/version.py
+++ b/src/version.py
@@ -1 +1 @@
-__version__ = "0f9ea9f3ba711cbaa5b97adac2f1e48190e343bf"
+__version__ = "04fc086ef4c0c1ff155c5213d4ee4f41ba79f784"