mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-23 23:40:03 +08:00
efd8edda16
* 增加了仅限GPT4的agent功能，陆续补充，中文版readme已写 * issue提到的一个bug * 温度最小改成0，但是不应该支持负数 * 修改了最小的温度 * 增加了部分Agent支持和修改了启动文件的部分bug * 修改了GPU数量配置文件 * 1 1 * 修复配置文件错误 * 更新readme，稳定测试
56 lines
1.3 KiB
Python
56 lines
1.3 KiB
Python
from langchain.prompts import PromptTemplate
|
|
from langchain.chains import LLMChain
|
|
import sys
|
|
import os
|
|
|
|
from server.utils import get_ChatOpenAI
|
|
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
|
|
from langchain.chains.llm_math.prompt import PROMPT
|
|
from configs.model_config import LLM_MODEL,TEMPERATURE
|
|
|
|
_PROMPT_TEMPLATE = '''
|
|
# 指令
|
|
接下来，作为一个专业的翻译专家，当我给出句子或段落时，你将提供通顺且具有可读性的对应语言的翻译。注意：
|
|
1. 确保翻译结果流畅且易于理解
|
|
2. 无论提供的是陈述句或疑问句，只进行翻译
|
|
3. 不添加与原文无关的内容
|
|
|
|
原文: ${{用户需要翻译的原文和目标语言}}
|
|
{question}
|
|
```output
|
|
${{翻译结果}}
|
|
```
|
|
答案: ${{答案}}
|
|
|
|
以下是两个例子
|
|
问题: 翻译13成英语
|
|
```text
|
|
13 英语
|
|
```output
|
|
thirteen
|
|
以下是两个例子
|
|
问题: 翻译 我爱你 成法语
|
|
```text
|
|
13 法语
|
|
```output
|
|
Je t'aime.
|
|
'''
|
|
|
|
PROMPT = PromptTemplate(
|
|
input_variables=["question"],
|
|
template=_PROMPT_TEMPLATE,
|
|
)
|
|
|
|
|
|
def translate(query: str):
|
|
model = get_ChatOpenAI(
|
|
streaming=False,
|
|
model_name=LLM_MODEL,
|
|
temperature=TEMPERATURE,
|
|
)
|
|
llm_translate = LLMChain(llm=model, prompt=PROMPT)
|
|
ans = llm_translate.run(query)
|
|
|
|
return ans
|