Get Embedder configuration from environment variables.
()
| 57 | |
| 58 | |
| 59 | def get_embedder_config() -> dict[str, Any]: |
| 60 | """Get Embedder configuration from environment variables.""" |
| 61 | openai_api_key = os.getenv("OPENAI_API_KEY") |
| 62 | openai_base_url = os.getenv("OPENAI_API_BASE", "https://api.openai.com/v1") |
| 63 | ollama_api_base = os.getenv("OLLAMA_API_BASE", "http://localhost:11434") |
| 64 | |
| 65 | # .env uses MOS_EMBEDDER_BACKEND |
| 66 | embedder_backend = os.getenv("MOS_EMBEDDER_BACKEND", "ollama") |
| 67 | |
| 68 | if embedder_backend == "universal_api": |
| 69 | return { |
| 70 | "backend": "universal_api", |
| 71 | "config": { |
| 72 | "provider": os.getenv("MOS_EMBEDDER_PROVIDER", "openai"), |
| 73 | "api_key": os.getenv("MOS_EMBEDDER_API_KEY", openai_api_key or "sk-xxxx"), |
| 74 | "model_name_or_path": os.getenv("MOS_EMBEDDER_MODEL", "text-embedding-3-large"), |
| 75 | "base_url": os.getenv("MOS_EMBEDDER_API_BASE", openai_base_url), |
| 76 | }, |
| 77 | } |
| 78 | else: # ollama |
| 79 | return { |
| 80 | "backend": "ollama", |
| 81 | "config": { |
| 82 | "model_name_or_path": os.getenv("MOS_EMBEDDER_MODEL", "nomic-embed-text:latest"), |
| 83 | "api_base": ollama_api_base, |
| 84 | }, |
| 85 | } |
| 86 | |
| 87 | |
| 88 | def get_chunker_config() -> dict[str, Any]: |
no outgoing calls
no test coverage detected