feat: Add salary negotiation AI coach endpoint

backend/app/api/routes/negotiate.py

@@ -0,0 +1,110 @@
+"""
+Salary Negotiation Coach - AI-powered negotiation strategy and scripts.
+"""
+import json
+import logging
+
+import litellm
+from fastapi import APIRouter
+from pydantic import BaseModel
+
+from app.api.deps import CurrentUser, DBSession
+from app.core.config import settings
+
+logger = logging.getLogger(__name__)
+
+router = APIRouter(prefix="/ai/negotiate", tags=["salary-negotiation"])
+
+NEGOTIATION_PROMPT = """You are an expert salary negotiation coach. Given the context about an offer and the candidate's situation, provide a comprehensive negotiation strategy.
+
+Return JSON:
+{
+  "assessment": "Is this offer competitive? Brief analysis.",
+  "should_negotiate": true/false,
+  "target_range": {"min": number, "max": number, "stretch": number},
+  "strategy": {
+    "approach": "collaborative|assertive|data-driven",
+    "timing": "When to negotiate",
+    "leverage_points": ["What gives you negotiation power"],
+    "risks": ["Potential downsides to be aware of"]
+  },
+  "scripts": {
+    "initial_response": "What to say when receiving the offer",
+    "counter_offer": "How to present your counter",
+    "if_they_push_back": "Response to resistance",
+    "closing": "How to close the deal"
+  },
+  "non_salary_items": [
+    {"item": "Signing bonus", "how_to_ask": "...", "typical_value": "..."},
+    {"item": "Equity/RSUs", "how_to_ask": "...", "typical_value": "..."},
+    {"item": "Remote flexibility", "how_to_ask": "...", "typical_value": "..."}
+  ],
+  "mistakes_to_avoid": ["..."],
+  "confidence_rating": "How strong your position is (1-10)"
+}"""
+
+
+class NegotiationRequest(BaseModel):
+    offer_salary: int
+    offer_equity: str | None = None
+    offer_bonus: str | None = None
+    role_title: str
+    company_name: str
+    your_current_salary: int | None = None
+    years_experience: int | None = None
+    competing_offers: bool = False
+    market_rate_min: int | None = None
+    market_rate_max: int | None = None
+    additional_context: str | None = None
+
+
+@router.post("")
+async def get_negotiation_strategy(data: NegotiationRequest, db: DBSession, user: CurrentUser):
+    """Get AI-powered salary negotiation strategy and scripts."""
+    context = f"""
+Offer Details:
+- Role: {data.role_title} at {data.company_name}
+- Base Salary: ${data.offer_salary:,}
+- Equity: {data.offer_equity or 'Not mentioned'}
+- Bonus: {data.offer_bonus or 'Not mentioned'}
+
+Candidate Context:
+- Current Salary: ${data.your_current_salary:,} if {data.your_current_salary} else 'Not disclosed'
+- Years Experience: {data.years_experience or 'Not specified'}
+- Competing Offers: {'Yes' if data.competing_offers else 'No'}
+- Market Rate: ${data.market_rate_min:,} - ${data.market_rate_max:,} if {data.market_rate_min and data.market_rate_max} else 'Unknown'
+
+Additional Context: {data.additional_context or 'None'}
+
+Provide a negotiation strategy."""
+
+    model = "ollama/llama3.2"
+    if settings.OPENAI_API_KEY:
+        model = "openai/gpt-4o"
+    elif settings.ANTHROPIC_API_KEY:
+        model = "anthropic/claude-sonnet-4-20250514"
+    elif settings.DEEPSEEK_API_KEY:
+        model = "deepseek/deepseek-chat"
+
+    try:
+        response = await litellm.acompletion(
+            model=model,
+            messages=[
+                {"role": "system", "content": NEGOTIATION_PROMPT},
+                {"role": "user", "content": context},
+            ],
+            response_format={"type": "json_object"},
+            temperature=0.4,
+            max_tokens=2500,
+        )
+        result = json.loads(response.choices[0].message.content)
+        result["offer_analyzed"] = {
+            "salary": data.offer_salary,
+            "company": data.company_name,
+            "role": data.role_title,
+        }
+        return result
+
+    except Exception as e:
+        logger.error(f"Negotiation coach failed: {e}")
+        return {"error": str(e)}