mirror of
https://github.com/hwchase17/langchain
synced 2024-10-29 17:07:25 +00:00
47 lines
919 B
Plaintext
47 lines
919 B
Plaintext
|
```python
|
||
|
from langchain.output_parsers import CommaSeparatedListOutputParser
|
||
|
from langchain.prompts import PromptTemplate, ChatPromptTemplate, HumanMessagePromptTemplate
|
||
|
from langchain.llms import OpenAI
|
||
|
from langchain.chat_models import ChatOpenAI
|
||
|
|
||
|
output_parser = CommaSeparatedListOutputParser()
|
||
|
```
|
||
|
|
||
|
|
||
|
```python
|
||
|
format_instructions = output_parser.get_format_instructions()
|
||
|
prompt = PromptTemplate(
|
||
|
template="List five {subject}.\n{format_instructions}",
|
||
|
input_variables=["subject"],
|
||
|
partial_variables={"format_instructions": format_instructions}
|
||
|
)
|
||
|
```
|
||
|
|
||
|
|
||
|
```python
|
||
|
model = OpenAI(temperature=0)
|
||
|
```
|
||
|
|
||
|
|
||
|
```python
|
||
|
_input = prompt.format(subject="ice cream flavors")
|
||
|
output = model(_input)
|
||
|
```
|
||
|
|
||
|
|
||
|
```python
|
||
|
output_parser.parse(output)
|
||
|
```
|
||
|
|
||
|
<CodeOutputBlock lang="python">
|
||
|
|
||
|
```
|
||
|
['Vanilla',
|
||
|
'Chocolate',
|
||
|
'Strawberry',
|
||
|
'Mint Chocolate Chip',
|
||
|
'Cookies and Cream']
|
||
|
```
|
||
|
|
||
|
</CodeOutputBlock>
|