add response check

okdshin · Nov 28, 2023 · 9cd8697 · 9cd8697
1 parent 5cb8f00
commit 9cd8697
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/flatline_lsp.py b/flatline_lsp.py
@@ -75,6 +75,7 @@ def forward(  # type: ignore
             f"http://{self.baseckend_server_host}:{self.baseckend_server_port}/v1/calc_next_token_logits",
             json=dict(input_tokens=input_ids[0].tolist()),
         )
+        res.raise_for_status()
         return CausalLMOutput(
             loss=None,
             logits=torch.FloatTensor(res.json()["next_token_logits"]).reshape(