load("@fbcode_macros//build_defs:python_library.bzl", "python_library")

oncall("llamafirewall")

python_library(
    name = "base_llm",
    srcs = ["base_llm.py"],
    # @lint-ignore BUCKLINT needed for OSS
    base_module = "llamafirewall.utils",
    deps = [
        "fbsource//third-party/pypi/openai:openai",
        "fbsource//third-party/pypi/pydantic:pydantic",
    ],
)

python_library(
    name = "utils",
    srcs = ["__init__.py"],
    # @lint-ignore BUCKLINT needed for OSS
    base_module = "llamafirewall.utils",
    deps = [
        ":base_llm",
    ],
)
