vimlm 0.0.1__py3-none-any.whl

vimlm/main.py

@@ -0,0 +1,72 @@
+import argparse
+import subprocess
+import tempfile
+import os
+from pathlib import Path
+
+VIM_CONFIG = """
+let s:watched_dir = expand('~/watched_dir')
+
+function! Monitor()
+    write
+    let response_path = s:watched_dir . '/response.md'
+    rightbelow vsplit | execute 'view ' . response_path
+    setlocal autoread
+    setlocal readonly
+    setlocal nobuflisted
+    filetype detect
+    syntax on
+    wincmd h
+    let s:monitor_timer = timer_start(1000, 'CheckForUpdates', {'repeat': -1})
+endfunction
+
+function! CheckForUpdates(timer)
+    let bufnum = bufnr(s:watched_dir . '/response.md')
+    if bufnum == -1
+        call timer_stop(s:monitor_timer)
+        return
+    endif
+    silent! checktime
+endfunction
+
+function! SaveUserInput()
+    let user_input = input('Ask LLM: ')
+    let user_file = s:watched_dir . '/user'
+    call writefile([user_input], user_file, 'w')
+    let current_file = expand('%:t')
+    let tree_file = s:watched_dir . '/tree'
+    call writefile([current_file], tree_file, 'w')
+endfunction
+
+vnoremap <c-l> :w! ~/watched_dir/yank<CR>:w! ~/watched_dir/context<CR>:call SaveUserInput()<CR>
+nnoremap <c-l> V:w! ~/watched_dir/yank<CR>:w! ~/watched_dir/context<CR>:call SaveUserInput()<CR>
+
+call Monitor()
+"""
+
+def main():
+    parser = argparse.ArgumentParser(description="VimLM - LLM-powered Vim assistant")
+    parser.add_argument("vim_args", nargs=argparse.REMAINDER, help="Additional Vim arguments")
+    args = parser.parse_args()
+    watch_dir = Path.home() / "watched_dir"
+    watch_dir.mkdir(exist_ok=True)
+    with tempfile.NamedTemporaryFile(mode='w', suffix='.vim', delete=False) as f:
+        f.write(VIM_CONFIG)
+        vim_script = f.name
+    try:
+        watcher = subprocess.Popen(
+            ["python", "-m", "vimlm.watcher"],
+            stdout=(watch_dir / "response.md").open("w"),
+            stderr=subprocess.STDOUT,
+        )
+        vim_command = ["vim", "-c", f"source {vim_script}"]
+        if args.vim_args:
+            vim_command.extend(args.vim_args)
+        subprocess.run(vim_command)
+    finally:
+        watcher.terminate()
+        watcher.wait()
+        os.unlink(vim_script)
+
+if __name__ == "__main__":
+    main()

vimlm/watcher.py

@@ -0,0 +1,59 @@
+import asyncio
+import subprocess
+import json
+import os
+from watchfiles import awatch
+from nanollama32 import Chat
+
+DEBUG = False
+NUM_TOKEN = 1000
+DIRECTORY = os.path.expanduser("~/watched_dir")  
+OUT_FILE = "response.md"
+LOG_FILE = "log.json"
+FILES = ["context", "yank", "user", "tree"]  
+
+out_path = os.path.join(DIRECTORY, OUT_FILE)
+log_path = os.path.join(DIRECTORY, LOG_FILE)
+os.makedirs(DIRECTORY, exist_ok=True)
+chat = Chat(variant='uncn_llama_32_3b_it')
+
+with open(out_path, "w", encoding="utf-8") as f:
+    f.write('LLM is ready')
+
+async def monitor_directory():
+    async for changes in awatch(DIRECTORY):
+        found_files = {os.path.basename(f) for _, f in changes}
+        if FILES[-1] in found_files and set(FILES).issubset(set(os.listdir(DIRECTORY))):
+            await process_files()
+
+async def process_files():
+    data = {}
+    for file in FILES:
+        path = os.path.join(DIRECTORY, file)
+        with open(path, "r", encoding="utf-8") as f:
+            data[file] = f.read().strip()
+        os.remove(path)  
+    data['ext'] = data['tree'].split('.')[-1]
+    if len(data['yank']) > 0:
+        str_template = "**{tree}**\n```{ext}\n{context}\n```\n\n```{ext}\n{yank}\n```\n\n{user}" if '\n' in data['yank'] else "**{tree}**\n```{ext}\n{context}\n```\n\n`{yank}` {user}"
+    else:
+        str_template = "**{tree}**\n```{ext}\n{context}\n\n```\n\n{user}"
+    prompt = str_template.format(**data)
+    with open(out_path, "w", encoding="utf-8") as f:
+        f.write('')
+    response = chat(prompt, max_new=NUM_TOKEN, verbose=DEBUG, stream=out_path)[0][:-10].strip()
+    with open(out_path, "w", encoding="utf-8") as f:
+        f.write(response)
+    chat.reset()
+    if DEBUG:
+        if os.path.exists(log_path):
+            with open(log_path, "r", encoding="utf-8") as log_f:
+                logs = json.load(log_f)
+        else:
+            logs = []
+        logs.append({'prompt':prompt, 'respone':response})
+        with open(log_path, "w", encoding="utf-8") as log_f:
+            json.dump(logs, log_f, indent=2)
+
+asyncio.run(monitor_directory())
+

vimlm-0.0.1.dist-info/METADATA

@@ -0,0 +1,55 @@
+Metadata-Version: 2.2
+Name: vimlm
+Version: 0.0.1
+Summary: VimLM - LLM-powered Vim assistant
+Home-page: https://github.com/JosefAlbers/vimlm
+Author: Josef Albers
+Author-email: albersj66@gmail.com
+Requires-Python: >=3.13.1
+Description-Content-Type: text/markdown
+Requires-Dist: nanollama==0.0.3
+Requires-Dist: watchfiles==1.0.4
+Dynamic: author
+Dynamic: author-email
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+# VimLM - LLM-powered Vim assistant
+
+## Features
+
+- Real-time code assistance using local LLMs
+- Context-aware suggestions based on your current file
+- Split-window interface showing LLM responses
+- Simple keybinding integration with Vim
+- Works completely offline with local models
+
+## Installation
+
+```zsh
+pip install vimlm
+```
+
+## Usage
+
+1. Start Vim with VimLM:
+
+```zsh
+vimlm your_file.js
+```
+
+2. Use the key bindings in Vim:
+- `Ctrl-L` in normal mode: Get suggestions for current line
+- `Ctrl-L` in visual mode: Get suggestions for selected code
+
+The LLM response will appear in a split window on the right side of your Vim interface.
+
+## Demo
+
+![vimlm](https://github.com/user-attachments/assets/4aa39efe-aa6d-4363-8fe1-cf964d7f849c)
+
+

vimlm-0.0.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+vimlm/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vimlm/main.py,sha256=oquliLkQSdPgPBf0sUBBSu7ZZAeerLMkms3H8kXBmIw,2164
+vimlm/watcher.py,sha256=4xI6LE-IhtgvLU8ho_pXCBL5vx8xRkkgQq9Iwr8-jeQ,2075
+vimlm-0.0.1.dist-info/METADATA,sha256=-4HQ3JOHYnfnCmVJ1Yej_7DKGOVzYrwJeQ9vjkV3Dzw,1246
+vimlm-0.0.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+vimlm-0.0.1.dist-info/entry_points.txt,sha256=s1QPno-YPS74cjm8JI_FeZqYpKO-QrTThGXUwl9shqk,42
+vimlm-0.0.1.dist-info/top_level.txt,sha256=I8GjqoiP--scYsO3AfLhha-6Ax9ci3IvbWvVbPv8g94,6
+vimlm-0.0.1.dist-info/RECORD,,

vimlm-0.0.1.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (75.8.0)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

vimlm-0.0.1.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+vimlm = vimlm.main:main

vimlm-0.0.1.dist-info/top_level.txt

@@ -0,0 +1 @@
+vimlm