diff --git a/packages/gen/gen_ai_hub/orchestration_v2/models/response.py b/packages/gen/gen_ai_hub/orchestration_v2/models/response.py index 238fdb4..91c9f14 100644 --- a/packages/gen/gen_ai_hub/orchestration_v2/models/response.py +++ b/packages/gen/gen_ai_hub/orchestration_v2/models/response.py @@ -5,7 +5,6 @@ from typing import List, Optional, Any, Literal, Union from pydantic import ConfigDict, Field -from gen_ai_hub.orchestration.models.response import ModuleResultsStreaming from gen_ai_hub.orchestration_v2.models.base import ResponseBaseModel from gen_ai_hub.orchestration_v2.models.message import ChatMessage, FunctionCall, ResponseChatMessage @@ -189,6 +188,7 @@ class StreamDelta(ResponseBaseModel): role: Optional[str] = None content: str tool_calls: Optional[List[StreamToolCall]] = None + refusal: Optional[str] = None class StreamLLMChoice(ResponseBaseModel): @@ -345,7 +345,7 @@ class SAPAPIErrorStreaming(ResponseBaseModel): code: int message: str location: str - intermediate_results: Optional[ModuleResultsStreaming] = None + intermediate_results: Optional[StreamModuleResults] = None headers: Optional[dict[str, str]] = None class CompletionPostResponse(ResponseBaseModel):