commitai 2.0.0__py3-none-any.whl → 2.2.2__py3-none-any.whl

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/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/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.dist-info → commitai-2.2.2.dist-info}/METADATA

@@ -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.2.2.dist-info/RECORD

@@ -0,0 +1,11 @@
+commitai/__init__.py,sha256=6977C2xbYqWwIRHhuFqmyoyI7MaMQmzb3SYtWjIJ4To,388
+commitai/agent.py,sha256=dxTHlxpYWuOQU9purJMoZCWhTszY4pkQhhnXXuWvmWk,8330
+commitai/chains.py,sha256=i5tQL9Qg-kP4yVCiQavhOLS9ljh72bxYALQ27Hoqivc,4612
+commitai/cli.py,sha256=I_3c2q15e3KniprWidqry8ZLJJXg-jAcT2w9wO00_L4,11666
+commitai/git.py,sha256=XWAloZWQuLrFHUyfh3SkOgLsL4kfKRtgj3fzuJjcL2A,1649
+commitai/template.py,sha256=PAS3BUjj6fcdsvSheopU_0xGCLPMj-vvFeZVCNRf0VM,2274
+commitai-2.2.2.dist-info/METADATA,sha256=sdVS1BuvMbvck6vWfNjv4tPrGyMI8RTff6bvhditrz8,13856
+commitai-2.2.2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+commitai-2.2.2.dist-info/entry_points.txt,sha256=qzWJQdPoR38mjQgRPRCB3tA7Kojtj3WrozlFWR4KhLY,128
+commitai-2.2.2.dist-info/licenses/LICENSE,sha256=wVkmSz0UMpGw0xYxk4AmkPLd_tVFcuszTdNIoq02tJA,1087
+commitai-2.2.2.dist-info/RECORD,,

commitai-2.0.0.dist-info/RECORD

@@ -1,11 +0,0 @@
-commitai/__init__.py,sha256=__b9FU1YSgHa5pyZG0hsgo2bQrNVAFPUMH2o-pBKzzE,388
-commitai/agent.py,sha256=fHh8d_ND9s0gqPaWqbHwmW9Kkt9dS5M6sRnGmhHb87Y,7037
-commitai/chains.py,sha256=i5tQL9Qg-kP4yVCiQavhOLS9ljh72bxYALQ27Hoqivc,4612
-commitai/cli.py,sha256=9-_y6STLCz-E7kHH6bX7CaV3aIxt8cdp-oa3IqbKDHY,11638
-commitai/git.py,sha256=XWAloZWQuLrFHUyfh3SkOgLsL4kfKRtgj3fzuJjcL2A,1649
-commitai/template.py,sha256=PAS3BUjj6fcdsvSheopU_0xGCLPMj-vvFeZVCNRf0VM,2274
-commitai-2.0.0.dist-info/METADATA,sha256=Vjrry5eOeDtht4XyJuloAzQJXOBJf6WEVnMEx0R9BbE,13868
-commitai-2.0.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-commitai-2.0.0.dist-info/entry_points.txt,sha256=qzWJQdPoR38mjQgRPRCB3tA7Kojtj3WrozlFWR4KhLY,128
-commitai-2.0.0.dist-info/licenses/LICENSE,sha256=wVkmSz0UMpGw0xYxk4AmkPLd_tVFcuszTdNIoq02tJA,1087
-commitai-2.0.0.dist-info/RECORD,,