logo
openai
repo-copy-icon

copied

You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions

3.3 KiB

ChatBot with OpenAI

author: Jael, Yuchen


Description

A chat-bot operator returns answer in text given input text. This operator is implemented with Completion method from OpenAI. Please note you need an OpenAI API key to access OpenAI.


Code Example

Use the default model to answer the question "Who are you?".

Write a pipeline with explicit inputs/outputs name specifications:

from towhee import pipe, ops, DataCollection

p = (
    pipe.input('question')
        .map('question', 'answer', 
             ops.chatbot.openai(api_key=OPENAI_API_KEY))
        .output('question', 'answer')
)

DataCollection(p('Who are you?')).show()


Factory Constructor

Create the operator via the following factory method:

chatbot.openai(model_name: str, api_key: str)

Parameters:

model_name: str

The model name in string, defaults to 'text-davinci-003'. Refer to OpenAI List Models for all supported model names.

api_key: str=None

The OpenAI API key in string, defaults to None.


Interface

The operator takes a piece of text in string as input. It returns a text answer.

__call__(txt)

Parameters:

text: str

​ The text in string.

Returns:

answer: str

​ The answer in string generated by model.


supported_model_names()

Get a list of supported model names.

More Resources

3.3 KiB

ChatBot with OpenAI

author: Jael, Yuchen


Description

A chat-bot operator returns answer in text given input text. This operator is implemented with Completion method from OpenAI. Please note you need an OpenAI API key to access OpenAI.


Code Example

Use the default model to answer the question "Who are you?".

Write a pipeline with explicit inputs/outputs name specifications:

from towhee import pipe, ops, DataCollection

p = (
    pipe.input('question')
        .map('question', 'answer', 
             ops.chatbot.openai(api_key=OPENAI_API_KEY))
        .output('question', 'answer')
)

DataCollection(p('Who are you?')).show()


Factory Constructor

Create the operator via the following factory method:

chatbot.openai(model_name: str, api_key: str)

Parameters:

model_name: str

The model name in string, defaults to 'text-davinci-003'. Refer to OpenAI List Models for all supported model names.

api_key: str=None

The OpenAI API key in string, defaults to None.


Interface

The operator takes a piece of text in string as input. It returns a text answer.

__call__(txt)

Parameters:

text: str

​ The text in string.

Returns:

answer: str

​ The answer in string generated by model.


supported_model_names()

Get a list of supported model names.

More Resources