mirror of
https://github.com/lightbroker/llmsecops-research.git
synced 2026-06-01 12:41:54 +02:00
Flask API
This commit is contained in:
Executable
+39
@@ -0,0 +1,39 @@
|
||||
#!/bin/bash
|
||||
|
||||
# Start Flask server in the background
|
||||
python -m src.api.controller &
|
||||
SERVER_PID=$!
|
||||
|
||||
# Function to check if server is up
|
||||
wait_for_server() {
|
||||
echo "Waiting for Flask server to start..."
|
||||
local max_attempts=30
|
||||
local attempt=0
|
||||
|
||||
while [ $attempt -lt $max_attempts ]; do
|
||||
if curl -s http://localhost:9998/ > /dev/null 2>&1; then
|
||||
echo "Server is up!"
|
||||
return 0
|
||||
fi
|
||||
|
||||
attempt=$((attempt + 1))
|
||||
echo "Attempt $attempt/$max_attempts - Server not ready yet, waiting..."
|
||||
sleep 1
|
||||
done
|
||||
|
||||
echo "Server failed to start after $max_attempts attempts"
|
||||
kill $SERVER_PID
|
||||
return 1
|
||||
}
|
||||
|
||||
# Wait for server to be ready
|
||||
wait_for_server || exit 1
|
||||
|
||||
# Make the actual request once server is ready
|
||||
echo "Making API request..."
|
||||
curl -X POST -i localhost:9998/api/conversations \
|
||||
-d '{ "prompt": "describe a random planet in our solar system in 10 words or less" }' \
|
||||
-H "Content-Type: application/json" || exit 1
|
||||
echo
|
||||
|
||||
exit 0
|
||||
Reference in New Issue
Block a user