highperfocused/danswer
1
0
Fork 0
mirror of https://github.com/danswer-ai/danswer.git synced 2025-07-03 11:11:45 +02:00
Code Issues Packages Projects Releases Wiki Activity

print sanitized api key to help troubleshoot (#4764)

Browse Source 
Co-authored-by: Richard Kuo (Onyx) <rkuo@onyx.app>
This commit is contained in:
rkuo-danswer
2025-05-24 15:27:37 -07:00
committed by GitHub
parent 0c7ba8e2ac
commit 23ff3476bc
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
backend/model_server/encoders.py
View File

@ -92,6 +92,7 @@ def format_embedding_error(
service_name: str, service_name: str,
model: str | None, model: str | None,
provider: EmbeddingProvider, provider: EmbeddingProvider,
sanitized_api_key: str | None = None,
status_code: int | None = None, status_code: int | None = None,
) -> str: ) -> str:
""" """
@ -103,6 +104,7 @@ def format_embedding_error(
f"{'HTTP error' if status_code else 'Exception'} embedding text with {service_name} - {detail}: " f"{'HTTP error' if status_code else 'Exception'} embedding text with {service_name} - {detail}: "
f"Model: {model} " f"Model: {model} "
f"Provider: {provider} " f"Provider: {provider} "
f"API Key: {sanitized_api_key} "
f"Exception: {error}" f"Exception: {error}"
) )
@ -133,6 +135,7 @@ class CloudEmbedding:
self.timeout = timeout self.timeout = timeout
self.http_client = httpx.AsyncClient(timeout=timeout) self.http_client = httpx.AsyncClient(timeout=timeout)
self._closed = False self._closed = False
self.sanitized_api_key = api_key[:4] + "********" + api_key[-4:]
async def _embed_openai( async def _embed_openai(
self, texts: list[str], model: str | None, reduced_dimension: int | None self, texts: list[str], model: str | None, reduced_dimension: int | None
@ -306,6 +309,7 @@ class CloudEmbedding:
str(self.provider), str(self.provider),
model_name or deployment_name, model_name or deployment_name,
self.provider, self.provider,
sanitized_api_key=self.sanitized_api_key,
status_code=e.response.status_code, status_code=e.response.status_code,
) )
logger.error(error_string) logger.error(error_string)
@ -317,7 +321,11 @@ class CloudEmbedding:
raise AuthenticationError(provider=str(self.provider)) raise AuthenticationError(provider=str(self.provider))
error_string = format_embedding_error( error_string = format_embedding_error(
e, str(self.provider), model_name or deployment_name, self.provider e,
str(self.provider),
model_name or deployment_name,
self.provider,
sanitized_api_key=self.sanitized_api_key,
) )
logger.error(error_string) logger.error(error_string)
logger.debug(f"Exception texts: {texts}") logger.debug(f"Exception texts: {texts}")