phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-03 01:03:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

use pydantic v2's model_dump() instead of dict()

Browse source
This commit is contained in:
Matthew Farrellee 2024-11-21 10:04:26 -05:00
parent 5fbfb9d854
commit 3ed2e816fa
1 changed files with 1 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/providers/remote/inference/nvidia/_openai_utils.py
View file

@ -118,7 +118,7 @@ def _convert_message(message: Message) -> Dict:
""" """
Convert a Message to an OpenAI API-compatible dictionary. Convert a Message to an OpenAI API-compatible dictionary.
""" """
out_dict = message.dict() out_dict = message.model_dump()
# Llama Stack uses role="ipython" for tool call messages, OpenAI uses "tool" # Llama Stack uses role="ipython" for tool call messages, OpenAI uses "tool"
if out_dict["role"] == "ipython": if out_dict["role"] == "ipython":
out_dict.update(role="tool") out_dict.update(role="tool")