Update config.py

config.py

@@ -1,16 +1,21 @@
 import os
 
 
+def to_bool(value: str, default: bool = False) -> bool:
+    if value is None:
+        return default
+    return str(value).strip().lower() in {"true", "1", "yes", "on"}
+
+
 class Settings:
     APP_NAME = "Chat7 CodeX Backend"
     APP_VERSION = "1.0.0"
 
-    PRIMARY_CODE_MODEL = os.getenv("PRIMARY_CODE_MODEL", "Qwen/Qwen2.5-Coder-7B-Instruct")
-    FALLBACK_CODE_MODEL = os.getenv("FALLBACK_CODE_MODEL", "deepseek-ai/deepseek-coder-6.7b-instruct")
+    PRIMARY_CODE_MODEL = os.getenv("PRIMARY_CODE_MODEL", "Qwen/Qwen2.5-7B-Instruct")
+    FALLBACK_CODE_MODEL = os.getenv("FALLBACK_CODE_MODEL", "Qwen/Qwen2.5-3B-Instruct")
 
     GITHUB_TOKEN = os.getenv("GITHUB_TOKEN", "")
     STACKOVERFLOW_KEY = os.getenv("STACKOVERFLOW_KEY", "")
-
     HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN", "")
 
     MODEL_TIMEOUT_SECONDS = int(os.getenv("MODEL_TIMEOUT_SECONDS", "90"))
@@ -24,16 +29,18 @@ class Settings:
     MAX_GITHUB_RESULTS = int(os.getenv("MAX_GITHUB_RESULTS", "5"))
     MAX_RETRIEVED_ITEMS = int(os.getenv("MAX_RETRIEVED_ITEMS", "8"))
 
-    ENABLE_STACK_SEARCH = os.getenv("ENABLE_STACK_SEARCH", "true").lower() == "true"
-    ENABLE_GITHUB_SEARCH = os.getenv("ENABLE_GITHUB_SEARCH", "true").lower() == "true"
-    ENABLE_RETRIEVAL_FOR_FIX = os.getenv("ENABLE_RETRIEVAL_FOR_FIX", "true").lower() == "true"
-    ENABLE_RETRIEVAL_FOR_DEBUG = os.getenv("ENABLE_RETRIEVAL_FOR_DEBUG", "true").lower() == "true"
-    ENABLE_RETRIEVAL_FOR_GENERATE = os.getenv("ENABLE_RETRIEVAL_FOR_GENERATE", "false").lower() == "true"
-    ENABLE_RETRIEVAL_FOR_EXPLAIN = os.getenv("ENABLE_RETRIEVAL_FOR_EXPLAIN", "false").lower() == "true"
+    ENABLE_STACK_SEARCH = to_bool(os.getenv("ENABLE_STACK_SEARCH", "true"), True)
+    ENABLE_GITHUB_SEARCH = to_bool(os.getenv("ENABLE_GITHUB_SEARCH", "true"), True)
+    ENABLE_RETRIEVAL_FOR_FIX = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_FIX", "true"), True)
+    ENABLE_RETRIEVAL_FOR_DEBUG = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_DEBUG", "true"), True)
+    ENABLE_RETRIEVAL_FOR_GENERATE = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_GENERATE", "false"), False)
+    ENABLE_RETRIEVAL_FOR_EXPLAIN = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_EXPLAIN", "false"), False)
 
     DEFAULT_TEMPERATURE = float(os.getenv("DEFAULT_TEMPERATURE", "0.2"))
     DEFAULT_TOP_P = float(os.getenv("DEFAULT_TOP_P", "0.9"))
 
+    MAX_OUTPUT_TOKENS = int(os.getenv("MAX_OUTPUT_TOKENS", "1200"))
+
     ALLOWED_TASK_TYPES = {
         "generate",
         "fix",