Upload 2 files

app.py

@@ -0,0 +1,69 @@
+import streamlit as st
+import asyncio
+from autogen import AssistantAgent, UserProxyAgent
+
+
+st.write("""# AutoGen Chat Agents""")
+
+class TrackableAssistantAgent(AssistantAgent):
+    def _process_received_message(self, message, sender, silent):
+        with st.chat_message(sender.name):
+            st.markdown(message)
+        return super()._process_received_message(message, sender, silent)
+
+
+class TrackableUserProxyAgent(UserProxyAgent):
+    def _process_received_message(self, message, sender, silent):
+        with st.chat_message(sender.name):
+            st.markdown(message)
+        return super()._process_received_message(message, sender, silent)
+
+
+selected_model = None
+selected_key = None
+with st.sidebar:
+    st.header("OpenAI Configuration")
+    selected_model = st.selectbox("Model", ['gpt-3.5-turbo', 'gpt-4'], index=1)
+    selected_key = st.text_input("API Key", type="password")
+
+with st.container():
+    # for message in st.session_state["messages"]:
+    #    st.markdown(message)
+
+    user_input = st.chat_input("Type something...")
+    if user_input:
+        if not selected_key or not selected_model:
+            st.warning(
+                'You must provide valid OpenAI API key and choose preferred model', icon="⚠️")
+            st.stop()
+
+        llm_config = {
+            "request_timeout": 600,
+            "config_list": [
+                {
+                    "model": selected_model,
+                    "api_key": selected_key
+                }
+            ]
+        }
+        # create an AssistantAgent instance named "assistant"
+        assistant = TrackableAssistantAgent(
+            name="assistant", llm_config=llm_config)
+
+        # create a UserProxyAgent instance named "user"
+        user_proxy = TrackableUserProxyAgent(
+            name="user", human_input_mode="NEVER", llm_config=llm_config)
+
+        # Create an event loop
+        loop = asyncio.new_event_loop()
+        asyncio.set_event_loop(loop)
+
+        # Define an asynchronous function
+        async def initiate_chat():
+            await user_proxy.a_initiate_chat(
+                assistant,
+                message=user_input,
+            )
+
+        # Run the asynchronous function within the event loop
+        loop.run_until_complete(initiate_chat())

requirements.txt

@@ -0,0 +1,59 @@
+aiohttp==3.8.6
+aiosignal==1.3.1
+altair==5.1.2
+async-timeout==4.0.3
+attrs==23.1.0
+blinker==1.6.3
+cachetools==5.3.2
+certifi==2023.7.22
+charset-normalizer==3.3.1
+click==8.1.7
+diskcache==5.6.3
+docker==6.1.3
+FLAML==2.1.1
+frozenlist==1.4.0
+gitdb==4.0.11
+GitPython==3.1.40
+idna==3.4
+importlib-metadata==6.8.0
+Jinja2==3.1.2
+jsonschema==4.19.1
+jsonschema-specifications==2023.7.1
+markdown-it-py==3.0.0
+MarkupSafe==2.1.3
+mdurl==0.1.2
+multidict==6.0.4
+numpy==1.26.1
+openai==0.28.1
+packaging==23.2
+pandas==2.1.2
+Pillow==10.1.0
+protobuf==4.24.4
+pyarrow==13.0.0
+pyautogen==0.1.13
+pydeck==0.8.1b0
+Pygments==2.16.1
+python-dateutil==2.8.2
+python-dotenv==1.0.0
+pytz==2023.3.post1
+referencing==0.30.2
+requests==2.31.0
+rich==13.6.0
+rpds-py==0.10.6
+six==1.16.0
+smmap==5.0.1
+streamlit==1.28.0
+tenacity==8.2.3
+termcolor==2.3.0
+toml==0.10.2
+toolz==0.12.0
+tornado==6.3.3
+tqdm==4.66.1
+typing_extensions==4.8.0
+tzdata==2023.3
+tzlocal==5.2
+urllib3==2.0.7
+validators==0.22.0
+websocket-client==1.6.4
+yarl==1.9.2
+zipp==3.17.0