Update github-repo-analyzer.py

github-repo-analyzer.py

@@ -6,7 +6,6 @@ from urllib.parse import urlparse
 import requests
 from github import Github
 from git import Repo
-import anthropic
 from collections import defaultdict
 import time
 import numpy as np
@@ -149,15 +148,15 @@ def analyze_pull_requests(github_repo, max_prs):
         'filtered_closed_prs': filtered_closed_prs
     }
 
-def call_llm(client, prompt, model="claude-3-5-sonnet-20240620", max_tokens=4096):
-    message = client.messages.create(
+def call_llm(client, prompt, model="nousresearch/hermes-3-llama-3.1-405b:free", max_tokens=4096):
+    message = client.chat.completions.create(
         max_tokens=max_tokens,
         model=model,
         messages=[
             {"role": "user", "content": prompt}
         ]
     )
-    return message.content[0].text
+    return message.content[0].message.content
 
 def safe_call_llm(client, prompt, retries=3):
     for attempt in range(retries):
@@ -603,79 +602,3 @@ def generate_report(repo_info, code_analysis, issues_analysis, pr_analysis, fina
     report += f"\n## Synthesis and Recommendations\n{final_analysis}\n"
     
     return report
-
-def main(repo_input, max_issues, max_prs):
-    github_token = os.environ.get("GITHUB_TOKEN")
-    if not github_token:
-        print("Error: GITHUB_TOKEN environment variable not set.")
-        sys.exit(1)
-
-    anthropic_api_key = os.environ.get("ANTHROPIC_API_KEY")
-    if not anthropic_api_key:
-        print("Error: ANTHROPIC_API_KEY environment variable not set.")
-        sys.exit(1)
-
-    owner, repo_name = get_repo_info(repo_input)
-    repo_url = f"https://github.com/{owner}/{repo_name}"
-    
-    g = Github(github_token)
-    github_repo = g.get_repo(f"{owner}/{repo_name}")
-
-    client = anthropic.Anthropic(api_key=anthropic_api_key)
-
-    with tempfile.TemporaryDirectory() as temp_dir:
-        try:
-            print(f"Cloning repository {owner}/{repo_name}...")
-            repo_path = clone_repo(owner, repo_name, temp_dir)
-
-            print("Analyzing code...")
-            code_analysis = analyze_code(repo_path)
-            code_analysis['llm_analysis'] = llm_analyze_code(client, code_analysis)
-
-            print(f"Analyzing issues (max {max_issues})...")
-            issues_data = analyze_issues(github_repo, max_issues)
-            issues_analysis = llm_analyze_issues(client, issues_data, repo_url)
-
-            print(f"Analyzing pull requests (max {max_prs})...")
-            prs_data = analyze_pull_requests(github_repo, max_prs)
-            pr_analysis = llm_analyze_prs(client, prs_data, repo_url)
-
-            print("Synthesizing findings...")
-            final_analysis = llm_synthesize_findings(
-                client, 
-                code_analysis.get('llm_analysis', ''),
-                issues_analysis.get('summary', ''),
-                pr_analysis.get('summary', '')
-            )
-
-            repo_info = {
-                "owner": owner,
-                "repo_name": repo_name,
-            }
-
-            print("Generating report...")
-            report = generate_report(repo_info, code_analysis, issues_analysis, pr_analysis, final_analysis)
-
-            print("\nAnalysis Report:")
-            print(report)
-
-            # Save the report to a file
-            with open(f"{owner}_{repo_name}_analysis.md", "w") as f:
-                f.write(report)
-            print(f"\nReport saved to {owner}_{repo_name}_analysis.md")
-
-        except Exception as e:
-            print(f"An error occurred: {str(e)}")
-            traceback.print_exc()
-        finally:
-            print("Cleaning up...")
-
-if __name__ == "__main__":
-    parser = argparse.ArgumentParser(description="Analyze a GitHub repository with limits on issues and PRs.")
-    parser.add_argument("repo", help="Repository slug (owner/repo) or URL")
-    parser.add_argument("--max_issues", type=int, default=10, help="Maximum number of issues to analyze")
-    parser.add_argument("--max_prs", type=int, default=10, help="Maximum number of pull requests to analyze")
-    
-    args = parser.parse_args()
-    
-    main(args.repo, args.max_issues, args.max_prs)