wip

Author: caroljung-cg

src/codegen/runner/clients/codebase_client.py

@@ -5,6 +5,7 @@
 from codegen.git.schemas.repo_config import RepoConfig
 from codegen.runner.clients.server_client import LocalServerClient
 from codegen.runner.models.apis import SANDBOX_SERVER_PORT
+from codegen.shared.configs.session_configs import config
 
 logger = logging.getLogger(__name__)
 
@@ -22,15 +23,23 @@ def __init__(self, repo_config: RepoConfig, git_access_token: str | None, host:
         self.git_access_token = git_access_token
         super().__init__(server_path=RUNNER_SERVER_PATH, host=host, port=port)
 
-    def _get_envs(self):
+    def _get_envs(self) -> dict:
         envs = super()._get_envs()
-        envs.update(
-            {
-                "CODEGEN_REPOSITORY__REPO_PATH": self.repo_config.repo_path,
-                "CODEGEN_REPOSITORY__REPO_NAME": self.repo_config.name,
-                "CODEGEN_REPOSITORY__FULL_NAME": self.repo_config.full_name,
-                "CODEGEN_REPOSITORY__LANGUAGE": self.repo_config.language.value,
-                "CODEGEN_SECRETS__GITHUB_TOKEN": self.git_access_token,
-            }
-        )
+        codebase_envs = {
+            "CODEGEN_REPOSITORY__REPO_PATH": self.repo_config.repo_path,
+            "CODEGEN_REPOSITORY__REPO_NAME": self.repo_config.name,
+            "CODEGEN_REPOSITORY__FULL_NAME": self.repo_config.full_name,
+            "CODEGEN_REPOSITORY__LANGUAGE": self.repo_config.language.value,
+        }
+        if self.git_access_token is not None:
+            codebase_envs["CODEGEN_SECRETS__GITHUB_TOKEN"] = self.git_access_token
+
+        envs.update(codebase_envs)
         return envs
+
+
+if __name__ == "__main__":
+    test_config = RepoConfig.from_repo_path("/Users/caroljung/git/codegen/codegen-agi")
+    test_config.full_name = "codegen-sh/codegen-agi"
+    client = CodebaseClient(test_config, config.secrets.github_token)
+    print(client.healthcheck())

src/codegen/runner/clients/server_client.py

@@ -36,7 +36,7 @@ def __del__(self):
             self._process.terminate()
             self._process.wait()
 
-    def _get_envs(self):
+    def _get_envs(self) -> dict:
         return os.environ.copy()
 
     def _start_server(self, server_path: str) -> None:
@@ -57,7 +57,7 @@ def _start_server(self, server_path: str) -> None:
         )
         self._wait_for_server()
 
-    def _wait_for_server(self, timeout: int = 60, interval: float = 0.1) -> None:
+    def _wait_for_server(self, timeout: int = 10, interval: float = 0.1) -> None:
         """Wait for the server to start by polling the health endpoint"""
         start_time = time.time()
         while (time.time() - start_time) < timeout:

src/codegen/runner/sandbox/server.py

@@ -39,7 +39,7 @@ async def lifespan(server: FastAPI):
     global runner
 
     try:
-        server_info = ServerInfo(repo_name=config.repository.full_name)
+        server_info = ServerInfo(repo_name=config.repository.full_name or config.repository.repo_name)
         logger.info(f"Starting up sandbox fastapi server for repo_name={server_info.repo_name}")
         repo_config = RepoConfig(
             name=config.repository.repo_name,