fix: CG-17050: skip codebase init if repo operator is none (#999)

# Motivation <!-- Why is this change necessary? --> # Content <!-- Please include a summary of the change --> # Testing <!-- How was the change tested? --> # Please check the following before marking your PR as ready for review - [ ] I have added tests for my changes - [ ] I have updated the documentation or added new documentation as needed

Author: caroljung-cg

src/codegen/sdk/core/codebase.py

@@ -25,7 +25,7 @@
 from codegen.configs.models.codebase import CodebaseConfig, PinkMode
 from codegen.configs.models.secrets import SecretsConfig
 from codegen.git.repo_operator.repo_operator import RepoOperator
-from codegen.git.schemas.enums import CheckoutResult, SetupOption
+from codegen.git.schemas.enums import CheckoutResult
 from codegen.git.schemas.repo_config import RepoConfig
 from codegen.git.utils.pr_review import CodegenPR
 from codegen.sdk._proxy import proxy_property
@@ -1337,7 +1337,6 @@ def from_repo(
  language: Literal["python", "typescript"] | ProgrammingLanguage | None = None,
  config: CodebaseConfig | None = None,
  secrets: SecretsConfig | None = None,
- setup_option: SetupOption | None = None,
  full_history: bool = False,
  ) -> "Codebase":
  """Fetches a codebase from GitHub and returns a Codebase instance.
@@ -1382,6 +1381,11 @@ def from_repo(
  else:
  # Ensure the operator can handle remote operations
  repo_operator = RepoOperator.create_from_commit(repo_path=repo_path, commit=commit, url=repo_url, full_name=repo_full_name, access_token=access_token)
+
+ if repo_operator is None:
+ logger.error("Failed to clone repository")
+ return None
+
  logger.info("Clone completed successfully")
 
  # Initialize and return codebase with proper context

tests/integration/codegen/git/codebase/conftest.py

@@ -2,12 +2,11 @@
 
 import pytest
 
-from codegen.git.schemas.enums import SetupOption
 from codegen.sdk.core.codebase import Codebase
 
 
 @pytest.fixture
 def codebase(tmpdir):
  os.chdir(tmpdir)
- codebase = Codebase.from_repo(repo_full_name="codegen-sh/Kevin-s-Adventure-Game", tmp_dir=tmpdir, language="python", setup_option=SetupOption.PULL_OR_CLONE)
+ codebase = Codebase.from_repo(repo_full_name="codegen-sh/Kevin-s-Adventure-Game", tmp_dir=tmpdir, language="python")
  yield codebase

tests/unit/codegen/sdk/core/test_codebase.py

@@ -2,8 +2,10 @@
 
 import pytest
 
+from codegen.configs.models.secrets import SecretsConfig
 from codegen.sdk.codebase.codebase_context import CodebaseContext
 from codegen.sdk.codebase.factory.get_session import get_codebase_session
+from codegen.sdk.core.codebase import Codebase
 
 
 @pytest.fixture(autouse=True)
@@ -39,3 +41,9 @@ def test_codeowners_property(context_mock, codebase):
  assert len(codebase.codeowners) == 1
  assert callable(codebase.codeowners[0].files_source)
  assert codebase.codeowners[0].files_source() == codebase.files.return_value
+
+
+def test_from_codebase_non_existent_repo(context_mock, tmpdir):
+ with get_codebase_session(tmpdir=tmpdir, files={"src/main.py": "print('Hello, world!')"}, verify_output=False) as codebase:
+ codebase = Codebase.from_repo("some-org/non-existent-repo", tmp_dir=tmpdir, secrets=SecretsConfig(github_token="some-token"))
+ assert codebase is None