import goodmem_client
from goodmem_client.models.reranker_creation_request import RerankerCreationRequest
from goodmem_client.models.reranker_response import RerankerResponse
from goodmem_client.rest import ApiException
from pprint import pprint

# Defining the host is optional and defaults to http://localhost:8080/v1/default
# See configuration.py for a list of all supported configuration parameters.
configuration = goodmem_client.Configuration(
    host = "http://localhost:8080/v1/default"
)

# The client must configure the authentication and authorization parameters
# in accordance with the API server security policy.
# Examples for each auth method are provided below, use the example that
# satisfies your auth use case.

# Configure API key authorization: ApiKeyAuth
configuration.api_key['ApiKeyAuth'] = os.environ["API_KEY"]

# Uncomment below to setup prefix (e.g. Bearer) for API key, if needed
# configuration.api_key_prefix['ApiKeyAuth'] = 'Bearer'

# Enter a context with an instance of the API client
with goodmem_client.ApiClient(configuration) as api_client:
    # Create an instance of the API class
    api_instance = goodmem_client.RerankersApi(api_client)
    reranker_creation_request = {"displayName":"BGE Cross-Encoder Reranker","description":"BAAI BGE reranker for cross-encoder semantic ranking","providerType":"TEI","endpointUrl":"http://bge.reranker.goodmem.ai:8010","apiPath":"/rerank","modelIdentifier":"BAAI/bge-reranker-base","supportedModalities":["TEXT"],"credentials":{"kind":"CREDENTIAL_KIND_API_KEY","apiKey":{"inlineSecret":"sk-demo-api-key"}},"labels":{"environment":"production","team":"search"}} # RerankerCreationRequest | Reranker configuration details

    try:
        # Create a new reranker
        api_response = api_instance.create_reranker(reranker_creation_request)
        print("The response of RerankersApi->create_reranker:\n")
        pprint(api_response)
    except Exception as e:
        print("Exception when calling RerankersApi->create_reranker: %s\n" % e)

import goodmem_client
from goodmem_client.rest import ApiException
from pprint import pprint

# Defining the host is optional and defaults to http://localhost:8080/v1/default
# See configuration.py for a list of all supported configuration parameters.
configuration = goodmem_client.Configuration(
    host = "http://localhost:8080/v1/default"
)

# The client must configure the authentication and authorization parameters
# in accordance with the API server security policy.
# Examples for each auth method are provided below, use the example that
# satisfies your auth use case.

# Configure API key authorization: ApiKeyAuth
configuration.api_key['ApiKeyAuth'] = os.environ["API_KEY"]

# Uncomment below to setup prefix (e.g. Bearer) for API key, if needed
# configuration.api_key_prefix['ApiKeyAuth'] = 'Bearer'

# Enter a context with an instance of the API client
with goodmem_client.ApiClient(configuration) as api_client:
    # Create an instance of the API class
    api_instance = goodmem_client.RerankersApi(api_client)
    id = '550e8400-e29b-41d4-a716-446655440000' # str | The unique identifier of the reranker to delete

    try:
        # Delete a reranker
        api_instance.delete_reranker(id)
    except Exception as e:
        print("Exception when calling RerankersApi->delete_reranker: %s\n" % e)

import goodmem_client
from goodmem_client.models.reranker_response import RerankerResponse
from goodmem_client.rest import ApiException
from pprint import pprint

# Defining the host is optional and defaults to http://localhost:8080/v1/default
# See configuration.py for a list of all supported configuration parameters.
configuration = goodmem_client.Configuration(
    host = "http://localhost:8080/v1/default"
)

# The client must configure the authentication and authorization parameters
# in accordance with the API server security policy.
# Examples for each auth method are provided below, use the example that
# satisfies your auth use case.

# Configure API key authorization: ApiKeyAuth
configuration.api_key['ApiKeyAuth'] = os.environ["API_KEY"]

# Uncomment below to setup prefix (e.g. Bearer) for API key, if needed
# configuration.api_key_prefix['ApiKeyAuth'] = 'Bearer'

# Enter a context with an instance of the API client
with goodmem_client.ApiClient(configuration) as api_client:
    # Create an instance of the API class
    api_instance = goodmem_client.RerankersApi(api_client)
    id = '550e8400-e29b-41d4-a716-446655440000' # str | The unique identifier of the reranker to retrieve

    try:
        # Get a reranker by ID
        api_response = api_instance.get_reranker(id)
        print("The response of RerankersApi->get_reranker:\n")
        pprint(api_response)
    except Exception as e:
        print("Exception when calling RerankersApi->get_reranker: %s\n" % e)

import goodmem_client
from goodmem_client.models.list_rerankers_response import ListRerankersResponse
from goodmem_client.rest import ApiException
from pprint import pprint

# Defining the host is optional and defaults to http://localhost:8080/v1/default
# See configuration.py for a list of all supported configuration parameters.
configuration = goodmem_client.Configuration(
    host = "http://localhost:8080/v1/default"
)

# The client must configure the authentication and authorization parameters
# in accordance with the API server security policy.
# Examples for each auth method are provided below, use the example that
# satisfies your auth use case.

# Configure API key authorization: ApiKeyAuth
configuration.api_key['ApiKeyAuth'] = os.environ["API_KEY"]

# Uncomment below to setup prefix (e.g. Bearer) for API key, if needed
# configuration.api_key_prefix['ApiKeyAuth'] = 'Bearer'

# Enter a context with an instance of the API client
with goodmem_client.ApiClient(configuration) as api_client:
    # Create an instance of the API class
    api_instance = goodmem_client.RerankersApi(api_client)
    owner_id = '550e8400-e29b-41d4-a716-446655440000' # str | Filter rerankers by owner ID. With LIST_RERANKER_ANY permission, omitting this shows all accessible rerankers; providing it filters by that owner. With LIST_RERANKER_OWN permission, only your own rerankers are shown regardless of this parameter (PERMISSION_DENIED if set to another user). (optional)
    provider_type = 'TEI' # str | Filter rerankers by provider type (e.g., OPENAI, TEI, VLLM, etc.) (optional)
    label_ = '?label.environment=production&label.team=search' # str | Filter by label value. Multiple label filters can be specified (e.g., ?label.environment=production&label.team=search) (optional)

    try:
        # List rerankers
        api_response = api_instance.list_rerankers(owner_id=owner_id, provider_type=provider_type, label_=label_)
        print("The response of RerankersApi->list_rerankers:\n")
        pprint(api_response)
    except Exception as e:
        print("Exception when calling RerankersApi->list_rerankers: %s\n" % e)

import goodmem_client
from goodmem_client.models.reranker_response import RerankerResponse
from goodmem_client.models.update_reranker_request import UpdateRerankerRequest
from goodmem_client.rest import ApiException
from pprint import pprint

# Defining the host is optional and defaults to http://localhost:8080/v1/default
# See configuration.py for a list of all supported configuration parameters.
configuration = goodmem_client.Configuration(
    host = "http://localhost:8080/v1/default"
)

# The client must configure the authentication and authorization parameters
# in accordance with the API server security policy.
# Examples for each auth method are provided below, use the example that
# satisfies your auth use case.

# Configure API key authorization: ApiKeyAuth
configuration.api_key['ApiKeyAuth'] = os.environ["API_KEY"]

# Uncomment below to setup prefix (e.g. Bearer) for API key, if needed
# configuration.api_key_prefix['ApiKeyAuth'] = 'Bearer'

# Enter a context with an instance of the API client
with goodmem_client.ApiClient(configuration) as api_client:
    # Create an instance of the API class
    api_instance = goodmem_client.RerankersApi(api_client)
    id = '550e8400-e29b-41d4-a716-446655440000' # str | The unique identifier of the reranker to update
    update_reranker_request = {"displayName":"Updated Production BGE Reranker","description":"Updated BAAI BGE cross-encoder reranker with enhanced configuration for production use","endpointUrl":"http://bge.reranker.goodmem.ai:8010","apiPath":"/rerank","modelIdentifier":"BAAI/bge-reranker-large","supportedModalities":["TEXT"],"credentials":{"kind":"CREDENTIAL_KIND_API_KEY","apiKey":{"inlineSecret":"updated-api-key-here"}},"version":"2.1.0","monitoringEndpoint":"https://monitoring.company.com/rerankers/status","replaceLabels":{"environment":"production","team":"search-platform","cost-center":"ml-infrastructure"}} # UpdateRerankerRequest | Updated reranker configuration details

    try:
        # Update a reranker
        api_response = api_instance.update_reranker(id, update_reranker_request)
        print("The response of RerankersApi->update_reranker:\n")
        pprint(api_response)
    except Exception as e:
        print("Exception when calling RerankersApi->update_reranker: %s\n" % e)