(args, model: LanguageModel)
| 2 | |
| 3 | |
| 4 | def build_runner(args, model: LanguageModel): |
| 5 | if model.model_style == LMStyle.OpenAIChat: |
| 6 | from lcb_runner.runner.oai_runner import OpenAIRunner |
| 7 | |
| 8 | return OpenAIRunner(args, model) |
| 9 | if model.model_style == LMStyle.OpenAIReason: |
| 10 | from lcb_runner.runner.oai_runner import OpenAIRunner |
| 11 | |
| 12 | return OpenAIRunner(args, model) |
| 13 | if model.model_style == LMStyle.Gemini: |
| 14 | from lcb_runner.runner.gemini_runner import GeminiRunner |
| 15 | |
| 16 | return GeminiRunner(args, model) |
| 17 | if model.model_style == LMStyle.Claude3: |
| 18 | from lcb_runner.runner.claude3_runner import Claude3Runner |
| 19 | |
| 20 | return Claude3Runner(args, model) |
| 21 | if model.model_style == LMStyle.Claude: |
| 22 | from lcb_runner.runner.claude_runner import ClaudeRunner |
| 23 | |
| 24 | return ClaudeRunner(args, model) |
| 25 | if model.model_style == LMStyle.MistralWeb: |
| 26 | from lcb_runner.runner.mistral_runner import MistralRunner |
| 27 | |
| 28 | return MistralRunner(args, model) |
| 29 | if model.model_style == LMStyle.CohereCommand: |
| 30 | from lcb_runner.runner.cohere_runner import CohereRunner |
| 31 | |
| 32 | return CohereRunner(args, model) |
| 33 | if model.model_style == LMStyle.DeepSeekAPI: |
| 34 | from lcb_runner.runner.deepseek_runner import DeepSeekRunner |
| 35 | |
| 36 | return DeepSeekRunner(args, model) |
| 37 | elif model.model_style in []: |
| 38 | raise NotImplementedError( |
| 39 | f"Runner for language model style {model.model_style} not implemented yet" |
| 40 | ) |
| 41 | else: |
| 42 | from lcb_runner.runner.vllm_runner import VLLMRunner |
| 43 | |
| 44 | return VLLMRunner(args, model) |
no test coverage detected