from openai import OpenAI

payi_headers = {
  "xProxy-Api-Key": YOUR_PAYI_KEY
}
payi_base_url = "https://api.pay-i.com" # Will vary for Enterprise deployments
payi_openai_url = payi_base_url + "/api/v1/proxy/openai/v1"

client = OpenAI(
  api_key=YOUR_OPENAI_KEY,
  base_url=payi_openai_url,
  default_headers=payi_headers
)

from openai import AzureOpenAI
import json
import os

#Read the API KEYs from the environment, replace the default values (the second argument) with your own keys if needed
aoai_key = os.getenv("AZURE_OPENAI_API_KEY", "YOUR_AZURE_OPENAI_API_KEY")
payi_api_key = os.getenv("PAYI_API_KEY", "YOUR_PAYI_API_KEY")

aoai_api_version = "2024-02-15-preview"

#Replace with your Azure OpenAI Model Name, e.g. "gpt-4o-2024-05-13"
#Note that the full model name is required by Azure OpenAI
aoai_model = "YOUR_AZURE_OPENAI_MODEL"

#Replace with your Azure OpenAI Deployment Name, e.g. "test-4o"
aoai_deployment = "YOUR_AZURE_OPENAI_DEPLOYMENT"

#Replace with your deployed Azure OpenAI endpoint URI
aoai_instance_uri = "YOUR_AZURE_OPENAI_ENDPOINT"

payi_headers = {
    "xProxy-api-key": payi_api_key,
    "xProxy-Provider-BaseUri": aoai_instance_uri,
    "xProxy-RouteAs-Resource": aoai_model,
}

payi_base_url = "https://api.pay-i.com"
payi_aoai_url = payi_base_url + "/api/v1/proxy/azure.openai"

oai_client = AzureOpenAI(
    api_key=aoai_key,
    api_version=aoai_api_version,
    azure_endpoint=payi_aoai_url,
    default_headers=payi_headers
)

from anthropic import Anthropic

payi_headers = {
  "xProxy-Api-Key": YOUR_PAYI_KEY
}

payi_base_url = "https://api.pay-i.com" # Will vary for Enterprise deployments
payi_anthropic_url = payi_base_url + "/api/v1/proxy/anthropic"

client = Anthropic(
  auth_token = YOUR_ANTHROPIC_KEY,
  base_url = payi_anthropic_url,  
  default_headers=payi_headers
)

import boto3
import json
import os
import urllib3

# Read the API KEYs from the environment, replace the default values (the second argument) with your own keys if needed
payi_api_key = os.getenv("PAYI_API_KEY", "YOUR_PAYI_API_KEY")

payi_base_url = "https://api.pay-i.com"

def HandlePayIParameters(params, context, **kwargs):
    context["extra_headers"] = params.pop("extra_headers", {})

def RedirectToPayi(request, event_name, **kwargs):
    if not event_name.startswith('request-created.bedrock-runtime'):
        return
    
    parsed_url = urllib3.util.parse_url(request.url)
    route_path = parsed_url.path
    request.url = f"{payi_base_url}/api/v1/proxy/aws.bedrock{route_path}"

    request.headers['xProxy-api-key'] = payi_api_key
    request.headers['xProxy-Provider-BaseUri'] = parsed_url.scheme + "://" + parsed_url.host
    extra_headers = request.context.get('extra_headers', {})
    for key, value in extra_headers.items():
        request.headers[key] = value


def RegisterCallbacks(client, model):
    # Process the extra_headers parameter passed to the bedrock runtime call before the AWS client validates the input parameters
    client.meta.events.register(f'provide-client-params.bedrock-runtime.{model}', HandlePayIParameters)
    client.meta.events.register(f'provide-client-params.bedrock-runtime.{model}', HandlePayIParameters) 

    # Redirect the request to the Pay-i endpoint after the request has been signed. Pass a unqiue_id to avoid registering the same callback multiple times.
    client.meta.events.register_last(f'request-created', RedirectToPayi, unique_id='RedirectToPayi')
    
# Substitute the region for your regional deployment
region_name = "us-west-2"

bedrock = boto3.client(
    'bedrock-runtime',
    region_name=region_name,
    )

# Register client callbacks to handle the Pay-i extra_headers parameter in the inference calls and redirect the request to the Pay-i endpoint
RegisterCallbacks(bedrock, 'InvokeModel')
RegisterCallbacks(bedrock, 'InvokeModelWithResponseStream')
RegisterCallbacks(bedrock, 'Converse')
RegisterCallbacks(bedrock, 'ConverseStream')

from payi import Payi

client = Payi(
    api_key=YOUR_PAYI_KEY
    base_url = YOUR_PAYI_ENDPOINT
)