(self, pretrained_model_name_or_path, api_key, temperature=0)
| 8 | class GeminiProEvaluator(Evaluator): |
| 9 | |
| 10 | def __init__(self, pretrained_model_name_or_path, api_key, temperature=0): |
| 11 | super(GeminiProEvaluator, self).__init__() |
| 12 | |
| 13 | gemini.configure( |
| 14 | api_key=api_key, |
| 15 | transport='rest' |
| 16 | ) |
| 17 | self.model = gemini.GenerativeModel(pretrained_model_name_or_path) |
| 18 | self.temperature = temperature |
| 19 | self.candidate_count = 1 |
| 20 | self.max_output_tokens = 2048 |
| 21 | |
| 22 | def format_prompt(self, prompt): |
| 23 | return prompt |
nothing calls this directly
no outgoing calls
no test coverage detected