fixed completions endpoint and readme details for dsl example (#33)

This commit is contained in:
Francisco Ingham
2023-07-03 13:55:43 -03:00
committed by GitHub
parent 8b7d28009e
commit 8dfec89c70
2 changed files with 8 additions and 8 deletions
+5 -5
View File
@@ -119,8 +119,8 @@ class Search(OpenAISchema):
id: int
query: str
task = (
ChatCompletion(name="Acme Inc Email Segmentation", model="gpt3.5-turbo-0613")
tasks = (
ChatCompletion(name="Acme Inc Email Segmentation", model="gpt-3.5-turbo-0613")
| m.ExpertSystem(task="Segment emails into search queries")
| MultiTask(subtask_class=Search)
| m.TaggedMessage(
@@ -139,8 +139,8 @@ task = (
# Its important that this just builds you request,
# all these | operators are overloaded and all we do is compile
# it to the openai kwargs
assert isinstance(task, ChatCompletion)
pprint(task.kwargs, indent=3)
assert isinstance(tasks, ChatCompletion)
pprint(tasks.kwargs, indent=3)
"""
{
"messages": [
@@ -189,7 +189,7 @@ pprint(task.kwargs, indent=3)
"function_call": {"name": "MultiSearch"},
"max_tokens": 1000,
"temperature": 0.1,
"model": "gpt3.5-turbo-0613"
"model": "gpt-3.5-turbo-0613"
}
"""
+3 -3
View File
@@ -92,14 +92,14 @@ class ChatCompletion(BaseModel):
def create(self):
kwargs = self.kwargs
completion = openai.Completion.create(**kwargs)
completion = openai.ChatCompletion.create(**kwargs)
if self.function:
return self.function.from_response(completion)
async def acreate(self):
kwargs = self.kwargs
completion = openai.Completion.acreate(**kwargs)
completion = openai.ChatCompletion.acreate(**kwargs)
if self.function:
return self.function.from_response(await completion)
return await completion