mirror of
https://github.com/kennethreitz/instructor.git
synced 2026-06-05 22:50:18 +00:00
Jason Liu
18 lines
422 B
Python
18 lines
422 B
Python
from instructor import OpenAISchema
|
|
from instructor.dsl import MultiTask
|
|
|
|
|
|
def test_multi_task():
|
|
class Search(OpenAISchema):
|
|
"""This is the search docstring"""
|
|
|
|
id: int
|
|
query: str
|
|
|
|
multitask = MultiTask(Search)
|
|
assert multitask.openai_schema["name"] == "MultiSearch"
|
|
assert (
|
|
multitask.openai_schema["description"]
|
|
== "Correct segmentation of `Search` tasks"
|
|
)
|