Merge pull request #591 from srtab/feat/repository-ctx-rename

Renamed RepositoryCtx to RuntimeCtx.

Author: srtab

README.md

@@ -7,8 +7,7 @@
 DAIV is an open-source automation assistant designed to enhance developer productivity. It integrates seamlessly with **GitLab** and **GitHub** repositories to streamline your development process. It uses AI agents and configurable actions to automate common software development tasks such as:
 
 - **Issue Addressing**: Planning and, after human approval, executing solutions directly from issue titles and descriptions.
-- **Code Review Assistance**: Automatically responding to reviewer comments, adjusting code, and improving pull requests on demand.
-- **Pipeline Fixer**: Repairing failed CI/CD pipeline jobs by suggesting fixes and applying them after human approval.
+- **Code Review Assistance**: Automatically responding to reviewer comments, adjusting code, repairing failed CI/CD pipelines and improving pull requests on demand.
 
 ## Key Features
 
@@ -135,7 +134,7 @@ DAIV is an open-source automation assistant designed to enhance developer produc
 
 - [x] 🐙 Add support to GitHub.
 - [x] 🤖 Add support to [AGENTS.md](https://agents.md/) format to guide agents.
-- [x] 🌐 Add support to custom MCP servers.
+- [ ] 🌐 Add support to custom MCP servers.
 - [x] 📊 Add an evaluation system to measure the quality of DAIV's agents.
 - [ ] 🔍 Add support to automated code review.
 - [ ] 🎨 Create a frontend to DAIV initial setup and configuration, dashboard with some metrics, a chat interface to interact with DAIV...

daiv/automation/agents/codebase_chat/agent.py

@@ -10,7 +10,7 @@
 
 from automation.agents import BaseAgent, ThinkingLevel
 from automation.agents.tools.toolkits import FileNavigationToolkit
-from codebase.context import get_repository_ctx
+from codebase.context import get_runtime_ctx
 
 from .conf import settings
 from .prompts import codebase_chat_system
@@ -20,7 +20,7 @@ class CodebaseChatAgent(BaseAgent[CompiledStateGraph]):
     """
     Agent for answering questions about specific repository.
 
-    Use `set_repository_ctx` to set the repository context before using this agent.
+    Use `set_runtime_ctx` to set the runtime context before using this agent.
     """
 
     async def compile(self) -> CompiledStateGraph:
@@ -37,7 +37,7 @@ async def compile(self) -> CompiledStateGraph:
             tools=FileNavigationToolkit.get_tools(),
             store=InMemoryStore(),
             prompt=ChatPromptTemplate.from_messages([codebase_chat_system, MessagesPlaceholder("messages")]).partial(
-                current_date_time=timezone.now().strftime("%d %B, %Y"), repository=get_repository_ctx().repo_id
+                current_date_time=timezone.now().strftime("%d %B, %Y"), repository=get_runtime_ctx().repo_id
             ),
             name=settings.NAME,
         ).with_config(RunnableConfig(recursion_limit=settings.RECURSION_LIMIT))

daiv/automation/agents/nodes.py

@@ -5,7 +5,7 @@
 from typing import TYPE_CHECKING
 
 from automation.agents.tools.sandbox import bash_tool
-from codebase.context import get_repository_ctx
+from codebase.context import get_runtime_ctx
 
 if TYPE_CHECKING:
     from langgraph.store.base import BaseStore
@@ -23,7 +23,7 @@ async def apply_format_code_node(store: BaseStore) -> str | None:
     Returns:
         str | None: The output of the last command, or None if format code is disabled.
     """
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
 
     if not ctx.config.sandbox.enabled or not ctx.config.sandbox.format_code:
         logger.info("Format code is disabled for this repository, skipping.")

daiv/automation/agents/plan_and_execute/agent.py

@@ -29,7 +29,7 @@
 )
 from automation.agents.utils import extract_images_from_text
 from automation.utils import has_file_changes
-from codebase.context import get_repository_ctx
+from codebase.context import get_runtime_ctx
 from core.constants import BOT_LABEL, BOT_NAME
 
 from .conf import settings
@@ -185,7 +185,7 @@ def __init__(
         self.plan_system_template = plan_system_template or plan_system
         self.skip_approval = skip_approval
         self.skip_format_code = skip_format_code
-        self.ctx = get_repository_ctx()
+        self.ctx = get_runtime_ctx()
         super().__init__(**kwargs)
 
     async def compile(self) -> CompiledStateGraph:

daiv/automation/agents/review_addressor/agent.py

@@ -17,7 +17,7 @@
 from automation.agents.plan_and_execute import PlanAndExecuteAgent
 from automation.agents.tools.toolkits import FileNavigationToolkit, MergeRequestToolkit, WebSearchToolkit
 from codebase.clients import RepoClient
-from codebase.context import get_repository_ctx
+from codebase.context import get_runtime_ctx
 from core.constants import BOT_NAME
 
 from .conf import settings
@@ -48,7 +48,7 @@ class ReplyReviewerAgent(BaseAgent[CompiledStateGraph]):
     """
 
     async def compile(self) -> CompiledStateGraph:
-        ctx = get_repository_ctx()
+        ctx = get_runtime_ctx()
 
         tools = FileNavigationToolkit.get_tools() + WebSearchToolkit.get_tools() + MergeRequestToolkit.get_tools()
 

daiv/automation/agents/schemas.py

@@ -8,7 +8,7 @@
 
 from codebase.base import ClientType
 from codebase.clients import RepoClient
-from codebase.context import get_repository_ctx
+from codebase.context import get_runtime_ctx
 from core.utils import extract_valid_image_mimetype, is_valid_url
 
 
@@ -35,7 +35,7 @@ async def from_images(images: list[Image]) -> list[ImageTemplate]:
         Returns:
             list[ImageTemplate]: The list of image templates.
         """
-        ctx = get_repository_ctx()
+        ctx = get_runtime_ctx()
         repo_client = RepoClient.create_instance()
         image_templates = []
 

daiv/automation/agents/tools/editing.py

@@ -9,7 +9,7 @@
 from automation.agents.utils import find_original_snippet
 from automation.utils import check_file_read, get_file_change, get_file_changes, register_file_change
 from codebase.base import FileChangeAction
-from codebase.context import get_repository_ctx
+from codebase.context import get_runtime_ctx
 
 logger = logging.getLogger("daiv.tools")
 
@@ -51,7 +51,7 @@ async def edit_tool(
     """  # noqa: E501
     logger.debug("[%s] Editing file '%s'", edit_tool.name, file_path)
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
     resolved_file_path = (ctx.repo_dir / file_path).resolve()
 
     if not resolved_file_path.exists() or not resolved_file_path.is_file():
@@ -121,7 +121,7 @@ async def write_tool(file_path: str, content: str, store: Annotated[Any, Injecte
     """  # noqa: E501
     logger.debug("[%s] Writing to file '%s'", write_tool.name, file_path)
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
     resolved_file_path = (ctx.repo_dir / file_path).resolve()
     file_exists = resolved_file_path.exists()
 
@@ -171,7 +171,7 @@ async def delete_tool(file_path: str, store: Annotated[Any, InjectedStore()] = N
     """  # noqa: E501
     logger.debug("[%s] Deleting file '%s'", delete_tool.name, file_path)
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
     resolved_file_path = (ctx.repo_dir / file_path).resolve()
 
     if not resolved_file_path.exists() or not resolved_file_path.is_file():
@@ -221,7 +221,7 @@ async def rename_tool(file_path: str, new_file_path: str, store: Annotated[Any,
     """  # noqa: E501
     logger.debug("[%s] Renaming file '%s' to '%s'", rename_tool.name, file_path, new_file_path)
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
 
     resolved_file_path = (ctx.repo_dir / file_path).resolve()
     resolved_new_file_path = (ctx.repo_dir / new_file_path).resolve()

daiv/automation/agents/tools/merge_request.py

@@ -7,7 +7,7 @@
 
 from codebase.clients import RepoClient
 from codebase.clients.utils import clean_job_logs
-from codebase.context import get_repository_ctx
+from codebase.context import get_runtime_ctx
 
 logger = logging.getLogger("daiv.tools")
 
@@ -35,7 +35,7 @@ def pipeline_tool(placeholder: str = "") -> str:
     Returns:
         str: JSON formatted pipeline/workflow information including status and job details.
     """
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
 
     logger.debug("[%s] Getting pipeline for merge request %d", pipeline_tool.name, ctx.merge_request_id)
 
@@ -120,7 +120,7 @@ def job_logs_tool(job_id: int, offset_from_end: int = 0, line_count: int = JOB_L
         offset_from_end,
     )
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
     client = RepoClient.create_instance()
 
     # Get job details to determine status

daiv/automation/agents/tools/navigation.py

@@ -13,7 +13,7 @@
 
 from automation.agents.schemas import ImageTemplate
 from automation.utils import register_file_read
-from codebase.context import get_repository_ctx
+from codebase.context import get_runtime_ctx
 from core.utils import extract_valid_image_mimetype
 
 logger = logging.getLogger("daiv.tools")
@@ -47,7 +47,7 @@ def glob_tool(pattern: str, path: str | None = None) -> str:
     """  # noqa: E501
     logger.debug("[%s] Finding files matching '%s' in %s", glob_tool.name, pattern, path or "repository root")
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
 
     root = ctx.repo_dir if path is None else (ctx.repo_dir / path).resolve()
 
@@ -116,7 +116,7 @@ def grep_tool(pattern: str, path: str | None = None, include: str | None = None)
         include,
     )
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
 
     root = ctx.repo_dir if path is None else (ctx.repo_dir / path).resolve()
 
@@ -151,7 +151,7 @@ def ls_tool(path: str) -> str:
     """  # noqa: E501
     logger.debug("[%s] Listing files in %s", ls_tool.name, path)
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
 
     root = (ctx.repo_dir / path).resolve()
 
@@ -193,7 +193,7 @@ async def read_tool(file_path: str, store: Annotated[Any, InjectedStore()] = Non
     """  # noqa: E501
     logger.debug("[%s] Reading file '%s'", read_tool.name, file_path)
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
     resolved_file_path = (ctx.repo_dir / file_path).resolve()
 
     if (

daiv/automation/agents/tools/sandbox.py

@@ -15,7 +15,7 @@
 
 from automation.utils import register_file_change
 from codebase.base import FileChangeAction
-from codebase.context import get_repository_ctx
+from codebase.context import get_runtime_ctx
 from core.conf import settings
 from core.sandbox import run_sandbox_commands, start_sandbox_session
 from core.sandbox.schemas import RunCommandResult, RunCommandsRequest, StartSessionRequest
@@ -132,7 +132,7 @@ async def bash_tool(commands: list[str], store: Annotated[Any, InjectedStore()])
 
     assert settings.SANDBOX_API_KEY is not None, "SANDBOX_API_KEY is not set"
 
-    ctx = get_repository_ctx()
+    ctx = get_runtime_ctx()
 
     # Start the sandbox session to ensure that the sandbox session is started before running the commands.
     # If the sandbox session already exists in the store, it will be reused.