mirror of
https://github.com/RYDE-WORK/Langchain-Chatchat.git
synced 2026-01-19 21:37:20 +08:00
2c8fc95f7a
* 更新上agent提示词代码 * 更新部分文档，修复了issue中提到的bge匹配超过1 的bug * 按需修改 * 解决了部分最新用户用依赖的bug，加了两个工具，移除google工具 * Agent大幅度优化 1. 修改了UI界面 (1)高亮所有没有进行agent对齐的模型， (2)优化输出体验和逻辑，使用markdown 2. 降低天气工具使用门槛 3. 依赖更新 (1) vllm 更新到0.2.0，增加了一些参数 (2) torch 建议更新到2.1 (3)pydantic不要更新到1.10.12
79 lines
1.6 KiB
Python
79 lines
1.6 KiB
Python
## 单独运行的时候需要添加
|
||
import sys
|
||
import os
|
||
sys.path.append(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))))
|
||
|
||
from langchain.prompts import PromptTemplate
|
||
from langchain.chains import LLMMathChain
|
||
from server.utils import get_ChatOpenAI
|
||
from configs.model_config import LLM_MODEL, TEMPERATURE
|
||
_PROMPT_TEMPLATE = """
|
||
将数学问题翻译成可以使用Python的numexpr库执行的表达式。使用运行此代码的输出来回答问题。
|
||
问题: ${{包含数学问题的问题。}}
|
||
```text
|
||
${{解决问题的单行数学表达式}}
|
||
```
|
||
...numexpr.evaluate(query)...
|
||
```output
|
||
${{运行代码的输出}}
|
||
```
|
||
答案: ${{答案}}
|
||
|
||
这是两个例子：
|
||
|
||
问题: 37593 * 67是多少？
|
||
```text
|
||
37593 * 67
|
||
```
|
||
...numexpr.evaluate("37593 * 67")...
|
||
```output
|
||
2518731
|
||
|
||
答案: 2518731
|
||
|
||
问题: 37593的五次方根是多少？
|
||
```text
|
||
37593**(1/5)
|
||
```
|
||
...numexpr.evaluate("37593**(1/5)")...
|
||
```output
|
||
8.222831614237718
|
||
|
||
答案: 8.222831614237718
|
||
|
||
|
||
问题: 2的平方是多少？
|
||
```text
|
||
2 ** 2
|
||
```
|
||
...numexpr.evaluate("2 ** 2")...
|
||
```output
|
||
4
|
||
|
||
答案: 4
|
||
|
||
|
||
现在，这是我的问题：
|
||
问题: {question}
|
||
"""
|
||
PROMPT = PromptTemplate(
|
||
input_variables=["question"],
|
||
template=_PROMPT_TEMPLATE,
|
||
)
|
||
|
||
|
||
def calculate(query: str):
|
||
model = get_ChatOpenAI(
|
||
streaming=False,
|
||
model_name=LLM_MODEL,
|
||
temperature=TEMPERATURE,
|
||
)
|
||
llm_math = LLMMathChain.from_llm(model, verbose=True, prompt=PROMPT)
|
||
ans = llm_math.run(query)
|
||
return ans
|
||
|
||
if __name__ == "__main__":
|
||
result = calculate("2的三次方")
|
||
print("答案:",result)
|
||
|