fix: Normalize openai client stop reason to make more robust

python/packages/autogen-core/src/autogen_core/models/_types.py

@@ -52,7 +52,7 @@ class RequestUsage:
     completion_tokens: int
 
 
-FinishReasons = Literal["stop", "length", "function_calls", "content_filter"]
+FinishReasons = Literal["stop", "length", "function_calls", "content_filter", "unknown"]
 
 
 @dataclass

python/packages/autogen-ext/src/autogen_ext/models/openai/_openai_client.py

@@ -30,6 +30,7 @@
     Image,
     MessageHandlerContext,
 )
+from autogen_core.models import FinishReasons
 from autogen_core.logging import LLMCallEvent
 from autogen_core.models import (
     AssistantMessage,
@@ -327,6 +328,21 @@
     return name
 
 
+def normalize_stop_reason(stop_reason: str | None) -> FinishReasons:
+    if stop_reason is None:
+        return "unknown"
+
+    # Convert to lower case
+    stop_reason = stop_reason.lower()
+
+    KNOWN_STOP_MAPPINGS: Dict[str, FinishReasons] = {
+        "end_turn": "stop",
+        "tool_calls": "function_calls",
+    }
+
+    return KNOWN_STOP_MAPPINGS.get(stop_reason, "unknown")
+
+
 class BaseOpenAIChatCompletionClient(ChatCompletionClient):
     def __init__(
         self,
@@ -747,8 +763,8 @@
         else:
             prompt_tokens = 0
 
-        if stop_reason is None:
-            raise ValueError("No stop reason found")
+        if stop_reason == "function_call":
+            raise ValueError("Function calls are not supported in this context")
 
         content: Union[str, List[FunctionCall]]
         if len(content_deltas) > 1:
@@ -770,13 +786,9 @@
             prompt_tokens=prompt_tokens,
             completion_tokens=completion_tokens,
         )
-        if stop_reason == "function_call":
-            raise ValueError("Function calls are not supported in this context")
-        if stop_reason == "tool_calls":
-            stop_reason = "function_calls"
 
         result = CreateResult(
-            finish_reason=stop_reason,  # type: ignore
+            finish_reason=normalize_stop_reason(stop_reason),
             content=content,
             usage=usage,
             cached=False,