Delete config.py
Browse files
config.py
DELETED
@@ -1,16 +0,0 @@
|
|
1 |
-
|
2 |
-
class ConfigConstants:
|
3 |
-
# Constants related to datasets and models
|
4 |
-
DATA_SET_NAMES = ['covidqa', 'cuad', 'techqa']#, 'delucionqa', 'emanual', 'expertqa', 'finqa', 'hagrid', 'hotpotqa', 'msmarco', 'pubmedqa', 'tatqa']
|
5 |
-
EMBEDDING_MODEL_NAME = "sentence-transformers/paraphrase-MiniLM-L3-v2"
|
6 |
-
RE_RANKER_MODEL_NAME = 'cross-encoder/ms-marco-electra-base'
|
7 |
-
GENERATION_MODEL_NAME = 'mixtral-8x7b-32768'
|
8 |
-
VALIDATION_MODEL_NAME = 'llama3-70b-8192'
|
9 |
-
DEFAULT_CHUNK_SIZE = 1000
|
10 |
-
CHUNK_OVERLAP = 200
|
11 |
-
|
12 |
-
class AppConfig:
|
13 |
-
def __init__(self, vector_store, gen_llm, val_llm):
|
14 |
-
self.vector_store = vector_store
|
15 |
-
self.gen_llm = gen_llm
|
16 |
-
self.val_llm = val_llm
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|