Skip to content

Commit cc401a0

Browse files
andrewm4894claude
andcommitted
fix: remove 'unknown' fallback from non-streaming to match original behavior
Non-streaming originally returned None when model wasn't in kwargs. Streaming keeps "unknown" fallback as that was the original behavior. 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <[email protected]>
1 parent 8dccb79 commit cc401a0

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

posthog/ai/utils.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -285,9 +285,7 @@ def call_llm_and_track_usage(
285285

286286
event_properties = {
287287
"$ai_provider": provider,
288-
"$ai_model": kwargs.get("model")
289-
or getattr(response, "model", None)
290-
or "unknown",
288+
"$ai_model": kwargs.get("model") or getattr(response, "model", None),
291289
"$ai_model_parameters": get_model_params(kwargs),
292290
"$ai_input": with_privacy_mode(
293291
ph_client, posthog_privacy_mode, sanitized_messages
@@ -398,9 +396,7 @@ async def call_llm_and_track_usage_async(
398396

399397
event_properties = {
400398
"$ai_provider": provider,
401-
"$ai_model": kwargs.get("model")
402-
or getattr(response, "model", None)
403-
or "unknown",
399+
"$ai_model": kwargs.get("model") or getattr(response, "model", None),
404400
"$ai_model_parameters": get_model_params(kwargs),
405401
"$ai_input": with_privacy_mode(
406402
ph_client, posthog_privacy_mode, sanitized_messages

0 commit comments

Comments
 (0)