Skip to content

Commit ec16c9f

Browse files
committed
fix: Update repository cloning logic.
1 parent d4c9d18 commit ec16c9f

File tree

2 files changed

+4
-12
lines changed

2 files changed

+4
-12
lines changed

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
44

55
[tool.poetry]
66
name = "readmeai"
7-
version = "0.4.019"
7+
version = "0.4.022"
88
description = "🚀 Generate beautiful README.md files from the terminal. Powered by OpenAI's GPT LLMs 💫"
99
authors = ["Eli <0x.eli.64s@gmail.com>"]
1010
license = "MIT"

readmeai/main.py

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
__package__ = "readmeai"
66

77
import asyncio
8-
import shutil
98
import traceback
109

1110
from readmeai.core import logger, model, preprocess
@@ -45,10 +44,9 @@ def main(
4544
config.api.temperature = temperature
4645
config.api.offline_mode = offline_mode
4746
config.git = GitConfig(repository=repository)
48-
(
49-
_,
50-
config.git.name,
51-
) = git.get_user_repository_name(repository)
47+
(user, config.git.name) = git.get_user_repository_name(repository)
48+
config.api.offline_mode = offline_mode
49+
config.git = GitConfig(repository=repository)
5250
if api_key is None and offline_mode is False:
5351
config.api.offline_mode = offline_mode
5452

@@ -73,11 +71,7 @@ async def readme_agent(
7371

7472
llm = model.OpenAIHandler(config)
7573

76-
temp_dir = None
7774
try:
78-
temp_dir = await asyncio.to_thread(
79-
git.clone_repo_to_temp_dir, repository
80-
)
8175
temp_dir = git.clone_repo_to_temp_dir(repository)
8276
tree_str = tree.generate_tree(temp_dir, repository)
8377
tree_str = tree.format_tree(name, tree_str)
@@ -130,8 +124,6 @@ async def readme_agent(
130124
)
131125

132126
finally:
133-
if temp_dir:
134-
await asyncio.to_thread(shutil.rmtree, temp_dir)
135127
await llm.close()
136128

137129
logger.info("README-AI execution complete.")

0 commit comments

Comments
 (0)