File Coverage

blib/lib/OpenAPI/Client/Pinecone.pm

Criterion	Covered	Total	%
statement	24	46	52.1
branch	0	12	0.0
condition	0	3	0.0
subroutine	8	11	72.7
pod	1	1	100.0
total	33	73	45.2

package OpenAPI::Client::Pinecone;

use strict;
use warnings;

use Carp;
use File::ShareDir ':ALL';
use File::Spec::Functions qw(catfile);

use Mojo::Base 'OpenAPI::Client';
use Mojo::URL;

our $VERSION = '0.03';

sub new {
    my ( $class, $specification ) = ( shift, shift );
    my $attrs = @_ == 1 ? shift : {@_};

    if ( !$ENV{PINECONE_API_KEY} ) {
        Carp::croak('PINECONE_API_KEY environment variable must be set');
    }

    if ( !$ENV{PINECONE_API_KEY} && !$attrs->{base_url} ) {
        Carp::croak('PINECONE_API_KEY environment variable must be set');
    }

    if ( !$specification ) {
        eval {
            $specification = dist_file( 'OpenAPI-Client-Pinecone', 'pinecone.yaml' );
            1;
        } or do {
            # Fallback to local share directory during development
            warn $@;
            $specification = catfile( 'share', 'pinecone.yaml' );
        };
    }

    my $self = $class->SUPER::new( $specification, %{$attrs} );

    $self->ua->on(
        start => sub {
            my ( $ua, $tx ) = @_;
            $tx->req->headers->header( 'Api-Key' => $ENV{PINECONE_API_KEY} );
        }
    );

    if ( !$attrs->{base_url} ) {
        $self->base_url( Mojo::URL->new("https://controller.$ENV{PINECONE_ENVIRONMENT}.pinecone.io") );
    }

    return $self;
}

# install snake case aliases

{
    my %snake_case_alias = (
        DescribeIndexStats => 'describe_index_stats',
        Query              => 'query',
        Delete             => 'delete_vector',
        Fetch              => 'fetch_vector',
        Update             => 'update_vector',
        Upsert             => 'upsert_vector',
    );

    for my $camel_case_method ( keys %snake_case_alias ) {
        no strict 'refs';
        *{"$snake_case_alias{$camel_case_method}"} = sub {
            my $self = shift;
            $self->$camel_case_method(@_);
        }
    }
}

1;

__END__