File Coverage

blib/lib/WebService/Ollama/Response.pm

Criterion	Covered	Total	%
statement	22	26	84.6
branch	6	10	60.0
condition	6	8	75.0
subroutine	4	5	80.0
pod	3	3	100.0
total	41	52	78.8

package WebService::Ollama::Response;

use Moo;

use JSON::Lines;

has [qw/
	done
	context
	total_duration
	load_duration
	model
	create_at
	eval_count
	eval_duration
	done_reason
	response
	prompt_eval_duration
	prompt_eval_count
	message
	status
	digest
	total
	completed
	version
	embeddings
	models
/] => (
	is => 'ro',
);

sub has_tool_calls {
	my ($self) = @_;
	my $calls = $self->extract_tool_calls;
	return scalar @$calls > 0;
}

sub extract_tool_calls {
	my ($self) = @_;
	my @calls;

	# Ollama returns tool_calls in the message
	if ($self->message && ref($self->message) eq 'HASH') {
		if (my $tc = $self->message->{tool_calls}) {
			push @calls, @$tc;
		}

		# Fallback: some models output tool calls as JSON in content
		if (!@calls && $self->message->{content}) {
			my $content = $self->message->{content};

			# Look for JSON tool call patterns - handle both "arguments" and "parameters"
			while ($content =~ /\{\s*"name"\s*:\s*"([^"]+)"\s*,\s*"(?:arguments|parameters)"\s*:\s*(\{[^}]*\})\s*\}/g) {
				my ($name, $args_json) = ($1, $2);
				my $args = eval {
					JSON::Lines->new->decode($args_json)->[0];
				} // {};
				push @calls, {
					function => {
						name      => $name,
						arguments => $args,
					},
				};
			}
		}
	}

	return \@calls;
}

sub json_response {
	my ($self) = shift;
	my $aoa = JSON::Lines->new->decode($self->response ? $self->response : $self->message->{content});
	return scalar @{$aoa} == 1 ? $aoa->[0] : $aoa;
}

1;

__END__