Skip to content

Commit

Permalink
feat:First version of Gemini chat evaluator class.
Browse files Browse the repository at this point in the history
  • Loading branch information
antononcube committed Mar 17, 2024
1 parent 38009b6 commit efd5171
Showing 1 changed file with 21 additions and 0 deletions.
21 changes: 21 additions & 0 deletions lib/LLM/Functions/EvaluatorChatGemini.rakumod
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
use LLM::Functions::EvaluatorChat;

# Should this class inherit from Evaluator or from EvaluatorChat
class LLM::Functions::EvaluatorChatGemini
is LLM::Functions::EvaluatorChat {

submethod TWEAK {
self.assitant-role = 'model';
self.system-role = 'user';
}

method prompt-texts-combiner($prompt, @texts, *%args) {
my @messages = self.LLM::Functions::EvaluatorChat::prompt-texts-combiner($prompt, @texts, |%args);

if @messages.elems2 && @messages[0].key eq @messages[1].key eq self.system-role {
@messages = [@messages.head, Pair.new(self.assitant-role, "OK."), @messages.tail(*-1)];
}

return @messages;
}
}

0 comments on commit efd5171

Please sign in to comment.