plusnarrative-phillip
commented
11 months ago ``from flask import Flask, request, jsonify from flask_socketio import SocketIO, emit import logging from autogen import AssistantAgent, UserProxyAgent, config_list_from_json import openai import autogen from flask_cors import CORS import json from datetime import datetime
app = Flask(name) CORS(app)
logging.basicConfig(level=logging.INFO) socketio = SocketIO(app, cors_allowed_origins="*")
openai.api_key = "" config_list = config_list_from_json(env_or_file="OAI_CONFIG_LIST") llm_config = {"config_list": config_list, "seed": 42}
strategist = autogen.AssistantAgent( name="strategist", system_message="""Strategist. With vast expertise in .""", llm_config=llm_config )
innovator = autogen.AssistantAgent( name="innovator", system_message="""Innovator. """, llm_config=llm_config )
critic = autogen.AssistantAgent( name="critic", system_message="""Critic. You.""", llm_config=llm_config )
user_proxy = UserProxyAgent("user_proxy") groupchat = autogen.GroupChat(agents=[user_proxy, strategist, innovator, critic], messages=[], max_round=20) manager = autogen.GroupChatManager(groupchat=groupchat, llm_config=llm_config) connected_clients = 0
@socketio.on('connect') def handle_connect(): global connected_clients connected_clients += 1 logging.info(f'Client connected. Total connected clients: {connected_clients}')
# Emitting initial connection message with timestamp
logging.info(f"[{datetime.now()}] Sending initial connection message to frontend")
emit('status', {'timestamp': str(datetime.now()), 'message': 'Connected to the server!'})
# Emitting test message with timestamp
logging.info(f"[{datetime.now()}] Sending test message to frontend")
emit('chat', {'timestamp': str(datetime.now()), 'message': 'This is a test message from the backend.'}, broadcast=True)@socketio.on('disconnect') def handle_disconnect(): global connected_clients connected_clients -= 1 logging.info(f"[{datetime.now()}] Client disconnected. Total connected clients: {connected_clients}")
@app.route('/api/chat', methods=['GET', 'POST']) def chat(): try:
Log when a request is received
logging.info(f"[{datetime.now()}] Received a request at /api/chat.")
user_message = request.json.get('message', '')
# Log the user's message
logging.info(f"[{datetime.now()}] User message: {user_message}")
# Initialize a chat session with the user's message
user_proxy.initiate_chat(manager, message=user_message)
# Loop to keep processing feedback and sending all agent responses to the frontend
while True:
last_message = manager.groupchat.messages[-1]
if last_message.sender != user_proxy:
response_data = {
'sender': last_message.sender.name,
'text': last_message.text
}
# Log the OpenAI response
logging.info(f"[{datetime.now()}] OpenAI response: {response_data}")
# Emit the response data via WebSocket to frontend
logging.info(f"[{datetime.now()}] Sending OpenAI response to frontend")
socketio.emit('chat', {
'timestamp': str(datetime.now()),
'sender': response_data['sender'],
'text': response_data['text']
}, room=request.sid)
# If we hit a feedback request, auto-reply and continue the loop to get the next message.
if isinstance(last_message, autogen.FeedbackRequest):
logging.info(f"[{datetime.now()}] Auto-replying to feedback request.")
manager.reply_to_feedback_request(auto_reply=True)
else:
break
# Log when message processing is completed
logging.info(f"[{datetime.now()}] Message processing completed.")
return jsonify(status="Message processed."), 200
except Exception as e:
# Log any errors that occur during message processing
logging.error(f"Error processing the chat request: {e}")
return jsonify(error=str(e)), 500if name == 'main': socketio.run(app, debug=True)```
sonichi
victordibia
bamboriz
ChristianWeyer
khawajaJunaid
dsaha21
Tech Stack:
Backend: Python (Flask, Socket.IO, AutoGen, OpenAI) Frontend: React Working Parts:
Flask and Socket.IO initialization. Frontend sends messages to backend via /api/chat. User's message processed by user_proxy agent. Agents generate responses. Issues:
Feedback loop: Post-response, the system asks for feedback (likely from AutoGen's user_proxy.initiate_chat). Response Delivery: Backend sends responses via Socket.IO, but frontend doesn't display them. Questions:
How does user_proxy.initiate_chat work? Can we handle or bypass the feedback loop? How should frontend handle/display responses from Socket.IO? How to finalize or confirm agent responses to prevent further prompts? Brief Flow:
Flask, SocketIO, OpenAI initialized. Agents (strategist, innovator, critic, user_proxy) instantiated. On /api/chat, user message sent to user_proxy, agent responses collected, and sent to frontend via Socket.IO.