fix: audit fixes per assignment spec

.env.example

@@ -2,12 +2,14 @@
 # Copy this file to .env and fill in ONE provider key.
 
 # --- LLM Provider (pick one) ---
-ANTHROPIC_API_KEY=sk-ant-...      # Claude 3 Haiku (default if set)
+GROQ_API_KEY=gsk_...              # Groq (llama-3.3-70b-versatile, fast + free)
+# ANTHROPIC_API_KEY=sk-ant-...    # Claude 3 Haiku
 # OPENAI_API_KEY=sk-...           # GPT-4o-mini
 # GOOGLE_API_KEY=...              # Gemini 1.5 Flash
 
 # --- Optional Overrides ---
-# LLM_PROVIDER=anthropic          # Force provider: anthropic | openai | google
+# LLM_PROVIDER=groq               # Force: groq | anthropic | openai | google
+# GROQ_MODEL=llama-3.3-70b-versatile
 # ANTHROPIC_MODEL=claude-3-haiku-20240307
 # OPENAI_MODEL=gpt-4o-mini
 # GOOGLE_MODEL=gemini-1.5-flash

README.md

@@ -25,7 +25,7 @@ Built for the ServiceHive / Inflx ML Intern assignment: _Social-to-Lead Agentic
 | Lead qualification | Progressive field collection (name → email → platform) |
 | Tool execution guard | `mock_lead_capture` fires **only** after all 3 fields collected |
 | State persistence | Full conversation history via LangGraph `AgentState` across turns |
-| Multi-LLM support | Claude 3 Haiku · GPT-4o-mini · Gemini 1.5 Flash |
+| Multi-LLM support | Groq llama-3.3-70b (default) · Claude 3 Haiku · GPT-4o-mini · Gemini 2.0 Flash |
 
 ---
 
@@ -44,9 +44,10 @@ pip install -r requirements.txt
 ```bash
 cp .env.example .env
 # Edit .env — add ONE of:
-#   ANTHROPIC_API_KEY=sk-ant-...   ← Claude 3 Haiku (default)
+#   GROQ_API_KEY=gsk_...           ← llama-3.3-70b-versatile (default, free + fast)
+#   ANTHROPIC_API_KEY=sk-ant-...   ← Claude 3 Haiku
 #   OPENAI_API_KEY=sk-...          ← GPT-4o-mini
-#   GOOGLE_API_KEY=...             ← Gemini 1.5 Flash
+#   GOOGLE_API_KEY=...             ← Gemini 2.0 Flash
 ```
 
 ### 3. Run
@@ -82,7 +83,7 @@ You:   alex@example.com
 Agent: Almost there! Which creator platform are you primarily on?
 
 You:   YouTube
-[LEAD CAPTURE] Lead captured successfully: Alex Johnson, alex@example.com, YouTube
+Lead captured successfully: Alex Johnson, alex@example.com, YouTube
 
 Agent: You're all set, Alex! Our team will reach out to alex@example.com shortly.
        Start your free trial at autostream.io/signup

agent/nodes.py

@@ -137,12 +137,13 @@ def generate_response(state: AgentState) -> dict:
     intent = state.get("intent", "unknown")
     collecting = state.get("collecting_lead", False)
 
-    # Build KB context block
-    ctx = state.get("rag_context", "")
+    # Build KB context block — only inject when current turn is an inquiry.
+    # Prevents stale context from leaking into greeting / collecting turns.
+    ctx = state.get("rag_context", "") if intent == "inquiry" else ""
     kb_context = (
         f"\nKnowledge Base Context:\n{ctx}\n"
         if ctx
-        else "\n[No specific context retrieved — answer from general knowledge about AutoStream.]\n"
+        else "\n[No KB context for this turn.]\n"
     )
 
     # Build lead collection status block

agent/tools.py

@@ -1,5 +1,3 @@
-def mock_lead_capture(name: str, email: str, platform: str) -> str:
-    """Mock lead capture API. Prints confirmation to stdout."""
-    msg = f"Lead captured successfully: {name}, {email}, {platform}"
-    print(f"\n[LEAD CAPTURE] {msg}")
-    return msg
+def mock_lead_capture(name: str, email: str, platform: str) -> None:
+    """Mock lead capture API — matches assignment spec exactly."""
+    print(f"Lead captured successfully: {name}, {email}, {platform}")

demo.py

@@ -5,7 +5,6 @@ Run: python demo.py
 """
 from __future__ import annotations
 import time
-import os
 from dotenv import load_dotenv
 
 load_dotenv()