github/OpenHands
1
0
Fork 0
mirror of https://github.com/OpenHands/OpenHands.git synced 2026-03-22 13:47:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add branch information to repository context to prevent unwanted branch switching (#9833)

Browse Source
This commit is contained in:
Rohit Malhotra
2025-08-01 00:25:36 -04:00
committed by GitHub
parent 1cdc38eafb
commit 287c34b3f3
11 changed files with 44 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
openhands/memory/memory.py
View File

@@ -181,6 +181,9 @@ class Memory:
if self.repository_info
and self.repository_info.repo_directory is not None
else '',
repo_branch=self.repository_info.branch_name
if self.repository_info and self.repository_info.branch_name is not None
else '',
repo_instructions=repo_instructions if repo_instructions else '',
runtime_hosts=self.runtime_info.available_hosts
if self.runtime_info and self.runtime_info.available_hosts is not None
@@ -322,10 +325,14 @@ class Memory:
return mcp_configs
def set_repository_info(self, repo_name: str, repo_directory: str) -> None:
def set_repository_info(
self, repo_name: str, repo_directory: str, branch_name: str | None = None
) -> None:
"""Store repository info so we can reference it in an observation."""
if repo_name or repo_directory:
self.repository_info = RepositoryInfo(repo_name, repo_directory)
self.repository_info = RepositoryInfo(
repo_name, repo_directory, branch_name
)
else:
self.repository_info = None