move wolfram out of its own file

This commit is contained in:
Luke Robles 2022-10-26 08:37:08 -07:00
parent a79c2409ab
commit a1a2687da8
2 changed files with 7 additions and 45 deletions

View File

@ -83,12 +83,15 @@ class ActualUtils(commands.Cog):
description="The query you want to pass to wolfram alpha",
)
async def wolfram(self, ctx, query):
import questions
import wolframalpha
client = wolframalpha.Client(os.getenv("wolfram_token"))
await ctx.defer()
await ctx.send_followup(
questions.answer_question(query),
)
try:
res = client.query(query)
return await ctx.send_followup(next(res.results).text)
except Exception:
return await ctx.send_followup("Sorry, I'm unable to answer that")
@commands.has_role("Track day gamers")
@commands.slash_command(

View File

@ -1,41 +0,0 @@
import os
import wolframalpha
import help_methods
def answer_question(message):
"""
answer_question(message)
Submits a request to the wolfram API and returns the response
If no answer is found, tries wikipedia. If that fails, apologizes
"""
client = wolframalpha.Client(os.getenv("wolfram_token"))
try:
res = client.query(message)
return next(res.results).text
except Exception:
return "Sorry, I'm unable to answer that"
return help_methods.get_help_message("message")
def open_ai(prompt):
import os
import openai
openai.api_key = os.getenv("OPENAI_API_KEY")
response = openai.Completion.create(
model="text-ada-001",
prompt=prompt,
temperature=0.7,
max_tokens=256,
top_p=1,
frequency_penalty=0,
presence_penalty=0,
)["choices"][0]["text"]
return response