vimlm 0.0.1__tar.gz

vimlm-0.0.1/PKG-INFO

@@ -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/README.md

@@ -0,0 +1,35 @@
+# 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/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

vimlm-0.0.1/setup.py

@@ -0,0 +1,25 @@
+from setuptools import setup, find_packages
+
+with open("requirements.txt") as f:
+    requirements = [l.strip() for l in f.readlines()]
+
+setup(
+    name="vimlm",
+    version="0.0.1",
+    author="Josef Albers",
+    author_email="albersj66@gmail.com",
+    readme='README.md',
+    description="VimLM - LLM-powered Vim assistant",
+    long_description=open('README.md', encoding='utf-8').read(),
+    long_description_content_type="text/markdown",
+    url="https://github.com/JosefAlbers/vimlm",
+    packages=find_packages(),
+    python_requires=">=3.13.1",
+    install_requires=requirements,
+    entry_points={
+        "console_scripts": [
+            "vimlm=vimlm.main:main",
+        ],
+    },
+)
+

vimlm-0.0.1/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-0.0.1/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/vimlm.egg-info/PKG-INFO

@@ -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/vimlm.egg-info/SOURCES.txt

@@ -0,0 +1,11 @@
+README.md
+setup.py
+vimlm/__init__.py
+vimlm/main.py
+vimlm/watcher.py
+vimlm.egg-info/PKG-INFO
+vimlm.egg-info/SOURCES.txt
+vimlm.egg-info/dependency_links.txt
+vimlm.egg-info/entry_points.txt
+vimlm.egg-info/requires.txt
+vimlm.egg-info/top_level.txt

vimlm-0.0.1/vimlm.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

vimlm-0.0.1/vimlm.egg-info/entry_points.txt

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

vimlm-0.0.1/vimlm.egg-info/requires.txt

@@ -0,0 +1,2 @@
+nanollama==0.0.3
+watchfiles==1.0.4

vimlm-0.0.1/vimlm.egg-info/top_level.txt

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