AgentFlow / app.py
xaochNYU's picture
Upload app.py
7bd9861 verified
raw
history blame contribute delete
997 Bytes
import chainlit as cl
import os
from langflow.load import load_flow_from_json
from langflow.processing.process import run_graph
google_api_key=os.environ.get("GOOGLE_API_KEY")
composio_api_key=os.environ.get("COMPOSIO_API_KEY")
TWEAKS = {
"ChatInput-2x8vF": {},
"Prompt-CLQNW": {},
"Agent-6ilGB": {"api_key":google_api_key},
"Agent-e7sUF": {"api_key":google_api_key},
"Agent-womuY": {"api_key":google_api_key},
"ComposioAPI-RfpkY": {"api_key":composio_api_key},
"Agent-g9eV8": {"api_key":google_api_key},
"ChatOutput-IIhYR": {},
"Prompt-Od4Ay": {},
"Agent-CjuzA": {"api_key":google_api_key}
}
@cl.on_message
async def main(message: cl.Message):
flow = load_flow_from_json("AgentFlow2.json",tweaks=TWEAKS)
flow.session_id="session_1"
flow.user_id="user_1"
response = await run_graph(flow, input_value=message.content, input_type="chat", output_type="chat")
await cl.Message(response[0].outputs[0].results["message"].text).send()