mirror of
https://github.com/kennethreitz/simplemind.git
synced 2026-06-05 14:50:16 +00:00
Barış Özmen
29 lines
1010 B
Python
29 lines
1010 B
Python
from _context import sm
|
|
|
|
|
|
class MathPlugin(sm.BasePlugin):
|
|
def pre_send_hook(self, conversation: sm.Conversation):
|
|
last_user_message = conversation.get_last_message(role="user")
|
|
if last_user_message is None:
|
|
return
|
|
if "calculate" in last_user_message.text.lower():
|
|
expression = last_user_message.text.lower().replace("calculate", "").strip()
|
|
try:
|
|
result = eval(expression)
|
|
conversation.add_message(
|
|
role="assistant", text=f"The result is {result}."
|
|
)
|
|
except Exception:
|
|
conversation.add_message(
|
|
role="assistant",
|
|
text="I'm sorry, I couldn't compute that expression. Please try again.",
|
|
)
|
|
|
|
|
|
conversation = sm.create_conversation(llm_model="gpt-4o", llm_provider="openai")
|
|
conversation.add_plugin(MathPlugin())
|
|
|
|
conversation.add_message("user", "Calculate 2 + 2 * 3")
|
|
|
|
print(conversation.send())
|