42 lines
833 B
Markdown
42 lines
833 B
Markdown
### Example: `sqlchat` (use like openai pypi package) <a name="example-sqlchat"></a>
|
|
|
|
```python
|
|
# Import sqlchat
|
|
import sqlchat
|
|
|
|
# sqlchat.Completion.create
|
|
# sqlchat.StreamCompletion.create
|
|
|
|
[...]
|
|
|
|
```
|
|
|
|
#### Example Chatbot
|
|
```python
|
|
messages = []
|
|
|
|
while True:
|
|
user = input('you: ')
|
|
|
|
sqlchat_cmpl = sqlchat.Completion.create(
|
|
prompt = user,
|
|
messages = messages
|
|
)
|
|
|
|
print('gpt:', sqlchat_cmpl.completion.choices[0].text)
|
|
|
|
messages.extend([
|
|
{'role': 'user', 'content': user },
|
|
{'role': 'assistant', 'content': sqlchat_cmpl.completion.choices[0].text}
|
|
])
|
|
```
|
|
|
|
#### Streaming Response:
|
|
|
|
```python
|
|
for response in sqlchat.StreamCompletion.create(
|
|
prompt = 'write python code to reverse a string',
|
|
messages = []):
|
|
|
|
print(response.completion.choices[0].text)
|
|
```
|