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

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

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

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

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

@@ -1,7 +1,6 @@
 import logging
 
 from langchain.agents import create_agent
-from langchain_core.messages import BaseMessage, get_buffer_string
 from langchain_openai import ChatOpenAI
 from pydantic import SecretStr
 
@@ -19,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_TEMPLATE.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
@@ -49,9 +48,7 @@ class OpenAIProvider(AIProvider):
             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,
@@ -82,6 +79,7 @@ class OpenAIProvider(AIProvider):
                 suggestion.message,
             )
             return suggestion
+
         suggestion = self._invoke(prompt)
 
         logger.info("Generated new commit suggestion.\n%s", suggestion)

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

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

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

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