Update code
Browse files
config.py
CHANGED
|
@@ -7,6 +7,20 @@ def to_bool(value: str, default: bool = False) -> bool:
|
|
| 7 |
return str(value).strip().lower() in {"true", "1", "yes", "on"}
|
| 8 |
|
| 9 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
class Settings:
|
| 11 |
APP_NAME = "Chat7 CodeX Backend"
|
| 12 |
APP_VERSION = "1.0.0"
|
|
@@ -18,16 +32,18 @@ class Settings:
|
|
| 18 |
STACKOVERFLOW_KEY = os.getenv("STACKOVERFLOW_KEY", "")
|
| 19 |
HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN", "")
|
| 20 |
|
| 21 |
-
MODEL_TIMEOUT_SECONDS =
|
| 22 |
-
SEARCH_TIMEOUT_SECONDS =
|
| 23 |
|
| 24 |
-
MAX_HISTORY_ITEMS =
|
| 25 |
-
MAX_CODE_CHARS =
|
| 26 |
-
MAX_ERROR_CHARS =
|
|
|
|
|
|
|
| 27 |
|
| 28 |
-
MAX_STACK_RESULTS =
|
| 29 |
-
MAX_GITHUB_RESULTS =
|
| 30 |
-
MAX_RETRIEVED_ITEMS =
|
| 31 |
|
| 32 |
ENABLE_STACK_SEARCH = to_bool(os.getenv("ENABLE_STACK_SEARCH", "true"), True)
|
| 33 |
ENABLE_GITHUB_SEARCH = to_bool(os.getenv("ENABLE_GITHUB_SEARCH", "true"), True)
|
|
@@ -35,11 +51,19 @@ class Settings:
|
|
| 35 |
ENABLE_RETRIEVAL_FOR_DEBUG = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_DEBUG", "true"), True)
|
| 36 |
ENABLE_RETRIEVAL_FOR_GENERATE = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_GENERATE", "false"), False)
|
| 37 |
ENABLE_RETRIEVAL_FOR_EXPLAIN = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_EXPLAIN", "false"), False)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 38 |
|
| 39 |
-
|
| 40 |
-
DEFAULT_TOP_P = float(os.getenv("DEFAULT_TOP_P", "0.9"))
|
| 41 |
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
|
| 44 |
ALLOWED_TASK_TYPES = {
|
| 45 |
"generate",
|
|
|
|
| 7 |
return str(value).strip().lower() in {"true", "1", "yes", "on"}
|
| 8 |
|
| 9 |
|
| 10 |
+
def to_int(value: str, default: int) -> int:
|
| 11 |
+
try:
|
| 12 |
+
return int(str(value).strip())
|
| 13 |
+
except Exception:
|
| 14 |
+
return default
|
| 15 |
+
|
| 16 |
+
|
| 17 |
+
def to_float(value: str, default: float) -> float:
|
| 18 |
+
try:
|
| 19 |
+
return float(str(value).strip())
|
| 20 |
+
except Exception:
|
| 21 |
+
return default
|
| 22 |
+
|
| 23 |
+
|
| 24 |
class Settings:
|
| 25 |
APP_NAME = "Chat7 CodeX Backend"
|
| 26 |
APP_VERSION = "1.0.0"
|
|
|
|
| 32 |
STACKOVERFLOW_KEY = os.getenv("STACKOVERFLOW_KEY", "")
|
| 33 |
HUGGINGFACE_API_TOKEN = os.getenv("HUGGINGFACE_API_TOKEN", "")
|
| 34 |
|
| 35 |
+
MODEL_TIMEOUT_SECONDS = to_int(os.getenv("MODEL_TIMEOUT_SECONDS", "90"), 90)
|
| 36 |
+
SEARCH_TIMEOUT_SECONDS = to_int(os.getenv("SEARCH_TIMEOUT_SECONDS", "20"), 20)
|
| 37 |
|
| 38 |
+
MAX_HISTORY_ITEMS = to_int(os.getenv("MAX_HISTORY_ITEMS", "5"), 5)
|
| 39 |
+
MAX_CODE_CHARS = to_int(os.getenv("MAX_CODE_CHARS", "25000"), 25000)
|
| 40 |
+
MAX_ERROR_CHARS = to_int(os.getenv("MAX_ERROR_CHARS", "8000"), 8000)
|
| 41 |
+
MAX_MESSAGE_CHARS = to_int(os.getenv("MAX_MESSAGE_CHARS", "6000"), 6000)
|
| 42 |
+
MAX_PREVIOUS_CONTEXT_CHARS = to_int(os.getenv("MAX_PREVIOUS_CONTEXT_CHARS", "6000"), 6000)
|
| 43 |
|
| 44 |
+
MAX_STACK_RESULTS = to_int(os.getenv("MAX_STACK_RESULTS", "5"), 5)
|
| 45 |
+
MAX_GITHUB_RESULTS = to_int(os.getenv("MAX_GITHUB_RESULTS", "5"), 5)
|
| 46 |
+
MAX_RETRIEVED_ITEMS = to_int(os.getenv("MAX_RETRIEVED_ITEMS", "8"), 8)
|
| 47 |
|
| 48 |
ENABLE_STACK_SEARCH = to_bool(os.getenv("ENABLE_STACK_SEARCH", "true"), True)
|
| 49 |
ENABLE_GITHUB_SEARCH = to_bool(os.getenv("ENABLE_GITHUB_SEARCH", "true"), True)
|
|
|
|
| 51 |
ENABLE_RETRIEVAL_FOR_DEBUG = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_DEBUG", "true"), True)
|
| 52 |
ENABLE_RETRIEVAL_FOR_GENERATE = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_GENERATE", "false"), False)
|
| 53 |
ENABLE_RETRIEVAL_FOR_EXPLAIN = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_EXPLAIN", "false"), False)
|
| 54 |
+
ENABLE_RETRIEVAL_FOR_REVIEW = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_REVIEW", "false"), False)
|
| 55 |
+
ENABLE_RETRIEVAL_FOR_REFACTOR = to_bool(os.getenv("ENABLE_RETRIEVAL_FOR_REFACTOR", "false"), False)
|
| 56 |
+
|
| 57 |
+
DEFAULT_TEMPERATURE = to_float(os.getenv("DEFAULT_TEMPERATURE", "0.2"), 0.2)
|
| 58 |
+
DEFAULT_TOP_P = to_float(os.getenv("DEFAULT_TOP_P", "0.9"), 0.9)
|
| 59 |
|
| 60 |
+
MAX_OUTPUT_TOKENS = to_int(os.getenv("MAX_OUTPUT_TOKENS", "1200"), 1200)
|
|
|
|
| 61 |
|
| 62 |
+
PRESERVE_SYMBOL_NAMES_BY_DEFAULT = to_bool(
|
| 63 |
+
os.getenv("PRESERVE_SYMBOL_NAMES_BY_DEFAULT", "true"),
|
| 64 |
+
True,
|
| 65 |
+
)
|
| 66 |
+
ENABLE_SCOPE_GUARD = to_bool(os.getenv("ENABLE_SCOPE_GUARD", "true"), True)
|
| 67 |
|
| 68 |
ALLOWED_TASK_TYPES = {
|
| 69 |
"generate",
|