main.py

import argparse
import os

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

from functions.call_function import call_function
from functions.function_bindings import function_tools
from functions.prompt import system_prompt
from lib.util import log_exit, parse_verbose_info


def main():
    try:
        load_dotenv()

        parser = argparse.ArgumentParser(description="Chatbot")
        parser.add_argument(
            "--verbose", action="store_true", help="Enable verbose output"
        )
        parser.add_argument("user_prompt", type=str, help="User prompt")
        args = parser.parse_args()

        api_key = os.environ.get("GEMINI_API_KEY")
        client = genai.Client(api_key=api_key)

        messages = [
            types.Content(role="user", parts=[types.Part(text=args.user_prompt)])
        ]
        for _ in range(20):
            response = client.models.generate_content(
                model="gemini-2.5-flash",
                contents=messages,
                config=types.GenerateContentConfig(
                    tools=[function_tools], system_instruction=system_prompt
                ),
            )
            if response.candidates:
                for candidate in response.candidates:
                    if candidate.content.parts is not None:
                        messages.append(
                            types.Content(role="user", parts=candidate.content.parts)
                        )
            if response.function_calls:
                for fc in response.function_calls:
                    function_call_result = call_function(fc, args.verbose)
                    if len(function_call_result.parts) == 0:
                        raise Exception("empty function call result parts")
                    if function_call_result.parts[0].function_response is None:
                        raise Exception("empty function call result parts response")
                    print(
                        f"-> {function_call_result.parts[0].function_response.response}"
                    )
                    messages.append(
                        types.Content(role="user", parts=function_call_result.parts)
                    )
            if response.function_calls is None and response.text is not None:
                response_message = response.text
                if args.verbose:
                    response_message = f"{response_message}\n{parse_verbose_info(args.user_prompt, response)}"
                log_exit(response_message, 0)
            if response.text is None and response.function_calls is None:
                log_exit("No response from model, exiting", 1)

    except Exception as e:
        raise RuntimeError("There was an error returning a response:", e)


if __name__ == "__main__":
    main()