mirror of
https://github.com/tiennm99/serena.git
synced 2026-07-25 06:21:34 +00:00
Michael Panchenko
efbea42e39
More things: - list tools using .subclasses instead of inspect - entrypoint for listing tools - util for listing selected tools - documentation
60 lines
2.1 KiB
Python
60 lines
2.1 KiB
Python
import os
|
|
from logging import Logger
|
|
|
|
from agno.models.anthropic.claude import Claude
|
|
from agno.models.google.gemini import Gemini
|
|
from dotenv import load_dotenv
|
|
from sensai.util import logging
|
|
from sensai.util.helper import mark_used
|
|
|
|
from serena.agent import SerenaAgent
|
|
|
|
mark_used(Gemini, Claude)
|
|
|
|
os.chdir(os.path.dirname(os.path.abspath(__file__)))
|
|
Logger.root.setLevel(logging.INFO)
|
|
|
|
load_dotenv()
|
|
|
|
project_file_path = "../myproject.yml"
|
|
serena_agent = SerenaAgent(project_file_path, 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 = os.path.join("tmp", "agent_storage.db")
|
|
os.makedirs(os.path.dirname(sql_db_path), exist_ok=True)
|
|
# delete the db file if it exists
|
|
if os.path.exists(sql_db_path):
|
|
os.remove(sql_db_path)
|
|
|
|
for tool_name in serena_agent.list_tool_names():
|
|
print(tool_name)
|
|
|
|
# model = Claude(id="claude-3-7-sonnet-20250219")
|
|
# # model = Gemini(id="gemini-2.5-pro-exp-03-25")
|
|
|
|
# agno_agent = Agent(
|
|
# name="Serena",
|
|
# model=model,
|
|
# storage=SqliteAgentStorage(table_name="serena_agent_sessions", db_file=sql_db_path),
|
|
# description="A fully-featured coding assistant",
|
|
# tools=[SerenaAgnoToolkit(serena_agent)], # type: ignore
|
|
# show_tool_calls=False,
|
|
# markdown=True,
|
|
# system_message=serena_agent.prompt_factory.create_system_prompt(),
|
|
# read_tool_call_history=False,
|
|
# telemetry=False,
|
|
# memory=AgentMemory(),
|
|
# add_history_to_messages=True,
|
|
# num_history_responses=100, # you might want to adjust this (expense vs. history awareness)
|
|
# )
|
|
|
|
# # The app object must be in the module scope so that the server can access it for hot reloading
|
|
# app = Playground(agents=[agno_agent]).get_app()
|
|
|
|
# if __name__ == "__main__":
|
|
# serve_playground_app("agno_agent:app", reload=False)
|