File size: 997 Bytes
c25b046 7bd9861 c25b046 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 | 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() |