This commit is contained in:
2026-04-02 16:17:48 +03:30
parent 1134147b7e
commit a67236d45c
8 changed files with 60 additions and 15 deletions
+20 -4
View File
@@ -12,6 +12,14 @@ from .config import RAGConfig, load_rag_config
logger = logging.getLogger(__name__)
def _mask_secret(value: str | None) -> str:
if not value:
return "<missing>"
if len(value) <= 8:
return "****"
return f"{value[:4]}...{value[-4:]}"
def get_embedding_client(config: RAGConfig | None = None) -> OpenAI:
"""
ساخت کلاینت OpenAI برای Embedding بر اساس provider فعال.
@@ -19,7 +27,7 @@ def get_embedding_client(config: RAGConfig | None = None) -> OpenAI:
"""
cfg = config or load_rag_config()
emb = cfg.embedding
logger.info(emb.provider)
logger.info("embedding provider=%s", emb.provider)
if emb.provider == "avalai":
env_var = emb.avalai_api_key_env or emb.api_key_env or "AVALAI_API_KEY"
@@ -29,7 +37,11 @@ def get_embedding_client(config: RAGConfig | None = None) -> OpenAI:
env_var = emb.api_key_env or "GAPGPT_API_KEY"
api_key = os.environ.get(env_var)
base_url = emb.base_url or "https://api.gapgpt.app/v1"
logger.info(api_key+" "+base_url)
logger.info(
"embedding base_url=%s api_key=%s",
base_url,
_mask_secret(api_key),
)
return OpenAI(api_key=api_key, base_url=base_url)
@@ -44,7 +56,7 @@ def get_chat_client(config: RAGConfig | None = None) -> OpenAI:
provider = llm.provider or cfg.embedding.provider
logger.info(provider)
logger.info("chat provider=%s", provider)
if provider == "avalai":
env_var = llm.avalai_api_key_env or llm.api_key_env or "AVALAI_API_KEY"
api_key = os.environ.get(env_var)
@@ -53,6 +65,10 @@ def get_chat_client(config: RAGConfig | None = None) -> OpenAI:
env_var = llm.api_key_env or "GAPGPT_API_KEY"
api_key = os.environ.get(env_var)
base_url = llm.base_url or "https://api.gapgpt.app/v1"
logger.info(api_key,base_url)
logger.info(
"chat base_url=%s api_key=%s",
base_url,
_mask_secret(api_key),
)
return OpenAI(api_key=api_key, base_url=base_url)