# llm_api.R

library(httr)
library(jsonlite)

# Function to send a chat completion request to the Groq API
send_chat_completion <- function(api_key, model, system_message, messages, max_tokens = NULL, temperature = 1, top_p = 1) {
  url <- "https://api.groq.com/openai/v1/chat/completions"
  
  # Set up the headers
  headers <- add_headers(
    `Content-Type` = "application/json",
    `Authorization` = paste("Bearer", api_key)
  )
  
  # Create the request body
  body <- list(
    model = model,
    messages = append(list(list(role = "system", content = system_message)), messages),
    temperature = temperature,
    top_p = top_p
  )
  
  if (!is.null(max_tokens)) {
    body$max_tokens <- max_tokens
  }
  
  # Convert body to JSON
  body_json <- toJSON(body, auto_unbox = TRUE, pretty = TRUE)
  
  # Send the POST request
  response <- POST(url, headers, body = body_json)
  
  # Get response content as text
  response_content <- content(response, as = "text", encoding = "UTF-8")
  
  # Check for HTTP errors
  if (http_error(response)) {
    error_message <- paste("API request failed [", status_code(response), "]: ", response_content)
    stop(error_message)
  }
  
  # Parse the response JSON without flattening
  response_json <- fromJSON(response_content, simplifyDataFrame = FALSE)
  
  # Return a list containing the response JSON, request body, and response content
  return(list(
    response_json = response_json,
    request_body = body_json,
    response_text = response_content
  ))
}