scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-11-05 07:37:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 233

style: apply black formatting to predibase and xecguard modules

Browse source 
Bahtya
This commit is contained in:
Bahtya 2026-04-27 12:23:23 +08:00
parent 9fc41ca7ce
commit 5c13008ebb
2 changed files with 16 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
litellm/llms/predibase/chat/transformation.py
View file

@ -175,7 +175,10 @@ class PredibaseConfig(BaseConfig):
completion_response["generated_text"]
)
if "details" in completion_response and "tokens" in completion_response["details"]:
if (
"details" in completion_response
and "tokens" in completion_response["details"]
):
model_response.choices[0].finish_reason = map_finish_reason(
completion_response["details"]["finish_reason"]
)
@ -203,7 +206,9 @@ class PredibaseConfig(BaseConfig):
and "best_of_sequences" in completion_response["details"]
):
choices_list = []
for idx, item in enumerate(completion_response["details"]["best_of_sequences"]):
for idx, item in enumerate(
completion_response["details"]["best_of_sequences"]
):
sum_logprob = 0
for token in item["tokens"]:
if token["logprob"] is not None:
@ -342,7 +347,9 @@ class PredibaseConfig(BaseConfig):
base_url = os.getenv("PREDIBASE_API_BASE", "")
completion_url = f"{base_url}/{tenant_id}/deployments/v2/llms/{model}"
should_stream = stream if stream is not None else optional_params.get("stream", False)
should_stream = (
stream if stream is not None else optional_params.get("stream", False)
)
if should_stream is True:
completion_url += "/generate_stream"
else:

9
litellm/proxy/guardrails/guardrail_hooks/xecguard/xecguard.py
View file

@ -212,8 +212,11 @@ class XecGuardGuardrail(CustomGuardrail):
isinstance(kwargs, dict)
and "litellm_params" in kwargs
and "metadata" in kwargs["litellm_params"]
and "standard_logging_guardrail_information"in kwargs["litellm_params"]["metadata"]
and kwargs["litellm_params"]["metadata"]["standard_logging_guardrail_information"]
and "standard_logging_guardrail_information"
in kwargs["litellm_params"]["metadata"]
and kwargs["litellm_params"]["metadata"][
"standard_logging_guardrail_information"
]
):
return kwargs, result
@ -585,4 +588,4 @@ class XecGuardGuardrail(CustomGuardrail):
return (
f"Blocked by XecGuard grounding: rules=[{rules_str}] "
f"trace_id={trace_id} rationale={rationale}"
)
)