package dev.simonscholz

data class Freelancer(
    val name: String,
    val skills: List<String>,
)

package dev.simonscholz

import jakarta.enterprise.context.ApplicationScoped

@ApplicationScoped
class FreelancerService {
    fun findFreelancersBySkills(skills: List<String>): List<Freelancer> =
        listOf(
            Freelancer("Simon Scholz", listOf("Kotlin", "Java", "Quarkus", "DDD").map { it.lowercase() }),
            Freelancer("John Go", listOf("Go", "Docker", "Kubernetes", "GCP").map { it.lowercase() }),
        ).filter { freelancer ->
            skills.any { skill -> freelancer.skills.contains(skill.lowercase()) }
        }
}

package dev.simonscholz

import io.quarkiverse.mcp.server.McpLog
import io.quarkiverse.mcp.server.Tool
import io.quarkiverse.mcp.server.ToolArg
import io.smallrye.common.annotation.RunOnVirtualThread
import jakarta.inject.Singleton

@Singleton
class FindFreelancerTool(
    private val freelancerService: FreelancerService,
) {
    @Tool(name = "freelancer", description = "Finds freelancers by their skills.")
    @RunOnVirtualThread
    fun findFreelancersBySkills(
        @ToolArg(description = "skills") skills: List<String>,
        log: McpLog,
    ): List<Freelancer> {
        log.info("Finding freelancers with skills: $skills")
        val freelancers = freelancerService.findFreelancersBySkills(skills)
        log.info("Found ${freelancers.size} freelancers with skills: ${freelancers.joinToString(",") { it.skills.joinToString(",") }}")
        return freelancers
    }
}

package dev.simonscholz

import dev.langchain4j.service.SystemMessage
import dev.langchain4j.service.UserMessage
import io.quarkiverse.langchain4j.RegisterAiService
import io.quarkiverse.langchain4j.mcp.runtime.McpToolBox

@RegisterAiService
interface FindFreelancersBySkillsAssistant {
    @SystemMessage(
        """
        You are an assistant that helps to find freelancers based on a list of skills.
        
        When a user asks to find a freelancer with a list of skills follow this one-step process:
        
        1. Use the 'freelancer' tool to get a list of freelancers that match those skills.
        
        Provide the user with the names of the freelancers found.
        
        Always use the 'freelancer' tool - never try to find freelancers yourself.
        Be helpful and provide a list of freelancers based on the result of the tool call.
    """,
    )
    @McpToolBox("freelancer")
    fun chat(
        @UserMessage message: String,
    ): String
}

package dev.simonscholz

import io.quarkus.logging.Log
import jakarta.ws.rs.GET
import jakarta.ws.rs.Path
import jakarta.ws.rs.Produces
import jakarta.ws.rs.QueryParam
import jakarta.ws.rs.core.MediaType

class FindFreelancersResource(
    private val freelancersBySkillsAssistant: FindFreelancersBySkillsAssistant,
) {
    @GET
    @Path("/ask")
    @Produces(MediaType.TEXT_PLAIN)
    fun askQuestion(
        @QueryParam("q") question: String?,
    ): String {
        val query = question ?: "Find me freelancers with skills in Kotlin and Quarkus."
        Log.info("Question: $query")
        val response = freelancersBySkillsAssistant.chat(query)
        Log.info("Response: $response")
        return response
    }
}

package dev.simonscholz

import dev.langchain4j.service.SystemMessage
import dev.langchain4j.service.UserMessage
import io.quarkiverse.langchain4j.RegisterAiService
import io.quarkiverse.langchain4j.mcp.runtime.McpToolBox
import org.eclipse.microprofile.faulttolerance.Fallback

@RegisterAiService
interface FindFreelancersBySkillsAssistant {
    @SystemMessage(
        """
        You are an assistant that helps to find freelancers based on a list of skills.
        
        When a user asks to find a freelancer with a list of skills follow this one-step process:
        
        1. Use the 'freelancer' tool to get a list of freelancers that match those skills.
        
        Provide the user with the names of the freelancers found.
        
        Always use the 'freelancer' tool - never try to find freelancers yourself.
        Be helpful and provide a list of freelancers based on the result of the tool call.
    """,
    )
    @McpToolBox("freelancer")
    @Fallback(fallbackMethod = "fallbackChat")
    fun chat(
        @UserMessage message: String,
    ): String

    fun fallbackChat(message: String): String = "I'm sorry, but I'm currently unable to find freelancers for you. Please try again later."
}