Upload 6 files

aworld/cmd/web_legacy/chat.py

@@ -0,0 +1,126 @@
+import asyncio
+import sys
+import streamlit as st
+from dotenv import load_dotenv
+import logging
+import os
+import traceback
+import importlib.util
+import utils
+import aworld.trace as trace
+from trace_net import generate_trace_graph_full
+from aworld.trace.base import get_tracer_provider
+
+load_dotenv(os.path.join(os.getcwd(), ".env"))
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+sys.path.insert(0, os.getcwd())
+
+
+def agent_page():
+    st.set_page_config(
+        page_title="AWorld Agent",
+        page_icon=":robot_face:",
+        layout="wide",
+    )
+
+    st.markdown(
+        """\
+        <style> 
+        .stAppHeader { display: none; }
+        
+        div[data-testid="stMarkdownContainer"] pre {
+            max-height: 300px;
+            overflow-y: auto;
+        }
+        div[data-testid="stMarkdownContainer"] img {
+            max-height: 500px;
+        }
+        </style>""",
+        unsafe_allow_html=True,
+    )
+
+    query_params = st.query_params
+    selected_agent_from_url = query_params.get("agent", None)
+
+    if "selected_agent" not in st.session_state:
+        st.session_state.selected_agent = selected_agent_from_url
+        logger.info(f"Initialized selected_agent from URL: {selected_agent_from_url}")
+
+    if selected_agent_from_url != st.session_state.selected_agent:
+        st.session_state.selected_agent = selected_agent_from_url
+
+    with st.sidebar:
+        st.title("AWorld Agents List")
+        for agent in utils.list_agents():
+            if st.button(agent):
+                st.query_params["agent"] = agent
+                st.session_state.selected_agent = agent
+                logger.info(f"selected_agent={st.session_state.selected_agent}")
+
+    if st.session_state.selected_agent:
+        agent_name = st.session_state.selected_agent
+        st.title(f"AWorld Agent: {agent_name}")
+
+        if prompt := st.chat_input("Input message here~"):
+
+            with st.chat_message("user"):
+                st.markdown(prompt)
+
+            with st.chat_message("assistant"):
+                agent_name = st.session_state.selected_agent
+                agent_package_path = utils.get_agent_package_path(agent_name)
+                agent_module_file = os.path.join(agent_package_path, "agent.py")
+                try:
+                    spec = importlib.util.spec_from_file_location(
+                        agent_name, agent_module_file
+                    )
+
+                    if spec is None or spec.loader is None:
+                        logger.error(
+                            f"Could not load spec for agent {agent_name} from {agent_module_file}"
+                        )
+                        st.error(f"Error: Could not load agent! {agent_name}")
+                        return
+
+                    agent_module = importlib.util.module_from_spec(spec)
+                    spec.loader.exec_module(agent_module)
+                except Exception as e:
+                    logger.error(
+                        f"Error loading agent {agent_name}, cwd:{os.getcwd()}, sys.path:{sys.path}: {traceback.format_exc()}"
+                    )
+                    st.error(f"Error: Could not load agent! {agent_name}")
+                    return
+
+                agent = agent_module.AWorldAgent()
+
+                async def markdown_generator():
+                    trace_id = None
+                    async with trace.span("start") as span:
+                        trace_id = span.get_trace_id()
+                        async for line in agent.run(prompt):
+                            st.write(line)
+                            await asyncio.sleep(0.1)
+
+                    get_tracer_provider().force_flush(5000)
+                    file_name = f"graph.{trace_id}.html"
+                    folder_name = "trace_data"
+                    generate_trace_graph_full(
+                        trace_id, folder_name=folder_name, file_name=file_name
+                    )
+                    view_page_url = f"/trace?trace_id={trace_id}"
+                    st.write(f"\n---\n[View Trace]({view_page_url})\n")
+
+                asyncio.run(markdown_generator())
+    else:
+        st.title("AWorld Agent Chat Assistant")
+        st.info("Please select an Agent from the left sidebar to start")
+
+
+try:
+    agent_page()
+except Exception as e:
+    logger.error(f">>> Error: {traceback.format_exc()}")
+    st.error(f"Error: {str(e)}")

aworld/cmd/web_legacy/main.py

@@ -0,0 +1,18 @@
+import streamlit as st
+from dotenv import load_dotenv
+import logging
+import os
+import aworld.trace as trace
+
+load_dotenv(os.path.join(os.getcwd(), ".env"))
+
+logging.basicConfig(level=logging.INFO)
+logger = logging.getLogger(__name__)
+
+trace.configure()
+
+chat = st.Page("chat.py", title="Chat", icon=":material/message:")
+trace = st.Page("trace.py", title="Trace")
+
+pg = st.navigation([chat, trace], position="hidden")
+pg.run()

aworld/cmd/web_legacy/trace.py

@@ -0,0 +1,50 @@
+import logging
+import traceback
+import streamlit as st
+import os
+
+logger = logging.getLogger(__name__)
+
+def view_page():
+    st.set_page_config(
+        page_title="HTML Viewer",
+        page_icon=":robot_face:",
+        layout="wide",
+    )
+
+    st.markdown(
+        "<style> .stAppHeader { display: none !important;} .stMainBlockContainer { padding: 5px 10px !important; } </style>",
+        unsafe_allow_html=True,
+    )
+
+    query_params = st.query_params
+    trace_id = query_params.get("trace_id", None)
+
+    side, main = st.columns([2, 8])
+
+    with side:
+        if st.button("Back To Chat"):
+            st.switch_page("chat.py")
+
+    with main:
+        if trace_id:
+            try:
+                st.header(f"Chat Trace Graph: {trace_id}")
+                folder_name = "trace_data"
+                file_name = f"graph.{trace_id}.html"
+                html_file_path = os.path.join(
+                    os.getcwd(), folder_name, file_name
+                )
+                with open(html_file_path, "r") as file:
+                    html_content = file.read()
+
+                st.components.v1.html(html_content, height=600, scrolling=True)
+            except Exception as e:
+                logger.error(f"Error: {traceback.format_exc()}")
+                st.write(f"Error: {traceback.format_exc()}")
+        else:
+            st.write("Parameter error!")
+
+
+if __name__ == "__main__":
+    view_page()

aworld/cmd/web_legacy/trace_net.py

@@ -0,0 +1,243 @@
+import requests
+import json
+import os
+from pyvis.network import Network
+from aworld.logs.util import logger
+from aworld.trace.base import get_tracer_provider_silent
+
+run_type_colors = {
+    "LLM": "#E6E6FA",
+    "TOOL": "#99FF99",   # green
+    "MCP": "#99FF99",
+    "AGENT": "#9999FF",  # blue
+    "OTHER": "#FFFF99"  # yellow
+}
+
+
+def get_span_color(span):
+    run_type = span.get('run_type', 'OTHER')
+    return run_type_colors.get(run_type, run_type_colors['OTHER'])
+
+
+def fetch_trace_data(trace_id=None):
+    try:
+        if trace_id:
+            response = requests.get(
+                f'http://localhost:7079/api/traces/{trace_id}')
+            response.raise_for_status()
+            return response.json() or {"root_span": []}
+    except requests.exceptions.RequestException as e:
+        logger.error(f"Error fetching trace data: {e}")
+        return {"root_span": []}
+
+
+def flatten_spans(node, result=None):
+    if result is None:
+        result = []
+    result.append(node)
+    for child in node.get('children', []):
+        flatten_spans(child, result)
+    return result
+
+
+def generate_trace_graph(trace_id=None):
+    tracer_provider = get_tracer_provider_silent()
+    if tracer_provider:
+        tracer_provider.force_flush(5000)
+    trace_data = fetch_trace_data(trace_id)
+    net = Network(height="200px", width="100%",
+                  notebook=False, cdn_resources='in_line')
+
+    net.set_options("""
+    {
+      "nodes": {
+        "font": {"size": 8}
+      },
+      "physics": {
+        "enabled": true,
+        "hierarchicalRepulsion": {
+          "centralGravity": 0.5,
+          "springLength": 150,
+          "nodeDistance": 120
+        }
+      },
+      "layout": {
+        "hierarchical": {
+          "enabled": true,
+          "direction": "LR",
+          "sortMethod": "directed",
+          "nodeSpacing": 50,
+          "levelSeparation": 100,
+          "blockShifting": true,
+          "edgeMinimization": true
+        }
+      }
+    }
+    """)
+    root_spans = trace_data['root_span']
+    all_spans = []
+    for span in root_spans:
+        all_spans.extend(flatten_spans(span))
+
+    spans = {span['span_id']: span for span in all_spans}
+    for span_id, span in spans.items():
+        title = f"{span['name']}\n({span['run_type']})"
+        net.add_node(span_id, label=title, title=title,
+                     shape='box', color=get_span_color(span))
+        if span.get('parent_id') and span['parent_id'] in spans:
+            net.add_edge(span['parent_id'], span_id, arrows='to')
+
+    net.show('trace_graph.html', notebook=False)
+    with open('trace_graph.html', 'a') as f:
+        f.write(f"""
+        <style>
+            #fullscreenBtn {{
+                position: absolute;
+                top: 10px;
+                right: 10px;
+                padding: 8px 12px;
+                background: #4CAF50;
+                color: white;
+                border: none;
+                border-radius: 4px;
+                cursor: pointer;
+                z-index: 1000;
+            }}
+        </style>
+        <div style="margin-top: 20px; padding: 20px; border-top: 1px solid #eee;">
+            <h3>Span Details</h3>
+            <pre id="spanDetails" style="max-height: 600px; overflow-y: auto;">Click on a node to view details</pre>
+        </div>
+        <script>
+            function moveNodesOnce() {{
+                var nodes = network.body.nodes;
+                var offsetX = 100;
+                var offsetY = 80;
+                for (var nodeId in nodes) {{
+                    if (nodes.hasOwnProperty(nodeId)) {{
+                        var node = nodes[nodeId];
+                        network.moveNode(nodeId, node.x + offsetX, node.y + offsetY);
+                    }}
+                }}
+                network.off("afterDrawing", moveNodesOnce);
+            }}
+            network.on("afterDrawing", moveNodesOnce);
+
+            network.on("click", function(params) {{
+                var nodeId = params.nodes[0];
+                var spanData = {json.dumps(spans)}[nodeId];
+                var displayData = {{}};
+                for (var key in spanData) {{
+                    if (key !== 'children') {{
+                        displayData[key] = spanData[key];
+                    }}
+                }}
+                document.getElementById("spanDetails").innerHTML = 
+                    JSON.stringify(displayData, null, 2);
+            }});
+        </script>
+        """)
+
+
+def generate_trace_graph_full(trace_id=None, folder_name="", file_name="trace_graph_full.html"):
+    trace_data = fetch_trace_data(trace_id)
+    net = Network(height="100%", width="100%",
+                  notebook=False, cdn_resources='in_line')
+
+    net.set_options("""
+    {
+      "nodes": {
+        "font": {"size": 8}
+      },
+      "physics": {
+        "enabled": true,
+        "hierarchicalRepulsion": {
+          "centralGravity": 0.5,
+          "springLength": 150,
+          "nodeDistance": 120
+        }
+      },
+      "layout": {
+        "hierarchical": {
+          "enabled": true,
+          "direction": "LR",
+          "sortMethod": "directed",
+          "nodeSpacing": 50,
+          "levelSeparation": 100,
+          "blockShifting": true,
+          "edgeMinimization": true
+        }
+      }
+    }
+    """)
+
+    root_spans = trace_data['root_span']
+    all_spans = []
+    for span in root_spans:
+        all_spans.extend(flatten_spans(span))
+    spans = {span['span_id']: span for span in all_spans}
+    for span_id, span in spans.items():
+        title = f"{span['name']}\n({span['run_type']})"
+        net.add_node(span_id, label=title, title=title,
+                     shape='box', color=get_span_color(span))
+        if span.get('parent_id') and span['parent_id'] in spans:
+            net.add_edge(span['parent_id'], span_id, arrows='to')
+
+    folder_path = os.path.join(os.getcwd(), folder_name)
+    os.makedirs(folder_path, exist_ok=True)
+    file_path = os.path.join(folder_path, file_name)
+    net.show(file_path, notebook=False)
+
+    with open(file_path, 'a') as f:
+        f.write(f"""
+        <style>
+            body {{ margin: 0; padding: 0; }}
+            #mynetwork {{
+                width: 100vw;
+                height: 100vh;
+            }}
+            #spanDetails {{
+                position: absolute;
+                right: 0;
+                top: 0;
+                width: 30%;
+                height: 100%;
+                padding: 20px;
+                background: white;
+                border-left: 1px solid #eee;
+                overflow-y: auto;
+                z-index: 1000;
+            }}
+        </style>
+        <div>
+            <h3>Span Details</h3>
+            <pre id="spanDetails" style="max-height: 100%; overflow-y: auto;">Click on a node to view details</pre>
+        </div>
+        <script>
+            function moveNodesOnce() {{
+                var nodes = network.body.nodes;
+                var offsetX = -250;
+                var offsetY = 0;
+                for (var nodeId in nodes) {{
+                    if (nodes.hasOwnProperty(nodeId)) {{
+                        var node = nodes[nodeId];
+                        network.moveNode(nodeId, node.x + offsetX, node.y + offsetY);
+                    }}
+                }}
+                network.off("afterDrawing", moveNodesOnce);
+            }}
+            network.on("afterDrawing", moveNodesOnce);
+            network.on("click", function(params) {{
+                var nodeId = params.nodes[0];
+                var spanData = {json.dumps(spans)}[nodeId];
+                var displayData = {{}};
+                for (var key in spanData) {{
+                    if (key !== 'children') {{
+                        displayData[key] = spanData[key];
+                    }}
+                }}
+                document.getElementById("spanDetails").innerHTML = 
+                    JSON.stringify(displayData, null, 2);
+            }});
+        </script>
+        """)

aworld/cmd/web_legacy/utils.py

@@ -0,0 +1,32 @@
+import os
+
+
+def list_agents():
+    agents_dir = os.path.join(os.getcwd(), "agent_deploy")
+
+    if not os.path.exists(agents_dir):
+        return []
+
+    try:
+        # 列出agents_dir下的所有目录
+        agents = []
+        for item in os.listdir(agents_dir):
+            item_path = os.path.join(agents_dir, item)
+            if os.path.isdir(item_path):
+                # 检查是否包含agent.py文件
+                agent_file = os.path.join(item_path, "agent.py")
+                if os.path.exists(agent_file):
+                    agents.append(item)
+        return agents
+    except OSError as e:
+        # 处理权限错误或其他文件系统错误
+        print(f"Error listing agents: {e}")
+        return []
+
+
+def get_agent_package_path(agent_name):
+    return os.path.join(
+        os.getcwd(),
+        "agent_deploy",
+        agent_name,
+    )

aworld/cmd/web_legacy/web_server.py

@@ -0,0 +1,9 @@
+import os
+import streamlit.web.bootstrap as bootstrap
+
+
+def run_web_server(port, args=None, **kwargs):
+    script = os.path.join(os.path.dirname(os.path.abspath(__file__)), "main.py")
+    kwargs = {**kwargs, "server.port": port}
+    bootstrap.load_config_options(flag_options=kwargs)
+    bootstrap.run(script, False, args, flag_options=kwargs)