Skip to content

Commit 77c217f

Browse files
rkuo-danswerRichard Kuo (Onyx)
authored andcommitted
print sanitized api key to help troubleshoot (onyx-dot-app#4764)
Co-authored-by: Richard Kuo (Onyx) <rkuo@onyx.app>
1 parent a8ecac0 commit 77c217f

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

backend/model_server/encoders.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,6 +92,7 @@ def format_embedding_error(
9292
service_name: str,
9393
model: str | None,
9494
provider: EmbeddingProvider,
95+
sanitized_api_key: str | None = None,
9596
status_code: int | None = None,
9697
) -> str:
9798
"""
@@ -103,6 +104,7 @@ def format_embedding_error(
103104
f"{'HTTP error' if status_code else 'Exception'} embedding text with {service_name} - {detail}: "
104105
f"Model: {model} "
105106
f"Provider: {provider} "
107+
f"API Key: {sanitized_api_key} "
106108
f"Exception: {error}"
107109
)
108110

@@ -133,6 +135,7 @@ def __init__(
133135
self.timeout = timeout
134136
self.http_client = httpx.AsyncClient(timeout=timeout)
135137
self._closed = False
138+
self.sanitized_api_key = api_key[:4] + "********" + api_key[-4:]
136139

137140
async def _embed_openai(
138141
self, texts: list[str], model: str | None, reduced_dimension: int | None
@@ -306,6 +309,7 @@ async def embed(
306309
str(self.provider),
307310
model_name or deployment_name,
308311
self.provider,
312+
sanitized_api_key=self.sanitized_api_key,
309313
status_code=e.response.status_code,
310314
)
311315
logger.error(error_string)
@@ -317,7 +321,11 @@ async def embed(
317321
raise AuthenticationError(provider=str(self.provider))
318322

319323
error_string = format_embedding_error(
320-
e, str(self.provider), model_name or deployment_name, self.provider
324+
e,
325+
str(self.provider),
326+
model_name or deployment_name,
327+
self.provider,
328+
sanitized_api_key=self.sanitized_api_key,
321329
)
322330
logger.error(error_string)
323331
logger.debug(f"Exception texts: {texts}")

0 commit comments

Comments
 (0)