mirror of
https://github.com/kennethreitz/instructor.git
synced 2026-06-05 22:50:18 +00:00
Jason Liu
31 lines
1.1 KiB
Python
31 lines
1.1 KiB
Python
from instructor import OpenAISchema, MultiTask
|
|
from instructor.dsl import ChatCompletion
|
|
from instructor.dsl import messages as m
|
|
from instructor.dsl.messages import messages as s
|
|
|
|
|
|
def test_chatcompletion_has_kwargs():
|
|
class Search(OpenAISchema):
|
|
id: int
|
|
query: str
|
|
|
|
task = (
|
|
ChatCompletion(name="Acme Inc Email Segmentation", model="gpt3.5-turbo-0613")
|
|
| s.SystemTask(task="Segment emails into search queries")
|
|
| MultiTask(subtask_class=Search)
|
|
| m.TaggedMessage(
|
|
tag="email",
|
|
content="Can you find the video I sent last week and also the post about dogs",
|
|
)
|
|
| m.TipsMessage(
|
|
tips=[
|
|
"When unsure about the correct segmentation, try to think about the task as a whole",
|
|
"If acronyms are used expand them to their full form",
|
|
"Use multiple phrases to describe the same thing",
|
|
]
|
|
)
|
|
| m.ChainOfThought()
|
|
)
|
|
assert isinstance(task, ChatCompletion)
|
|
assert isinstance(task.kwargs, dict)
|