Spaces:
Running on Zero
Running on Zero
Refactor config.py to separate Hub and local paths
Browse files
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/
|
|
|
|
| 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"))
|