bria-vgl 0.1.0__tar.gz

bria_vgl-0.1.0/.gitignore

@@ -0,0 +1,200 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# UV
+#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#uv.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
+.pdm.toml
+.pdm-python
+.pdm-build/
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+.idea/
+
+# Abstra
+# Abstra is an AI-powered process automation framework.
+# Ignore directories containing user credentials, local state, and settings.
+# Learn more at https://abstra.io/docs
+.abstra/
+
+# Visual Studio Code
+#  Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore 
+#  that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
+#  and can be added to the global gitignore or merged into this file. However, if you prefer, 
+#  you could uncomment the following to ignore the enitre vscode folder
+ .vscode/
+
+# Ruff stuff:
+.ruff_cache/
+
+# PyPI configuration file
+.pypirc
+
+# Cursor
+#  Cursor is an AI-powered code editor. `.cursorignore` specifies files/directories to
+#  exclude from AI features like autocomplete and code analysis. Recommended for sensitive data
+#  refer to https://docs.cursor.com/context/ignore-files
+.cursorignore
+.cursorindexingignore
+
+vertex_ai_creds.json
+
+local_data/
+# scratch tree written by examples/styles_examples/develop_a_style.py
+.vgl-style-dev/

bria_vgl-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 BRIA AI
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

bria_vgl-0.1.0/PKG-INFO

@@ -0,0 +1,151 @@
+Metadata-Version: 2.5
+Name: bria-vgl
+Version: 0.1.0
+Summary: Turn a prompt, or images plus a prompt, into a structured JSON scene description (VGL) for image models like FIBO.
+Project-URL: Homepage, https://github.com/Bria-AI/vgl
+Project-URL: Documentation, https://github.com/Bria-AI/vgl#readme
+Project-URL: Issues, https://github.com/Bria-AI/vgl/issues
+Project-URL: Changelog, https://github.com/Bria-AI/vgl/blob/dev/CHANGELOG.md
+Author: Bria AI
+License-Expression: MIT
+License-File: LICENSE
+Keywords: bria,fibo,image generation,llm,prompt,structured output,vgl
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Multimedia :: Graphics
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Classifier: Typing :: Typed
+Requires-Python: >=3.10
+Requires-Dist: boltons>=20.0.0
+Requires-Dist: litellm>=1.80.11
+Requires-Dist: openai>=2.20.0
+Requires-Dist: pillow>=9.1.0
+Requires-Dist: pydantic>=2.0
+Requires-Dist: pyyaml>=6.0.1
+Provides-Extra: all
+Requires-Dist: boto3>=1.34; extra == 'all'
+Requires-Dist: google-auth>=2.0; extra == 'all'
+Provides-Extra: bedrock
+Requires-Dist: boto3>=1.34; extra == 'bedrock'
+Provides-Extra: vertex
+Requires-Dist: google-auth>=2.0; extra == 'vertex'
+Description-Content-Type: text/markdown
+
+# VGL — Visual Generation Language
+
+VGL turns a text prompt, or images plus a prompt, into a structured JSON description of a scene: objects, background, lighting, aesthetics, camera, and text. The result is a typed Pydantic model that an image model like FIBO takes as its prompt.
+
+## Install
+
+```bash
+pip install bria-vgl
+```
+
+The default LLM is Gemini through Google AI Studio. For other providers, install the extra:
+
+```bash
+pip install 'bria-vgl[vertex]'     # Gemini on Vertex AI
+pip install 'bria-vgl[bedrock]'    # Claude on AWS Bedrock
+```
+
+## Authenticate
+
+| Provider | `llm=` | Set |
+|---|---|---|
+| Google AI Studio (default) | `gemini/gemini-3.5-flash-lite` | `GEMINI_API_KEY` ([get one](https://aistudio.google.com/apikey)) |
+| Vertex AI | `vertex_ai/gemini-3.5-flash-lite` | `VERTEXAI_PROJECT`, `VERTEXAI_LOCATION`, `GOOGLE_APPLICATION_CREDENTIALS` (a service account JSON) |
+| AWS Bedrock | e.g. `bedrock/us.anthropic.claude-sonnet-4-5-20250929-v1:0` | the usual AWS credentials and `AWS_REGION_NAME` |
+
+`llm` takes any [litellm model id](https://docs.litellm.ai/docs/providers), but the prompts are tuned for Gemini.
+
+## Quickstart
+
+### Create a scene from a prompt
+
+```python
+import vgl
+from vgl.dialects.thales import Generate
+
+scene = vgl.create(Generate(prompt="a red fox in snow at dusk"))
+print(scene.short_description)
+print(scene.model_dump_json(indent=2, exclude_none=True))
+```
+
+### Edit from images
+
+```python
+from PIL import Image
+from vgl.dialects.thales import EditImages
+
+room = Image.open("room.jpg")
+lamp = Image.open("lamp.jpg")
+
+scene = vgl.modify(EditImages(prompt="make it night", images=[room]))
+scene = vgl.modify(EditImages(prompt="put the lamp from <image_2> in <image_1>", images=[room, lamp]))
+print(scene.edit_instruction)
+```
+
+Up to 4 images. The prompt refers to them as `<image_1>` to `<image_4>`, in list order.
+
+### Edit a masked area
+
+```python
+from vgl.dialects.thales import EditImageWithMask
+
+mask = Image.open("mask.png")  # white is the area to change; same size as the image
+scene = vgl.modify(EditImageWithMask(prompt="replace it with a cat", image=room, mask=mask))
+```
+
+### Edit with the image's VGL
+
+If you have the VGL that an image was generated from, pass it as `input_image_vgl`. The edit then keeps everything the prompt doesn't change, rather than re-describing the image from scratch. vgl doesn't check that it matches the image.
+
+```python
+scene = vgl.create(Generate(prompt="a cosy living room"))
+image = ...  # generate the image from `scene` with FIBO
+night = vgl.modify(EditImages(prompt="make it night", images=[image], input_image_vgl=scene))
+```
+
+Runnable versions are in [`examples/`](https://github.com/Bria-AI/vgl/tree/dev/examples).
+
+## Options
+
+```python
+vgl.create(payload, llm="vertex_ai/gemini-3.5-flash-lite", seed=7, style_preset_prompt="Muted film colours, soft grain.")
+vgl.modify(payload, llm=..., seed=...)
+```
+
+- `llm`: the litellm model id; see [Authenticate](#authenticate).
+- `seed`: makes a call repeatable, on models that accept one.
+- `style_preset_prompt`: style guidelines added to the system prompt. Applies to `create` only; `modify` ignores it.
+vgl logs under the `vgl` logger hierarchy (`vgl.gateway`, `vgl.prompting`, …). Configure it the standard way — e.g. `logging.getLogger("vgl").setLevel(logging.INFO)` and attach your own handler; add a `logging.Filter` if you need to inject contextual fields.
+
+A Gemini call that keeps failing (3 attempts) or hits a rate limit (429, not retried) moves to `gemini-3.1-flash-lite` on the same provider.
+
+## Errors
+
+Everything vgl raises inherits from `vgl.exceptions.VGLError`, except a bad payload, which raises `pydantic.ValidationError` before any LLM call. The classes below all live in `vgl.exceptions`.
+
+| Error | Means |
+|---|---|
+| `AuthenticationError` | the provider rejected your credentials, e.g. a missing or wrong API key |
+| `RateLimitError` | the provider returned 429 (quota exhausted), on the Gemini fallback too |
+| `LLMError` | any other provider failure, after retries and the fallback |
+| `InvalidResponseError` | the model's reply isn't a valid VGL |
+| `MaxTokensError` | the model ran out of output tokens mid-reply |
+| `PromptNotFoundError` | a bundled prompt is missing; a packaging fault, please report it |
+
+## Dialects
+
+A dialect is a small variant of VGL, with its own model, payloads, and prompts. The payload you import decides the dialect. vgl ships the Thales dialect: no aesthetic scores, and every edit states its change in `edit_instruction`.
+
+## More
+
+- [Best practices](https://github.com/Bria-AI/vgl/blob/dev/docs/best_practices.md): how images are prepared, masks, and getting consistent results.
+- [Contributing](https://github.com/Bria-AI/vgl/blob/dev/CONTRIBUTING.md)

bria_vgl-0.1.0/README.md

@@ -0,0 +1,112 @@
+# VGL — Visual Generation Language
+
+VGL turns a text prompt, or images plus a prompt, into a structured JSON description of a scene: objects, background, lighting, aesthetics, camera, and text. The result is a typed Pydantic model that an image model like FIBO takes as its prompt.
+
+## Install
+
+```bash
+pip install bria-vgl
+```
+
+The default LLM is Gemini through Google AI Studio. For other providers, install the extra:
+
+```bash
+pip install 'bria-vgl[vertex]'     # Gemini on Vertex AI
+pip install 'bria-vgl[bedrock]'    # Claude on AWS Bedrock
+```
+
+## Authenticate
+
+| Provider | `llm=` | Set |
+|---|---|---|
+| Google AI Studio (default) | `gemini/gemini-3.5-flash-lite` | `GEMINI_API_KEY` ([get one](https://aistudio.google.com/apikey)) |
+| Vertex AI | `vertex_ai/gemini-3.5-flash-lite` | `VERTEXAI_PROJECT`, `VERTEXAI_LOCATION`, `GOOGLE_APPLICATION_CREDENTIALS` (a service account JSON) |
+| AWS Bedrock | e.g. `bedrock/us.anthropic.claude-sonnet-4-5-20250929-v1:0` | the usual AWS credentials and `AWS_REGION_NAME` |
+
+`llm` takes any [litellm model id](https://docs.litellm.ai/docs/providers), but the prompts are tuned for Gemini.
+
+## Quickstart
+
+### Create a scene from a prompt
+
+```python
+import vgl
+from vgl.dialects.thales import Generate
+
+scene = vgl.create(Generate(prompt="a red fox in snow at dusk"))
+print(scene.short_description)
+print(scene.model_dump_json(indent=2, exclude_none=True))
+```
+
+### Edit from images
+
+```python
+from PIL import Image
+from vgl.dialects.thales import EditImages
+
+room = Image.open("room.jpg")
+lamp = Image.open("lamp.jpg")
+
+scene = vgl.modify(EditImages(prompt="make it night", images=[room]))
+scene = vgl.modify(EditImages(prompt="put the lamp from <image_2> in <image_1>", images=[room, lamp]))
+print(scene.edit_instruction)
+```
+
+Up to 4 images. The prompt refers to them as `<image_1>` to `<image_4>`, in list order.
+
+### Edit a masked area
+
+```python
+from vgl.dialects.thales import EditImageWithMask
+
+mask = Image.open("mask.png")  # white is the area to change; same size as the image
+scene = vgl.modify(EditImageWithMask(prompt="replace it with a cat", image=room, mask=mask))
+```
+
+### Edit with the image's VGL
+
+If you have the VGL that an image was generated from, pass it as `input_image_vgl`. The edit then keeps everything the prompt doesn't change, rather than re-describing the image from scratch. vgl doesn't check that it matches the image.
+
+```python
+scene = vgl.create(Generate(prompt="a cosy living room"))
+image = ...  # generate the image from `scene` with FIBO
+night = vgl.modify(EditImages(prompt="make it night", images=[image], input_image_vgl=scene))
+```
+
+Runnable versions are in [`examples/`](https://github.com/Bria-AI/vgl/tree/dev/examples).
+
+## Options
+
+```python
+vgl.create(payload, llm="vertex_ai/gemini-3.5-flash-lite", seed=7, style_preset_prompt="Muted film colours, soft grain.")
+vgl.modify(payload, llm=..., seed=...)
+```
+
+- `llm`: the litellm model id; see [Authenticate](#authenticate).
+- `seed`: makes a call repeatable, on models that accept one.
+- `style_preset_prompt`: style guidelines added to the system prompt. Applies to `create` only; `modify` ignores it.
+vgl logs under the `vgl` logger hierarchy (`vgl.gateway`, `vgl.prompting`, …). Configure it the standard way — e.g. `logging.getLogger("vgl").setLevel(logging.INFO)` and attach your own handler; add a `logging.Filter` if you need to inject contextual fields.
+
+A Gemini call that keeps failing (3 attempts) or hits a rate limit (429, not retried) moves to `gemini-3.1-flash-lite` on the same provider.
+
+## Errors
+
+Everything vgl raises inherits from `vgl.exceptions.VGLError`, except a bad payload, which raises `pydantic.ValidationError` before any LLM call. The classes below all live in `vgl.exceptions`.
+
+| Error | Means |
+|---|---|
+| `AuthenticationError` | the provider rejected your credentials, e.g. a missing or wrong API key |
+| `RateLimitError` | the provider returned 429 (quota exhausted), on the Gemini fallback too |
+| `LLMError` | any other provider failure, after retries and the fallback |
+| `InvalidResponseError` | the model's reply isn't a valid VGL |
+| `MaxTokensError` | the model ran out of output tokens mid-reply |
+| `PromptNotFoundError` | a bundled prompt is missing; a packaging fault, please report it |
+
+## Dialects
+
+A dialect is a small variant of VGL, with its own model, payloads, and prompts. The payload you import decides the dialect. vgl ships the Thales dialect: no aesthetic scores, and every edit states its change in `edit_instruction`.
+
+## More
+
+- [Best practices](https://github.com/Bria-AI/vgl/blob/dev/docs/best_practices.md): how images are prepared, masks, and getting consistent results.
+- [Contributing](https://github.com/Bria-AI/vgl/blob/dev/CONTRIBUTING.md)

bria_vgl-0.1.0/pyproject.toml

@@ -0,0 +1,125 @@
+[project]
+name = "bria-vgl"
+version = "0.1.0"
+description = "Turn a prompt, or images plus a prompt, into a structured JSON scene description (VGL) for image models like FIBO."
+readme = "README.md"
+license = "MIT"
+license-files = ["LICENSE"]
+authors = [{ name = "Bria AI" }]
+keywords = ["vgl", "fibo", "bria", "image generation", "prompt", "structured output", "llm"]
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Developers",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+    "Topic :: Multimedia :: Graphics",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+    "Typing :: Typed",
+]
+
+requires-python = ">=3.10"
+# vgl is a base package: every spec stays an open lower bound, floored at the oldest
+# version whose API vgl actually uses. A hard pin here becomes a resolution conflict for
+# every consumer, who then has to carry an override.
+dependencies = [
+    # One function, boltons.iterutils.remap, stable for years.
+    "boltons>=20.0.0",
+    # The whole surface used is Image.Image/new/open/composite, Image.Resampling and
+    # .convert/.resize/.save. The newest of those, Image.Resampling, landed in 9.1.
+    "pillow>=9.1.0",
+    # The VGL models and payloads. v2-only API, so 2.0 is the honest floor.
+    "pydantic>=2.0",
+    # Floor is a version vgl is known to run on, not the newest release.
+    "litellm>=1.80.11",
+    # Imported for the base class of litellm's provider errors. Floor matches litellm's own.
+    "openai>=2.20.0",
+    # NOT used by vgl. litellm imports yaml at base import time
+    # (integrations/dotprompt/prompt_manager.py) but only declares pyyaml under its `cli`
+    # and `proxy` extras, so without this a clean install of vgl fails on `import vgl`
+    # with ModuleNotFoundError: yaml. Verified against litellm 1.90.0. Remove only once
+    # litellm declares it, or stops importing it unconditionally.
+    "pyyaml>=6.0.1",
+]
+
+[project.urls]
+Homepage = "https://github.com/Bria-AI/vgl"
+Documentation = "https://github.com/Bria-AI/vgl#readme"
+Issues = "https://github.com/Bria-AI/vgl/issues"
+Changelog = "https://github.com/Bria-AI/vgl/blob/dev/CHANGELOG.md"
+
+[project.optional-dependencies]
+# litellm reaches Vertex by doing `import google.auth` to read GOOGLE_APPLICATION_CREDENTIALS,
+# but declares it only under its own extras — so nothing installs it and a clean vgl dies on the
+# first Vertex call. Narrower than litellm's own suggestion of google-cloud-aiplatform, which is
+# a far larger tree; google-auth is what litellm actually imports and is verified sufficient.
+vertex = ["google-auth>=2.0"]
+
+# litellm reaches Bedrock through boto3 but declares it under no extra of its own, so the failure
+# is the same shape as the Vertex one: fine until the first call. vgl dropped boto3 in PR #45 as
+# an unused direct import, which it still is — this is litellm's need, declared honestly.
+bedrock = ["boto3>=1.34"]
+
+# Every optional dependency, for a consumer that does not want to track which provider needs what.
+all = ["bria-vgl[vertex]", "bria-vgl[bedrock]"]
+
+# Local development only — PEP 735 groups are not published in the wheel, which is why the
+# Vertex dependency above is an extra instead: a consumer has to be able to install it.
+[dependency-groups]
+# google-auth is the `vertex` extra's dependency; the Vertex validation tests import it.
+test = ["pytest>=8.0.0", "google-auth>=2.0"]
+
+lint = ["pyright>=1.1.408", "ruff>=0.15.12"]
+
+# The smoke test loads a .env; python-dotenv otherwise arrives only transitively, via litellm.
+examples = ["python-dotenv>=1.0"]
+
+dev = [
+    { include-group = "test" },
+    { include-group = "lint" },
+    { include-group = "examples" },
+]
+
+[build-system]
+# 1.27 is the first to support the PEP 639 license fields.
+requires = ["hatchling>=1.27"]
+build-backend = "hatchling.build"
+
+
+[tool.hatch.metadata]
+allow-direct-references = true
+
+[tool.hatch.build.targets.wheel]
+packages = ["src/vgl"]
+
+[tool.hatch.build]
+include = ["src/**/*"]
+
+
+
+[tool.ruff.lint]
+select = ["C4", "E", "F", "I", "PERF", "UP"]
+ignore = ["PERF203"]
+
+[tool.ruff]
+line-length = 160
+target-version = "py310"
+
+[tool.pytest.ini_options]
+log_cli = true
+python_files = ["*_tests.py"]
+testpaths = ["tests"]
+markers = [
+    "unit: isolated, no I/O beyond tmp_path and the bundled package data",
+    "integration: several real components together, still no network",
+]
+
+[tool.pyright]
+enableExperimentalFeatures = true
+exclude = ["**/.venv", "**/node_modules", "**/__pycache__"]
+
+[tool.commitizen]
+name = "cz_conventional_commits"

bria_vgl-0.1.0/src/vgl/__init__.py

@@ -0,0 +1,9 @@
+from vgl._version import __version__
+from vgl.api import DEFAULT_LLM, create, modify
+
+__all__ = [
+    "__version__",
+    "DEFAULT_LLM",
+    "create",
+    "modify",
+]

bria_vgl-0.1.0/src/vgl/_version.py

@@ -0,0 +1,6 @@
+from importlib.metadata import PackageNotFoundError, version
+
+try:
+    __version__ = version("vgl")
+except PackageNotFoundError:  # run from a source checkout without installing
+    __version__ = "0.0.0"

bria_vgl-0.1.0/src/vgl/api.py

@@ -0,0 +1,58 @@
+import os
+
+from litellm.exceptions import AuthenticationError as LiteLLMAuthenticationError
+from litellm.exceptions import RateLimitError as LiteLLMRateLimitError
+from openai import OpenAIError
+
+from vgl import gateway
+from vgl.exceptions import AuthenticationError, LLMError, RateLimitError
+from vgl.payload import Payload, T
+from vgl.prompting import render_prompts
+from vgl.utils.vertex import validate_vertex_env
+
+DEFAULT_LLM = "gemini/gemini-3.5-flash-lite"
+
+
+def create(payload: Payload[T], *, llm: str = DEFAULT_LLM, style_preset_prompt: str | None = None, seed: int | None = None) -> T:
+    """A new VGL. ``style_preset_prompt`` is composed into the system prompt."""
+    if not payload.creates:
+        raise TypeError(f"{type(payload).__name__} edits a VGL; pass it to vgl.modify()")
+    return _run(payload, llm=llm, seed=seed, style_preset=style_preset_prompt)
+
+
+def modify(payload: Payload[T], *, llm: str = DEFAULT_LLM, style_preset_prompt: str | None = None, seed: int | None = None) -> T:
+    """An edited VGL. ``style_preset_prompt`` is accepted and ignored: presets apply to create only."""
+    if payload.creates:
+        raise TypeError(f"{type(payload).__name__} creates a VGL; pass it to vgl.create()")
+    return _run(payload, llm=llm, seed=seed, style_preset=None)
+
+
+def _check_gemini_key(llm: str) -> None:
+    """litellm reports a missing key as a connection error, which the gateway would retry."""
+    if llm.startswith("gemini/") and not (os.environ.get("GEMINI_API_KEY") or os.environ.get("GOOGLE_API_KEY")):
+        raise AuthenticationError(f"{llm} needs GEMINI_API_KEY or GOOGLE_API_KEY set")
+
+
+def _run(payload: Payload[T], *, llm: str, seed: int | None, style_preset: str | None) -> T:
+    validate_vertex_env(llm)
+    _check_gemini_key(llm)
+    request = payload.to_request()
+    system_prompt, user_prompt = render_prompts(request.prompts, request.context, style_preset)
+    try:
+        text = gateway.call(
+            model_name=llm,
+            system_prompt=system_prompt,
+            user_prompt=user_prompt,
+            images=list(request.images),
+            seed=seed,
+            response_schema=request.response_schema,
+            temperature=request.temperature,
+        )
+    except LiteLLMAuthenticationError as error:
+        raise AuthenticationError(str(error)) from error
+    except LiteLLMRateLimitError as error:
+        raise RateLimitError(str(error)) from error
+    # Every litellm provider error subclasses openai's.
+    except OpenAIError as error:
+        raise LLMError(str(error)) from error
+    return payload.parse(text)

bria_vgl-0.1.0/src/vgl/dialects/__init__.py

@@ -0,0 +1,5 @@
+"""Public import paths for vgl's dialects. Each payload you import here selects its dialect."""
+
+from vgl.dialects import thales
+
+__all__ = ["thales"]