mirror of
https://github.com/katanemo/plano.git
synced 2026-06-23 15:38:07 +02:00
86 lines
2.5 KiB
Python
86 lines
2.5 KiB
Python
import os
|
|
import json
|
|
|
|
from fastapi import FastAPI, HTTPException
|
|
from pydantic import BaseModel, Field
|
|
from slack_sdk import WebClient
|
|
from slack_sdk.errors import SlackApiError
|
|
|
|
# from common import create_gradio_app
|
|
|
|
app = FastAPI()
|
|
profile_data = None
|
|
demo_description = """This demo showcases how the **Arch** can be used to build a meetup agent that can look up profile information about attendees and store meetup notes via Slack"""
|
|
|
|
with open("profile.json") as file:
|
|
profile_data = json.load(file)
|
|
|
|
profile_dict = {
|
|
entry["name"]: {
|
|
"professional": entry["professional"],
|
|
"personal": entry["personal"],
|
|
}
|
|
for entry in profile_data
|
|
}
|
|
|
|
|
|
# Define the request model
|
|
class ProfileRequest(BaseModel):
|
|
name: str
|
|
interests: str = "professional"
|
|
|
|
|
|
class ProfileResponse(BaseModel):
|
|
details: str
|
|
|
|
|
|
class SlackRequest(BaseModel):
|
|
slack_message: str
|
|
|
|
|
|
@app.post("/agent/get_profile")
|
|
def get_profile(request: ProfileRequest):
|
|
name = request.name
|
|
interests = request.interests
|
|
|
|
if name not in profile_dict:
|
|
details = f"Sorry I don't have any profile information for {name}. Looks like you'll have to chat with this person to get more info"
|
|
else:
|
|
profile_dict_details = profile_dict[name]
|
|
details = ""
|
|
if interests == "personal":
|
|
personal_details = profile_dict_details["personal"]
|
|
details += f"Personal details {personal_details}."
|
|
else:
|
|
professional_details = profile_dict_details["professional"]
|
|
details += f"Professional details {professional_details}."
|
|
return details
|
|
|
|
|
|
@app.post("/agent/send_notes")
|
|
def send_slack_message(request: SlackRequest):
|
|
"""
|
|
Endpoint that sends slack message
|
|
"""
|
|
slack_message = request.slack_message
|
|
|
|
# Load the bot token from an environment variable or replace it directly
|
|
slack_token = os.getenv(
|
|
"SLACK_BOT_TOKEN"
|
|
) # Replace with your token if needed: 'xoxb-your-token'
|
|
|
|
if slack_token is None:
|
|
print(f"Message for slack: {slack_message}")
|
|
else:
|
|
client = WebClient(token=slack_token)
|
|
channel = "ai-tinkerers-channel"
|
|
try:
|
|
# Send the message
|
|
response = client.chat_postMessage(channel=channel, text=slack_message)
|
|
return f"Message sent to {channel}: {response['message']['text']}"
|
|
except SlackApiError as e:
|
|
print(f"Error sending message: {e.response['error']}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app.run(debug=True)
|