Skip to content

Commit 20b3c19

Browse files
committed
Update lite_llm.py
1 parent c727c3f commit 20b3c19

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/google/adk/models/lite_llm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1352,7 +1352,7 @@ def _model_response_to_generate_content_response(
13521352
if finish_reason:
13531353
# If LiteLLM already provides a FinishReason enum (e.g., for Gemini), use
13541354
# it directly. Otherwise, map the finish_reason string to the enum.
1355-
_set_finish_reason(llm_response, finish_reason)
1355+
llm_response.finish_reason = _map_finish_reason(finish_reason)
13561356
if response.get("usage", None):
13571357
llm_response.usage_metadata = types.GenerateContentResponseUsageMetadata(
13581358
prompt_token_count=response["usage"].get("prompt_tokens", 0),

0 commit comments

Comments
 (0)