Spaces:
Runtime error
Runtime error
File size: 6,353 Bytes
58e0d8e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 |
from crewai import Agent
import streamlit as st
from utils import rnd_id, fix_columns_width
from streamlit import session_state as ss
from db_utils import save_agent, delete_agent
from llms import llm_providers_and_models, create_llm
from datetime import datetime
class MyAgent:
def __init__(self, id=None, role=None, backstory=None, goal=None, temperature=None, allow_delegation=False, verbose=False, cache= None, llm_provider_model=None, max_iter=None, created_at=None, tools=None):
self.id = id or "A_" + rnd_id()
self.role = role or "Senior Researcher"
self.backstory = backstory or "Driven by curiosity, you're at the forefront of innovation, eager to explore and share knowledge that could change the world."
self.goal = goal or "Uncover groundbreaking technologies in AI"
self.temperature = temperature or 0.1
self.allow_delegation = allow_delegation if allow_delegation is not None else False
self.verbose = verbose if verbose is not None else True
self.llm_provider_model = llm_providers_and_models()[0] if llm_provider_model is None else llm_provider_model
self.created_at = created_at or datetime.now().isoformat()
self.tools = tools or []
self.max_iter = max_iter or 25
self.cache = cache if cache is not None else True
self.edit_key = f'edit_{self.id}'
if self.edit_key not in ss:
ss[self.edit_key] = False
@property
def edit(self):
return ss[self.edit_key]
@edit.setter
def edit(self, value):
ss[self.edit_key] = value
def get_crewai_agent(self) -> Agent:
llm = create_llm(self.llm_provider_model, temperature=self.temperature)
tools = [tool.create_tool() for tool in self.tools]
return Agent(
role=self.role,
backstory=self.backstory,
goal=self.goal,
allow_delegation=self.allow_delegation,
verbose=self.verbose,
max_iter=self.max_iter,
cache=self.cache,
tools=tools,
llm=llm
)
def delete(self):
ss.agents = [agent for agent in ss.agents if agent.id != self.id]
delete_agent(self.id)
def get_tool_display_name(self, tool):
first_param_name = tool.get_parameter_names()[0] if tool.get_parameter_names() else None
first_param_value = tool.parameters.get(first_param_name, '') if first_param_name else ''
return f"{tool.name} ({first_param_value if first_param_value else tool.tool_id})"
def is_valid(self, show_warning=False):
for tool in self.tools:
if not tool.is_valid(show_warning=show_warning):
if show_warning:
st.warning(f"Tool {tool.name} is not valid")
return False
return True
def validate_llm_provider_model(self):
available_models = llm_providers_and_models()
if self.llm_provider_model not in available_models:
self.llm_provider_model = available_models[0]
def draw(self, key=None):
self.validate_llm_provider_model()
expander_title = f"{self.role[:60]} -{self.llm_provider_model.split(':')[1]}" if self.is_valid() else f"❗ {self.role[:20]} -{self.llm_provider_model.split(':')[1]}"
if self.edit:
with st.expander(f"Agent: {self.role}", expanded=True):
with st.form(key=f'form_{self.id}' if key is None else key):
self.role = st.text_input("Role", value=self.role)
self.backstory = st.text_area("Backstory", value=self.backstory)
self.goal = st.text_area("Goal", value=self.goal)
self.allow_delegation = st.checkbox("Allow delegation", value=self.allow_delegation)
self.verbose = st.checkbox("Verbose", value=self.verbose)
self.cache = st.checkbox("Cache", value=self.cache)
self.llm_provider_model = st.selectbox("LLM Provider and Model", options=llm_providers_and_models(), index=llm_providers_and_models().index(self.llm_provider_model))
self.temperature = st.slider("Temperature", value=self.temperature, min_value=0.0, max_value=1.0)
self.max_iter = st.number_input("Max Iterations", value=self.max_iter, min_value=1, max_value=100)
enabled_tools = [tool for tool in ss.tools]
selected_tools = st.multiselect(
"Select Tools",
[self.get_tool_display_name(tool) for tool in enabled_tools],
default=[self.get_tool_display_name(tool) for tool in self.tools],
key=f"{self.id}_tools{key}"
)
submitted = st.form_submit_button("Save")
if submitted:
self.tools = [tool for tool in enabled_tools if self.get_tool_display_name(tool) in selected_tools]
self.set_editable(False)
else:
fix_columns_width()
with st.expander(expander_title, expanded=False):
st.markdown(f"**Role:** {self.role}")
st.markdown(f"**Backstory:** {self.backstory}")
st.markdown(f"**Goal:** {self.goal}")
st.markdown(f"**Allow delegation:** {self.allow_delegation}")
st.markdown(f"**Verbose:** {self.verbose}")
st.markdown(f"**Cache:** {self.cache}")
st.markdown(f"**LLM Provider and Model:** {self.llm_provider_model}")
st.markdown(f"**Temperature:** {self.temperature}")
st.markdown(f"**Max Iterations:** {self.max_iter}")
st.markdown(f"**Tools:** {[self.get_tool_display_name(tool) for tool in self.tools]}")
self.is_valid(show_warning=True)
col1, col2 = st.columns(2)
with col1:
st.button("Edit", on_click=self.set_editable, args=(True,), key=rnd_id())
with col2:
st.button("Delete", on_click=self.delete, key=rnd_id())
def set_editable(self, edit):
self.edit = edit
save_agent(self)
if not edit:
st.rerun() |