norbert/watsonx-openai-proxy
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: pass tool_call_id through to watsonx API

Browse Source 
watsonx requires tool_call_id on role=tool messages.
Added field to ChatMessage model and passthrough in transformer.
This commit is contained in:
Norbert
2026-02-23 18:46:46 +00:00
parent 592b34cff0
commit c6232d262d
2 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/models/openai_models.py
View File

@@ -16,6 +16,7 @@ class ChatMessage(BaseModel):
name: Optional[str] = None name: Optional[str] = None
function_call: Optional[Dict[str, Any]] = None function_call: Optional[Dict[str, Any]] = None
tool_calls: Optional[List[Dict[str, Any]]] = None tool_calls: Optional[List[Dict[str, Any]]] = None
tool_call_id: Optional[str] = None
class FunctionCall(BaseModel): class FunctionCall(BaseModel):
@@ -85,6 +86,7 @@ class ChatCompletionChunkDelta(BaseModel):
content: Optional[Union[str, List[Dict[str, Any]]]] = None content: Optional[Union[str, List[Dict[str, Any]]]] = None
function_call: Optional[Dict[str, Any]] = None function_call: Optional[Dict[str, Any]] = None
tool_calls: Optional[List[Dict[str, Any]]] = None tool_calls: Optional[List[Dict[str, Any]]] = None
tool_call_id: Optional[str] = None
class ChatCompletionChunkChoice(BaseModel): class ChatCompletionChunkChoice(BaseModel):

3
app/utils/transformers.py
View File

@@ -47,6 +47,9 @@ def transform_messages_to_watsonx(messages: List[ChatMessage]) -> List[Dict[str,
if msg.function_call: if msg.function_call:
watsonx_msg["function_call"] = msg.function_call watsonx_msg["function_call"] = msg.function_call
if msg.tool_call_id:
watsonx_msg["tool_call_id"] = msg.tool_call_id
watsonx_messages.append(watsonx_msg) watsonx_messages.append(watsonx_msg)
return watsonx_messages return watsonx_messages