File Coverage

blib/lib/WebService/Jina.pm

Criterion	Covered	Total	%
statement	14	66	21.2
branch	0	28	0.0
condition	0	24	0.0
subroutine	5	11	45.4
pod	6	6	100.0
total	25	135	18.5

package WebService::Jina;

use 5.006;
use strict;
use warnings;

our $VERSION = '0.05';

use WebService::Jina::UA;
use Moo;

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

has ua => (
	is => 'ro',
	default => sub {
		return WebService::Jina::UA->new(
			api_key => $_[0]->api_key
		);
	}
);

sub deepsearch {
	my ($self, %args) = @_;

	$args{model} ||= 'jina-deepsearch-v1';

	if (! $args{messages}) {
		die 'No messages defined for deepsearch';
	}

	$args{stream} = $args{stream} ? \1 : \0;

	$args{reasoning_effort} ||= "medium";

	$self->ua->post(
		url => 'https://deepsearch.jina.ai/v1/chat/completions',
		headers => delete $args{headers},
		data => \%args
	);
}

sub reader {
	my ($self, %args) = @_;

	if (! $args{url}) {
		die "no url defined for reader";
	}

	$self->ua->post(
		url => "https://r.jina.ai/",
		headers => delete $args{headers},
		data => \%args
	);
}

sub embedding {
	my ($self, %args) = @_;

	$args{model} ||= 'jina-clip-v2';

	$args{dimensions} ||= 1024;

	$args{normalized} //= 1;
	$args{normalized} = $args{normalized} ? \0 : \1;

	$args{embedding_type} ||= 'float';

	if (! $args{input}) {
		die 'no input defined for embedding';
	}

	for my $input (@{$args{input}}) {
                if ($input->{image_file}) {
                        $input->{image} = $self->ua->base64_images([delete $input->{image_file}])->[0];
                }
        }

	$self->ua->post(
		url => 'https://api.jina.ai/v1/embeddings',
		headers => delete $args{headers},
		data => \%args,
	);
}

sub rerank {
	my ($self, %args) = @_;

	$args{model} ||= 'jina-reranker-v2-base-multilingual';

	$args{top_n} ||= 3;

	if (! $args{query}) {
		die 'no query defined for rerank';
	}

	if (! $args{documents}) {
		die 'no documents defiend for rerank';
	}

	$self->ua->post(
		url => 'https://api.jina.ai/v1/rerank',
		headers => delete $args{headers},
		data => \%args
	);

}

sub classify {
	my ($self, %args) = @_;

	$args{model} ||= 'jina-clip-v2';

	if (! $args{input}) {
		die 'no input defined for classify';
	}

	for my $input (@{$args{input}}) {
                if ($input->{image_file}) {
                        $input->{image} = $self->ua->base64_images([delete $input->{image_file}])->[0];
                }
        }

	if (! $args{labels}) {
		die 'no labels defined for classify';
	}

	$self->ua->post(
		url => 'https://api.jina.ai/v1/classify',
		headers => delete $args{headers},
		data => \%args
	);
}

sub segment {
	my ($self, %args) = @_;

	$args{return_tokens} //= 1;
	$args{return_tokens} = $args{return_tokens} ? \1 : \0;

	$args{return_chunks} //= 1;
	$args{return_chunks} = $args{return_chunks} ? \1 : \0;

	$args{max_chunk_length} ||= 1000;

	if (! $args{content}) {
		die 'no content defined for segment';
	}

	$self->ua->post(
		url => 'https://api.jina.ai/v1/segment',
		headers => delete $args{headers},
		data => \%args
	);
}


1;

__END__