참고소스 수정본
This commit is contained in:
28
참고/instructor-main/examples/mistral/mistral.py
Normal file
28
참고/instructor-main/examples/mistral/mistral.py
Normal file
@@ -0,0 +1,28 @@
|
||||
from pydantic import BaseModel
|
||||
from mistralai.client import MistralClient
|
||||
from instructor import from_mistral
|
||||
from instructor.mode import Mode
|
||||
import os
|
||||
|
||||
|
||||
class UserDetails(BaseModel):
|
||||
name: str
|
||||
age: int
|
||||
|
||||
|
||||
# enables `response_model` in chat call
|
||||
client = MistralClient(api_key=os.environ.get("MISTRAL_API_KEY"))
|
||||
instructor_client = from_mistral(
|
||||
client=client,
|
||||
model="mistral-large-latest",
|
||||
mode=Mode.TOOLS,
|
||||
max_tokens=1000,
|
||||
)
|
||||
|
||||
resp = instructor_client.messages.create(
|
||||
response_model=UserDetails,
|
||||
messages=[{"role": "user", "content": "Jason is 10"}],
|
||||
temperature=0,
|
||||
)
|
||||
|
||||
print(resp)
|
||||
Reference in New Issue
Block a user