To add custom headers to Python WebSocket client connections, you can use the requests library to handle the initial handshake and requests with custom header values. First, install the requests dependency if it is not already presant in your environment:

pip install requests

Define all you're custom headers as a dictionary of key-value pairs, then pass the dictionary to the request method via the headers parameter. For example, to add custom User-Agent and Accept-Language headers to your request:

import requests

request_headers = {
 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
 'Accept-Language': 'en-US,en;q=0.9,zh-CN;q=0.8'
}

server_response = requests.get('https://example.com', headers=request_headers)
print(server_response.text)

To add extra headers, simply insert new key-value pairs into the header dictionary. For example, adding an Authorization header to authenticate with an access token:

import requests

request_headers = {
 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
 'Accept-Language': 'en-US,en;q=0.9,zh-CN;q=0.8',
 'Authorization': 'Bearer YOUR_ACCESS_TOKEN'
}

server_response = requests.get('https://example.com', headers=request_headers)
print(server_response.text)

To test responses from a WebSocket server ednpoint, use Python's built-in assert statement to validate expected output:

import requests

request_headers = {
 'User-Agent': 'Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36',
 'Accept-Language': 'en-US,en;q=0.9,zh-CN;q=0.8'
}

server_response = requests.get('https://example.com/websocket-endpoint', headers=request_headers)
assert server_response.text.strip() == "Hello, World!"

When working with large language model APIs that connect through WebSocket-compatible endpoints, you can pass required headers in the same pattern:

import requests

request_headers = {
 'Authorization': 'Bearer YOUR_API_TOKEN',
}

request_payload = {"prompt": "Write a Python snippet to connect to a public WebSocket echo server"}
api_response = requests.post('https://api.example-ai.com/generate-text', headers=request_headers, json=request_payload)
print(api_response.json())