Nihal2000 commited on
Commit
1992efa
·
verified ·
1 Parent(s): befe9e4

Update config.py

Browse files
Files changed (1) hide show
  1. config.py +5 -4
config.py CHANGED
@@ -7,16 +7,17 @@ load_dotenv()
7
 
8
  class Config:
9
  # API Keys
10
- ANTHROPIC_API_KEY: Optional[str] = os.getenv("ANTHROPIC_API_KEY")
11
  MISTRAL_API_KEY: Optional[str] = os.getenv("MISTRAL_API_KEY")
12
  HUGGINGFACE_API_KEY: Optional[str] = os.getenv("HUGGINGFACE_API_KEY", os.getenv("HF_TOKEN"))
13
- OPENAI_API_KEY: Optional[str] = os.getenv("OPENAI_API_KEY")
 
 
 
14
 
15
  # Model Configuration
16
  EMBEDDING_MODEL: str = os.getenv("EMBEDDING_MODEL", "sentence-transformers/all-MiniLM-L6-v2")
17
- ANTHROPIC_MODEL: str = os.getenv("ANTHROPIC_MODEL", "claude-3-haiku-20240307") # Using faster model
18
  MISTRAL_MODEL: str = os.getenv("MISTRAL_MODEL", "mistral-small-latest") # Using smaller model
19
- OPENAI_MODEL: str = os.getenv("OPENAI_MODEL", "gpt-4o-mini")
20
 
21
  # Vector Store Configuration
22
  VECTOR_STORE_PATH: str = os.getenv("VECTOR_STORE_PATH", "./data/vector_store")
 
7
 
8
  class Config:
9
  # API Keys
10
+ NEBIUS_API_KEY: Optional[str] = os.getenv("NEBIUS_API_KEY")
11
  MISTRAL_API_KEY: Optional[str] = os.getenv("MISTRAL_API_KEY")
12
  HUGGINGFACE_API_KEY: Optional[str] = os.getenv("HUGGINGFACE_API_KEY", os.getenv("HF_TOKEN"))
13
+
14
+ # NEBIUS Configuration (OpenAI OSS models)
15
+ NEBIUS_BASE_URL: str = os.getenv("NEBIUS_BASE_URL", "https://api.studio.nebius.com/v1/")
16
+ NEBIUS_MODEL: str = os.getenv("NEBIUS_MODEL", "openai/gpt-oss-120b")
17
 
18
  # Model Configuration
19
  EMBEDDING_MODEL: str = os.getenv("EMBEDDING_MODEL", "sentence-transformers/all-MiniLM-L6-v2")
 
20
  MISTRAL_MODEL: str = os.getenv("MISTRAL_MODEL", "mistral-small-latest") # Using smaller model
 
21
 
22
  # Vector Store Configuration
23
  VECTOR_STORE_PATH: str = os.getenv("VECTOR_STORE_PATH", "./data/vector_store")