tools/linkedin_job_search.py

from smolagents.tools import Tool
import requests
from typing import Dict, List

class LinkedInJobSearchTool(Tool):
    name = "linkedin_job_search"
    description = "Searches for jobs on LinkedIn based on job title, location, and work mode (remote, hybrid, in-office)."
    
    inputs = {
        "position": {"type": "string", "description": "Job title (e.g., Data Scientist)"},
        "location": {"type": "string", "description": "City or country (e.g., New York)"},
        "work_mode": {"type": "string", "description": "remote, hybrid, in-office"}
    }
    
    output_type = "dict"  # ✅ Use "dict" instead of "list[dict]"

    def forward(self, position: str, location: str, work_mode: str) -> Dict:
        """
        Searches LinkedIn for job postings using SerpAPI.
        """
        SERPAPI_KEY = "2f660e5a696e7d1d08662085b95f83a61224476ec19558de3c68218baf346e43"  # Replace with your API key
        base_url = "https://serpapi.com/search"

        params = {
            "engine": "google_jobs",
            "q": f"{position} {work_mode} jobs",
            "location": location,
            "hl": "en",
            "api_key": SERPAPI_KEY
        }

        response = requests.get(base_url, params=params)

        if response.status_code == 200:
            data = response.json()
            job_results = data.get("jobs_results", [])

            formatted_jobs = [
                {
                    "title": job["title"],
                    "company": job.get("company_name", "N/A"),
                    "location": job.get("location", "N/A"),
                    "posted_date": job.get("detected_extensions", {}).get("posted_at", "N/A"),
                    "link": job.get("job_id", "N/A")
                }
                for job in job_results
            ]

            return {"jobs": formatted_jobs}  # ✅ Wrap list inside a dictionary

        else:
            return {"error": f"Error {response.status_code}: {response.text}"}