genmedia-izumi-agent/pyproject.toml at main · GoogleCloudPlatform/genmedia-izumi-agent · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
# Copyright 2025 Google LLC
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

[project]
name = "genmedia-izumi-agent"
version = "0.3.0"
description = "Combined repository for GenMedia mediagent-kit and reference demos."
authors = [
    {name = "Khanh LeViet", email = "khanhlvg@google.com"},
    {name = "Daniel Vlasic", email = "dvlasic@google.com"},
    {name = "Ye Jin", email = "jinye@google.com"},
    {name = "Rhyard Zhu", email = "shoudongzhu@google.com"},
]
dependencies = [
    "beautifulsoup4",
    "fastapi",
    "firebase-admin",
    "google-adk==1.29.0",

    "google-cloud-aiplatform[adk,agent-engines]>=1.132.0",
    "google-cloud-storage",
    "google-cloud-texttospeech",
    "google-genai",
    "moviepy",
    "opentelemetry-exporter-gcp-trace",
    "python-dotenv",
    "requests",
    "uvicorn",
]

requires-python = ">=3.11,<3.14"

[dependency-groups]
dev = [
    "hatch",
    "jupyter",
    "keyrings.google-artifactregistry-auth",
    "langchain-google-vertexai",
    "locust",
    "nest-asyncio",
    "pytest",
    "pytest-cov",
    "black",
    "pylint",
    "uv",
    "pre-commit",
]

[project.optional-dependencies]

lint = [
    "codespell",
    "mypy",
    "pylint",
    "types-pyyaml",
    "types-requests",
]

[tool.black]
line-length = 88
target-version = ['py311']

[tool.mypy]
disallow_untyped_calls = true
disallow_untyped_defs = true
disallow_incomplete_defs = true
no_implicit_optional = true
check_untyped_defs = true
disallow_subclassing_any = true
warn_incomplete_stub = true
warn_redundant_casts = true
warn_unused_ignores = true
warn_unreachable = true
follow_imports = "silent"
ignore_missing_imports = true
explicit_package_bases = true
disable_error_code = ["misc", "no-untyped-call", "no-any-return"]

exclude = [".venv", "samples", "notebooks", "tmp"]

[tool.codespell]
ignore-words-list = "rouge"
skip = "./locust_env/*,uv.lock,.venv,./frontend,**/*.ipynb"


[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"


[tool.pytest.ini_options]
pythonpath = [".", "demos", "tests"]
asyncio_default_fixture_loop_scope = "function"
norecursedirs = "samples tests/examples/load_test tmp"
markers = ["integration: marks tests as integration tests"]
testpaths = ["tests"]
python_files = ["test_*.py"]
asyncio_mode = "auto"

[tool.hatch.build.targets.wheel]
packages = ["mediagent_kit"]

[tool.hatch.metadata]
allow-direct-references = true

[[tool.uv.index]]
url = "https://pypi.org/simple"
default = true