fix: init command

src/codegen/cli/workspace/initialize_workspace.py

@@ -36,7 +36,7 @@
         Tuple of (codegen_folder, docs_folder, examples_folder)
     """
     repo = get_git_repo()
     REPO_PATH = Path(repo.workdir)
     CODEGEN_FOLDER = REPO_PATH / CODEGEN_DIR
     PROMPTS_FOLDER = REPO_PATH / PROMPTS_DIR
     DOCS_FOLDER = REPO_PATH / DOCS_DIR
@@ -59,7 +59,7 @@
         CODEMODS_DIR.mkdir(parents=True, exist_ok=True)
 
         # Initialize virtual environment
         status_obj.update(f"   {'Creating' if isinstance(status, str) else 'Checking'} virtual environment...")
         venv = VenvManager()
         if not venv.is_initialized():
             venv.create_venv()
@@ -78,44 +78,43 @@
         if not repo:
             rich.print("No git repository found. Please run this command in a git repository.")
         else:
             status_obj.update(f"   {'Updating' if isinstance(status, Status) else status} .gitignore...")
             modify_gitignore(CODEGEN_FOLDER)
 
             # Create or update config.toml with basic repo info
-            if not session:  # Only create if session doesn't exist (it handles config itself)
-                org_name, repo_name = get_git_organization_and_repo(repo)
-                config = {}
-                if CONFIG_PATH.exists():
-                    config = toml.load(CONFIG_PATH)
-                config.update(
-                    {
-                        "organization_name": config.get("organization_name", org_name),
-                        "repo_name": config.get("repo_name", repo_name),
-                    }
-                )
-                CONFIG_PATH.write_text(toml.dumps(config))
+            org_name, repo_name = get_git_organization_and_repo(repo)
+            config = {}
+            if CONFIG_PATH.exists():
+                config = toml.load(CONFIG_PATH)
+            config.update(
+                {
+                    "organization_name": config.get("organization_name", org_name),
+                    "repo_name": config.get("repo_name", repo_name),
+                }
+            )
+            CONFIG_PATH.write_text(toml.dumps(config))
 
             # Create notebook template
             create_notebook(JUPYTER_DIR)
 
         # Only fetch docs and examples if requested and session is provided
         if fetch_docs and session:
             status_obj.update("Fetching latest docs & examples...")
             shutil.rmtree(DOCS_FOLDER, ignore_errors=True)
             shutil.rmtree(EXAMPLES_FOLDER, ignore_errors=True)
 
             DOCS_FOLDER.mkdir(parents=True, exist_ok=True)
             EXAMPLES_FOLDER.mkdir(parents=True, exist_ok=True)
 
             response = RestAPI(session.token).get_docs()
             populate_api_docs(DOCS_FOLDER, response.docs, status_obj)
             populate_examples(session, EXAMPLES_FOLDER, response.examples, status_obj)
 
             # Set programming language
             if programming_language:
                 session.config.programming_language = programming_language
             else:
                 session.config.programming_language = str(response.language)
 
             session.write_config()