diff --git a/requirements.txt b/requirements.txt index 766b000..d371f86 100644 --- a/requirements.txt +++ b/requirements.txt @@ -3,3 +3,4 @@ streamlit openai requests flask_swagger_ui +expr.py \ No newline at end of file diff --git a/slop_with_models.py b/slop_with_models.py index cbe14c1..30d8de1 100644 --- a/slop_with_models.py +++ b/slop_with_models.py @@ -2,6 +2,7 @@ from flask import Flask, request, jsonify from datetime import datetime import os import json +import expr from threading import Lock from openai import OpenAI import logging @@ -118,7 +119,7 @@ tools = { "arguments": [ { "name": "expression", "type": "str", "description": "the math expression to evaluate."} ], - "execute": lambda params: {"result": eval(params["expression"])}, + "execute": lambda params: {"result": expr(params["expression"])}, }, "greet": { "id": "greet",