chat-ui-with-agent-examples / agents /code_writing_agent.py
ccm's picture
Removing second agent
e5cbdce
raw
history blame
696 Bytes
import os
import smolagents
import smolagents.models
def create_code_writing_agent():
"""
Create a code-writing agent without any extra tools.
"""
return smolagents.CodeAgent(
name="code_writing_agent_without_tools",
model=smolagents.models.OpenAIServerModel(
model_id=os.getenv("AGENT_MODEL", ""),
api_base=os.getenv("UPSTREAM_OPENAI_BASE", "").rstrip("/"),
api_key=os.getenv("OPENAI_API_KEY"),
),
tools=[], # no extra tools
add_base_tools=False,
max_steps=4,
verbosity_level=int(
os.getenv("AGENT_VERBOSITY", "1")
), # quieter by default; override via env
)