scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-09 03:18:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 108

style: ruff format

Browse source
This commit is contained in:
Onur Cakmak 2026-09-19 16:19:44 -04:00
parent 5e187903a4
commit 0ceb56820c
2 changed files with 2 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/chatgpt/common_utils.py
View file

@ -281,8 +281,7 @@ def get_chatgpt_session_id(litellm_params: object) -> str | None:
# carry internal metadata under the latter. Generated ids are skipped,
# not returned, so a caller-supplied stable prompt_cache_key still wins.
generated: Final = any(
isinstance(params.get(name), dict)
and params[name].get(SESSION_ID_GENERATED_METADATA_KEY)
isinstance(params.get(name), dict) and params[name].get(SESSION_ID_GENERATED_METADATA_KEY)
for name in ("metadata", "litellm_metadata")
)
if not generated:

5
tests/unit/llms/chatgpt/test_session_id.py
View file

@ -15,10 +15,7 @@ from litellm.llms.chatgpt.common_utils import (
def test_explicit_session_ids_win():
assert get_chatgpt_session_id({"session_id": "s", "prompt_cache_key": "k"}) == "s"
assert get_chatgpt_session_id({"litellm_session_id": "ls", "prompt_cache_key": "k"}) == "ls"
assert (
get_chatgpt_session_id({"metadata": {"session_id": "ms"}, "prompt_cache_key": "k"})
== "ms"
)
assert get_chatgpt_session_id({"metadata": {"session_id": "ms"}, "prompt_cache_key": "k"}) == "ms"
def test_prompt_cache_key_becomes_session_id():