fix

Signed-off-by: Kenta Kozuka <kenta.kozuka@gmail.com>

Author: kentakozuka

.github/workflow/main_workflow.yaml .github/workflows/main_workflow.yaml.github/workflow/main_workflow.yaml renamed to .github/workflows/main_workflow.yaml

@@ -27,9 +27,10 @@ runs:
       uses: actions/setup-python@v4
       with:
         python-version: "3.11"
+
     - name: Install Python library
       run: |
-        poetry install --no-interaction
+        poetry install --no-root --no-interaction
       shell: bash
 
     - name: Run review
@@ -38,4 +39,4 @@ runs:
       env:
         PERSONAL_ACCESS_TOKEN: ${{ inputs.PERSONAL_ACCESS_TOKEN }}
         OPENAI_API_KEY: ${{ inputs.OPENAI_API_KEY }}
-        GITHUB_ISSUE_NUMBER: ${{ github.event.issue.number }}"
+        GITHUB_ISSUE_NUMBER: ${{ github.event.issue.number }}

action.yaml

@@ -1,16 +1,21 @@
 [tool.poetry]
-authors = ["Kenta Kozuka <kenta.kozuka@gmail.com>"]
-description = ""
 name = "github-pr-review"
-packages = [{include = "github_pr_review"}]
-readme = "README.md"
 version = "0.1.0"
+description = ""
+authors = ["Kenta Kozuka <kenta.kozuka@gmail.com>"]
+readme = "README.md"
+packages = [{include = "github_pr_review"}]
 
 [tool.poetry.dependencies]
-langchain = "^0.0.194"
-openai = "^0.27.8"
 python = "^3.11"
+pygithub = "^1.58.2"
+openai = "^0.27.8"
+langchain = "^0.0.194"
+
+
+[tool.poetry.group.dev.dependencies]
+black = "^23.3.0"
 
 [build-system]
-build-backend = "poetry.core.masonry.api"
 requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

poetry.lock

@@ -1,32 +1,77 @@
 import os
 
 from github import Github
-from src.review.review import review
+import openai
+from langchain.chat_models import ChatOpenAI
+from langchain.schema import (
+    SystemMessage,
+    HumanMessage,
+)
+
+CHAR_SOFT_LIMIT = 9000
+FILE_SYSTEM_EN = 'You are an experienced software developer. You will act as a reviewer for a GitHub Pull Request titled "{}".'
+FILE_QUESTION_EN = "The following is a GitHub patch. Please summarize the key changes and identify potential problems. Start with the most important findings.\n\n"
+SUMMARY_SYSTEM_EN = "Here is a set of summaries for software source code patches. Each summary starts with a ------ line. Please write an overall summary considering all the individual summary. Please present the potential issues and errors first, following by the most important findings, in your summary.\n\n{}"
+
+FILE_SYSTEM_JP = 'あなたは経験豊富なソフトウェア開発者です。"{}"というタイトルのGitHub Pull Requestのレビュアーをしてもらいます。'
+FILE_QUESTION_JP = "以下はGitHubのパッチです。主要な変更点をまとめ、潜在的な問題点を特定してください。最も重要な発見から始めてください。\n\n"
+SUMMARY_SYSTEM_JP = "ここでは、ソフトウェアのソースコードパッチに関する要約を紹介します。それぞれの要約は、------の行で始まります。個々の要約をすべて考慮した全体的な要約を作成してください。潜在的な問題やエラーを最初に提示し、次に最も重要な発見を要約してください。\n\n{}"
 
 
 def main():
     pat = os.getenv("PERSONAL_ACCESS_TOKEN")
     repo_name = os.getenv("GITHUB_REPOSITORY")
     issue_number = os.getenv("GITHUB_ISSUE_NUMBER")
-    openai_api_key = os.getenv("OPENAI_API_KEY")
 
     github_client = Github(pat)
     repo = github_client.get_repo(repo_name)
-    issue = repo.get_issue(number=int(issue_number))
+    issue = repo.get_issue(int(issue_number))
+
+    if not issue.pull_request:
+        return
+
+    openai.api_key = os.getenv("OPENAI_API_KEY")
+    chat = ChatOpenAI(temperature=0.5, model_name="gpt-3.5-turbo", request_timeout=600)
+    system = FILE_SYSTEM_JP.format(issue.title)
+
+    # Iterate all patches.
+    reviews = []
+    reviews_summary = ""
+    review_details = ""
+    files = issue.as_pull_request().get_files()
+    for file in files:
+        question = FILE_QUESTION_JP + file.patch[:CHAR_SOFT_LIMIT]
+        print(question)
 
-    comments = issue.get_comments()
+        messages = [
+            SystemMessage(content=system),
+            HumanMessage(content=question),
+        ]
+        ai_message = chat(messages)
+        # For summary.
+        if len(reviews_summary) < CHAR_SOFT_LIMIT:
+            reviews_summary += "------\n"
+            reviews_summary += ai_message.content
+            reviews_summary += "\n"
+        # For details.
+        review_details += "\n---\n\n**{}**\n".format(file.filename)
+        review_details += "{}\n".format(ai_message.content)
 
-    prompt = prompt_template.format(
-        issue_title=issue.title, comments_section=comments_section
-    )
+    # Summarize all patches.
+    question = SUMMARY_SYSTEM_JP.format(reviews_summary)
+    messages = [
+        SystemMessage(content=system),
+        HumanMessage(content=question),
+    ]
+    print(question)
 
-		# Truncate prompt to 4000 characters because 4000 token is the limit.
-    prompt = prompt[:4000]
-    print(prompt)
-    review_text = review(prompt, openai_api_key)
+    ai_message = chat(messages)
+    resp = ai_message.content
+    resp += "\n\n## Details\n\n"
+    resp += review_details
 
-    issue.create_comment(f"\n{review_text}")
+    issue.create_comment(resp)
 
 
 if __name__ == "__main__":
-    main()
+    main()