cmt-cli 1.0.1__tar.gz → 1.0.3__tar.gz

{cmt_cli-1.0.1 → cmt_cli-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: cmt-cli
-Version: 1.0.1
+Version: 1.0.3
 Summary: AI-powered Git commit message generator
 Project-URL: Homepage, https://github.com/bijaydas/cmt
 Project-URL: Documentation, https://github.com/bijaydas/cmt#readme

{cmt_cli-1.0.1 → cmt_cli-1.0.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "cmt-cli"
-version = "1.0.1"
+version = "1.0.3"
 description = "AI-powered Git commit message generator"
 readme = "README.md"
 requires-python = ">=3.12"

{cmt_cli-1.0.1 → cmt_cli-1.0.3}/src/cmt/ai/openai.py

@@ -1,14 +1,11 @@
 import logging
-from typing import cast
 
 from langchain.agents import create_agent
-from langchain.agents.middleware.types import InputAgentState
-from langchain_core.messages import BaseMessage, get_buffer_string
 from langchain_openai import ChatOpenAI
 from pydantic import SecretStr
 
 from cmt.ai.cache import CommitMessageCache
-from cmt.ai.prompt import COMMIT_PROMPT, COMMIT_SYSTEM_PROMPT
+from cmt.ai.prompt import COMMIT_PROMPT_TEMPLATE, COMMIT_SYSTEM_PROMPT
 from cmt.ai.provider import AIProvider
 from cmt.core.settings import settings
 from cmt.models.changes import AnalysisResult, StagedChangeSet, StagedFile
@@ -21,20 +18,20 @@ class OpenAIProvider(AIProvider):
     def __init__(self) -> None:
         self.config = settings.get()
 
-    def _build_prompt(
-        self, changes: StagedChangeSet, analysis: AnalysisResult
-    ) -> list[BaseMessage]:
+    def _build_prompt(self, changes: StagedChangeSet, analysis: AnalysisResult):
         staged_files = self._process_files(changes.files)
         staged_diffs = changes.diff
 
-        return COMMIT_PROMPT.format_messages(
-            total_files=len(changes.files),
-            added_files=analysis.added_files,
-            modified_files=analysis.modified_files,
-            deleted_files=analysis.deleted_files,
-            renamed_files=analysis.renamed_files,
-            staged_files=staged_files,
-            staged_diffs=staged_diffs,
+        return COMMIT_PROMPT_TEMPLATE.invoke(
+            {
+                "total_files": len(changes.files),
+                "added_files": analysis.added_files,
+                "modified_files": analysis.modified_files,
+                "deleted_files": analysis.deleted_files,
+                "renamed_files": analysis.renamed_files,
+                "staged_files": staged_files,
+                "staged_diffs": staged_diffs,
+            }
         )
 
     @staticmethod
@@ -45,15 +42,13 @@ class OpenAIProvider(AIProvider):
 
         return output.strip()
 
-    def _invoke(self, prompt: list[BaseMessage]) -> CommitSuggestion:
+    def _invoke(self, prompt) -> CommitSuggestion:
         model = ChatOpenAI(
             model=self.config.model,
             api_key=SecretStr(self.config.api_key),
             timeout=30,
         )
-        logger.info(
-            "Invoking %s model with prompt:\n%s", self.config.model, get_buffer_string(prompt)
-        )
+        logger.info("Invoking %s model with prompt:\n%s", self.config.model, prompt.to_string())
 
         agent = create_agent(
             model=model,
@@ -61,7 +56,7 @@ class OpenAIProvider(AIProvider):
             response_format=CommitSuggestion,
         )
 
-        result = agent.invoke(cast(InputAgentState, prompt))
+        result = agent.invoke(prompt)
 
         return result["structured_response"]
 
@@ -86,11 +81,16 @@ class OpenAIProvider(AIProvider):
             return suggestion
 
         suggestion = self._invoke(prompt)
+
         logger.info("Generated new commit suggestion.\n%s", suggestion)
 
         cache.set(change_set.diff, self.config.model, suggestion)
         logger.info("Cached new commit suggestion.")
 
+        suggestion = CommitSuggestion(
+            message=suggestion.message, description=suggestion.description
+        )
+
         return suggestion
 
     @staticmethod

{cmt_cli-1.0.1 → cmt_cli-1.0.3}/src/cmt/ai/prompt.py

@@ -59,14 +59,4 @@ Renamed: {renamed_files}
 Generate the commit message now.
 """
 
-COMMIT_PROMPT1 = ChatPromptTemplate.from_messages(
-    [
-        ("human", COMMIT_DATA_PROMPT),
-    ]
-)
-
-COMMIT_PROMPT = ChatPromptTemplate(
-    [
-        ("human", COMMIT_DATA_PROMPT),
-    ]
-)
+COMMIT_PROMPT_TEMPLATE = ChatPromptTemplate.from_template(COMMIT_DATA_PROMPT)

{cmt_cli-1.0.1 → cmt_cli-1.0.3}/src/cmt/cli.py

@@ -1,5 +1,4 @@
 import logging
-import sys
 from importlib.metadata import version
 
 import typer
@@ -66,6 +65,7 @@ def suggest() -> None:
         open_ai = OpenAIProvider()
 
         commit = open_ai.generate_commit_message(staged_files, analysis_result)
+
         commit_command = OpenAIProvider.commit_command(commit)
 
         logger.info("Suggested commit:\n\n%s", commit_command)

{cmt_cli-1.0.1 → cmt_cli-1.0.3}/uv.lock

@@ -175,7 +175,7 @@ wheels = [
 
 [[package]]
 name = "cmt-cli"
-version = "0.2.0"
+version = "1.0.3"
 source = { editable = "." }
 dependencies = [
     { name = "langchain" },