echo "Waiting for inference to be ready..."
timeout=300
while [ $timeout -gt 0 ]; do
  inference_status=$(kubectl get workspace workspace-aikit-test -o jsonpath='{.status.conditions[?(@.type=="InferenceReady")].status}' 2>/dev/null || echo "")
  if [ "$inference_status" = "True" ]; then
    echo "InferenceReady condition is True"
    break
  fi
  echo "Waiting for InferenceReady condition... (status: $inference_status, $timeout seconds remaining)"
  sleep 10
  timeout=$((timeout - 10))
done

if [ $timeout -le 0 ]; then
  echo "Timeout waiting for InferenceReady condition"
  exit 1
fi

kubectl port-forward service/workspace-aikit-test 8080:80 &
PORT_FORWARD_PID=$!

# Wait for port-forward to be ready
sleep 5

# Test that we get expected model information
echo 'Validating model response contains expected data...'
MODELS_RESPONSE=$(curl -s http://localhost:8080/v1/models)
echo "Models response: $MODELS_RESPONSE"

# Extract the first model name from the response
MODEL_NAME=$(echo "$MODELS_RESPONSE" | grep -o '"id":[[:space:]]*"[^"]*"' | head -1 | sed 's/"id":[[:space:]]*"\([^"]*\)"/\1/')
echo "Extracted model name: $MODEL_NAME"

# Check if response contains 'llama'
if echo "$MODELS_RESPONSE" | grep -i 'llama'; then
  echo 'Model response validation passed!'
else
  echo 'Model response validation failed - no llama model found'
  kill $PORT_FORWARD_PID
  exit 1
fi

echo 'Testing completion response...'
COMPLETION_RESPONSE=$(curl -s -X POST \
  -H 'Content-Type: application/json' \
  -d "{
    \"model\": \"$MODEL_NAME\",
    \"messages\": [{\"role\": \"user\", \"content\": \"hello\"}]
  }" \
  http://localhost:8080/v1/chat/completions)
echo "Completion response: $COMPLETION_RESPONSE"

# Check if we get some response with expected fields
if echo "$COMPLETION_RESPONSE" | grep -E '(choices|text|usage)'; then
  echo 'Completion response validation passed!'
else
  echo 'Completion response validation failed - missing expected fields'
  kill $PORT_FORWARD_PID
  exit 1
fi

# Clean up port forwarding
kill $PORT_FORWARD_PID