codekingpro/portable-devtools

1import base64
2import os
3import numpy as np
4from typing import TYPE_CHECKING
5
6DEFAULT_CHROMA_EMBED_URL = "https://embed.trychroma.com"
7
8
9def get_chroma_embed_url() -> str:
10    """Return embed base URL from CHROMA_EMBED_URL or default without trailing slash."""
11    return (os.environ.get("CHROMA_EMBED_URL") or DEFAULT_CHROMA_EMBED_URL).rstrip("/")
12
13if TYPE_CHECKING:
14    from chromadb.api.shared_system_client import SharedSystemClient
15
16
17def _get_shared_system_client() -> "type[SharedSystemClient]":
18    """Lazy import of SharedSystemClient to avoid circular imports."""
19    from chromadb.api.shared_system_client import SharedSystemClient
20
21    return SharedSystemClient
22
23
24def decode_embedding(b64_string: str):
25    """Decode a base64-encoded int8 embedding."""
26    return np.frombuffer(base64.b64decode(b64_string), dtype=np.int8).astype(np.float32)
27