(episode, msg,)
| 71 | |
| 72 | |
| 73 | def run_episode(episode, msg,): |
| 74 | global _llm,_tokenizer |
| 75 | outputs = _llm.chat(image=None, msgs=msg, system_prompt=SYSTEM_PROMPT, tokenizer=_tokenizer, temperature=0.1,top_p=0.3,n=1,) |
| 76 | episode["pred"] = extract_and_validate_json(outputs) |
| 77 | return episode |
| 78 | |
| 79 | |
| 80 | def extract_and_validate_json(input_string): |
nothing calls this directly
no test coverage detected