From 121a3dd44b60f9fb5afe84b4d23dc3b78a405680 Mon Sep 17 00:00:00 2001 From: Tim Pietrusky Date: Wed, 13 Aug 2025 12:20:44 +0200 Subject: [PATCH] fix: parse value for RAW_OPENAI_OUTPUT correctly --- src/engine.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/src/engine.py b/src/engine.py index 073f0d5..99d9132 100644 --- a/src/engine.py +++ b/src/engine.py @@ -178,7 +178,12 @@ class OpenAIvLLMEngine(vLLMEngine): self.response_role = os.getenv("OPENAI_RESPONSE_ROLE") or "assistant" self.lora_adapters = self._load_lora_adapters() asyncio.run(self._initialize_engines()) - self.raw_openai_output = bool(int(os.getenv("RAW_OPENAI_OUTPUT", 1))) + # Handle both integer and boolean string values for RAW_OPENAI_OUTPUT + raw_output_env = os.getenv("RAW_OPENAI_OUTPUT", "1") + if raw_output_env.lower() in ('true', 'false'): + self.raw_openai_output = raw_output_env.lower() == 'true' + else: + self.raw_openai_output = bool(int(raw_output_env)) def _load_lora_adapters(self): adapters = []