From 98f887af0de5fb2e40944a94e3d761b8be472312 Mon Sep 17 00:00:00 2001 From: Eli Date: Sun, 27 Oct 2024 03:36:33 -0500 Subject: [PATCH] Fix tomllib py311 import bug. --- pyproject.toml | 2 +- readmeai/templates/quickstart.py | 8 ++++++-- 2 files changed, 7 insertions(+), 3 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index b94c459f..4f25f899 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "readmeai" -version = "0.5.99.post4" +version = "0.5.99.post5" description = "Automated README file generator, powered by AI." authors = ["Eli "] license = "MIT" diff --git a/readmeai/templates/quickstart.py b/readmeai/templates/quickstart.py index 95a74192..f6771295 100644 --- a/readmeai/templates/quickstart.py +++ b/readmeai/templates/quickstart.py @@ -1,11 +1,15 @@ from pathlib import Path from string import Template -import tomllib - from readmeai.config.settings import ConfigLoader from readmeai.generators.quickstart import QuickStartGenerator from readmeai.ingestion.models import QuickStart, RepositoryContext +from readmeai.utils.helpers import is_available + +if is_available("tomllib"): # pragma: no cover + import tomllib +elif is_available("tomli"): # pragma: no cover + import tomli as tomllib class QuickStartBuilder: