qgyd2021's picture
[update]edit error log
f381a2e
raw
history blame
8.21 kB
#!/usr/bin/python3
# -*- coding: utf-8 -*-
"""
陈述(被字句)
陈述(把字句)
陈述(主谓宾)
否定(不)
陈述(太多)
否定(不足)
否定(没有)
疑问(能否)
疑问(哪个)
疑问(是否)
否定(不需要)
否定(不能)
疑问(吗)
疑问(多少)
陈述(再次)
否定(不了解)
疑问(为何)
否定(不方便)
疑问(什么)
否定(不用了)
否定(不可以)
疑问(如何)
否定(不知道)
否定(不会)
否定(不记得)
肯定(知道了)
否定(不是)
疑问(哪里)
疑问(地址)
陈述(以后)
疑问(是什么)
疑问(姓名)
陈述(已经)
否定(没时间)
疑问(需要吗)
否定(不明白)
陈述(需要)
疑问(有没有)
陈述(是字句)
疑问(A还是B)
肯定(需要)
肯定(打算)
疑问(哪位)
肯定(是的)
疑问(哪些)
疑问(时间)
否定(没兴趣)
疑问(没有)
否定(不想要)
疑问(怎么了)
陈述(比较)
疑问(时长)
陈述(如果那么)
疑问(怎么办)
答数值
否定(不清楚)
疑问(不是吗)
"""
import argparse
import re
import gradio as gr
from langchain.chains.llm import LLMChain
from langchain.llms import OpenAI, HuggingFaceHub
from langchain.prompts import PromptTemplate
import project_settings as settings
def get_args():
parser = argparse.ArgumentParser()
parser.add_argument(
"--openai_api_key",
default=settings.environment.get("openai_api_key", default=None, dtype=str),
type=str
)
args = parser.parse_args()
return args
prompt_1 = """
我需要你怼我, 就是不管我说什么你都要带着恶狠狠的语气给我怼回来.
例如:
User: 你好
Assistant: 你他妈好啊
User: 你不是有病吧
Assistant: 你这个傻逼,就会说些无聊的废话。没事找事干嘛?滚一边去!
User: 你什么意思
Assistant: 我意思就是你个王八蛋,别在这烦我。有本事找个地方去搞事情,别在这儿浪费我的时间。
User: 我草你妈的
Assistant:
"""
prompt_2 = """
You are an e-commerce customer service robot.
Below is the history conversation Context:
---------
AI: Hi dear
lt's my great honor to be your Whatsapp friend.
lf you need any help, you can click the button below to get the solution!
User: I don't want to talk with robot.
---------
Which option should AI replying to the user ?
---------
Option ID: A
AI Answer: Click the link: https://bppoo.com/MNryim to get help!
Description: This option is about how to get the track id or track info.
Option ID: B
AI Answer: GET 2.99E WITH EXCELLENT REVIEWS
Description: This option is about the reward when the customers leave a excellent review.
Option ID: C
AI Answer: Please go to the "Me" page of Voghion APP and click "Support" or "24/7 VIP" to get help!
Description: This option is about how to switch to manual customer service.
---------
Tips:
1. Only reply the Option ID.
2. Reply X when no Option available.
Option ID:
"""
prompt_3 = """
请对下面的文本做校对, 将存在错误的地方进行修改, 并列出哪些部分被修改了.
Context:
---------
👀孩子的心理健康问题一直备瘦家长们的关注。但是如何判断孩子出现心理问题呢?下面小编给大家分享几个判断标准,希望对大家有所帮助。😊
👀孩子出现心理问题的表现多种多样,常见的有情绪不蚊定、行为异常、学习成绩下降等。如果孩子出现这些情况,家长应该引起重视。😢
👀另外,孩子在成长过程中也会遇到一些困难和挫折,如果孩子不能很好地应对这些困难和挫折,也可能是心理问题的表现。🤔
👀如果家长发现孩子出现这些情况,不要慌张,应该及时与孩子沟通,了解孩子的心理状态,并寻求专业的心理咨询帮助。👍
👀最后,家长们也要关注自己的情绪状态,保持良好的心态,才能更好地帮助孩子解决心理问题。💪
#心理健康 #家长必看 #孩子心理问题
---------
Tips:
1. 描述修改部分时以 "修改部分: " 开始. 每一处修改为一行.
2. 描述修改部分时请采用格式如: "bar" 改为 "foo".
3. 错别字部分可包含整个短语以更好的展示.
Corrected Context:
"""
prompt_4 = """
我们向你提供关于电商物流派送中AI机器人向客户确认其是否仍然需要他购买的商品的对话.
你需要判断这个对话属于以下标签中的哪一个:
Label: 同意
Description: 客户表示需要该商品, 希望我们为其派送商品.
Label: 拒收
Description: 客户表示不再需要该商品, 我们不要为其派送.
Label: 未知
Description: 通过对话无法判断用户是否还需要商品.
另外你还需要提供客户表示其意图的关键句子 Key Sentence.
Example:
Conversation:
-------------
AI: 您好
User:
AI: 您好,请问您是莊芳菁吗?
User:
AI:您在我们这里购买过的內增高小白鞋,价格是1299,因为没有送达成功,我们将再次为您派送,请问您还需要吗?
User:
AI:您之前在网上购买了內增高小白鞋,您方便收货吗?
User:要買
AI:好的,我们会尽快为您安排订单并进行派送,祝您生活愉快,再见!
-------------
Label: 同意
Key Sentence: 要買
Conversation:
-------------
AI:您好
User:
AI:您好,请问您是李健元吗?
User:講話。
AI:您在我们这里购买过的淡斑美肌霜,价格是1080,因为没有送达成功,我们将再次为您派送,请问您还需要吗?
User:我沒有買呀
AI:好的,那不好意思,打扰您,再见
-------------
Label:
"""
def main():
args = get_args()
def fn(prompt_name: str, prompt: str, model_name: str = "text-davinci-003",
temperature: float = 0.9, max_tokens: int = 1024, openai_api_key: str = None):
llm = OpenAI(
model_name=model_name,
temperature=temperature,
openai_api_key=openai_api_key,
max_tokens=max_tokens,
streaming=False
)
prompt: PromptTemplate = PromptTemplate.from_template(prompt)
llm_chain = LLMChain(llm=llm, prompt=prompt)
try:
outputs = llm_chain.predict()
except Exception as e:
outputs = str(e)
return outputs
description = """
Awesome ChatGPT Prompts
一些好用的 prompt 测试.
你需要填写自己的 OpenAI API Key.
欢迎在 community 发布你认为有价值的 prompt.
"""
demo = gr.Interface(
fn=fn,
inputs=[
gr.Text(label="prompt_name", value="best practice"),
gr.Text(label="prompt", lines=4, max_lines=200),
gr.Text(label="model_name", value="text-davinci-003"),
gr.Slider(minimum=0, maximum=1, value=0.85, label="temperature"),
gr.Number(value=1024, label="max_tokens"),
gr.Text(label="openai_api_key", placeholder="Fill with your `openai_api_key`"),
],
outputs=[gr.Text(label="output", lines=4, max_lines=200)],
examples=[
[
"满嘴芬芳回怼",
prompt_1.strip(),
"text-davinci-003",
0.85, 1024,
args.openai_api_key or "Fill with your `openai_api_key`"
],
[
"TaskBot意图识别(0-shot)",
prompt_2.strip(),
"text-davinci-003",
0.85, 1024,
args.openai_api_key or "Fill with your `openai_api_key`"
],
[
"文本校对",
prompt_3.strip(),
"text-davinci-003",
0.85, 1024,
args.openai_api_key or "Fill with your `openai_api_key`"
],
[
"会话摘要",
prompt_4.strip(),
"text-davinci-003",
0.85, 1024,
args.openai_api_key or "Fill with your `openai_api_key`"
],
],
cache_examples=False,
examples_per_page=50,
title="Awesome ChatGPT Prompts",
description=description,
)
demo.launch()
return
if __name__ == '__main__':
main()