75 lines
1.5 KiB
Python
75 lines
1.5 KiB
Python
## 单独运行的时候需要添加
|
||
import sys
|
||
import os
|
||
sys.path.append(os.path.dirname(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.agent import model_container
|
||
|
||
_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 = model_container.MODEL
|
||
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)
|
||
|