call_function.py

import os
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 , get_file_content
from functions.run_python_file import schema_run_python_file , run_python_file
from functions.write_file import schema_write_file , write_file


available_functions = types.Tool(
    function_declarations=[
        schema_get_files_info,
        schema_get_file_content,
        schema_run_python_file,
        schema_write_file,
    ]
)



def call_function(function_call_part, verbose=False):
    if verbose:
        print(f"Calling function: {function_call_part.name}({function_call_part.args})")
    else:
        print(f" - Calling function: {function_call_part.name}")
    
    functions = {
        "get_files_info": get_files_info,
        "get_file_content": get_file_content,
        "run_python_file": run_python_file,
        "write_file": write_file,
    }

    #working dirctory
    working_directory = './calculator'
    


    if function_call_part.name in functions:
        func = functions[function_call_part.name]
        function_result = func(working_directory,**function_call_part.args)

    if function_call_part.name not in functions:
        return types.Content(
            role="tool" ,
            parts=[
                types.Part.from_function_response(
                    name=function_call_part.name,
                    response={"error": f"Unknown function: {function_call_part.name}"},
                    )
                ],
            )

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