远兮 commited on
Commit
a9098a9
·
1 Parent(s): 9acaed2

add CommaSeparatedListOutputParser

Browse files
Files changed (1) hide show
  1. parser_list_output.ipynb +100 -0
parser_list_output.ipynb ADDED
@@ -0,0 +1,100 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "cells": [
3
+ {
4
+ "cell_type": "code",
5
+ "execution_count": 6,
6
+ "metadata": {},
7
+ "outputs": [],
8
+ "source": [
9
+ "from langchain.output_parsers import CommaSeparatedListOutputParser\n",
10
+ "from langchain.prompts import PromptTemplate, ChatPromptTemplate, HumanMessagePromptTemplate\n",
11
+ "from langchain.llms import OpenAI\n",
12
+ "from langchain.chat_models import ChatOpenAI"
13
+ ]
14
+ },
15
+ {
16
+ "cell_type": "code",
17
+ "execution_count": 7,
18
+ "metadata": {},
19
+ "outputs": [],
20
+ "source": [
21
+ "output_parser = CommaSeparatedListOutputParser()"
22
+ ]
23
+ },
24
+ {
25
+ "cell_type": "code",
26
+ "execution_count": 8,
27
+ "metadata": {},
28
+ "outputs": [],
29
+ "source": [
30
+ "format_instructions = output_parser.get_format_instructions()\n",
31
+ "prompt = PromptTemplate(\n",
32
+ " template=\"五个 {subject}.\\n{format_instructions}\",\n",
33
+ " input_variables=[\"subject\"],\n",
34
+ " partial_variables={\"format_instructions\": format_instructions}\n",
35
+ ")"
36
+ ]
37
+ },
38
+ {
39
+ "cell_type": "code",
40
+ "execution_count": 9,
41
+ "metadata": {},
42
+ "outputs": [],
43
+ "source": [
44
+ "model = OpenAI(temperature=0)"
45
+ ]
46
+ },
47
+ {
48
+ "cell_type": "code",
49
+ "execution_count": 10,
50
+ "metadata": {},
51
+ "outputs": [],
52
+ "source": [
53
+ "_input = prompt.format(subject=\"编程语言\")\n",
54
+ "output = model(_input)"
55
+ ]
56
+ },
57
+ {
58
+ "cell_type": "code",
59
+ "execution_count": 11,
60
+ "metadata": {},
61
+ "outputs": [
62
+ {
63
+ "data": {
64
+ "text/plain": [
65
+ "['Java', 'C', 'C++', 'Python', 'JavaScript']"
66
+ ]
67
+ },
68
+ "execution_count": 11,
69
+ "metadata": {},
70
+ "output_type": "execute_result"
71
+ }
72
+ ],
73
+ "source": [
74
+ "output_parser.parse(output)"
75
+ ]
76
+ }
77
+ ],
78
+ "metadata": {
79
+ "kernelspec": {
80
+ "display_name": "base",
81
+ "language": "python",
82
+ "name": "python3"
83
+ },
84
+ "language_info": {
85
+ "codemirror_mode": {
86
+ "name": "ipython",
87
+ "version": 3
88
+ },
89
+ "file_extension": ".py",
90
+ "mimetype": "text/x-python",
91
+ "name": "python",
92
+ "nbconvert_exporter": "python",
93
+ "pygments_lexer": "ipython3",
94
+ "version": "3.10.10"
95
+ },
96
+ "orig_nbformat": 4
97
+ },
98
+ "nbformat": 4,
99
+ "nbformat_minor": 2
100
+ }