CultriX commited on
Commit
9bf9067
·
1 Parent(s): f9752a0
Files changed (1) hide show
  1. config/settings.py +28 -0
config/settings.py ADDED
@@ -0,0 +1,28 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import os
2
+ from typing import Optional
3
+ from pydantic import BaseModel
4
+ from dotenv import load_dotenv
5
+
6
+ load_dotenv()
7
+
8
+ class Config(BaseModel):
9
+ # OpenRouter API configuration
10
+ OPENROUTER_API_KEY: Optional[str] = os.getenv("OPENROUTER_API_KEY")
11
+ OPENROUTER_BASE_URL: str = "https://openrouter.ai/api/v1"
12
+
13
+ # Model configuration - using free models when possible
14
+ EXTRACTION_MODEL: str = "qwen/qwen3-32b" # Free model
15
+ BACKUP_MODEL: str = "qwen/qwen3-32b" # Backup free model
16
+
17
+ # Processing configuration
18
+ CHUNK_SIZE: int = 2000
19
+ CHUNK_OVERLAP: int = 200
20
+ MAX_FILE_SIZE_MB: int = 10
21
+
22
+ # Graph configuration
23
+ MAX_ENTITIES: int = 100
24
+ MAX_RELATIONSHIPS: int = 200
25
+ ENTITY_IMPORTANCE_THRESHOLD: float = 0.3
26
+
27
+ class Config:
28
+ env_file = ".env"