update python code tool

agents/tools/python_code_tool.py

@@ -13,7 +13,7 @@ Please write Python script to fulfill the following requirement:
 {input}
 ---
 
-Only output the code section without including code block symbols.
+Only output the code section with code block.
 """
 
 generate_python_code_promopt = PromptTemplate(input_variables=["input"], template=generate_python_code,)
@@ -26,10 +26,16 @@ def generate_and_excute_python_code(input: str) -> str:
     '''useful for when you need to generate python code and excute it'''
     answer_code = generate_code_chain.run(input)
     print(answer_code)
-    start = answer_code.find('python') + len('python')
-    end = answer_code.rfind('```')
-    code_content = answer_code[start:end].strip()
-
+    code_content = answer_code
+    if('```python' in answer_code): 
+        start = answer_code.find('```python') + len('```python')
+        end = answer_code.rfind('```')
+        code_content = answer_code[start:end].strip()
+    elif("```" in answer_code): 
+        start = answer_code.find('```') + len('```')
+        end = answer_code.rfind('```')
+        code_content = answer_code[start:end].strip()
+    
     print(code_content)
     python_repl = PythonREPL()
     result = python_repl.run(code_content)