repositories
loading repo index
repositories
loading repo index
repository
loading code, commits, and activity
public Clawd ADK gateway launch mirror
stars
latest
clone command
git clone gitlawb://did:key:z6Mkq5mY...iFZ5/my-project-publ...git clone gitlawb://did:key:z6Mkq5mY.../my-project-publ...2fa351d6docs: add automaton and perps launch sources16d ago| #1 | import logging |
| #2 | import os |
| #3 | from typing import Optional |
| #4 | |
| #5 | from langchain_nvidia_ai_endpoints import NVIDIAEmbeddings |
| #6 | |
| #7 | from embedchain.config import BaseEmbedderConfig |
| #8 | from embedchain.embedder.base import BaseEmbedder |
| #9 | from embedchain.models import VectorDimensions |
| #10 | |
| #11 | logger = logging.getLogger(__name__) |
| #12 | |
| #13 | |
| #14 | class NvidiaEmbedder(BaseEmbedder): |
| #15 | def __init__(self, config: Optional[BaseEmbedderConfig] = None): |
| #16 | if "NVIDIA_API_KEY" not in os.environ: |
| #17 | raise ValueError("NVIDIA_API_KEY environment variable must be set") |
| #18 | |
| #19 | super().__init__(config=config) |
| #20 | |
| #21 | model = self.config.model or "nvolveqa_40k" |
| #22 | logger.info(f"Using NVIDIA embedding model: {model}") |
| #23 | embedder = NVIDIAEmbeddings(model=model) |
| #24 | embedding_fn = BaseEmbedder._langchain_default_concept(embedder) |
| #25 | self.set_embedding_fn(embedding_fn=embedding_fn) |
| #26 | |
| #27 | vector_dimension = self.config.vector_dimension or VectorDimensions.NVIDIA_AI.value |
| #28 | self.set_vector_dimension(vector_dimension=vector_dimension) |
| #29 |