fastmcp-agents-library-agent-simple-code 0.5.7__py3-none-any.whl → 0.5.8__py3-none-any.whl

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,7 +171,7 @@ 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)

{fastmcp_agents_library_agent_simple_code-0.5.7.dist-info → fastmcp_agents_library_agent_simple_code-0.5.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: fastmcp-agents-library-agent-simple-code
-Version: 0.5.7
+Version: 0.5.8
 Summary: Agents for Simple Code
 Requires-Dist: fastmcp-ai-agent-bridge-pydantic-ai>=0.1.2
 Requires-Dist: gitpython>=3.1.44

{fastmcp_agents_library_agent_simple_code-0.5.7.dist-info → fastmcp_agents_library_agent_simple_code-0.5.8.dist-info}/RECORD

@@ -4,8 +4,8 @@ fastmcp_agents/library/agent/simple_code/archive/workspace.py.disabled,sha256=d2
 fastmcp_agents/library/agent/simple_code/helpers/filesystem.py,sha256=ced0f82524a1355cb6cc2a2a5f61b61a6c507455e49cdfdb2d0bf12695586757,3771
 fastmcp_agents/library/agent/simple_code/helpers/git.py,sha256=7d3264b3b29df8025752018a51e677201f8761c07f549559260de06e6a88502f,262
 fastmcp_agents/library/agent/simple_code/implement.py,sha256=5267dbc76ca6fdebeb76282846068cc96901503e3004da5ac65cb04e242c5e94,5801
-fastmcp_agents/library/agent/simple_code/investigate.py,sha256=c17855e96e54c426b8c77dbf378d3bb1080ffe88bd39a65ccc9bfe1a80987851,6460
-fastmcp_agents_library_agent_simple_code-0.5.7.dist-info/WHEEL,sha256=03f80d698bb4300c27bcafc3987ea73ef427fcb365e59e808e50d041a7f87b89,78
-fastmcp_agents_library_agent_simple_code-0.5.7.dist-info/entry_points.txt,sha256=6a34307eece27892df63ae0af789d4ee3b839693877c185288303c27b92b5ada,166
-fastmcp_agents_library_agent_simple_code-0.5.7.dist-info/METADATA,sha256=d7b5af111b2cd83faf728ace6518c73ebeaadb8eeb9c7df0c148bf87d1f5e687,314
-fastmcp_agents_library_agent_simple_code-0.5.7.dist-info/RECORD,,
+fastmcp_agents/library/agent/simple_code/investigate.py,sha256=4c3ab2c0011c4c60eb3f24df097c9b676a25fe3c5917889b860f230b1a03148c,7180
+fastmcp_agents_library_agent_simple_code-0.5.8.dist-info/WHEEL,sha256=03f80d698bb4300c27bcafc3987ea73ef427fcb365e59e808e50d041a7f87b89,78
+fastmcp_agents_library_agent_simple_code-0.5.8.dist-info/entry_points.txt,sha256=6a34307eece27892df63ae0af789d4ee3b839693877c185288303c27b92b5ada,166
+fastmcp_agents_library_agent_simple_code-0.5.8.dist-info/METADATA,sha256=168944da9e725c74b31778ae397fcede4707d630586736c7631891ec785e6dd1,314
+fastmcp_agents_library_agent_simple_code-0.5.8.dist-info/RECORD,,