add wiki stuff

This commit is contained in:
crate 2025-05-22 17:09:48 -05:00
parent 3945eb6763
commit ffd642fd2a
3 changed files with 16 additions and 1 deletions

View File

@ -7,7 +7,7 @@ from garfpy import(
logger, is_private,
kroger_token, find_store, search_product,
picture_time, process_image_requests, generate_chat,
aod_message)
aod_message, wiki)
gapikey = config.GIF_TOKEN
@ -39,6 +39,7 @@ async def on_message(message):
if message.content.lower().startswith("hey garfield") or isinstance(message.channel, discord.DMChannel):
question = message.content[12:] if message.content.lower().startswith("hey garfield") else message.content
answer = await generate_chat(question)
logger.info(f"Chat Request - User: {user}, Server: {server}, Prompt: {question}")
await message.channel.send(answer)
if message.content.lower().startswith('garfpic '):
@ -49,6 +50,11 @@ async def on_message(message):
await message.channel.send(f"`Please wait... image generation queued: {prompt}`")
await picture_time(message, prompt)
if message.content.lower().startswith('wiki '):
search_term = message.content[5:]
summary = await wikisum(search_term)
await message.channel.send(summary)
if message.content.lower().startswith("garfping "):
try:
query = message.content.split()

View File

@ -10,3 +10,4 @@ from .garfai import(
)
from .iputils import is_private
from .aod import aod_message
from wiki import wikisum

View File

@ -1 +1,9 @@
import wikipedia
async def wikisum(search_term):
try:
summary = wikipedia.summary(search_term)
return summary
except wikipedia.exceptions.DisambiguationError as e:
return e