File Coverage

blib/lib/WebService/Chroma.pm

Criterion	Covered	Total	%
statement	23	34	67.6
branch			n/a
condition			n/a
subroutine	8	14	57.1
pod	5	5	100.0
total	36	53	67.9

package WebService::Chroma;

our $VERSION = '0.08';

use 5.006;
use strict;
use warnings;

use Moo;
use WebService::Chroma::UA;
use WebService::Chroma::Tenant;

use MooX::ValidateSubs;
use Types::Standard qw/Str ArrayRef HashRef/;

validate_subs (
	create_tenant => {
		params => {
			name => [Str]
		}
	},
	get_tenant => {
		params => {
			name => [Str]
		}
	},
	get_database => {
		params => {
			tenant => [Str],
			name => [Str]
		}
	},
	get_collection => {
		params => {
			tenant => [Str],
			db => [Str],
			name => [Str]
		}
	}
);

has base_url => (
	is => 'ro',
	lazy => 1,
	default => sub { 'http://localhost:8000' }
);

has embeddings_class => (
	is => 'ro',
	lazy => 1,
);

has embeddings_api_key => (
	is => 'ro',
	lazy => 1,
);

has embeddings_base_url => (
	is => 'ro',
	lazy => 1,
);

has embeddings_model => (
	is => 'ro',
	lazy => 1,
);

has ua => (
	is => 'ro',
	builder => sub {
		WebService::Chroma::UA->new(
			base_url => $_[0]->base_url,
			embeddings_class => $_[0]->embeddings_class,
			embeddings_model => $_[0]->embeddings_model,
			embeddings_base_url => $_[0]->embeddings_base_url,
			embeddings_api_key => $_[0]->embeddings_api_key
		);
	}
);

sub version {
	my ($self) = @_;
	return $self->ua->get(url => '/api/v2/version');
}

sub reset {
	my ($self) = @_;
	return $self->ua->post(url => '/api/v2/reset');
}

sub heartbeat {
	my ($self) = @_;
	return $self->ua->get(url => '/api/v2/heartbeat');
}

sub pre_flight_checks {
	my ($self) = @_;
	return $self->ua->get(url => '/api/v2/pre-flight-checks');
}

sub auth_identity {
	my ($self) = @_;
	return $self->ua->get(url => '/api/v2/auth/identity');
}

sub create_tenant {
	my ($self, %data) = @_;

	my $tenant = $self->ua->post(
		url => '/api/v2/tenants',
		data => \%data
	);

	return $self->get_tenant(%data);
}

sub get_tenant {
	my ($self, %data) = @_;

	my $tenant = $self->ua->get(url => '/api/v2/tenants/' . $data{name});

	return WebService::Chroma::Tenant->new(
		ua => $self->ua,
		%{$tenant}
	);
}

sub get_database {
	my ($self, %data) = @_;
	return $self->get_tenant(name => $data{tenant})->get_database(name => $data{name});
}

sub get_collection {
	my ($self, %data) = @_;
	return $self->get_tenant(name => $data{tenant})->get_database(name => $data{db})->get_collection(name => $data{name});
}

1;

__END__