phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(feat) set custom_llm_provider for embedding hidden params

Browse source
This commit is contained in:
ishaan-jaff 2024-01-12 17:35:08 -08:00
parent f85b64b85c
commit 7f37d7e44f
1 changed files with 2 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

2
litellm/main.py
View file

@ -2512,6 +2512,8 @@ def embedding(
else: else:
args = locals() args = locals()
raise ValueError(f"No valid embedding model args passed in - {args}") raise ValueError(f"No valid embedding model args passed in - {args}")
if response is not None:
response._hidden_params["custom_llm_provider"] = custom_llm_provider
return response return response
except Exception as e: except Exception as e:
## LOGGING ## LOGGING