hashan-7 commited on
Commit
ad78733
·
verified ·
1 Parent(s): c61c167

Update code

Browse files
Files changed (1) hide show
  1. config.py +35 -11
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 = int(os.getenv("MODEL_TIMEOUT_SECONDS", "90"))
22
- SEARCH_TIMEOUT_SECONDS = int(os.getenv("SEARCH_TIMEOUT_SECONDS", "20"))
23
 
24
- MAX_HISTORY_ITEMS = int(os.getenv("MAX_HISTORY_ITEMS", "5"))
25
- MAX_CODE_CHARS = int(os.getenv("MAX_CODE_CHARS", "25000"))
26
- MAX_ERROR_CHARS = int(os.getenv("MAX_ERROR_CHARS", "8000"))
 
 
27
 
28
- MAX_STACK_RESULTS = int(os.getenv("MAX_STACK_RESULTS", "5"))
29
- MAX_GITHUB_RESULTS = int(os.getenv("MAX_GITHUB_RESULTS", "5"))
30
- MAX_RETRIEVED_ITEMS = int(os.getenv("MAX_RETRIEVED_ITEMS", "8"))
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
- DEFAULT_TEMPERATURE = float(os.getenv("DEFAULT_TEMPERATURE", "0.2"))
40
- DEFAULT_TOP_P = float(os.getenv("DEFAULT_TOP_P", "0.9"))
41
 
42
- MAX_OUTPUT_TOKENS = int(os.getenv("MAX_OUTPUT_TOKENS", "1200"))
 
 
 
 
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",