Update z_generate.py

z_generate.py

@@ -1,3 +1,4 @@
+import ast
 from huggingface_hub import InferenceClient
 import os 
 from typing import List
@@ -154,18 +155,23 @@ Question: {question}""".format(context=context, question=query),
 
 
 def parse(value: str) -> List[int]:
+    """
+    Extracts a list of numbers from the given string.
+
+    Parameters:
+        value (str): The input string containing the list of numbers.
+
+    Returns:
+        list: A list of numbers if found, otherwise an empty list.
+    """
     try:
-        # Convert the string to a Python list using eval safely with literal_eval
-        from ast import literal_eval
-        parsed_value = literal_eval(value)
-        
-        # Ensure it's a list of numbers
-        if isinstance(parsed_value, list) and all(isinstance(i, (int, float)) for i in parsed_value):
-            return parsed_value
-        else:
-            print("The input string is not a valid list of numbers.")
-    except Exception as e:
-        print(f"Invalid input string: {value}. Error: {e}")
+        # Find the substring that looks like a list
+        start = value.index('[')
+        end = value.index(']')
+        # Extract and parse it into a Python list
+        return ast.literal_eval(value[start:end+1])
+    except (ValueError, SyntaxError):
+        # Return an empty list if parsing fails
         return []