vimlm 0.1.1a0__py3-none-any.whl → 0.1.2__py3-none-any.whl

{vimlm-0.1.1a0.dist-info → vimlm-0.1.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: vimlm
-Version: 0.1.1a0
+Version: 0.1.2
 Summary: VimLM - LLM-powered Vim assistant
 Home-page: https://github.com/JosefAlbers/vimlm
 Author: Josef Albers
@@ -20,7 +20,7 @@ Dynamic: requires-dist
 Dynamic: requires-python
 Dynamic: summary
 
-# VimLM - AI-Powered Coding Assistant for Vim
+# VimLM - AI-Powered Coding Assistant for Vim/NeoVim
 
 ![VimLM Demo](https://raw.githubusercontent.com/JosefAlbers/VimLM/main/assets/captioned_vimlm.gif)
 
@@ -38,8 +38,8 @@ Get started quickly with the [tutorial](tutorial.md).
 
 ## Requirements
 - Apple Silicon (M-series)
-- Python 3.12.8
-- Vim 9.1
+- Python v3.12.8
+- Vim v9.1 or NeoVim v0.10.4
 
 ## Quick Start
 ```bash
@@ -70,7 +70,7 @@ def quicksort(arr):
     return quicksort(left) + middle + quicksort(right)
 ```
 
-2. Use `Ctrl-j` to generate and insert the code (or `Ctrl-l` to trigger → `Ctrl-p` to insert)
+2. Use `Ctrl-j` to autocomplete
 
 ### **Repository-Level Code Completion**
 
@@ -205,3 +205,4 @@ Edit `~/vimlm/cfg.json`:
 ## License
 
 Apache 2.0 - See [LICENSE](LICENSE) for details.
+

vimlm-0.1.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+vimlm.py,sha256=_XhyfZgHpK2W6OKWp-6Ruif-oRfIUn8ZMesM3u18vVU,34092
+vimlm-0.1.2.dist-info/LICENSE,sha256=f1xgK8fAXg_intwnbc9nLkHf7ODPLtgpHs7DetQHOro,11343
+vimlm-0.1.2.dist-info/METADATA,sha256=lj82Tm60yrL2dOgXu9vbChC0vDfoJBA8hdpT5rMs5DY,6323
+vimlm-0.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+vimlm-0.1.2.dist-info/entry_points.txt,sha256=mU5V4MYsuIzCc6YB-Ro-6USSHWN5vHw8UDnTEoq0isw,36
+vimlm-0.1.2.dist-info/top_level.txt,sha256=I8GjqoiP--scYsO3AfLhha-6Ax9ci3IvbWvVbPv8g94,6
+vimlm-0.1.2.dist-info/RECORD,,

vimlm.py

@@ -28,13 +28,12 @@ import tempfile
 from pathlib import Path
 from string import Template
 import re
-
 import sys
 import tty
 import termios
 
 DEFAULTS = dict(
-    LLM_MODEL = "mlx-community/Qwen2.5-Coder-3B-Instruct-4bit", # None | "mlx-community/DeepSeek-R1-Distill-Qwen-7B-4bit" | "mlx-community/deepseek-r1-distill-qwen-1.5b" |  "mlx-community/phi-4-4bit" (8.25gb) |  "mlx-community/Qwen2.5-Coder-14B-Instruct-4bit" (8.31gb) |  "mlx-community/Qwen2.5-Coder-3B-Instruct-4bit" (1.74gb)
+    LLM_MODEL = "mlx-community/Qwen2.5-Coder-3B-Instruct-4bit", # None | "mlx-community/DeepSeek-R1-Distill-Qwen-7B-4bit" | "mlx-community/deepseek-r1-distill-qwen-1.5b" |  "mlx-community/phi-4-4bit" (8.25gb) |  "mlx-community/Qwen2.5-Coder-14B-Instruct-4bit" (8.31gb) |  "mlx-community/Qwen2.5-Coder-3B-Instruct-4bit" (1.74gb) | "mlx-community/phi-4-4bit" (8.25gb)
     FIM_MODEL = "mlx-community/Qwen2.5-Coder-0.5B-4bit", # None | "mlx-community/Qwen2.5-Coder-32B-4bit" |  "mlx-community/Qwen2.5-Coder-0.5B-4bit" (278mb)
     NUM_TOKEN = 2000,
     USE_LEADER = False,
@@ -43,8 +42,8 @@ DEFAULTS = dict(
     SHOW_USER = False, 
     SEP_CMD = '!',
     THINK = ('<think>', '</think>'),
-    VERSION = '0.1.0',
-    DEBUG = True,
+    VERSION = '0.1.2',
+    DEBUG = False,
 )
 
 DATE_FORM = "%Y_%m_%d_%H_%M_%S"
@@ -555,10 +554,10 @@ function! SaveUserInput(prompt)
         return
     endif
     let user_file = s:watched_dir . '/user'
-    call writefile([user_input], user_file, 'w')
+    call writefile([user_input], user_file)
     let current_file = expand('%:p')
     let tree_file = s:watched_dir . '/tree'
-    call writefile([current_file], tree_file, 'w')
+    call writefile([current_file], tree_file)
     call ScrollToTop()
 endfunction
 
@@ -577,9 +576,9 @@ function! NormalPrompt()
 endfunction
 
 function! FollowUpPrompt()
-    call writefile([], s:watched_dir . '/yank', 'w')
-    call writefile([], s:watched_dir . '/context', 'w')
-    call writefile([], s:watched_dir . '/followup', 'w')
+    call writefile([], s:watched_dir . '/yank')
+    call writefile([], s:watched_dir . '/context')
+    call writefile([], s:watched_dir . '/followup')
     call SaveUserInput('... ')
 endfunction
 
@@ -651,7 +650,7 @@ function! VimLM(...) range
     while filereadable(wip_file)
         sleep 100m
     endwhile
-    call writefile([], wip_file, 'w')
+    call writefile([], wip_file)
     let user_input = join(a:000, ' ')
     if empty(user_input)
         echo "Usage: :VimLM <prompt> [!command1] [!command2] ..."
@@ -663,10 +662,10 @@ function! VimLM(...) range
     silent execute "'<,'>w! " . s:watched_dir . "/yank"
     silent execute "w! " . s:watched_dir . "/context"
     let user_file = s:watched_dir . '/user'
-    call writefile([user_input], user_file, 'w')
+    call writefile([user_input], user_file)
     let current_file = expand('%:p')
     let tree_file = s:watched_dir . '/tree'
-    call writefile([current_file], tree_file, 'w')
+    call writefile([current_file], tree_file)
     call ScrollToTop()
 endfunction
 
@@ -683,7 +682,9 @@ function! SplitAtCursorInInsert()
     endif
     let suffix_lines = []
     let suffix_part = strpart(current_line, col - 1)
-    if !empty(suffix_part)
+    if empty(suffix_part)
+        call add(suffix_lines, "")
+    else
         call add(suffix_lines, suffix_part)
     endif
     if line_num < len(lines)
@@ -691,11 +692,11 @@ function! SplitAtCursorInInsert()
     endif
     call writefile(prefix_lines, s:watched_dir . '/context', 'b')
     call writefile(suffix_lines, s:watched_dir . '/yank', 'b')
-    call writefile([], s:watched_dir . '/fim', 'w')
-    call writefile([], s:watched_dir . '/user', 'w')
+    call writefile([], s:watched_dir . '/fim')
+    call writefile([], s:watched_dir . '/user')
     let current_file = expand('%:p')
     let tree_file = s:watched_dir . '/tree'
-    call writefile([current_file], tree_file, 'w')
+    call writefile([current_file], tree_file)
     call ScrollToTop()
 endfunction
 
@@ -705,8 +706,10 @@ function! InsertResponse()
         echoerr "Response file not found: " . response_path
         return
     endif
-    let content = readfile(response_path)
+    let content = readfile(response_path, 'b')
     let text = join(content, "\n")
+    let saved_z = getreg('z')
+    let saved_ztype = getregtype('z')
     call setreg('z', text)
     let col = col('.')
     let line = getline('.')
@@ -715,11 +718,12 @@ function! InsertResponse()
     else
         normal! "zgP
     endif
+    call setreg('z', saved_z, saved_ztype)
 endfunction
 
 function! TabInInsert()
     let wip_file = s:watched_dir . '/wip'
-    call writefile([], wip_file, 'w')
+    call writefile([], wip_file)
     call SplitAtCursorInInsert()
     while filereadable(wip_file)
         sleep 10m

vimlm-0.1.1a0.dist-info/RECORD

@@ -1,7 +0,0 @@
-vimlm.py,sha256=s9IT_-uaP2YmbtM4NapC55ZkH9731_eWtwYr0Vs4Z3k,33955
-vimlm-0.1.1a0.dist-info/LICENSE,sha256=f1xgK8fAXg_intwnbc9nLkHf7ODPLtgpHs7DetQHOro,11343
-vimlm-0.1.1a0.dist-info/METADATA,sha256=TrcJfO5jhyjNzTAPKdSHYRPk_13itOn8pS24MMpe03w,6361
-vimlm-0.1.1a0.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-vimlm-0.1.1a0.dist-info/entry_points.txt,sha256=mU5V4MYsuIzCc6YB-Ro-6USSHWN5vHw8UDnTEoq0isw,36
-vimlm-0.1.1a0.dist-info/top_level.txt,sha256=I8GjqoiP--scYsO3AfLhha-6Ax9ci3IvbWvVbPv8g94,6
-vimlm-0.1.1a0.dist-info/RECORD,,