Update config.py
Browse files
config.py
CHANGED
|
@@ -7,16 +7,17 @@ load_dotenv()
|
|
| 7 |
|
| 8 |
class Config:
|
| 9 |
# API Keys
|
| 10 |
-
|
| 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 |
|
| 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")
|