This commit is contained in:
Aidan Do 2024-11-26 10:00:38 +00:00
parent 1801aa145d
commit f946d23d8e
2 changed files with 74 additions and 0 deletions

39
2.py Normal file
View file

@ -0,0 +1,39 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from openai import OpenAI
client = OpenAI(base_url="http://localhost:8000/v1", api_key="fake")
completion = client.chat.completions.create(
model="meta-llama/Llama-3.2-3B-Instruct",
messages=[
{
"role": "user",
"content": "Write me a haiku about coding",
},
],
extra_body={
"response_format": {
"type": "json_schema",
"json_schema": {
"name": "name_of_response_format",
"schema": {
"type": "object",
"properties": {
"completion_message": {
"type": "string",
}
},
"required": ["completion_message"],
},
"strict": True,
},
}
},
)
print(completion.choices[0].message.content)