fix: Update repository cloning logic.

Author: eli64s

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "readmeai"
-version = "0.4.019"
+version = "0.4.025"
 description = "🚀 Generate beautiful README.md files from the terminal. Powered by OpenAI's GPT LLMs 💫"
 authors = ["Eli <0x.eli.64s@gmail.com>"]
 license = "MIT"

readmeai/main.py

@@ -5,7 +5,6 @@
 __package__ = "readmeai"
 
 import asyncio
-import shutil
 import traceback
 
 from readmeai.core import logger, model, preprocess
@@ -45,10 +44,8 @@ def main(
     config.api.temperature = temperature
     config.api.offline_mode = offline_mode
     config.git = GitConfig(repository=repository)
-    (
-        _,
-        config.git.name,
-    ) = git.get_user_repository_name(repository)
+    _, config.git.name = git.get_user_repository_name(repository)
+
     if api_key is None and offline_mode is False:
         config.api.offline_mode = offline_mode
 
@@ -73,11 +70,7 @@ async def readme_agent(
 
     llm = model.OpenAIHandler(config)
 
-    temp_dir = None
     try:
-        temp_dir = await asyncio.to_thread(
-            git.clone_repo_to_temp_dir, repository
-        )
         temp_dir = git.clone_repo_to_temp_dir(repository)
         tree_str = tree.generate_tree(temp_dir, repository)
         tree_str = tree.format_tree(name, tree_str)
@@ -130,8 +123,6 @@ async def readme_agent(
         )
 
     finally:
-        if temp_dir:
-            await asyncio.to_thread(shutil.rmtree, temp_dir)
         await llm.close()
 
     logger.info("README-AI execution complete.")

readmeai/utils/git.py

@@ -3,7 +3,6 @@
 import os
 import platform
 import re
-import shutil
 import tempfile
 from pathlib import Path
 from typing import Optional
@@ -94,14 +93,14 @@ def get_github_file_link(file: str, repo: str, user_repo_name: str) -> str:
     return base_urls["github.com"]
 
 
-def get_user_repository_name(url_or_path) -> (str, str):
+def get_user_repository_name(url_or_path):
     """
     Extract username and repository name from a
     GitHub, Bitbucket, or GitLab URL or local path.
     """
 
     if os.path.exists(url_or_path):
-        return os.path.basename(url_or_path)
+        return "local", os.path.basename(url_or_path)
 
     patterns = {
         "github": r"https?://github.com/([^/]+)/([^/]+)",