fix: chat completions ci tests (#1630)

This commit is contained in:
cthomas
2025-04-08 15:13:18 -07:00
committed by GitHub
parent 258727cb6a
commit d0334b8a65
2 changed files with 3 additions and 0 deletions

View File

@@ -160,6 +160,7 @@ class ChatCompletionsStreamingInterface(AgentChunkStreamingInterface):
message_id: str,
message_date: datetime,
expect_reasoning_content: bool = False,
name: Optional[str] = None,
message_index: int = 0,
) -> None:
"""

View File

@@ -54,6 +54,7 @@ class AgentChunkStreamingInterface(ABC):
message_id: str,
message_date: datetime,
expect_reasoning_content: bool = False,
name: Optional[str] = None,
message_index: int = 0,
):
"""Process a streaming chunk from an OpenAI-compatible server"""
@@ -105,6 +106,7 @@ class StreamingCLIInterface(AgentChunkStreamingInterface):
message_id: str,
message_date: datetime,
expect_reasoning_content: bool = False,
name: Optional[str] = None,
message_index: int = 0,
):
assert len(chunk.choices) == 1, chunk