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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.5
 Name: cmt-cli
-Version: 1.0.1
+Version: 1.0.2
 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.2}/pyproject.toml

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

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

@@ -1,14 +1,12 @@
 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
@@ -27,7 +25,7 @@ class OpenAIProvider(AIProvider):
         staged_files = self._process_files(changes.files)
         staged_diffs = changes.diff
 
-        return COMMIT_PROMPT.format_messages(
+        return COMMIT_PROMPT_TEMPLATE.format_messages(
             total_files=len(changes.files),
             added_files=analysis.added_files,
             modified_files=analysis.modified_files,
@@ -45,7 +43,7 @@ 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),
@@ -61,7 +59,7 @@ class OpenAIProvider(AIProvider):
             response_format=CommitSuggestion,
         )
 
-        result = agent.invoke(cast(InputAgentState, prompt))
+        result = agent.invoke(prompt)
 
         return result["structured_response"]
 
@@ -84,13 +82,17 @@ class OpenAIProvider(AIProvider):
                 suggestion.message,
             )
             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.2}/src/cmt/ai/prompt.py

@@ -65,7 +65,13 @@ COMMIT_PROMPT1 = ChatPromptTemplate.from_messages(
     ]
 )
 
-COMMIT_PROMPT = ChatPromptTemplate(
+COMMIT_PROMPT_TEMPLATE2 = ChatPromptTemplate(
+    [
+        ("human", COMMIT_DATA_PROMPT),
+    ]
+)
+
+COMMIT_PROMPT_TEMPLATE = ChatPromptTemplate.from_messages(
     [
         ("human", COMMIT_DATA_PROMPT),
     ]

{cmt_cli-1.0.1 → cmt_cli-1.0.2}/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.2}/uv.lock

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