test.pyā¢1 kB
import os
import requests
import json
from dotenv import load_dotenv
# Load environment variables from .env file
load_dotenv()
# Get API key from environment
api_key = os.getenv("FIREWORKS_API_KEY")
if not api_key:
raise ValueError("FIREWORKS_API_KEY not found in environment variables")
url = "https://api.fireworks.ai/inference/v1/chat/completions"
payload = {
"model": "accounts/socialnetwork0/deployedModels/prompt-score-v1-ih2o2qfu",
"max_tokens": 4000,
"top_p": 1,
"top_k": 40,
"presence_penalty": 0,
"frequency_penalty": 0,
"temperature": 0.6,
"messages": [
{
"role": "user",
"content": "Hello, how are you?"
}
]
}
headers = {
"Accept": "application/json",
"Content-Type": "application/json",
"Authorization": f"Bearer {api_key}"
}
response = requests.request("POST", url, headers=headers, data=json.dumps(payload))
print(f"Status Code: {response.status_code}")
print(f"Response: {response.text}")