commitai 2.0.0__tar.gz → 2.2.2__tar.gz

{commitai-2.0.0 → commitai-2.2.2}/.pre-commit-config.yaml

@@ -12,7 +12,7 @@ repos:
 
   - repo: https://github.com/astral-sh/ruff-pre-commit
     # Ruff version. Must be aligned with the version in pyproject.toml
-    rev: v0.4.4 # Choose a specific Ruff version
+    rev: v0.15.0 # Match local version
     hooks:
       # Run the linter. Applies fixes including import sorting, etc.
       - id: ruff
@@ -27,10 +27,11 @@ repos:
         # Ensure mypy runs with the necessary dependencies installed
         additional_dependencies:
           - "click>=8.0,<9.0"
-          - "langchain>=0.1.0,<=0.3.25"
-          - "langchain-core>=0.1.0,<=0.3.58"
-          - "langchain-community>=0.0.20,<=0.3.23"
-          - "langchain-google-genai~=2.1.4"
+          - "langchain>=1.0"
+          - "langchain-core>=1.0"
+          - "langchain-community>=0.4.0"
+          - "langchain-google-genai>=1.0"
+          - "langgraph>=1.0.0"
           - "pydantic>=2.0,<3.0"
           - "types-setuptools"
         args: [--config-file=pyproject.toml] # Point mypy to the config

{commitai-2.0.0 → commitai-2.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: commitai
-Version: 2.0.0
+Version: 2.2.2
 Summary: Commitai helps you generate git commit messages using AI
 Project-URL: Bug Tracker, https://github.com/lguibr/commitai/issues
 Project-URL: Documentation, https://github.com/lguibr/commitai/blob/main/README.md
@@ -40,18 +40,19 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Topic :: Software Development :: Version Control :: Git
 Classifier: Topic :: Utilities
 Requires-Python: >=3.10
-Requires-Dist: click<9.0,>=8.0
-Requires-Dist: langchain-community<=0.3.23,>=0.0.20
-Requires-Dist: langchain-core<=0.3.58,>=0.1.0
-Requires-Dist: langchain-google-genai~=2.1.4
-Requires-Dist: langchain<=0.3.25,>=0.1.0
-Requires-Dist: pydantic<3.0,>=2.0
+Requires-Dist: click>=8.1
+Requires-Dist: langchain-community>=0.4.0
+Requires-Dist: langchain-core>=1.0
+Requires-Dist: langchain-google-genai>=1.0
+Requires-Dist: langchain>=1.0
+Requires-Dist: langgraph>=1.0.0
+Requires-Dist: pydantic>=2.0
 Provides-Extra: test
-Requires-Dist: langchain-google-genai~=2.1.4; extra == 'test'
+Requires-Dist: langchain-google-genai>=2.0.0; extra == 'test'
 Requires-Dist: mypy>=1.9.0; extra == 'test'
 Requires-Dist: pytest-cov>=3.0; extra == 'test'
 Requires-Dist: pytest>=7.0; extra == 'test'
-Requires-Dist: ruff==0.4.4; extra == 'test'
+Requires-Dist: ruff>=0.4.4; extra == 'test'
 Requires-Dist: types-setuptools; extra == 'test'
 Description-Content-Type: text/markdown
 

{commitai-2.0.0 → commitai-2.2.2}/commitai/__init__.py

@@ -3,7 +3,7 @@
 
 # This __version__ string is read by hatchling during the build process
 # Make sure to update it for new releases.
-__version__ = "2.0.0"
+__version__ = "2.2.2"
 
 # The importlib.metadata approach is generally for reading the version
 # of an *already installed* package at runtime. We don't need it here

{commitai-2.0.0 → commitai-2.2.2}/commitai/agent.py

@@ -3,11 +3,11 @@ import os
 import subprocess
 from typing import Any, Dict, Type
 
-from langchain.agents import AgentExecutor, create_tool_calling_agent
+# from langchain.agents import AgentExecutor, create_tool_calling_agent # Removed
 from langchain_core.language_models import BaseChatModel
-from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
 from langchain_core.runnables import Runnable
 from langchain_core.tools import BaseTool
+from langgraph.prebuilt import create_react_agent
 from pydantic import BaseModel, Field
 
 # --- TOOLS ---
@@ -157,6 +157,9 @@ class TodoMiddleware:
 # --- AGENT ---
 
 
+# --- AGENT ---
+
+
 def create_commit_agent(llm: BaseChatModel) -> Runnable:
     # 1. Init Tools
     tools = [ReadOnlyShellTool(), FileSearchTool(), FileReadTool()]
@@ -186,18 +189,22 @@ Protocol:
 3. If clarification is needed, explore files.
 4. Final Answer MUST be ONLY the commit message.
 """
-    prompt = ChatPromptTemplate.from_messages(
-        [
-            ("system", system_prompt),
-            MessagesPlaceholder("chat_history", optional=True),
-            ("human", "Generate the commit message."),
-            MessagesPlaceholder("agent_scratchpad"),
-        ]
-    )
+    # Note: create_react_agent handles the prompt internally or via state_modifier.
+    # We can pass a system string or a function. Since our prompt depends on dynamic
+    # variables (diff, explanation, etc.), we need to inject them. LangGraph's
+    # prebuilt agent usually takes a static system message. However, we can use the
+    # 'messages' state. But to keep it simple and compatible with existing 'invoke'
+    # interface: We will format the system prompt in the wrapper and pass it as the
+    # first message.
+
+    # Actually, create_react_agent supports 'state_modifier'.
+    # If we pass a formatted string, it works as system prompt.
+
+    # 4. Construct Graph
+    # We don't construct the graph with ALL variables pre-bound if they change per run.
+    # Instead, we'll format the prompt in the pipeline and pass it to the agent.
 
-    # 4. Construct Agent
-    agent = create_tool_calling_agent(llm, tools, prompt)
-    agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=False)
+    agent_graph = create_react_agent(llm, tools)
 
     # 5. Pipeline with Middleware
     def run_pipeline(inputs: Dict[str, Any]) -> str:
@@ -210,11 +217,34 @@ Protocol:
         state.setdefault("explanation", "None")
         state.setdefault("summary", "None")
         state.setdefault("todo_str", "None")
-        state.setdefault("chat_history", [])
+
+        # Format System Prompt
+        formatted_system_prompt = system_prompt.format(
+            explanation=state["explanation"],
+            todo_str=state["todo_str"],
+            summary=state["summary"],
+            diff=state.get("diff", ""),
+        )
 
         # Run Agent
-        result = agent_executor.invoke(state)
-        return str(result["output"])
+        # LangGraph inputs: {"messages": [{"role": "user", "content": ...}]}
+        # We inject the system prompt as a SystemMessage or just update the state.
+        # create_react_agent primarily looks at 'messages'.
+
+        from langchain_core.messages import HumanMessage, SystemMessage
+
+        messages = [
+            SystemMessage(content=formatted_system_prompt),
+            HumanMessage(content="Generate the commit message."),
+        ]
+
+        # Invoke graph
+        # result is a dict with 'messages'
+        result = agent_graph.invoke({"messages": messages})
+
+        # Extract last message content
+        last_message = result["messages"][-1]
+        return str(last_message.content)
 
     # Wrap in RunnableLambda to expose 'invoke'
     from langchain_core.runnables import RunnableLambda

{commitai-2.0.0 → commitai-2.2.2}/commitai/cli.py

@@ -106,7 +106,8 @@ def _handle_commit(commit_message: str, commit_flag: bool) -> None:
     final_commit_message = commit_message
     if not commit_flag:
         click.secho(
-            f"\n📝 Generated Commit Message:\n{'-'*40}\n{commit_message}\n{'-'*40}\n",
+            f"\n📝 Generated Commit Message:\n{'-' * 40}\n"
+            f"{commit_message}\n{'-' * 40}\n",
             fg="green",
         )
 
@@ -197,7 +198,7 @@ def generate_message(
     template: Optional[str],
     add: bool,
     model: str,
-    deep: bool,
+    deep: bool = False,
 ) -> None:
     explanation = " ".join(description)
 

{commitai-2.0.0 → commitai-2.2.2}/pyproject.toml

@@ -7,7 +7,7 @@ build-backend = "hatchling.build"
 name = "commitai"
 # Make sure to update version in commitai/__init__.py as well
 
-version = "2.0.0"
+version = "2.2.2"
 
 description = "Commitai helps you generate git commit messages using AI"
 readme = "README.md"
@@ -29,12 +29,13 @@ classifiers = [
     "Topic :: Utilities",
 ]
 dependencies = [
-    "click>=8.0,<9.0",
-    "langchain>=0.1.0,<=0.3.25",
-    "langchain-core>=0.1.0,<=0.3.58",
-    "langchain-community>=0.0.20,<=0.3.23",
-    "langchain-google-genai~=2.1.4",
-    "pydantic>=2.0,<3.0",
+    "click>=8.1",
+    "langchain>=1.0",
+    "langchain-community>=0.4.0",
+    "langchain-core>=1.0",
+    "langchain-google-genai>=1.0",
+    "langgraph>=1.0.0",
+    "pydantic>=2.0",
 ]
 
 [project.urls]
@@ -52,9 +53,8 @@ test = [
     "pytest-cov>=3.0", # Still needed for coverage run command
     "mypy>=1.9.0",
     "types-setuptools",
-    # Pin ruff version to match pre-commit hook
-    "ruff==0.4.4",
-    "langchain-google-genai~=2.1.4", # Keep google genai here for mypy in pre-commit
+    "ruff>=0.4.4",
+    "langchain-google-genai>=2.0.0", # Keep google genai here for mypy in pre-commit
 ]
 
 [tool.hatch.version]

{commitai-2.0.0 → commitai-2.2.2}/tests/test_agent_chains.py

@@ -62,17 +62,25 @@ def test_todo_scanner_middleware(mock_llm):
 
 def test_create_commit_agent(mock_llm):
     # This tests the factory function
-    agent_executor = create_commit_agent(mock_llm)
-    assert agent_executor is not None
-    # We can try to invoke it if we mock enough stuff
-    # But just creating it covers the definition lines.
+    # Mocking create_react_agent to avoid actual graph compilation
+    with patch("commitai.agent.create_react_agent") as mock_create_graph:
+        mock_graph = MagicMock()
+        mock_create_graph.return_value = mock_graph
+        agent_executor = create_commit_agent(mock_llm)
+        assert agent_executor is not None
 
 
 def test_agent_run(mock_llm):
     # E2E-ish test of the agent logic with mocks
-    with patch("commitai.agent.AgentExecutor") as MockExecutor:
-        mock_executor_instance = MockExecutor.return_value
-        mock_executor_instance.invoke.return_value = {"output": "Final Commit Message"}
+    with patch("commitai.agent.create_react_agent") as mock_create_graph:
+        mock_graph = MagicMock()
+        mock_create_graph.return_value = mock_graph
+
+        # Determine strict return structure for LangGraph invoke
+        # It yields a dict with "messages" list
+        last_message = MagicMock()
+        last_message.content = "Final Commit Message"
+        mock_graph.invoke.return_value = {"messages": [last_message]}
 
         agent_runnable = create_commit_agent(mock_llm)
         result = agent_runnable.invoke({"diff": "diff", "explanation": "expl"})

{commitai-2.0.0 → commitai-2.2.2}/tests/test_cli.py

@@ -41,6 +41,7 @@ def mock_generate_deps(tmp_path):
             "commitai.cli.get_current_branch_name", return_value="main"
         ) as mock_branch,
         patch("commitai.cli.create_commit") as mock_commit,
+        # Update mock target for agent creation
         patch("commitai.cli.create_commit_agent") as mock_create_agent,
         patch("click.edit") as mock_edit,
         patch("click.clear"),
@@ -57,10 +58,10 @@ def mock_generate_deps(tmp_path):
 
         mock_google_instance = mock_google_class_in_cli.return_value
 
-        # Agent Mock
-        mock_agent_instance = MagicMock()
-        mock_agent_instance.invoke.return_value = "Generated commit message"
-        mock_create_agent.return_value = mock_agent_instance
+        # Agent Mock (RunnableLambda now)
+        mock_agent_runnable = MagicMock()
+        mock_agent_runnable.invoke.return_value = "Generated commit message"
+        mock_create_agent.return_value = mock_agent_runnable
 
         if mock_google_class_in_cli is not None:
             mock_google_instance.spec = ActualChatGoogleGenerativeAI
@@ -93,7 +94,7 @@ def mock_generate_deps(tmp_path):
             "path_exists": mock_path_exists,
             "commit_msg_path": fake_commit_msg_path,
             "create_agent": mock_create_agent,
-            "agent_instance": mock_agent_instance,
+            "agent_instance": mock_agent_runnable,  # Still useful alias for tests
             "confirm": mock_confirm,
         }
 
@@ -381,9 +382,9 @@ def test_generate_edit_error_io(mock_generate_deps):
     # Check exit code is 1
     result = runner.invoke(cli, ["generate", "--no-review", "Test explanation"])
 
-    assert (
-        result.exit_code == 1
-    ), f"Expected exit code 1, got {result.exit_code}. Output: {result.output}"
+    assert result.exit_code == 1, (
+        f"Expected exit code 1, got {result.exit_code}. Output: {result.output}"
+    )
     # The IOError bubbles up to the outer exception handler
     assert "Error handling user input: Read permission denied" in result.output
     mock_generate_deps["commit"].assert_not_called()