gl-kp commited on
Commit
e34262f
·
verified ·
1 Parent(s): dabaff0

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. config/config.py +29 -0
config/config.py ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from dotenv import load_dotenv
3
+
4
+ # Load environment variables
5
+ load_dotenv()
6
+
7
+ # Configuration class
8
+ class Config:
9
+ API_KEY = os.getenv('API_KEY')
10
+ OPENAI_API_BASE = os.getenv("OPENAI_API_BASE")
11
+ GROQ_API_KEY = os.getenv('GROQ_API_KEY')
12
+ MEM0_API_KEY = os.getenv('mem0')
13
+
14
+ # Model configurations
15
+ EMBEDDING_MODEL = "text-embedding-ada-002"
16
+ CHAT_MODEL = "gpt-4o-mini"
17
+ LLAMA_GUARD_MODEL = "meta-llama/llama-guard-4-12b"
18
+
19
+ # Vector store configuration
20
+ COLLECTION_NAME = "nutritional_hypotheticals"
21
+ PERSIST_DIRECTORY = "./nutritional_db"
22
+
23
+ # Agent configuration
24
+ GROUNDEDNESS_THRESHOLD = 0.4
25
+ PRECISION_THRESHOLD = 0.7
26
+ MAX_ITERATIONS = 2
27
+ RETRIEVAL_K = 3
28
+
29
+ config = Config()