diff --git a/src/codegen/extensions/langchain/tools.py b/src/codegen/extensions/langchain/tools.py index 933cb628d..e4ff30394 100644 --- a/src/codegen/extensions/langchain/tools.py +++ b/src/codegen/extensions/langchain/tools.py @@ -1,6 +1,5 @@ """Langchain tools for workspace operations.""" -import json from typing import Callable, ClassVar, Literal, Optional from langchain_core.tools.base import BaseTool @@ -462,7 +461,6 @@ def __init__(self, codebase: Codebase) -> None: def _run(self, pr_id: int) -> str: result = view_pr(self.codebase, pr_id) return result.render() - return json.dumps(result, indent=2) class GithubCreatePRCommentInput(BaseModel): @@ -791,4 +789,4 @@ def _run( end=end, count=count, ) - return json.dumps(result, indent=2) + return result.render()