call_function.py

import os
import sys
from dotenv import load_dotenv
from google import genai
from google.genai import types

from functions.get_files_info import get_files_info
from functions.get_file_content import get_file_content
from functions.write_file import write_file
from functions.run_python_file import run_python_file

FUNCTIONS = {
    "get_files_info": get_files_info,
    "get_file_content": get_file_content,
    "write_file": write_file,
    "run_python_file": run_python_file
}

def call_function(function_call_part, verbose=False):
    fn_name = function_call_part.name
    kwargs = dict(function_call_part.args)
    kwargs["working_directory"] = "./calculator"

    if verbose:
        print(f"Calling function: {fn_name}({function_call_part.args})")
    else:
        print(f" - Calling function: {fn_name}")

    fn = FUNCTIONS.get(fn_name)

    if fn is None:
        return types.Content(
            role="tool",
            parts=[
                types.Part.from_function_response(
                    name=fn_name,
                    response={"error": f"Unknown function: {fn_name}"},
                )
            ],
        )
    
    result = fn(**kwargs)

    return types.Content(
        role="tool",
        parts=[
            types.Part.from_function_response(
                name=fn_name,
                response={"result": result},
            )
        ],
    )