Update agent.py

agent.py

@@ -2,10 +2,13 @@ import os
 import requests
 from typing import List
 from crewai import Agent, Task, Crew, Process
-from google import genai  # Gemini client
 from dotenv import load_dotenv
 
+# ✅ Gemini client (modern import)
+from google import genai
+
 load_dotenv()
+
 # ---------------------------------
 # CONFIG
 # ---------------------------------
@@ -19,10 +22,10 @@ if not GOOGLE_API_KEY:
 client = genai.Client(api_key=GOOGLE_API_KEY)
 
 # ---------------------------------
-# HELPER: Simple GitHub Repo Fetcher (no embeddings)
+# SIMPLE GITHUB FETCHER (no embeddings)
 # ---------------------------------
 def fetch_repo_files(repo_url: str, max_files: int = 10) -> List[str]:
-    """Fetch first few code/text files from a GitHub repo using the REST API."""
+    """Fetch a few code/text files from a GitHub repo using the REST API."""
     try:
         owner_repo = repo_url.strip().split("github.com/")[-1]
         api_url = f"https://api.github.com/repos/{owner_repo}/contents"
@@ -42,16 +45,20 @@ def fetch_repo_files(repo_url: str, max_files: int = 10) -> List[str]:
 
 
 def fetch_file_content(url: str) -> str:
+    """Fetch raw file text safely."""
     try:
-        return requests.get(url).text
+        res = requests.get(url, timeout=10)
+        res.raise_for_status()
+        return res.text
     except Exception as e:
-        return f"⚠️ Could not fetch file: {url}\nError: {e}"
+        return f"⚠️ Could not fetch {url}\nError: {e}"
 
 # ---------------------------------
-# Gemini LLM Wrapper
+# GEMINI WRAPPER
 # ---------------------------------
 class GeminiLLM:
-    def __init__(self, model):
+    """CrewAI-compatible Gemini LLM wrapper."""
+    def __init__(self, model: str):
         self.model = model
 
     def generate(self, prompt: str) -> str:
@@ -59,54 +66,55 @@ class GeminiLLM:
             res = client.models.generate_content(
                 model=self.model,
                 contents=prompt,
-                generation_config={"temperature": 0.7, "max_output_tokens": 2048}
+                generation_config={"temperature": 0.6, "max_output_tokens": 2048},
             )
             return res.text
         except Exception as e:
             return f"⚠️ Gemini Error: {e}"
 
+# instantiate global LLM
 gemini_llm = GeminiLLM(MODEL_NAME)
 
 # ---------------------------------
 # AGENTS
 # ---------------------------------
-def make_agents():
+def make_agents(repo_url: str):
     repo_mapper = Agent(
         role="Repository Mapper",
-        goal="Map the project’s structure and identify its core technologies.",
-        backstory="You are skilled at reading GitHub repositories and summarizing their structure.",
+        goal="Map project structure, detect tech stack, and summarize key components.",
+        backstory="You analyze folder trees and dependencies for architecture insights.",
         llm=gemini_llm,
         verbose=True,
     )
 
     code_reviewer = Agent(
         role="Code Reviewer",
-        goal="Perform code reviews to identify potential issues and refactors.",
-        backstory="A senior engineer reviewing open-source codebases with actionable advice.",
+        goal="Perform pragmatic code reviews with clear, actionable feedback.",
+        backstory="An experienced engineer providing concise improvement tips.",
         llm=gemini_llm,
         verbose=True,
     )
 
     security_auditor = Agent(
         role="Security Auditor",
-        goal="Find potential security risks in code and suggest fixes.",
-        backstory="You think like an attacker but report like a professional auditor.",
+        goal="Find and describe security issues, secrets, or risky dependencies.",
+        backstory="You think like an attacker but document like a pro auditor.",
         llm=gemini_llm,
         verbose=True,
     )
 
     doc_explainer = Agent(
         role="Documentation Explainer",
-        goal="Explain what the repo does and how to run it.",
-        backstory="You make technical systems understandable.",
+        goal="Explain repository purpose, architecture, and how to run or contribute.",
+        backstory="You make complex projects understandable for new contributors.",
         llm=gemini_llm,
         verbose=True,
     )
 
     manager = Agent(
         role="Engineering Manager",
-        goal="Merge all insights into a final cohesive report.",
-        backstory="You coordinate team outputs into a polished result.",
+        goal="Coordinate all other agents and compile a cohesive final report.",
+        backstory="A seasoned manager merging all insights into one structured summary.",
         allow_delegation=True,
         llm=gemini_llm,
         verbose=True,
@@ -119,37 +127,42 @@ def make_agents():
 # ---------------------------------
 def make_tasks(repo_url: str, brief: str = ""):
     repo_files = fetch_repo_files(repo_url)
-    file_contents = "\n\n".join([fetch_file_content(u) for u in repo_files[:5]])
+    file_contents = "\n\n".join(fetch_file_content(f) for f in repo_files[:5])
 
-    context = f"Repository: {repo_url}\n{brief}\nFetched files:\n{', '.join(repo_files[:5])}\n\n{file_contents[:6000]}"
+    context = (
+        f"Repository: {repo_url}\n"
+        f"{'Brief: ' + brief if brief else ''}\n"
+        f"Fetched files: {', '.join(repo_files[:5])}\n\n"
+        f"{file_contents[:5000]}"
+    )
 
     t_map = Task(
-        description=f"{context}\n\nCreate a summary of the repository’s structure, dependencies, and frameworks.",
-        expected_output="Markdown repo overview (sections: Structure, Tech, Dependencies).",
+        description=f"{context}\n\nMap structure, dependencies, and key technologies.",
+        expected_output="Markdown summary: Structure | Frameworks | Key Files.",
         agent_role="Repository Mapper",
     )
 
     t_review = Task(
-        description=f"{context}\n\nPerform a detailed code review and suggest refactors and improvements.",
-        expected_output="Actionable code review notes with example snippets.",
+        description=f"{context}\n\nPerform a detailed review and suggest refactors.",
+        expected_output="Code review bullets grouped by improvement type.",
         agent_role="Code Reviewer",
     )
 
     t_sec = Task(
-        description=f"{context}\n\nPerform a security audit on visible files.",
-        expected_output="Table of Security Issues | Risk | Mitigation.",
+        description=f"{context}\n\nPerform security audit of visible files.",
+        expected_output="Table: Issue | Evidence | Risk | Fix.",
         agent_role="Security Auditor",
     )
 
     t_doc = Task(
         description=f"{context}\n\nExplain what this repo does and how to run it.",
-        expected_output="Simple explanation + setup instructions.",
+        expected_output="Architecture overview + Quickstart guide.",
         agent_role="Documentation Explainer",
     )
 
     t_merge = Task(
-        description="Combine all reports into one well-structured Markdown summary with a title and TOC.",
-        expected_output="Final comprehensive Markdown report.",
+        description="Merge all reports into one well-structured Markdown file with title, TOC, and clear sections.",
+        expected_output="Final cohesive Markdown report.",
         agent_role="Engineering Manager",
     )
 
@@ -159,19 +172,14 @@ def make_tasks(repo_url: str, brief: str = ""):
 # RUNNER
 # ---------------------------------
 def run_repo_review(repo_url: str, brief: str = "") -> str:
-    # create agents (manager returned as last item)
     repo_mapper, reviewer, auditor, explainer, manager = make_agents(repo_url)
-    # create tasks
     t_map, t_review, t_sec, t_doc, t_merge = make_tasks(repo_url, brief)
 
-    # IMPORTANT: do NOT include the manager in the agents list
-    worker_agents = [repo_mapper, reviewer, auditor, explainer]
-
     crew = Crew(
-        agents=worker_agents,
+        agents=[repo_mapper, reviewer, auditor, explainer],  # manager excluded
         tasks=[t_map, t_review, t_sec, t_doc, t_merge],
         process=Process.hierarchical,
-        manager_agent=manager,   # manager passed separately
+        manager_agent=manager,
         verbose=True,
     )