| 229 | |
| 230 | |
| 231 | def construct_llm_config(args: argparse.Namespace) -> lm_config.LMConfig: |
| 232 | llm_config = lm_config.LMConfig( |
| 233 | provider=args.provider, model=args.model, mode=args.mode |
| 234 | ) |
| 235 | if args.provider in {'openai', 'llama'}: |
| 236 | llm_config.gen_config["temperature"] = args.temperature |
| 237 | llm_config.gen_config["top_p"] = args.top_p |
| 238 | llm_config.gen_config["context_length"] = args.context_length |
| 239 | llm_config.gen_config["max_tokens"] = args.max_tokens |
| 240 | llm_config.gen_config["stop_token"] = args.stop_token |
| 241 | llm_config.gen_config["max_obs_length"] = args.max_obs_length |
| 242 | else: |
| 243 | raise NotImplementedError(f"provider {args.provider} not implemented") |
| 244 | return llm_config |
| 245 | |
| 246 | |
| 247 | def construct_agent(args: argparse.Namespace) -> Agent: |