try/catchon empty commits

Author: tawsifkamal

src/codegen/git/repo_operator/repo_operator.py

@@ -166,8 +166,14 @@ def git_cli(self) -> GitCLI:
         return git_cli
 
     @property
-    def head_commit(self) -> GitCommit:
-        return self.git_cli.head.commit
+    def head_commit(self) -> GitCommit | None:
+        try:
+            return self.git_cli.head.commit
+        except ValueError as e:
+            if "Reference at 'refs/heads/main' does not exist" in str(e):
+                logger.info("No commit exists in the repository yet")
+                return None
+            raise
 
     @property
     def git_diff(self) -> str:

src/codegen/sdk/core/codebase.py

@@ -898,6 +898,10 @@ def checkout(
             result = self._op.checkout_commit(commit_hash=commit)
         if result == CheckoutResult.SUCCESS:
             logger.info(f"Checked out {branch or commit}")
+            if self._op.head_commit is None:
+                logger.info("No commit exists in the repository yet")
+                return CheckoutResult.SUCCESS
+
             self.sync_to_commit(self._op.head_commit)
         elif result == CheckoutResult.NOT_FOUND:
             logger.info(f"Could not find branch {branch or commit}")
@@ -1382,13 +1386,25 @@ def from_repo(
 
             # Initialize and return codebase with proper context
             logger.info("Initializing Codebase...")
-            project = ProjectConfig.from_repo_operator(
-                repo_operator=repo_operator,
-                programming_language=ProgrammingLanguage(language.upper()) if language else None,
-            )
-            codebase = Codebase(projects=[project], config=config, secrets=secrets)
-            logger.info("Codebase initialization complete")
-            return codebase
+            try:
+                project = ProjectConfig.from_repo_operator(
+                    repo_operator=repo_operator,
+                    programming_language=ProgrammingLanguage(language.upper()) if language else None,
+                )
+                codebase = Codebase(projects=[project], config=config, secrets=secrets)
+                logger.info("Codebase initialization complete")
+                return codebase
+            except ValueError as e:
+                if "Reference at 'refs/heads/main' does not exist" in str(e):
+                    logger.warning("Repository appears to be empty (no main branch). Creating empty codebase.")
+                    project = ProjectConfig(
+                        repo_operator=repo_operator,
+                        programming_language=ProgrammingLanguage(language.upper()) if language else None,
+                    )
+                    codebase = Codebase(projects=[project], config=config, secrets=secrets)
+                    logger.info("Empty codebase initialization complete")
+                    return codebase
+                raise
         except Exception as e:
             logger.exception(f"Failed to initialize codebase: {e}")
             raise
@@ -1435,6 +1451,12 @@ def from_string(
 
         with tempfile.TemporaryDirectory(prefix="codegen_") as tmp_dir:
             logger.info(f"Using directory: {tmp_dir}")
+
+            # Initialize git repo to avoid "not in a git repository" error
+            import subprocess
+
+            subprocess.run(["git", "init"], cwd=tmp_dir, check=True, capture_output=True)
+
             codebase = CodebaseFactory.get_codebase_from_files(repo_path=tmp_dir, files=files, programming_language=prog_lang)
             logger.info("Codebase initialization complete")
             return codebase