Coding Review Assistant🧑‍🚀

const { createToolCallingAgent, AgentExecutor } = require("langchain/agents"); const { LangchainToolSet } = require("composio-core"); const { ChatOpenAI } = require("@langchain/openai"); const { ChatPromptTemplate } = require("@langchain/core/prompts"); async function main(inputs) { const llm = new ChatOpenAI({ model: "gpt-4o", apiKey: inputs.OPENAI_API_KEY, }); const toolset = new LangchainToolSet({ apiKey: inputs.COMPOSIO_API_KEY }); const entity = await toolset.client.getEntity(inputs.entityId); const prReviewTools = await toolset.getActions({ actions: ["GITHUB_GET_A_PULL_REQUEST", "GITHUB_CREATE_A_REVIEW_COMMENT_FOR_A_PULL_REQUEST"], }, entity.id); const prSummaryTools = await toolset.getActions({ actions: ["GITHUB_GET_A_PULL_REQUEST", "GITHUB_CREATE_AN_ISSUE_COMMENT"] },entity.id); const codeReviewerAgent = await createToolCallingAgent({ llm, tools: prReviewTools, prompt: ChatPromptTemplate.fromMessages([ ["system", `You are an experienced code reviewer. Your task is to review the provided file diff and give constructive feedback. Follow these steps: 1. Identify if the file contains significant logic changes. Typically, such files have modifications in function logic. 2. Summarize the changes in the diff in clear and concise English, within 100 words. Focus on the essence of the changes without repeating the first step. 3. Provide actionable suggestions if there are any issues in the code. Avoid unnecessary comments unless there are clear problems or typos. <format_instructions_for_review_comment> Review the file <name> change and provide: 1. An assessment of whether the file contains major logic changes. 2. A brief summary of the diff changes, within 100 words. 3. Constructive suggestions if there are any issues in the code. Avoid unnecessary comments unless there are clear problems or typos. Follow the instructions for review comments: <format_instructions_for_review_comment> Here is the diff content: \`\`\` <text> NOTE: Use the \`github_pulls_create_review_comment\` tool to post comments highlighting specific sections of the PR changes. \`\`\``], ["placeholder", "{chat_history}"], ["human", "{input}"], ["placeholder", "{agent_scratchpad}"], ]), }); const prSummariserAgent = await createToolCallingAgent({ llm, tools: prSummaryTools, prompt: ChatPromptTemplate.fromMessages([ ["system", `Act as a Code Reviewer Assistant. I want you to provide some information about the below Pull Request (PR) to help reviewers understand it better and review it faster. The items I want you to provide are: - Describe the changes of this PR and its objective. - Categorize this PR into one of the following types: Feature, Fix, Refactor, Perf, Doc, Test, Ci, Style, Housekeeping - If it's a feature/refactor PR, list the important change files which you believe contain the major logical changes of this PR. Below is information about this PR I can provide to you: PR Metadata: \`\`\`text <metadata_of_the_pr> \`\`\` Change Files (with status): \`\`\`text <change_files_in_the_pr> \`\`\` Code change summaries (if this PR contains no code files, this will be empty): \`\`\`text <code_summaries_of_the_pr> \`\`\` INSTRUCTIONS: 1. Read the PR step by step and then make comments. 2. DON'T PRIORTISE FORMATTING ISSUES `], ["placeholder", "{chat_history}"], ["human", "{input}"], ["placeholder", "{agent_scratchpad}"], ]), }); const codeReviewerExecutor = new AgentExecutor({ agent: codeReviewerAgent, tools: prReviewTools, verbose: true }); const prSummariserExecutor = new AgentExecutor({ agent: prSummariserAgent, tools: prSummaryTools, verbose: true }); const reviewResult = await codeReviewerExecutor.invoke({input: `Review the provided PR and create review comments if there is any serious issue with the changes. Ensure the comment on review PR is posted successfully if an actual comment isn't needed, just ignore it. USE github_pulls_create_review_comment to COMMENT ON THE CHANGES. DO NOT COMMENT UNCESSARILY IF THERE IS NO SERIOUS ISSUE OR A TYPO. THE URL OF THE PR is: ${inputs.PR_URL}`}) console.log(reviewResult.output); const summaryResult = await prSummariserExecutor.invoke({input: `Summarize the provided PR and post an comment in github PR with the changes summary. Ensure the summary is posted successfully before completing the task using github_issues_create_comment tool. THE URL OF THE PR is: ${inputs.PR_URL} EXPECTED OUTCOME: A github comment is posted with the changes summary in the specified PR using github_issues_create_comment tool. `}); return `## PR Review Comments: ${reviewResult.output} ## PR Summary: ${summaryResult.output} ` }; module.exports = { main };