move model init

mahaloz · Feb 5, 2025 · 216e193 · 216e193
1 parent 79a1d44
commit 216e193
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dailalib/api/litellm/litellm_api.py b/dailalib/api/litellm/litellm_api.py
@@ -39,13 +39,13 @@ def __init__(
         self._use_config = use_config
         # default values
         self._api_key = None
+        self.model = model
         # default to openai api key if not provided
         if api_key or os.getenv("OPENAI_API_KEY"):
             self.api_key = api_key or os.getenv("OPENAI_API_KEY")
         elif not self._use_config:
             self.api_key = None
 
-        self.model = model
         self.prompt_style = prompt_style
         self.fit_to_tokens = fit_to_tokens
         self.chat_use_ctx = chat_use_ctx