call_function.py

from google import genai
from google.genai import types
from functions.get_files_info import schema_get_files_info, get_files_info
from functions.get_file_content import schema_get_file_content_info, get_file_content
from functions.write_file import schema_write_file_info, write_file
from functions.run_python_file import schema_run_python_file_info, run_python_file

available_functions = types.Tool(
    function_declarations = [
        schema_get_files_info,
        schema_get_file_content_info,
        schema_write_file_info,
        schema_run_python_file_info
    ],
)

def call_function(function_call, verbose=False):
    # This is a placeholder implementation. In a real implementation, you would have logic here to call the appropriate function based on the function_call.name and function_call.args
    if verbose:
        print(f"Calling function: {function_call.name}({function_call.args})")
    else:
        print(f" - Calling function: {function_call.name}")
    
    function_map = {
        "get_file_content" : get_file_content,
        "get_files_info" : get_files_info,
        "write_file" : write_file,
        "run_python_file" : run_python_file
    }
    
    function_name = function_call.name or ""
    if function_name not in function_map:
        return types.Content(
        role="tool",
        parts=[
            types.Part.from_function_response(
                name=function_name,
                response={"error": f"Unknown function: {function_name}"},
            )
        ],
        )
        
    args = dict(function_call.args) if function_call.args is not None else {}
    args["working_directory"] = "./calculator"
    function_result = function_map[function_name](**args)
    return types.Content(
    role="tool",
    parts=[
        types.Part.from_function_response(
            name=function_name,
            response={"result": function_result},
        )
    ],
)