update wren_ibis

Canner · Jul 15, 2024 · b2633fc · b2633fc
1 parent 8753022
commit b2633fc
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/wren-ai-service/src/providers/engine/wren.py b/wren-ai-service/src/providers/engine/wren.py
@@ -55,8 +55,10 @@ async def dry_run_sql(
             "source": os.getenv("WREN_IBIS_SOURCE"),
             "manifest": os.getenv("WREN_IBIS_MANIFEST"),
             "connection_info": orjson.loads(
-                os.getenv("WREN_IBIS_CONNECTION_INFO", "{}")
-            ),
+                base64.b64decode(os.getenv("WREN_IBIS_CONNECTION_INFO")).decode()
+            )
+            if os.getenv("WREN_IBIS_CONNECTION_INFO")
+            else {},
         },
     ) -> Tuple[bool, Optional[Dict[str, Any]]]:
         async with session.post(