Merge branch 'develop' into jmeza-cg-9949-run-generate-docs-generation-only-in-ci

Author: jemeza-codegen

.github/workflows/unit-tests.yml .github/workflows/test.yml.github/workflows/unit-tests.yml renamed to .github/workflows/test.yml

@@ -1,24 +1,38 @@
-name: Unit Tests
+name: Tests
 
 on:
   push:
     branches:
       - "develop"
-  pull_request:
+  pull_request_target:
     types: [ opened, synchronize, reopened, labeled ]
     branches:
       - "develop"
   workflow_dispatch:
 
 jobs:
+  access-check:
+    runs-on: ubuntu-latest
+    steps:
+      - uses: actions-cool/check-user-permission@v2
+        with:
+          require: write
+          username: ${{ github.triggering_actor }}
+          error-if-missing: true
+
   unit-tests:
+    needs: access-check
     runs-on: ubuntu-latest-8
+    environment: testing
     steps:
       - uses: actions/checkout@v4
         with:
           fetch-depth: 0
+          ref: ${{ github.event.pull_request.head.sha }}
+
       - name: Setup environment
         uses: ./.github/actions/setup-environment
+
       - name: Run ATS and Tests
         uses: ./.github/actions/run-ats
         timeout-minutes: 15
@@ -30,9 +44,11 @@ jobs:
           codecov_flags: unit-tests
 
   codemod-tests:
+    needs: access-check
     # TODO: re-enable when this check is a develop required check
     if: false
     runs-on: ubuntu-latest-32
+    environment: testing
     strategy:
       matrix:
         sync_graph: [ true, false ]
@@ -48,10 +64,15 @@ jobs:
     name: "Codemod tests ${{matrix.size}}: Sync Graph=${{matrix.sync_graph}}"
     steps:
       - uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.pull_request.head.sha }}
+
       - name: Setup environment
         uses: ./.github/actions/setup-environment
+
       - name: Cache oss-repos
         uses: ./.github/actions/setup-oss-repos
+
       - name: Run ATS and Tests
         uses: ./.github/actions/run-ats
         timeout-minutes: 15
@@ -66,11 +87,15 @@ jobs:
           GITHUB_WORKSPACE: $GITHUB_WORKSPACE
 
   parse-tests:
-    runs-on: ubuntu-latest-32
+    needs: access-check
     if: contains(github.event.pull_request.labels.*.name, 'parse-tests') || github.event_name == 'push' || github.event_name == 'workflow_dispatch'
-    environment: parse-tests
+    runs-on: ubuntu-latest-32
+    environment: testing
     steps:
       - uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.pull_request.head.sha }}
+
       - name: Setup environment
         uses: ./.github/actions/setup-environment
 
@@ -91,10 +116,12 @@ jobs:
             -n auto \
             -o junit_suite_name="${{github.job}}" \
             tests/integration/codemod/test_parse.py
+
       - uses: ./.github/actions/report
         with:
           flag: no-flag
           codecov_token: ${{ secrets.CODECOV_TOKEN }}
+
       - name: Notify parse tests failure
         uses: slackapi/slack-github-action@v2.0.0
         if: failure() && github.event_name == 'push' && false
@@ -132,11 +159,17 @@ jobs:
             }
 
   integration-tests:
+    needs: access-check
     runs-on: ubuntu-latest-16
+    environment: testing
     steps:
       - uses: actions/checkout@v4
+        with:
+          ref: ${{ github.event.pull_request.head.sha }}
+
       - name: Setup environment
         uses: ./.github/actions/setup-environment
+
       - name: Test with pytest
         timeout-minutes: 5
         env:
@@ -147,6 +180,7 @@ jobs:
             -n auto \
             -o junit_suite_name="${{github.job}}" \
             tests/integration/codegen
+
       - uses: ./.github/actions/report
         with:
           flag: integration-tests

src/codegen/configs/models/codebase.py

@@ -15,6 +15,7 @@ def __init__(self, prefix: str = "CODEBASE", *args, **kwargs) -> None:
     full_range_index: bool = False
     ignore_process_errors: bool = True
     disable_graph: bool = False
+    disable_file_parse: bool = False
     generics: bool = True
     import_resolution_overrides: dict[str, str] = Field(default_factory=lambda: {})
     ts_dependency_manager: bool = False

src/codegen/extensions/langchain/agent.py

@@ -13,14 +13,14 @@
 from .tools import (
     CreateFileTool,
     DeleteFileTool,
-    EditFileTool,
     ListDirectoryTool,
     MoveSymbolTool,
+    RelaceEditTool,
     RenameFileTool,
     ReplacementEditTool,
     RevealSymbolTool,
     SearchTool,
-    SemanticEditTool,
+    # SemanticEditTool,
     ViewFileTool,
 )
 
@@ -62,14 +62,15 @@ def create_codebase_agent(
         ViewFileTool(codebase),
         ListDirectoryTool(codebase),
         SearchTool(codebase),
-        EditFileTool(codebase),
+        # EditFileTool(codebase),
         CreateFileTool(codebase),
         DeleteFileTool(codebase),
         RenameFileTool(codebase),
         MoveSymbolTool(codebase),
         RevealSymbolTool(codebase),
-        SemanticEditTool(codebase),
+        # SemanticEditTool(codebase),
         ReplacementEditTool(codebase),
+        RelaceEditTool(codebase),
         # SemanticSearchTool(codebase),
         # =====[ Github Integration ]=====
         # Enable Github integration

src/codegen/extensions/langchain/tools.py

@@ -16,6 +16,7 @@
     linear_search_issues_tool,
 )
 from codegen.extensions.tools.link_annotation import add_links_to_message
+from codegen.extensions.tools.relace_edit import relace_edit
 from codegen.extensions.tools.replacement_edit import replacement_edit
 from codegen.extensions.tools.reveal_symbol import reveal_symbol
 from codegen.extensions.tools.search import search
@@ -37,6 +38,7 @@
     view_file,
     view_pr,
 )
+from ..tools.relace_edit_prompts import RELACE_EDIT_PROMPT
 from ..tools.semantic_edit_prompts import FILE_EDIT_PROMPT
 
 
@@ -729,9 +731,10 @@ def get_workspace_tools(codebase: Codebase) -> list["BaseTool"]:
         RevealSymbolTool(codebase),
         RunBashCommandTool(),  # Note: This tool doesn't need the codebase
         SearchTool(codebase),
-        SemanticEditTool(codebase),
+        # SemanticEditTool(codebase),
         SemanticSearchTool(codebase),
         ViewFileTool(codebase),
+        RelaceEditTool(codebase),
         # Github
         GithubCreatePRTool(codebase),
         GithubCreatePRCommentTool(codebase),
@@ -788,3 +791,47 @@ def _run(
             count=count,
         )
         return result.render()
+
+
+# Brief description for the Relace Edit tool
+_RELACE_EDIT_BRIEF = """Tool for file editing using the Relace Instant Apply API.
+This high-speed code generation engine optimizes for real-time performance at 2000 tokens/second.
+
+Provide an edit snippet that describes the changes you want to make, with helpful comments to indicate unchanged sections, like so:
+```
+// ... keep existing imports ...
+
+// Add new function
+function calculateDiscount(price, discountPercent) {
+  return price * (discountPercent / 100);
+}
+
+// ... keep existing code ...
+```
+
+The API will merge your edit snippet with the existing code to produce the final result.
+The API key will be automatically retrieved from the RELACE_API environment variable.
+"""
+
+
+class RelaceEditInput(BaseModel):
+    """Input for Relace editing."""
+
+    filepath: str = Field(..., description="Path of the file relative to workspace root")
+    edit_snippet: str = Field(..., description=RELACE_EDIT_PROMPT)
+
+
+class RelaceEditTool(BaseTool):
+    """Tool for editing files using the Relace Instant Apply API."""
+
+    name: ClassVar[str] = "relace_edit"
+    description: ClassVar[str] = _RELACE_EDIT_BRIEF
+    args_schema: ClassVar[type[BaseModel]] = RelaceEditInput
+    codebase: Codebase = Field(exclude=True)
+
+    def __init__(self, codebase: Codebase) -> None:
+        super().__init__(codebase=codebase)
+
+    def _run(self, filepath: str, edit_snippet: str) -> str:
+        result = relace_edit(self.codebase, filepath, edit_snippet)
+        return result.render()