quant/galaxis-agent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor: replace GitHub/Linear/Slack with Gitea/Discord in server.py

Browse Source
This commit is contained in:
머니페니 2026-03-20 15:04:48 +09:00
parent 4382499071
commit 64e54a7392
1 changed files with 45 additions and 50 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

95
agent/server.py
View File

@ -1,4 +1,4 @@
"""Main entry point and CLI loop for Open SWE agent."""
"""Main entry point and CLI loop for galaxis-agent."""
# ruff: noqa: E402
# Suppress deprecation warnings from langchain_core (e.g., Pydantic V1 on Python 3.14+)
@ -24,7 +24,6 @@ warnings.filterwarnings("ignore", message=".*Pydantic V1.*", category=UserWarnin
# Now safe to import agent (which imports LangChain modules)
from deepagents import create_deep_agent
from deepagents.backends.protocol import SandboxBackendProtocol
from langsmith.sandbox import SandboxClientError
from .middleware import (
ToolErrorMiddleware,
@ -35,13 +34,12 @@ from .middleware import (
from .prompt import construct_system_prompt
from .tools import (
commit_and_open_pr,
discord_reply,
fetch_url,
github_comment,
gitea_comment,
http_request,
linear_comment,
slack_thread_reply,
)
from .utils.auth import resolve_github_token
from .utils.auth import get_gitea_token
from .utils.model import make_model
from .utils.sandbox import create_sandbox
@ -52,7 +50,7 @@ SANDBOX_CREATION_TIMEOUT = 180
SANDBOX_POLL_INTERVAL = 1.0
from .utils.agents_md import read_agents_md_in_sandbox
from .utils.github import (
from .utils.git_utils import (
_CRED_FILE_PATH,
cleanup_git_credentials,
git_has_uncommitted_changes,
@ -64,19 +62,23 @@ from .utils.sandbox_paths import aresolve_repo_dir, aresolve_sandbox_work_dir
from .utils.sandbox_state import SANDBOX_BACKENDS, get_sandbox_id_from_metadata
class SandboxConnectionError(Exception):
"""Raised when the sandbox connection is lost or unreachable."""
async def _clone_or_pull_repo_in_sandbox( # noqa: PLR0915
sandbox_backend: SandboxBackendProtocol,
owner: str,
repo: str,
github_token: str | None = None,
gitea_token: str | None = None,
) -> str:
"""Clone a GitHub repo into the sandbox, or pull if it already exists.
"""Clone a Gitea repo into the sandbox, or pull if it already exists.
Args:
sandbox_backend: The sandbox backend to execute commands in (LangSmithBackend)
owner: GitHub repo owner
repo: GitHub repo name
github_token: GitHub access token (from agent auth or env var)
sandbox_backend: The sandbox backend to execute commands in
owner: Gitea repo owner
repo: Gitea repo name
gitea_token: Gitea access token
Returns:
Path to the cloned/updated repo directory
@ -84,21 +86,33 @@ async def _clone_or_pull_repo_in_sandbox( # noqa: PLR0915
logger.info("_clone_or_pull_repo_in_sandbox called for %s/%s", owner, repo)
loop = asyncio.get_event_loop()
token = github_token
token = gitea_token
if not token:
msg = "No GitHub token provided"
msg = "No Gitea token provided"
logger.error(msg)
raise ValueError(msg)
work_dir = await aresolve_sandbox_work_dir(sandbox_backend)
repo_dir = await aresolve_repo_dir(sandbox_backend, repo)
clean_url = f"https://github.com/{owner}/{repo}.git"
clean_url = f"http://gitea:3000/{owner}/{repo}.git"
cred_helper_arg = f"-c credential.helper='store --file={_CRED_FILE_PATH}'"
safe_repo_dir = shlex.quote(repo_dir)
safe_clean_url = shlex.quote(clean_url)
logger.info("Resolved sandbox work dir to %s", work_dir)
# Set up git credentials using store file
await loop.run_in_executor(
None,
sandbox_backend.execute,
f'echo "http://agent:{token}@gitea:3000" > /tmp/.git-credentials',
)
await loop.run_in_executor(
None,
sandbox_backend.execute,
"git config --global credential.helper 'store --file=/tmp/.git-credentials'",
)
is_git_repo = await loop.run_in_executor(None, is_valid_git_repo, sandbox_backend, repo_dir)
if not is_git_repo:
@ -125,7 +139,6 @@ async def _clone_or_pull_repo_in_sandbox( # noqa: PLR0915
logger.info("Repo is clean, pulling latest changes from %s/%s", owner, repo)
await loop.run_in_executor(None, setup_git_credentials, sandbox_backend, token)
try:
pull_result = await loop.run_in_executor(
None,
@ -149,7 +162,6 @@ async def _clone_or_pull_repo_in_sandbox( # noqa: PLR0915
return repo_dir
logger.info("Cloning repo %s/%s to %s", owner, repo, repo_dir)
await loop.run_in_executor(None, setup_git_credentials, sandbox_backend, token)
try:
result = await loop.run_in_executor(
None,
@ -177,13 +189,9 @@ async def _recreate_sandbox(
repo_owner: str,
repo_name: str,
*,
github_token: str | None,
gitea_token: str | None,
) -> tuple[SandboxBackendProtocol, str]:
"""Recreate a sandbox and clone the repo after a connection failure.
Clears the stale cache entry, sets the SANDBOX_CREATING sentinel,
creates a fresh sandbox, and clones the repo.
"""
"""Recreate a sandbox and clone the repo after a connection failure."""
SANDBOX_BACKENDS.pop(thread_id, None)
await client.threads.update(
thread_id=thread_id,
@ -192,7 +200,7 @@ async def _recreate_sandbox(
try:
sandbox_backend = await asyncio.to_thread(create_sandbox)
repo_dir = await _clone_or_pull_repo_in_sandbox(
sandbox_backend, repo_owner, repo_name, github_token
sandbox_backend, repo_owner, repo_name, gitea_token
)
except Exception:
logger.exception("Failed to recreate sandbox after connection failure")
@ -202,14 +210,7 @@ async def _recreate_sandbox(
async def _wait_for_sandbox_id(thread_id: str) -> str:
"""Wait for sandbox_id to be set in thread metadata.
Polls thread metadata until sandbox_id is set to a real value
(not the creating sentinel).
Raises:
TimeoutError: If sandbox creation takes too long
"""
"""Wait for sandbox_id to be set in thread metadata."""
elapsed = 0.0
while elapsed < SANDBOX_CREATION_TIMEOUT:
sandbox_id = await get_sandbox_id_from_metadata(thread_id)
@ -251,8 +252,8 @@ async def get_agent(config: RunnableConfig) -> Pregel: # noqa: PLR0915
tools=[],
).with_config(config)
github_token, new_encrypted = await resolve_github_token(config, thread_id)
config["metadata"]["github_token_encrypted"] = new_encrypted
gitea_token = await get_gitea_token()
config["metadata"]["gitea_token"] = gitea_token
sandbox_backend = SANDBOX_BACKENDS.get(thread_id)
sandbox_id = await get_sandbox_id_from_metadata(thread_id)
@ -270,15 +271,15 @@ async def get_agent(config: RunnableConfig) -> Pregel: # noqa: PLR0915
logger.info("Pulling latest changes for repo %s/%s", repo_owner, repo_name)
try:
repo_dir = await _clone_or_pull_repo_in_sandbox(
sandbox_backend, repo_owner, repo_name, github_token
sandbox_backend, repo_owner, repo_name, gitea_token
)
except SandboxClientError:
except SandboxConnectionError:
logger.warning(
"Cached sandbox is no longer reachable for thread %s, recreating sandbox",
thread_id,
)
sandbox_backend, repo_dir = await _recreate_sandbox(
thread_id, repo_owner, repo_name, github_token=github_token
thread_id, repo_owner, repo_name, gitea_token=gitea_token
)
except Exception:
logger.exception("Failed to pull repo in cached sandbox")
@ -297,7 +298,7 @@ async def get_agent(config: RunnableConfig) -> Pregel: # noqa: PLR0915
if repo_owner and repo_name:
logger.info("Cloning repo %s/%s into sandbox", repo_owner, repo_name)
repo_dir = await _clone_or_pull_repo_in_sandbox(
sandbox_backend, repo_owner, repo_name, github_token
sandbox_backend, repo_owner, repo_name, gitea_token
)
logger.info("Repo cloned to %s", repo_dir)
@ -342,15 +343,15 @@ async def get_agent(config: RunnableConfig) -> Pregel: # noqa: PLR0915
logger.info("Pulling latest changes for repo %s/%s", repo_owner, repo_name)
try:
repo_dir = await _clone_or_pull_repo_in_sandbox(
sandbox_backend, repo_owner, repo_name, github_token
sandbox_backend, repo_owner, repo_name, gitea_token
)
except SandboxClientError:
except SandboxConnectionError:
logger.warning(
"Existing sandbox is no longer reachable for thread %s, recreating sandbox",
thread_id,
)
sandbox_backend, repo_dir = await _recreate_sandbox(
thread_id, repo_owner, repo_name, github_token=github_token
thread_id, repo_owner, repo_name, gitea_token=gitea_token
)
except Exception:
logger.exception("Failed to pull repo in existing sandbox")
@ -362,9 +363,6 @@ async def get_agent(config: RunnableConfig) -> Pregel: # noqa: PLR0915
msg = "Cannot proceed: no repo was cloned. Set 'repo.owner' and 'repo.name' in the configurable config"
raise RuntimeError(msg)
linear_issue = config["configurable"].get("linear_issue", {})
linear_project_id = linear_issue.get("linear_project_id", "")
linear_issue_number = linear_issue.get("linear_issue_number", "")
agents_md = await read_agents_md_in_sandbox(sandbox_backend, repo_dir)
logger.info("Returning agent with sandbox for thread %s", thread_id)
@ -372,17 +370,14 @@ async def get_agent(config: RunnableConfig) -> Pregel: # noqa: PLR0915
model=make_model("anthropic:claude-opus-4-6", temperature=0, max_tokens=20_000),
system_prompt=construct_system_prompt(
repo_dir,
linear_project_id=linear_project_id,
linear_issue_number=linear_issue_number,
agents_md=agents_md,
),
tools=[
http_request,
fetch_url,
commit_and_open_pr,
linear_comment,
slack_thread_reply,
github_comment,
gitea_comment,
discord_reply,
],
backend=sandbox_backend,
middleware=[