Delete app.py
Browse files
app.py
DELETED
@@ -1,120 +0,0 @@
|
|
1 |
-
##########################################################################
|
2 |
-
#
|
3 |
-
#
|
4 |
-
# Waiting on https://github.com/microsoft/autogen/issues/527 to be solved
|
5 |
-
#
|
6 |
-
#
|
7 |
-
##########################################################################
|
8 |
-
|
9 |
-
|
10 |
-
from typing import Dict, Optional, Union
|
11 |
-
|
12 |
-
from autogen import Agent, AssistantAgent, UserProxyAgent, config_list_from_json
|
13 |
-
import chainlit as cl
|
14 |
-
|
15 |
-
from dotenv import load_dotenv
|
16 |
-
import os
|
17 |
-
|
18 |
-
# Load environment variables from .env file
|
19 |
-
load_dotenv()
|
20 |
-
TASK = "Plot a chart of NVDA stock price change YTD and save it on disk."
|
21 |
-
|
22 |
-
|
23 |
-
async def ask_helper(func, **kwargs):
|
24 |
-
res = await func(**kwargs).send()
|
25 |
-
while not res:
|
26 |
-
res = await func(**kwargs).send()
|
27 |
-
return res
|
28 |
-
|
29 |
-
|
30 |
-
class ChainlitAssistantAgent(AssistantAgent):
|
31 |
-
async def a_send(
|
32 |
-
self,
|
33 |
-
message: Union[Dict, str],
|
34 |
-
recipient: Agent,
|
35 |
-
request_reply: Optional[bool] = None,
|
36 |
-
silent: Optional[bool] = False,
|
37 |
-
) -> bool:
|
38 |
-
await cl.Message(
|
39 |
-
content=f'*Sending message to "{recipient.name}":*\n\n{message}',
|
40 |
-
author="AssistantAgent",
|
41 |
-
).send()
|
42 |
-
await super(ChainlitAssistantAgent, self).a_send(
|
43 |
-
message=message,
|
44 |
-
recipient=recipient,
|
45 |
-
request_reply=request_reply,
|
46 |
-
silent=silent,
|
47 |
-
)
|
48 |
-
|
49 |
-
|
50 |
-
class ChainlitUserProxyAgent(UserProxyAgent):
|
51 |
-
async def get_human_input(self, prompt: str) -> str:
|
52 |
-
if prompt.startswith(
|
53 |
-
"Provide feedback to assistant. Press enter to skip and use auto-reply"
|
54 |
-
):
|
55 |
-
res = await ask_helper(
|
56 |
-
cl.AskActionMessage,
|
57 |
-
content="Continue or provide feedback?",
|
58 |
-
actions=[
|
59 |
-
cl.Action(
|
60 |
-
name="continue", value="continue", label="β
Continue"
|
61 |
-
),
|
62 |
-
cl.Action(
|
63 |
-
name="feedback",
|
64 |
-
value="feedback",
|
65 |
-
label="π¬ Provide feedback",
|
66 |
-
),
|
67 |
-
cl.Action(
|
68 |
-
name="exit",
|
69 |
-
value="exit",
|
70 |
-
label="π Exit Conversation"
|
71 |
-
),
|
72 |
-
],
|
73 |
-
)
|
74 |
-
if res.get("value") == "continue":
|
75 |
-
return ""
|
76 |
-
if res.get("value") == "exit":
|
77 |
-
return "exit"
|
78 |
-
|
79 |
-
reply = await ask_helper(
|
80 |
-
cl.AskUserMessage, content=prompt, timeout=60)
|
81 |
-
|
82 |
-
return reply["content"].strip()
|
83 |
-
|
84 |
-
async def a_send(
|
85 |
-
self,
|
86 |
-
message: Union[Dict, str],
|
87 |
-
recipient: Agent,
|
88 |
-
request_reply: Optional[bool] = None,
|
89 |
-
silent: Optional[bool] = False,
|
90 |
-
):
|
91 |
-
await cl.Message(
|
92 |
-
content=f'*Sending message to "{recipient.name}"*:\n\n{message}',
|
93 |
-
author="UserProxyAgent",
|
94 |
-
).send()
|
95 |
-
await super(ChainlitUserProxyAgent, self).a_send(
|
96 |
-
message=message,
|
97 |
-
recipient=recipient,
|
98 |
-
request_reply=request_reply,
|
99 |
-
silent=silent,
|
100 |
-
)
|
101 |
-
|
102 |
-
|
103 |
-
@cl.on_chat_start
|
104 |
-
async def on_chat_start():
|
105 |
-
config_list = config_list_from_json(env_or_file="OAI_CONFIG_LIST")
|
106 |
-
assistant = ChainlitAssistantAgent(
|
107 |
-
"assistant", llm_config={"config_list": config_list}
|
108 |
-
)
|
109 |
-
user_proxy = ChainlitUserProxyAgent(
|
110 |
-
"user_proxy",
|
111 |
-
code_execution_config={
|
112 |
-
"work_dir": "workspace",
|
113 |
-
"use_docker": False,
|
114 |
-
},
|
115 |
-
)
|
116 |
-
await cl.Message(content=f"Starting agents on task: {TASK}...").send()
|
117 |
-
await user_proxy.a_initiate_chat(
|
118 |
-
assistant,
|
119 |
-
message=TASK,
|
120 |
-
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|