mirror of
https://github.com/tiennm99/serena.git
synced 2026-07-15 08:28:09 +00:00
83 lines
3.0 KiB
Python
83 lines
3.0 KiB
Python
import argparse
|
|
import os
|
|
from logging import Logger
|
|
from pathlib import Path
|
|
|
|
from agno.agent.agent import Agent
|
|
from agno.memory.agent import AgentMemory
|
|
from agno.models.anthropic.claude import Claude
|
|
from agno.models.google.gemini import Gemini
|
|
from agno.playground.playground import Playground
|
|
from agno.playground.serve import serve_playground_app
|
|
from agno.storage.sqlite import SqliteStorage
|
|
from dotenv import load_dotenv
|
|
from sensai.util import logging
|
|
from sensai.util.helper import mark_used
|
|
|
|
from serena.agent import SerenaAgent
|
|
from serena.agno import SerenaAgnoToolkit
|
|
|
|
mark_used(Gemini, Claude)
|
|
|
|
os.chdir(Path(__file__).parent)
|
|
Logger.root.setLevel(logging.INFO)
|
|
|
|
load_dotenv()
|
|
|
|
|
|
parser = argparse.ArgumentParser(description="Serena coding assistant")
|
|
parser.add_argument("--project-file", required=True, help="Path to the project file, either absolute or relative to the root directory")
|
|
args = parser.parse_args()
|
|
|
|
project_file = Path(args.project_file).resolve()
|
|
# If project file path is relative, make it absolute by joining with project root
|
|
if not project_file.is_absolute():
|
|
# Get the project root directory (parent of scripts directory)
|
|
project_root = Path(__file__).parent.parent
|
|
project_file = project_root / args.project_file
|
|
|
|
# Ensure the path is normalized and absolute
|
|
project_file = project_file.resolve()
|
|
|
|
|
|
serena_agent = SerenaAgent(str(project_file), start_language_server=True)
|
|
|
|
# Even though we don't want to keep history between sessions,
|
|
# for the agno ui to work as a conversation, we a persistent storage on disk
|
|
# This storage should be deleted between sessions.
|
|
# Note that this might collide with custom options for the agent, like adding vector-search based tools.
|
|
# See here for an explanation: https://www.reddit.com/r/agno/comments/1jk6qea/regarding_the_built_in_memory/
|
|
sql_db_path = Path("tmp") / "agent_storage.db"
|
|
sql_db_path.parent.mkdir(exist_ok=True)
|
|
# delete the db file if it exists
|
|
if sql_db_path.exists():
|
|
sql_db_path.unlink()
|
|
|
|
# model = Claude(id="claude-3-7-sonnet-20250219")
|
|
# Or use any other model supported by agno, e.g. the newest Gemini:
|
|
model = Gemini(id="gemini-2.5-pro-exp-03-25")
|
|
|
|
agno_agent = Agent(
|
|
name="Serena",
|
|
model=model,
|
|
# See explanation above on why storage is needed
|
|
storage=SqliteStorage(table_name="serena_agent_sessions", db_file=str(sql_db_path)),
|
|
description="A fully-featured coding assistant",
|
|
tools=[SerenaAgnoToolkit(serena_agent)], # type: ignore
|
|
# The tool calls will be shown in the UI anyway since whether to show them is configurable per tool
|
|
# To see detailed logs, you should use the serena logger (configure it in the project file path)
|
|
show_tool_calls=False,
|
|
markdown=True,
|
|
system_message=serena_agent.prompt_factory.create_system_prompt(),
|
|
telemetry=False,
|
|
memory=AgentMemory(),
|
|
add_history_to_messages=True,
|
|
num_history_responses=100, # you might want to adjust this (expense vs. history awareness)
|
|
)
|
|
|
|
app = Playground(agents=[agno_agent]).get_app()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
serve_playground_app("agno_agent:app", reload=False)
|