From 93a52bf4bca548d99412aea11ea8b819c013a251 Mon Sep 17 00:00:00 2001
From: leavesster <11785335+leavesster@users.noreply.github.com>
Date: Tue, 17 Mar 2026 10:45:07 +0800
Subject: [PATCH] feat(oocana): add fusion_url and oomol_env to OOMOL_LLM_ENV
 TypedDict

---
 oocana/oocana/context.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/oocana/oocana/context.py b/oocana/oocana/context.py
index ff07bb0..9978bd5 100644
--- a/oocana/oocana/context.py
+++ b/oocana/oocana/context.py
@@ -170,8 +170,10 @@ class OOMOL_LLM_ENV(TypedDict):
     """{basUrl}/v1 openai compatible endpoint
     """
     base_url_v1: str
+    fusion_url: str
     api_key: str
     models: list[str]
+    oomol_env: str
 
 class HostInfo(TypedDict):
     gpu_vendor: str
@@ -315,6 +317,8 @@ def oomol_llm_env(self) -> OOMOL_LLM_ENV:
             "base_url_v1": os.getenv("OOMOL_LLM_BASE_URL_V1", ""),
             "api_key": os.getenv("OOMOL_LLM_API_KEY", ""),
             "models": os.getenv("OOMOL_LLM_MODELS", "").split(","),
+            "fusion_url": os.getenv("OOMOL_LLM_FUSION_URL", ""),
+            "oomol_env": os.getenv("OOMOL_ENV", "prod"),
         }
 
         for key, value in oomol_llm_env.items():