kennethreitz/instructor
1
0
Fork 0
mirror of https://github.com/kennethreitz/instructor.git synced 2026-06-05 22:50:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

clean up tests

Browse Source
This commit is contained in:
Jason Liu
2023-10-22 17:46:25 -04:00
parent 08d08aef26
commit 6f2e3d56c6
2 changed files with 5 additions and 64 deletions
Download Patch File Download Diff File Expand all files Collapse all files
instructor/distil.py
+1 -1
View File
@@ -172,7 +172,7 @@ class Instructions:
if finetune_format == FinetuneFormat.MESSAGES:
openai_kwargs = self.openai_kwargs(name, fn, args, kwargs, base_model)
openai_kwargs.append(
openai_kwargs["messages"].append(
{
"role": "assistant",
"function_call": {
tests/test_multitask.py
+4 -63
View File
@@ -10,67 +10,8 @@ def test_multi_task():
query: str
multitask = MultiTask(Search)
assert multitask.openai_schema == {
"description": "Correct segmentation of `Search` tasks",
"name": "MultiSearch",
"parameters": {
"$defs": {
"Search": {
"properties": {
"id": {"type": "integer"},
"query": {"type": "string"},
},
"required": ["id", "query"],
"description": "This is the search docstring",
"type": "object",
}
},
"properties": {
"tasks": {
"description": "Correctly segmented list of `Search` tasks",
"items": {"$ref": "#/$defs/Search"},
"type": "array",
}
},
"required": ["tasks"],
"type": "object",
},
}
def test_multi_task_with_name_and_desc():
class Search(OpenAISchema):
"""This is the search docstring"""
id: int
query: str
multitask = MultiTask(
subtask_class=Search, name="MyCustomName", description="MyCustomDesc"
assert multitask.openai_schema["name"] == "MultiSearch"
assert (
multitask.openai_schema["description"]
== "Correct segmentation of `Search` tasks"
)
assert multitask.openai_schema == {
"description": "MyCustomDesc",
"name": "MultiMyCustomName",
"parameters": {
"$defs": {
"Search": {
"properties": {
"id": {"type": "integer"},
"query": {"type": "string"},
},
"required": ["id", "query"],
"description": "This is the search docstring",
"type": "object",
}
},
"properties": {
"tasks": {
"description": "Correctly segmented list of `MyCustomName` tasks",
"items": {"$ref": "#/$defs/Search"},
"type": "array",
}
},
"required": ["tasks"],
"type": "object",
},
}