fastmcp-agents-library-agent-simple-code 0.5.1__tar.gz → 0.5.8__tar.gz

{fastmcp_agents_library_agent_simple_code-0.5.1 → fastmcp_agents_library_agent_simple_code-0.5.8}/PKG-INFO

@@ -1,7 +1,10 @@
-Metadata-Version: 2.4
+Metadata-Version: 2.3
 Name: fastmcp-agents-library-agent-simple-code
-Version: 0.5.1
+Version: 0.5.8
 Summary: Agents for Simple Code
-Requires-Python: >=3.13
 Requires-Dist: fastmcp-ai-agent-bridge-pydantic-ai>=0.1.2
 Requires-Dist: gitpython>=3.1.44
+Requires-Dist: fastmcp-agents-library-mcp
+Requires-Python: >=3.13
+Description-Content-Type: text/markdown
+

{fastmcp_agents_library_agent_simple_code-0.5.1 → fastmcp_agents_library_agent_simple_code-0.5.8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "fastmcp-agents-library-agent-simple-code"
-version = "0.5.1"
+version = "0.5.8"
 description = "Agents for Simple Code"
 readme = "README.md"
 requires-python = ">=3.13"
@@ -8,10 +8,13 @@ requires-python = ">=3.13"
 dependencies = [
     "fastmcp-ai-agent-bridge-pydantic-ai>=0.1.2",
     "gitpython>=3.1.44",
+    "fastmcp-agents-library-mcp",
 ]
 
 [project.scripts]
-simple-code = "fastmcp_agents.library.agent.simple_code:main"
+simple-code = "fastmcp_agents.library.agent.simple_code:run"
+fastmcp-agents-library-agent-simple-code = "fastmcp_agents.library.agent.simple_code:run"
+
 
 [tool.uv.sources]
 fastmcp-agents-library-mcp = { workspace = true }
@@ -26,15 +29,13 @@ dev = [
 ]
 
 [build-system]
-requires = ["hatchling"]
-build-backend = "hatchling.build"
+requires = ["uv_build>=0.8.2,<0.9.0"]
+build-backend = "uv_build"
 
+[tool.uv.build-backend]
+module-name = "fastmcp_agents.library.agent.simple_code"
 
-[tool.hatch.build.targets.wheel]
-packages = ["src/fastmcp_agents"]
 
-[tool.hatch.build.targets.sdist]
-packages = ["src/fastmcp_agents"]
 
 [tool.pytest.ini_options]
 env = ["COLUMNS=120"]

{fastmcp_agents_library_agent_simple_code-0.5.1 → fastmcp_agents_library_agent_simple_code-0.5.8/src}/fastmcp_agents/library/agent/simple_code/investigate.py

@@ -11,7 +11,7 @@ from typing import Annotated, Any, Literal
 from fastmcp import FastMCP
 from fastmcp.tools.tool import Tool
 from fastmcp_ai_agent_bridge.pydantic_ai import FastMCPToolset
-from git import Repo
+from git.repo import Repo
 from pydantic import AnyHttpUrl, BaseModel, Field, RootModel
 from pydantic_ai import Agent
 
@@ -84,6 +84,27 @@ class InvestigationRecommendation(BaseModel):
     proposed_lines: FileLines = Field(default=..., description="The proposed lines of code in the file with their line numbers.")
 
 
+class BranchInfo(BaseModel):
+    """A repository info."""
+
+    name: str
+    commit_sha: str
+
+    @classmethod
+    def from_repo(cls, repo: Repo) -> "BranchInfo":
+        """Create a branch info from a repository."""
+        return cls(name=repo.active_branch.name, commit_sha=repo.head.commit.hexsha)
+
+    @classmethod
+    def from_dir(cls, directory: Path) -> "BranchInfo | None":
+        """Create a branch info from a directory."""
+        try:
+            repo: Repo = Repo(path=directory)
+            return cls.from_repo(repo)
+        except Exception:
+            return None
+
+
 class InvestigationResponse(BaseModel):
     """An investigation response."""
 
@@ -123,17 +144,20 @@ CODE_REPOSITORY_TYPE = Annotated[
 ]
 
 
-async def investigate_code(task: str, code_repository: CODE_REPOSITORY_TYPE) -> InvestigationResponse:
+async def investigate_code(task: str, code_repository: CODE_REPOSITORY_TYPE) -> tuple[InvestigationResponse, BranchInfo | None]:
     """Perform a code investigation."""
 
     with tempfile.TemporaryDirectory(delete=False) as temp_dir:
         # We only actually use the tempdir if we are cloning a git repository
         if isinstance(code_repository, AnyHttpUrl):
             Repo.clone_from(url=str(code_repository), to_path=temp_dir, single_branch=True, depth=1)
+
             code_repository = Path(temp_dir)
         if code_repository is None:
             code_repository = Path.cwd()
 
+        branch_info: BranchInfo | None = BranchInfo.from_dir(directory=code_repository)
+
         directory_locked_toolset = FastMCPToolset.from_mcp_config(
             mcp_config={"filesystem": read_only_filesystem_mcp(root_dir=code_repository)}
         )
@@ -147,12 +171,18 @@ async def investigate_code(task: str, code_repository: CODE_REPOSITORY_TYPE) ->
 
         run_result = await code_investigation_agent.run(user_prompt=[task, repo_info], toolsets=[directory_locked_toolset])
 
-        return run_result.output
+        return run_result.output, branch_info
 
 
 investigate_code_tool = Tool.from_function(fn=investigate_code)
 
 server.add_tool(tool=investigate_code_tool)
 
+
+def run():
+    """Run the agent."""
+    server.run(transport="stdio")
+
+
 if __name__ == "__main__":
     server.run(transport="sse")

fastmcp_agents_library_agent_simple_code-0.5.1/.gitignore

@@ -1,183 +0,0 @@
-# Byte-compiled / optimized / DLL files
-__pycache__/
-*.py[cod]
-*$py.class
-
-# C extensions
-*.so
-
-# Distribution / packaging
-.Python
-build/
-develop-eggs/
-dist/
-downloads/
-eggs/
-.eggs/
-lib/
-lib64/
-parts/
-sdist/
-var/
-wheels/
-share/python-wheels/
-*.egg-info/
-.installed.cfg
-*.egg
-MANIFEST
-
-# PyInstaller
-#  Usually these files are written by a python script from a template
-#  before PyInstaller builds the exe, so as to inject date/other infos into it.
-*.manifest
-*.spec
-
-# Installer logs
-pip-log.txt
-pip-delete-this-directory.txt
-
-# Unit test / coverage reports
-htmlcov/
-.tox/
-.nox/
-.coverage
-.coverage.*
-.cache
-nosetests.xml
-coverage.xml
-*.cover
-*.py,cover
-.hypothesis/
-.pytest_cache/
-cover/
-
-# Translations
-*.mo
-*.pot
-
-# Django stuff:
-*.log
-local_settings.py
-db.sqlite3
-db.sqlite3-journal
-
-# Flask stuff:
-instance/
-.webassets-cache
-
-# Scrapy stuff:
-.scrapy
-
-# Sphinx documentation
-docs/_build/
-
-# PyBuilder
-.pybuilder/
-target/
-
-# Jupyter Notebook
-.ipynb_checkpoints
-
-# IPython
-profile_default/
-ipython_config.py
-
-# pyenv
-#   For a library or package, you might want to ignore these files since the code is
-#   intended to run in multiple environments; otherwise, check them in:
-# .python-version
-
-# pipenv
-#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
-#   However, in case of collaboration, if having platform-specific dependencies or dependencies
-#   having no cross-platform support, pipenv may install dependencies that don't work, or not
-#   install all needed dependencies.
-#Pipfile.lock
-
-# UV
-#   Similar to Pipfile.lock, it is generally recommended to include uv.lock in version control.
-#   This is especially recommended for binary packages to ensure reproducibility, and is more
-#   commonly ignored for libraries.
-#uv.lock
-
-# poetry
-#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
-#   This is especially recommended for binary packages to ensure reproducibility, and is more
-#   commonly ignored for libraries.
-#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
-#poetry.lock
-
-# pdm
-#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
-#pdm.lock
-#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
-#   in version control.
-#   https://pdm.fming.dev/latest/usage/project/#working-with-version-control
-.pdm.toml
-.pdm-python
-.pdm-build/
-
-# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
-__pypackages__/
-
-# Celery stuff
-celerybeat-schedule
-celerybeat.pid
-
-# SageMath parsed files
-*.sage.py
-
-# Environments
-.env
-.env.*
-.venv
-env/
-venv/
-ENV/
-env.bak/
-venv.bak/
-
-# Spyder project settings
-.spyderproject
-.spyproject
-
-# Rope project settings
-.ropeproject
-
-# mkdocs documentation
-/site
-
-# mypy
-.mypy_cache/
-.dmypy.json
-dmypy.json
-
-# Pyre type checker
-.pyre/
-
-# pytype static type analyzer
-.pytype/
-
-# Cython debug symbols
-cython_debug/
-
-# PyCharm
-#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
-#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
-#  and can be added to the global gitignore or merged into this file.  For a more nuclear
-#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
-#.idea/
-
-# Ruff stuff:
-.ruff_cache/
-
-# PyPI configuration file
-.pypirc
-secrets
-
-prompt_suggestions
-agent-working-dir
-
-**/*.bak
-
-storage