mirror of
https://github.com/FoundationAgents/MetaGPT.git
synced 2026-04-26 01:06:27 +02:00
delete spo app comment
This commit is contained in:
xiangjinyu
parent
3ca2c1756d
commit
8208b37463
1 changed files with 27 additions and 75 deletions
|
|
@ -1,22 +1,20 @@
|
|||
from loguru import logger as _logger
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
import streamlit as st
|
||||
import yaml
|
||||
import os
|
||||
from loguru import logger as _logger
|
||||
|
||||
root_path = Path(__file__).parent.parent.parent.parent
|
||||
sys.path.append(str(root_path))
|
||||
sys.path.append(str(Path(__file__).parents[3]))
|
||||
|
||||
from metagpt.ext.spo.components.optimizer import PromptOptimizer
|
||||
from metagpt.ext.spo.utils.llm_client import SPO_LLM
|
||||
from metagpt.const import METAGPT_ROOT
|
||||
from metagpt.const import METAGPT_ROOT # noqa: E402
|
||||
from metagpt.ext.spo.components.optimizer import PromptOptimizer # noqa: E402
|
||||
from metagpt.ext.spo.utils.llm_client import SPO_LLM # noqa: E402
|
||||
|
||||
|
||||
def load_yaml_template(template_path):
|
||||
if template_path.exists():
|
||||
with open(template_path, 'r', encoding='utf-8') as f:
|
||||
with open(template_path, "r", encoding="utf-8") as f:
|
||||
return yaml.safe_load(f)
|
||||
return {"prompt": "", "requirements": "", "count": None, "faq": [{"question": "", "answer": ""}]}
|
||||
|
||||
|
|
@ -25,32 +23,20 @@ def save_yaml_template(template_path, data):
|
|||
template_format = {
|
||||
"prompt": str(data.get("prompt", "")),
|
||||
"requirements": str(data.get("requirements", "")),
|
||||
"count": data.get("count"), # 保持None值
|
||||
"count": data.get("count"),
|
||||
"faq": [
|
||||
{
|
||||
"question": str(faq.get("question", "")).strip(),
|
||||
"answer": str(faq.get("answer", "")).strip()
|
||||
}
|
||||
{"question": str(faq.get("question", "")).strip(), "answer": str(faq.get("answer", "")).strip()}
|
||||
for faq in data.get("faq", [])
|
||||
]
|
||||
],
|
||||
}
|
||||
|
||||
template_path.parent.mkdir(parents=True, exist_ok=True)
|
||||
|
||||
with open(template_path, 'w', encoding='utf-8') as f:
|
||||
yaml.dump(
|
||||
template_format,
|
||||
f,
|
||||
allow_unicode=True,
|
||||
sort_keys=False,
|
||||
default_flow_style=False,
|
||||
indent=2
|
||||
)
|
||||
|
||||
with open(template_path, "w", encoding="utf-8") as f:
|
||||
yaml.dump(template_format, f, allow_unicode=True, sort_keys=False, default_flow_style=False, indent=2)
|
||||
|
||||
|
||||
def main():
|
||||
|
||||
st.title("SPO | Self-Supervised Prompt Optimization 🤖")
|
||||
|
||||
# Sidebar for configurations
|
||||
|
|
@ -67,29 +53,23 @@ def main():
|
|||
template_name = st.selectbox("Select Template", existing_templates)
|
||||
else:
|
||||
template_name = st.text_input("New Template Name")
|
||||
if template_name and not template_name.endswith('.yaml'):
|
||||
if template_name and not template_name.endswith(".yaml"):
|
||||
template_name = f"{template_name}"
|
||||
|
||||
# LLM Settings
|
||||
st.subheader("LLM Settings")
|
||||
opt_model = st.selectbox(
|
||||
"Optimization Model",
|
||||
["claude-3-5-sonnet-20240620", "gpt-4o", "gpt-4o-mini", "deepseek-chat"],
|
||||
index=0
|
||||
"Optimization Model", ["claude-3-5-sonnet-20240620", "gpt-4o", "gpt-4o-mini", "deepseek-chat"], index=0
|
||||
)
|
||||
opt_temp = st.slider("Optimization Temperature", 0.0, 1.0, 0.7)
|
||||
|
||||
eval_model = st.selectbox(
|
||||
"Evaluation Model",
|
||||
["gpt-4o-mini", "claude-3-5-sonnet-20240620", "gpt-4o", "deepseek-chat"],
|
||||
index=0
|
||||
"Evaluation Model", ["gpt-4o-mini", "claude-3-5-sonnet-20240620", "gpt-4o", "deepseek-chat"], index=0
|
||||
)
|
||||
eval_temp = st.slider("Evaluation Temperature", 0.0, 1.0, 0.3)
|
||||
|
||||
exec_model = st.selectbox(
|
||||
"Execution Model",
|
||||
["gpt-4o-mini", "claude-3-5-sonnet-20240620", "gpt-4o", "deepseek-chat"],
|
||||
index=0
|
||||
"Execution Model", ["gpt-4o-mini", "claude-3-5-sonnet-20240620", "gpt-4o", "deepseek-chat"], index=0
|
||||
)
|
||||
exec_temp = st.slider("Execution Temperature", 0.0, 1.0, 0.0)
|
||||
|
||||
|
|
@ -105,13 +85,13 @@ def main():
|
|||
template_path = settings_path / f"{template_name}.yaml"
|
||||
template_data = load_yaml_template(template_path)
|
||||
|
||||
if 'current_template' not in st.session_state or st.session_state.current_template != template_name:
|
||||
if "current_template" not in st.session_state or st.session_state.current_template != template_name:
|
||||
st.session_state.current_template = template_name
|
||||
st.session_state.faqs = template_data.get('faq', [])
|
||||
st.session_state.faqs = template_data.get("faq", [])
|
||||
|
||||
# Edit template sections
|
||||
prompt = st.text_area("Prompt", template_data.get('prompt', ''), height=100)
|
||||
requirements = st.text_area("Requirements", template_data.get('requirements', ''), height=100)
|
||||
prompt = st.text_area("Prompt", template_data.get("prompt", ""), height=100)
|
||||
requirements = st.text_area("Requirements", template_data.get("requirements", ""), height=100)
|
||||
|
||||
# FAQ section
|
||||
st.subheader("FAQ Examples")
|
||||
|
|
@ -128,17 +108,11 @@ def main():
|
|||
|
||||
with col1:
|
||||
question = st.text_area(
|
||||
f"Question {i + 1}",
|
||||
st.session_state.faqs[i].get('question', ''),
|
||||
key=f"q_{i}",
|
||||
height=100
|
||||
f"Question {i + 1}", st.session_state.faqs[i].get("question", ""), key=f"q_{i}", height=100
|
||||
)
|
||||
with col2:
|
||||
answer = st.text_area(
|
||||
f"Answer {i + 1}",
|
||||
st.session_state.faqs[i].get('answer', ''),
|
||||
key=f"a_{i}",
|
||||
height=100
|
||||
f"Answer {i + 1}", st.session_state.faqs[i].get("answer", ""), key=f"a_{i}", height=100
|
||||
)
|
||||
with col3:
|
||||
if st.button("🗑️", key=f"delete_{i}"):
|
||||
|
|
@ -149,12 +123,7 @@ def main():
|
|||
|
||||
# Save template button
|
||||
if st.button("Save Template"):
|
||||
new_template_data = {
|
||||
"prompt": prompt,
|
||||
"requirements": requirements,
|
||||
"count": None,
|
||||
"faq": new_faqs
|
||||
}
|
||||
new_template_data = {"prompt": prompt, "requirements": requirements, "count": None, "faq": new_faqs}
|
||||
|
||||
save_yaml_template(template_path, new_template_data)
|
||||
|
||||
|
|
@ -162,49 +131,32 @@ def main():
|
|||
st.success(f"Template saved to {template_path}")
|
||||
|
||||
st.subheader("Current Template Preview")
|
||||
preview_data = {
|
||||
"prompt": prompt,
|
||||
"requirements": requirements,
|
||||
"count": None,
|
||||
"faq": new_faqs
|
||||
}
|
||||
preview_data = {"prompt": prompt, "requirements": requirements, "count": None, "faq": new_faqs}
|
||||
st.code(yaml.dump(preview_data, allow_unicode=True), language="yaml")
|
||||
|
||||
# 创建一个固定的容器来显示日志
|
||||
st.subheader("Optimization Logs")
|
||||
log_container = st.empty()
|
||||
|
||||
class StreamlitSink:
|
||||
def write(self, message):
|
||||
# 获取当前日志内容
|
||||
current_logs = st.session_state.get('logs', [])
|
||||
current_logs = st.session_state.get("logs", [])
|
||||
current_logs.append(message.strip())
|
||||
st.session_state.logs = current_logs
|
||||
|
||||
# 使用 code 块显示日志
|
||||
log_container.code(
|
||||
"\n".join(current_logs),
|
||||
language="plaintext"
|
||||
)
|
||||
log_container.code("\n".join(current_logs), language="plaintext")
|
||||
|
||||
# 配置loguru日志
|
||||
streamlit_sink = StreamlitSink()
|
||||
_logger.remove()
|
||||
|
||||
# 添加过滤器,只捕获 PromptOptimizer 相关的日志
|
||||
def prompt_optimizer_filter(record):
|
||||
# 检查日志记录是否来自 PromptOptimizer 模块
|
||||
return "optimizer" in record["name"].lower()
|
||||
|
||||
_logger.add(
|
||||
streamlit_sink.write,
|
||||
format="{time:YYYY-MM-DD HH:mm:ss.SSS} | {level: <8} | {name}:{function}:{line} - {message}",
|
||||
filter=prompt_optimizer_filter # 添加过滤器
|
||||
)
|
||||
_logger.add(
|
||||
METAGPT_ROOT / "logs/{time:YYYYMMDD}.txt",
|
||||
level="DEBUG"
|
||||
filter=prompt_optimizer_filter,
|
||||
)
|
||||
_logger.add(METAGPT_ROOT / "logs/{time:YYYYMMDD}.txt", level="DEBUG")
|
||||
|
||||
# Start optimization button
|
||||
if st.button("Start Optimization"):
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue