tooluniverse 0.2.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

tooluniverse/utils.py

@@ -1,5 +1,64 @@
 import yaml
 import json
+import re
+import hashlib
+import os
+from typing import Dict, Any, Union, List
+from huggingface_hub import hf_hub_download
+from pydantic._internal._model_construction import ModelMetaclass
+
+
+def download_from_hf(tool_config):
+    # Extract dataset configuration
+    hf_parameters = tool_config.get("hf_dataset_path")
+    relative_local_path = hf_parameters.get("save_to_local_dir")
+
+    # Compute absolute path to save locally
+    if not os.path.isabs(relative_local_path):
+        project_root = os.path.dirname(os.path.dirname(os.path.dirname(__file__)))
+        absolute_local_dir = os.path.join(project_root, relative_local_path)
+    else:
+        absolute_local_dir = relative_local_path
+
+    # Ensure the directory exists
+    os.makedirs(absolute_local_dir, exist_ok=True)
+
+    # Download the CSV from Hugging Face Hub
+    try:
+        # Prepare download arguments
+        download_args = {
+            "repo_id": hf_parameters.get("repo_id"),
+            "filename": hf_parameters.get("path_in_repo"),
+            "repo_type": "dataset",
+            "local_dir": absolute_local_dir,
+        }
+
+        # Only add token if it's not None and not empty
+        token = hf_parameters.get("token")
+        if token is not None and token.strip():
+            download_args["token"] = token
+        else:
+            download_args["token"] = False
+
+        downloaded_path = hf_hub_download(**download_args)
+
+        # The downloaded file path is returned by hf_hub_download
+        result = {"success": True, "local_path": downloaded_path}
+    except Exception as e:
+        result = {"success": False, "error": str(e)}
+
+    return result
+
+
+def get_md5(input_str):
+    # Create an MD5 hash object
+    md5_hash = hashlib.md5()
+
+    # Encode the string and update the hash object
+    md5_hash.update(input_str.encode("utf-8"))
+
+    # Return the hexadecimal MD5 digest
+    return md5_hash.hexdigest()
 
 
 def yaml_to_dict(yaml_file_path):
@@ -13,7 +72,7 @@ def yaml_to_dict(yaml_file_path):
         dict: Dictionary representation of the YAML file content.
     """
     try:
-        with open(yaml_file_path, 'r') as file:
+        with open(yaml_file_path, "r") as file:
             yaml_dict = yaml.safe_load(file)
             return yaml_dict
     except FileNotFoundError:
@@ -32,7 +91,7 @@ def read_json_list(file_path):
     Returns:
     list: A list of dictionaries containing the JSON objects.
     """
-    with open(file_path, 'r') as file:
+    with open(file_path, "r") as file:
         data = json.load(file)
     return data
 
@@ -45,18 +104,30 @@ def evaluate_function_call(tool_definition, function_call):
         "number": float,
         "boolean": bool,
         "array": list,
-        "object": dict
+        "object": dict,
+        "pydantic": ModelMetaclass,
     }
 
     # Check if the function name matches
     if tool_definition["name"] != function_call["name"]:
         return False, "Function name does not match."
 
+    # Handle the case where properties is None (no arguments expected)
+    if tool_definition["parameter"]["properties"] is None:
+        # If properties is None, the function should not have any arguments
+        if function_call.get("arguments") and len(function_call["arguments"]) > 0:
+            return False, "This function does not accept any arguments."
+        return True, "Function call is valid."
+
     # Check if all required parameters are present
-    required_params = [key for key, value in tool_definition["parameter"]
-                       ["properties"].items() if value.get("required", False)]
+    required_params = [
+        key
+        for key, value in tool_definition["parameter"]["properties"].items()
+        if value.get("required", False)
+    ]
     missing_params = [
-        param for param in required_params if param not in function_call["arguments"]]
+        param for param in required_params if param not in function_call["arguments"]
+    ]
     if missing_params:
         return False, f"Missing required parameters: {missing_params}"
 
@@ -69,12 +140,37 @@ def evaluate_function_call(tool_definition, function_call):
         if param not in valid_params:
             invalid_params.append(param)
         else:
-            expected_type = valid_params[param]["type"]
+            param_schema = valid_params[param]
+
+            # Handle both simple and complex parameter schemas
+            expected_type = None
+
+            # Case 1: Simple schema with direct "type" field
+            if "type" in param_schema:
+                expected_type = param_schema["type"]
+
+            # Case 2: Complex schema with "anyOf" (common in MCP tools)
+            elif "anyOf" in param_schema:
+                # Extract the primary type from anyOf, ignoring null types
+                for type_option in param_schema["anyOf"]:
+                    if type_option.get("type") and type_option["type"] != "null":
+                        expected_type = type_option["type"]
+                        break
+
+            # If we still don't have a type, skip validation for this parameter
+            if not expected_type:
+                continue
+
             if expected_type not in type_map:
                 return False, f"Unsupported parameter type: {expected_type}"
-            if not isinstance(value, type_map[expected_type]):
-                type_mismatches.append(
-                    (param, expected_type, type(value).__name__))
+
+            # Special handling for 'number' type which should accept both int and float
+            if expected_type == "number":
+                if not isinstance(value, (int, float)):
+                    type_mismatches.append((param, expected_type, type(value).__name__))
+            else:
+                if not isinstance(value, type_map[expected_type]):
+                    type_mismatches.append((param, expected_type, type(value).__name__))
 
     if invalid_params:
         return False, f"Invalid parameters provided: {invalid_params}"
@@ -84,6 +180,7 @@ def evaluate_function_call(tool_definition, function_call):
 
     return True, "Function call is valid."
 
+
 def evaluate_function_call_from_toolbox(toolbox, function_call):
     tool_name = function_call["name"]
     this_tool_dec = toolbox.get_one_tool_by_one_name(tool_name)
@@ -91,9 +188,11 @@ def evaluate_function_call_from_toolbox(toolbox, function_call):
         return False, "Tool not found."
     results, results_message = evaluate_function_call(this_tool_dec, function_call)
     return results, results_message
-    
 
-def compare_function_calls(pred_function_call, gt_function_call, compare_arguments=True, compare_value=True):
+
+def compare_function_calls(
+    pred_function_call, gt_function_call, compare_arguments=True, compare_value=True
+):
     # Extracting the name and arguments from the predicted function call
     pred_name = pred_function_call["name"]
     pred_arguments = pred_function_call["arguments"]
@@ -111,7 +210,10 @@ def compare_function_calls(pred_function_call, gt_function_call, compare_argumen
         if set(pred_arguments.keys()) != set(gt_arguments.keys()):
             missing_in_pred = set(gt_arguments.keys()) - set(pred_arguments.keys())
             missing_in_gt = set(pred_arguments.keys()) - set(gt_arguments.keys())
-            return False, f"Argument keys do not match. Missing in predicted: {missing_in_pred}, Missing in ground truth: {missing_in_gt}"
+            return (
+                False,
+                f"Argument keys do not match. Missing in predicted: {missing_in_pred}, Missing in ground truth: {missing_in_gt}",
+            )
     if compare_value:
         # Compare argument values
         mismatched_values = []
@@ -125,12 +227,12 @@ def compare_function_calls(pred_function_call, gt_function_call, compare_argumen
     return True, "Function calls match."
 
 
-def extract_function_call_json(lst, return_message=False, verbose=True):
+def extract_function_call_json(lst, return_message=False, verbose=True, format="llama"):
     if type(lst) is dict:
         if return_message:
             return lst, ""
         return lst
-    result_str = ''.join(lst)
+    result_str = "".join(lst)
     if verbose:
         print("\033[1;34mPossible LLM outputs for function call:\033[0m", result_str)
     try:
@@ -140,33 +242,259 @@ def extract_function_call_json(lst, return_message=False, verbose=True):
         return function_call_json
     except json.JSONDecodeError:
         try:
-            index_start = result_str.find(
-                '[TOOL_CALLS]') 
-            index_end = result_str.find('</s>')
-            if index_end == -1:
-                index_end = result_str.find('<|eom_id|>')
-            if index_end == -1:
-                function_call_str = result_str[index_start+ len('[TOOL_CALLS]'):]
-            else:
-                function_call_str = result_str[index_start+ len('[TOOL_CALLS]'):index_end]
-            # print("function_call_str", function_call_str)
-            function_call_json = json.loads(function_call_str.strip())
+            if format == "llama":
+                index_start = result_str.find("[TOOL_CALLS]")
+                index_end = result_str.find("</s>")
+                if index_end == -1:
+                    index_end = result_str.find("<|eom_id|>")
+                if index_end == -1:
+                    function_call_str = result_str[index_start + len("[TOOL_CALLS]") :]
+                else:
+                    function_call_str = result_str[
+                        index_start + len("[TOOL_CALLS]") : index_end
+                    ]
+                # print("function_call_str", function_call_str)
+                function_call_json = json.loads(function_call_str.strip())
+            elif format == "qwen":
+                index_start = result_str.find("<tool_call>")
+                function_call_str = result_str[index_start:]
+
+                pattern = re.compile(r"<tool_call>(.*?)</tool_call>", re.DOTALL)
+                matches = pattern.findall(function_call_str)
+                function_call_json = []
+
+                for match in matches:
+                    # Clean up the JSON string
+                    json_str = match.strip()
+                    data = json.loads(json_str)
+                    function_call_json.append(data)
+
             if return_message:
                 message = result_str[:index_start]
                 return function_call_json, message
             return function_call_json
-        except json.JSONDecodeError:
-            try:
-                print("Multiple function calls not implemented for 'llama' format.")
-                index_start = result_str.find(
-                    '<functioncall>') + len('<functioncall>')
-                index_end = result_str.find('</functioncall>')
-                function_call_str = result_str[index_start:index_end]
-                # function_call_str = function_call_str.replace("'", '"')
-                function_call_json = json.loads(function_call_str.strip())
-                return function_call_json
-            except json.JSONDecodeError as e:
-                print("Not a function call:", e)
-                if return_message:
-                    return None, result_str
-                return None
+
+        except json.JSONDecodeError as e:
+            print("Not a function call:", e)
+            if return_message:
+                return None, result_str
+            return None
+
+
+def validate_query(query: Dict[str, Any]) -> bool:
+    """
+    Validate a query dictionary for required fields and structure.
+
+    Args:
+        query (Dict[str, Any]): The query dictionary to validate
+
+    Returns:
+        bool: True if query is valid, False otherwise
+    """
+    if not isinstance(query, dict):
+        return False
+
+    # Check for basic required fields (customize based on your needs)
+    required_fields = ["query", "parameters"]
+    for field in required_fields:
+        if field not in query:
+            return False
+
+    # Additional validation logic can be added here
+    return True
+
+
+def normalize_gene_symbol(gene_symbol: str) -> str:
+    """
+    Normalize a gene symbol to standard format.
+
+    Args:
+        gene_symbol (str): The gene symbol to normalize
+
+    Returns:
+        str: Normalized gene symbol
+    """
+    if not isinstance(gene_symbol, str):
+        return str(gene_symbol)
+
+    # Convert to uppercase and strip whitespace
+    normalized = gene_symbol.strip().upper()
+
+    # Remove common prefixes/suffixes if needed
+    # This is a basic implementation - customize as needed
+    normalized = re.sub(r"^GENE[-_]?", "", normalized)
+    normalized = re.sub(r"[-_]?GENE$", "", normalized)
+
+    return normalized
+
+
+def format_api_response(
+    response_data: Any, format_type: str = "json"
+) -> Union[str, Dict[str, Any], List[Any]]:
+    """
+    Format API response data into a standardized format.
+
+    Args:
+        response_data (Any): The response data to format
+        format_type (str): The desired output format ('json', 'pretty', 'minimal')
+
+    Returns:
+        Union[str, Dict[str, Any]]: Formatted response
+    """
+    if format_type == "json":
+        if isinstance(response_data, (dict, list)):
+            return response_data
+        else:
+            return {"data": response_data, "status": "success"}
+
+    elif format_type == "pretty":
+        if isinstance(response_data, dict):
+            return json.dumps(response_data, indent=2, ensure_ascii=False)
+        else:
+            return json.dumps(
+                {"data": response_data, "status": "success"},
+                indent=2,
+                ensure_ascii=False,
+            )
+
+    elif format_type == "minimal":
+        if isinstance(response_data, dict) and "data" in response_data:
+            return response_data["data"]
+        else:
+            return response_data
+
+    else:
+        return response_data
+
+
+def validate_hook_config(config: Dict[str, Any]) -> bool:
+    """
+    Validate hook configuration for correctness and completeness.
+
+    This function checks that the hook configuration contains all required
+    fields and that the structure is valid for the hook system.
+
+    Args:
+        config (Dict[str, Any]): Hook configuration to validate
+
+    Returns:
+        bool: True if configuration is valid, False otherwise
+    """
+    try:
+        # Check for required top-level fields
+        if not isinstance(config, dict):
+            return False
+
+        # Validate global settings if present
+        if "global_settings" in config:
+            global_settings = config["global_settings"]
+            if not isinstance(global_settings, dict):
+                return False
+
+        # Validate hooks array
+        if "hooks" in config:
+            hooks = config["hooks"]
+            if not isinstance(hooks, list):
+                return False
+
+            for hook in hooks:
+                if not validate_hook_conditions(hook.get("conditions", {})):
+                    return False
+
+        # Validate tool-specific hooks
+        if "tool_specific_hooks" in config:
+            tool_hooks = config["tool_specific_hooks"]
+            if not isinstance(tool_hooks, dict):
+                return False
+
+            for _tool_name, tool_config in tool_hooks.items():
+                if not isinstance(tool_config, dict):
+                    return False
+                if "hooks" in tool_config:
+                    for hook in tool_config["hooks"]:
+                        if not validate_hook_conditions(hook.get("conditions", {})):
+                            return False
+
+        # Validate category hooks
+        if "category_hooks" in config:
+            category_hooks = config["category_hooks"]
+            if not isinstance(category_hooks, dict):
+                return False
+
+            for _category_name, category_config in category_hooks.items():
+                if not isinstance(category_config, dict):
+                    return False
+                if "hooks" in category_config:
+                    for hook in category_config["hooks"]:
+                        if not validate_hook_conditions(hook.get("conditions", {})):
+                            return False
+
+        return True
+
+    except Exception:
+        return False
+
+
+def validate_hook_conditions(conditions: Dict[str, Any]) -> bool:
+    """
+    Validate hook trigger conditions.
+
+    This function checks that the hook conditions are properly structured
+    and contain valid operators and thresholds.
+
+    Args:
+        conditions (Dict[str, Any]): Hook conditions to validate
+
+    Returns:
+        bool: True if conditions are valid, False otherwise
+    """
+    try:
+        if not isinstance(conditions, dict):
+            return False
+
+        # Validate output length conditions
+        if "output_length" in conditions:
+            length_condition = conditions["output_length"]
+            if not isinstance(length_condition, dict):
+                return False
+
+            # Check for required fields
+            if "threshold" not in length_condition:
+                return False
+            if "operator" not in length_condition:
+                return False
+
+            # Validate threshold is numeric
+            threshold = length_condition["threshold"]
+            if not isinstance(threshold, (int, float)) or threshold < 0:
+                return False
+
+            # Validate operator
+            operator = length_condition["operator"]
+            if operator not in [">", ">=", "<", "<="]:
+                return False
+
+        # Validate content type conditions
+        if "content_type" in conditions:
+            content_type = conditions["content_type"]
+            if not isinstance(content_type, str):
+                return False
+            if content_type not in ["json", "text", "xml", "csv"]:
+                return False
+
+        # Validate tool type conditions
+        if "tool_type" in conditions:
+            tool_type = conditions["tool_type"]
+            if not isinstance(tool_type, str):
+                return False
+
+        # Validate tool name conditions
+        if "tool_name" in conditions:
+            tool_name = conditions["tool_name"]
+            if not isinstance(tool_name, str):
+                return False
+
+        return True
+
+    except Exception:
+        return False