gitpr-cli 0.0.23__py3-none-any.whl → 0.0.24__py3-none-any.whl

{gitpr_cli-0.0.23.dist-info → gitpr_cli-0.0.24.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gitpr-cli
-Version: 0.0.23
+Version: 0.0.24
 Summary: Automação de PRs, Commits e Code Review com IA (Gemini e DeepSeek)
 Author-email: Natan Fiuza <contato@natanfiuza.dev.br>
 Requires-Python: >=3.10

{gitpr_cli-0.0.23.dist-info → gitpr_cli-0.0.24.dist-info}/RECORD

@@ -1,23 +1,23 @@
-gitpr_cli-0.0.23.dist-info/licenses/LICENSE,sha256=UQzLC6mwqb-ZNDkBBAqOxC-SMQcEID0gaM6oFcu6u48,27030
+gitpr_cli-0.0.24.dist-info/licenses/LICENSE,sha256=UQzLC6mwqb-ZNDkBBAqOxC-SMQcEID0gaM6oFcu6u48,27030
 src/__init__.py,sha256=U6ksarernNcryHO20aj4rNG7w0thT7xdMs5KzBJyPvE,28
-src/ai_providers.py,sha256=3O91HmlnqE1o5SV0seley_gGH0-3BImVIluHIMpq9lU,3354
+src/ai_providers.py,sha256=rf4IJTG6P8nLKU-C-BdUdy68JvhnsgzvhEsJdp-9El0,3477
 src/blame_engine.py,sha256=056Rp5JkQU2cFB4b3lmIx99xjwfLj-tc7jwyevDoik8,10383
 src/cache.py,sha256=XggMTy00GlZfeF1VoLEYCdxL5h0eTg11TJfPIRPx7fs,3201
-src/config.py,sha256=QpKKpoP0RHpSl2s1zJDfx6wETKqJ8wLtn6T1zAnznis,6621
+src/config.py,sha256=J-Jg8jVCdtLsf9h4UZNrepS3bx09loOKzgZSWMELIQ8,6823
 src/core.py,sha256=eIpnTttGQhUwrEm7Xh59HYVu59LQcprf7K-IwN7kNYk,17965
 src/i18n.py,sha256=Bn1OrhCYEaT0kUidu5SeTu6kiIuWOZ4OjB-7mIb6RtY,3185
 src/issue_engine.py,sha256=QLEEHp9l6_QiaaLLrFtBU_-LkkxPE2Gs5-TLEYEBWCA,3613
 src/linter_engine.py,sha256=D04wvOgRa-vZEYknm-z-Cg7V_3gnoQz_64PMorWnISo,4414
-src/main.py,sha256=mzSYqNF7YgdI6vsuwWSgqUks_tA7Zn5gMk6fkebZFIY,27682
+src/main.py,sha256=TEmIUTaqX1oAPmknmu7Eq4p06KioMO8EvommuEOiDqI,27708
 src/security.py,sha256=Li0zM8ZURTK2nKRHIt49TWLO4DIddBEHzXPrVycv0Rs,1343
 src/spinner.py,sha256=UM-53CfrfQ6Ymg4-d2_pYgMaDdiEV69DFetaQtCyJzo,6055
 src/tui_issue.py,sha256=gcAJCIfMcdeMjrFJUWQQvdgBv7fnOBtxQxwdBk2Sk40,1789
-src/updater.py,sha256=giLCq9ygRXGhLbVqwLEDZrO2szEIk5nFUMl9NhiZZ90,5886
+src/updater.py,sha256=9huKjoz4eRC_i1pu5Lcw6_JNwLaS_drIVw7l6xiSE3o,5886
 src/ui/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 src/ui/help_screen.py,sha256=tjbokXhwyzLnkOC0S56NbpX-78hcFQnvsQMpNU1wt0c,1862
 src/ui/issue_app.py,sha256=8MdBcOVdCHPFx6EUhs80CiZ2G-29KulskkCz9qMZ_Lg,4737
-gitpr_cli-0.0.23.dist-info/METADATA,sha256=2rnIIo3C6OqfV1EZHF5YQNIp1u9FS4U8BJI5TM9FMVM,15488
-gitpr_cli-0.0.23.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
-gitpr_cli-0.0.23.dist-info/entry_points.txt,sha256=mh_4F_idM4cj0XhfRGBDQb5Nnku93YJ-dyKh5SufwlE,39
-gitpr_cli-0.0.23.dist-info/top_level.txt,sha256=74rtVfumQlgAPzR5_2CgYN24MB0XARCg0t-gzk6gTrM,4
-gitpr_cli-0.0.23.dist-info/RECORD,,
+gitpr_cli-0.0.24.dist-info/METADATA,sha256=36jmbuAbz0a1aCIr72fzPhXc-jKoqQglk8brhDRulh8,15488
+gitpr_cli-0.0.24.dist-info/WHEEL,sha256=K260EYznzXsJYBQGqmI8VTxEdiZYNvDZwW9cBh9-_MA,91
+gitpr_cli-0.0.24.dist-info/entry_points.txt,sha256=mh_4F_idM4cj0XhfRGBDQb5Nnku93YJ-dyKh5SufwlE,39
+gitpr_cli-0.0.24.dist-info/top_level.txt,sha256=74rtVfumQlgAPzR5_2CgYN24MB0XARCg0t-gzk6gTrM,4
+gitpr_cli-0.0.24.dist-info/RECORD,,

src/ai_providers.py

@@ -34,11 +34,13 @@ def call_ai_model(provider, api_key, api_model, prompt, system_instruction, quie
                     )
                     result_text = response.text
 
-                elif provider == "deepseek":
-                    # DeepSeek is 100% compatible with the OpenAI library
-                    client = OpenAI(api_key=api_key, base_url="https://api.deepseek.com")
+                elif provider in ["deepseek", "ollama"]:
+                    # DeepSeek and Ollama are 100% compatible with the OpenAI library.
+                    base_url = "https://api.deepseek.com" if provider == "deepseek" else "http://localhost:11434/v1"
+                    client = OpenAI(api_key=api_key, base_url=base_url)
+                    
                     response = client.chat.completions.create(
-                        model=api_model,  # e.g.: "deepseek-chat"
+                        model=api_model, 
                         messages=[
                             {"role": "system", "content": system_instruction},
                             {"role": "user", "content": prompt}

src/config.py

@@ -18,6 +18,8 @@ DEFAULT_CONFIG = {
     "GEMINI_API_MODEL_SECONDARY": "gemini-flash-lite-latest",
     "DEEPSEEK_API_MODEL_PRIMARY": "deepseek-v4-pro",
     "DEEPSEEK_API_MODEL_SECONDARY": "deepseek-v4-flash",
+    "OLLAMA_API_MODEL_PRIMARY": "llama3",
+    "OLLAMA_API_MODEL_SECONDARY": "llama3",
     "OUTPUT_FILE_NAME": "{branch}_{datetime}_PR_DESC.md",
     "OUTPUT_FILE_NAME_REVIEW": "{branch}_{datetime}_PR_REVIEW.txt",
     "OUTPUT_FILE_NAME_FULLREVIEW": "{branch}_{datetime}_PR_FULLREVIEW.txt",
@@ -39,6 +41,8 @@ def get_api_key(provider):
         encrypted_key = os.getenv("GEMINI_API_KEY_ENCRYPTED")
     elif provider == "deepseek":
         encrypted_key = os.getenv("DEEPSEEK_API_KEY_ENCRYPTED")
+    elif provider == "ollama":
+        return "ollama-local" # Olama does not require authentication!
     else:
         return None
 
@@ -86,7 +90,7 @@ def setup_environment():
         click.secho(__("🤖 Welcome to GitPR! Let's configure your AI engine."), fg="cyan", bold=True)
         provider = click.prompt(
             __("Which artificial intelligence do you want to use as default?"),
-            type=click.Choice(['gemini', 'deepseek'], case_sensitive=False),
+            type=click.Choice(['gemini', 'deepseek', 'ollama'], case_sensitive=False),
             default='gemini'
         ).lower()
         set_key(ENV_FILE, "DEFAULT_AI_PROVIDER", provider)

src/main.py

@@ -96,7 +96,7 @@ HELP_MAP: dict[str, dict[str, str]] = {
     'provider': {
         'url': get_doc_url('providers-ia.md'),
         'title': __('AI Provider Selection (--provider)'),
-        'description': __('Forces the use of a specific AI provider for this execution: gemini (Google Gemini) or deepseek (DeepSeek). Temporarily overrides the default provider defined in the .env file.'),
+        'description': __('Forces the use of a specific AI provider for this execution: gemini (Google Gemini), deepseek (DeepSeek) or ollama (Local). Temporarily overrides the default provider defined in the .env file.'),
     },
 }
 
@@ -133,7 +133,7 @@ HELP_PRIORITY: dict[str, int] = {
 @click.option('-b', '--blame', type=str, help=__("Analyzes the origin of a business rule (e.g., file.py:10-20 or just file.py)."))
 @click.option('-ht', '--history', is_flag=True, help=__("Uses the entire branch history (Git Log + PR Cache) as context to generate the issue."))
 @click.option('-is', '--issue', is_flag=True, help=__("Generates a standardized Issue from current changes and opens the interactive interface."))
-@click.option('-p', '--provider', type=click.Choice(['gemini', 'deepseek']), help=__("Forces the use of a specific AI provider for this execution."))
+@click.option('-p', '--provider', type=click.Choice(['gemini', 'deepseek', 'ollama']), help=__("Forces the use of a specific AI provider for this execution."))
 @click.option('-h', '--help', 'help_flag', is_flag=True, help=__("Shows this message and exits. Use with another flag for contextual help (e.g., -h --issue)."))
 def cli(commit, review, fullreview, linter, skill, update, installhooks, hook, quiet, provider, input, blame, history, issue, help_flag):
     """

src/updater.py

@@ -7,8 +7,8 @@ import click
 from datetime import datetime
 
 # Current version of your local executable (Update this on every new build!)
-__version__ = "0.0.22"  # GitPR current version
-__lang_version__ = "v0.0.1"  # Language dictionary version control
+__version__ = "0.0.24"  # GitPR current version
+__lang_version__ = "v0.0.2"  # Language dictionary version control
 GITHUB_API_URL = "https://api.github.com/repos/natanfiuza/gitpr/releases/latest"
 PYPI_API_URL = "https://pypi.org/pypi/gitpr-cli/json"