Crackershoot commited on
Commit
b306efc
·
verified ·
1 Parent(s): 889b4c2

Create config.py

Browse files
Files changed (1) hide show
  1. config.py +27 -0
config.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ '''
2
+ Configuration settings for the Dox application.
3
+ '''
4
+ import os
5
+
6
+ # --- Secrets ---
7
+ OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
8
+
9
+ # --- Agent Configuration ---
10
+ AGENT_MODEL = "gpt-4.1-mini"
11
+ AGENT_TEMP = 0.2
12
+
13
+ # --- Vector Database Configuration ---
14
+ LANCEDB_URI = "tmp/lancedb"
15
+ LANCEDB_TABLE = "pdf_documents_prod"
16
+ EMBEDDER_ID = "text-embedding-3-small"
17
+
18
+ # --- Data Sources ---
19
+ PDF_SOURCES = {
20
+ "Working with Hugging Face": "https://media.datacamp.com/cms/working-with-hugging-face.pdf",
21
+ "AI Agents Cheat Sheet": "https://media.datacamp.com/cms/ai-agents-cheat-sheet.pdf",
22
+ "Introduction to SQL with AI": "https://media.datacamp.com/cms/introduction-to-sql-with-ai-1.pdf",
23
+ "Azure CLI Cheat Sheet": "https://media.datacamp.com/legacy/image/upload/v1719844709/Marketing/Blog/Azure_CLI_Cheat_Sheet.pdf"
24
+ }
25
+
26
+ # --- Local Cache ---
27
+ PDF_CACHE_DIR = "pdf_cache"