Update agents/tools/ai_tools.py

agents/tools/ai_tools.py

@@ -55,7 +55,6 @@ class AITools():
             
             file = client.files.upload(file=file_path)
 
-            ###
             while True:
                 media_file = client.files.get(name=file.name)
                 if media_file.state == "ACTIVE":
@@ -63,7 +62,6 @@ class AITools():
                 elif media_file.state == "FAILED":
                     raise RuntimeError("Media file processing failed")
                 time.sleep(1)            
-            ###
 
             response = client.models.generate_content(
                 model=model,
@@ -75,7 +73,7 @@ class AITools():
                 )
             )
 
-            result = response.text.strip()
+            result = response.text
             print(f"🛠️ AITools: {tool_name}: result={result}")
             return result
         except Exception as e:
@@ -85,7 +83,7 @@ class AITools():
     def _extract_execution_result(response):
         for part in response.candidates[0].content.parts:
             if part.code_execution_result is not None:
-                return part.code_execution_result.output.strip()
+                return part.code_execution_result.output
 
         return None
 
@@ -118,7 +116,7 @@ class AITools():
                 )
             )
 
-            result = response.text.strip()
+            result = response.text
             print(f"🛠️ AITools: web_search_tool: result={result}")
             return result
         except Exception as e:
@@ -162,7 +160,7 @@ class AITools():
                     instruction=question,
                     url=url_str,
                     command_type="act" # TODO: act, extract, observe
-                ).strip()
+                )
 
                 print(f"🛠️ AITools: web_browser_tool: result={result}")
                 return result
@@ -248,7 +246,7 @@ class AITools():
                         thinking_level=THINKING_LEVEL_YOUTUBE_ANALYSIS
                     )
                 )
-            ).strip()
+            )
 
             print(f"🛠️ AITools: youtube_analysis_tool: result={result}")
             return result
@@ -299,7 +297,7 @@ class AITools():
                 )
             )
           
-            result = response.text.strip()
+            result = response.text
             print(f"🛠️ AITools: document_analysis_tool: result={result}")
             return result
         except Exception as e:
@@ -435,7 +433,7 @@ class AITools():
 
             for part in response.parts:
                 if part.text is not None:
-                    result = part.text.strip()
+                    result = part.text
                     break
 
             board = chess.Board(result) # FEN validation
@@ -498,7 +496,7 @@ class AITools():
             
             for part in response.parts:
                 if part.text is not None:
-                    result = part.text.strip()
+                    result = part.text
                     break
 
             print(f"🛠️ AITools: algebraic_notation_tool: result={result}")
@@ -537,7 +535,7 @@ class AITools():
                 )
             )
     
-            result = response.text.strip()
+            result = response.text
             print(f"🛠️ AITools: final_answer_tool: result={result}")
             return result
         except Exception as e: