55__package__ = "readmeai"
66
77import asyncio
8- import shutil
98import traceback
109
1110from 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