m97j commited on
Commit
3e43d9a
·
1 Parent(s): 596fcaa

Refactor config.py to separate Hub and local paths

Browse files
Files changed (1) hide show
  1. config.py +12 -1
config.py CHANGED
@@ -3,18 +3,29 @@ import os
3
 
4
  HF_TOKEN = os.environ.get("HF_TOKEN")
5
  HF_MODEL_HUB = os.environ.get("HF_MODEL_HUB", "m97j/pragmatic-search")
 
6
  FALLBACK_DATASET_ID = os.environ.get("FALLBACK_DATASET_ID", "m97j/pls-datasets")
7
  SESSIONS_DATASET_ID = os.environ.get("HF_SESSIONS_DATASET_ID", "pls-assistant-sessions")
 
 
 
 
 
8
  LLM_LOCAL_PATH = os.environ.get("HF_LLM_LOCAL_PATH", "./llm/model_int8.pt")
9
  CONFIG_LOCAL_PATH = os.environ.get("HF_CONFIG_LOCAL_PATH", "./llm/config.json")
10
- RERANKER_LOCAL_PATH = os.environ.get("RERANKER_LOCAL_PATH", "./reranker/8bit_quantized_bge-reranker-v2-m3.onnx")
 
11
  LLM_LOCAL_DIR = os.path.dirname(LLM_LOCAL_PATH)
12
  CONFIG_LOCAL_DIR = os.path.dirname(CONFIG_LOCAL_PATH)
 
 
13
  HF_LLM_REPO = os.environ.get("HF_LLM_REPO", "Qwen/Qwen3-8B")
14
  HF_RERANKER_REPO = os.environ.get("HF_RERANKER_REPO", "BAAI/bge-reranker-v2-m3")
 
15
  GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
16
  GOOGLE_CSE_ID = os.environ.get("GOOGLE_CSE_ID")
17
  TAVILY_SEARCH_API_KEY = os.environ.get("TAVILY_SEARCH_API_KEY")
18
  SERPER_API_KEY = os.environ.get("SERPER_API_KEY")
19
  RAG_API_URL = os.environ.get("RAG_API_URL", "fallback")
 
20
  REFINE_THRESHOLD = float(os.environ.get("REFINE_THRESHOLD", "0.7"))
 
3
 
4
  HF_TOKEN = os.environ.get("HF_TOKEN")
5
  HF_MODEL_HUB = os.environ.get("HF_MODEL_HUB", "m97j/pragmatic-search")
6
+
7
  FALLBACK_DATASET_ID = os.environ.get("FALLBACK_DATASET_ID", "m97j/pls-datasets")
8
  SESSIONS_DATASET_ID = os.environ.get("HF_SESSIONS_DATASET_ID", "pls-assistant-sessions")
9
+
10
+ HF_LLM_FILENAME = os.environ.get("HF_LLM_FILENAME", "llm/model_int8.pt")
11
+ HF_CONFIG_FILENAME = os.environ.get("HF_CONFIG_FILENAME", "llm/config.json")
12
+ HF_RERANKER_FILENAME = os.environ.get("HF_RERANKER_FILENAME", "reranker/model_quantized.onnx")
13
+
14
  LLM_LOCAL_PATH = os.environ.get("HF_LLM_LOCAL_PATH", "./llm/model_int8.pt")
15
  CONFIG_LOCAL_PATH = os.environ.get("HF_CONFIG_LOCAL_PATH", "./llm/config.json")
16
+ RERANKER_LOCAL_PATH = os.environ.get("RERANKER_LOCAL_PATH", "./reranker/model_quantized.onnx")
17
+
18
  LLM_LOCAL_DIR = os.path.dirname(LLM_LOCAL_PATH)
19
  CONFIG_LOCAL_DIR = os.path.dirname(CONFIG_LOCAL_PATH)
20
+ RERANKER_LOCAL_DIR = os.path.dirname(RERANKER_LOCAL_PATH)
21
+
22
  HF_LLM_REPO = os.environ.get("HF_LLM_REPO", "Qwen/Qwen3-8B")
23
  HF_RERANKER_REPO = os.environ.get("HF_RERANKER_REPO", "BAAI/bge-reranker-v2-m3")
24
+
25
  GOOGLE_API_KEY = os.environ.get("GOOGLE_API_KEY")
26
  GOOGLE_CSE_ID = os.environ.get("GOOGLE_CSE_ID")
27
  TAVILY_SEARCH_API_KEY = os.environ.get("TAVILY_SEARCH_API_KEY")
28
  SERPER_API_KEY = os.environ.get("SERPER_API_KEY")
29
  RAG_API_URL = os.environ.get("RAG_API_URL", "fallback")
30
+
31
  REFINE_THRESHOLD = float(os.environ.get("REFINE_THRESHOLD", "0.7"))