vimlm 0.0.3__py3-none-any.whl → 0.0.4__py3-none-any.whl

{vimlm-0.0.3.dist-info → vimlm-0.0.4.dist-info}/METADATA

@@ -1,14 +1,14 @@
 Metadata-Version: 2.2
 Name: vimlm
-Version: 0.0.3
+Version: 0.0.4
 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
+Requires-Python: >=3.12.8
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: nanollama==0.0.3
+Requires-Dist: nanollama==0.0.3b0
 Requires-Dist: watchfiles==1.0.4
 Dynamic: author
 Dynamic: author-email
@@ -21,11 +21,9 @@ Dynamic: summary
 
 # VimLM - Vim Language Model Assistant for privacy-conscious developers
 
-![vimlm](https://github.com/user-attachments/assets/4aa39efe-aa6d-4363-8fe1-cf964d7f849c)
+![vimlm](https://github.com/user-attachments/assets/67a97048-48df-40c1-9109-53c759e85d96)
 
-VimLM brings AI-powered assistance directly into Vim, keeping your workflow smooth and uninterrupted. Instead of constantly switching between your editor and external tools, VimLM provides real-time, context-aware suggestions, explanations, and code insights—all within Vim.
-
-Unlike proprietary cloud-based AI models, VimLM runs entirely offline, ensuring complete privacy, data security, and control. You’re not just using a tool—you own it. Fine-tune, modify, or extend it as needed, without reliance on big tech platforms.
+An LLM-powered coding companion for Vim, inspired by GitHub Copilot/Cursor. Integrates contextual code understanding, summarization, and AI assistance directly into your Vim workflow.
 
 ## Features
 
@@ -70,15 +68,15 @@ vimlm your_file.js
     - Example prompt: "AJAX-ify this app !@#$ ~/scrap/jph00/hypermedia-applications.summ.md"
 
 5. **Follow-Up**: After initial response:
-    - `Ctrl-R`: Continue thread
+    - `Ctrl-r`: Continue thread
     - Example follow-up: "In Manifest V3"
 
 ## Key Bindings
 
 | Binding    | Mode          | Action                                 |
 |------------|---------------|----------------------------------------|
-| `Ctrl-L`   | Normal/Visual | Send current file + selection to LLM   |
-| `Ctrl-R`   | Normal        | Continue conversation                  |
+| `Ctrl-l`   | Normal/Visual | Send current file + selection to LLM   |
+| `Ctrl-r`   | Normal        | Continue conversation                  |
 | `Esc`      | Prompt        | Cancel input                           |
 
 

vimlm-0.0.4.dist-info/RECORD

@@ -0,0 +1,7 @@
+vimlm.py,sha256=kCJphqs6B-ZUGzeikiW3u629Nw5Ih3DlDJV1rsKGXAY,13410
+vimlm-0.0.4.dist-info/LICENSE,sha256=f1xgK8fAXg_intwnbc9nLkHf7ODPLtgpHs7DetQHOro,11343
+vimlm-0.0.4.dist-info/METADATA,sha256=EdZyb9SyLpy0qyK_jILHa7nrd5G8lPxOm6Xfy7NMfiQ,2832
+vimlm-0.0.4.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+vimlm-0.0.4.dist-info/entry_points.txt,sha256=mU5V4MYsuIzCc6YB-Ro-6USSHWN5vHw8UDnTEoq0isw,36
+vimlm-0.0.4.dist-info/top_level.txt,sha256=I8GjqoiP--scYsO3AfLhha-6Ax9ci3IvbWvVbPv8g94,6
+vimlm-0.0.4.dist-info/RECORD,,

vimlm.py

@@ -26,23 +26,35 @@ import tempfile
 from pathlib import Path
 from string import Template
 
-DEBUG = False 
+DEBUG = True
 NUM_TOKEN = 2000
 SEP_CMD = '!@#$'
-CACHE_DIR = os.path.expanduser("~/vimlm")
+VIMLM_DIR = os.path.expanduser("~/vimlm")
 WATCH_DIR = os.path.expanduser("~/vimlm/watch_dir")
+CFG_FILE = "cfg.json"
 LOG_FILE = "log.json"
-CACHE_FILE = "cache.json"
+LTM_FILE = "cache.json"
 OUT_FILE = "response.md"
-FILES = ["context", "yank", "user", "tree"]  
-LOG_PATH = os.path.join(CACHE_DIR, LOG_FILE)
-CACHE_PATH = os.path.join(CACHE_DIR, CACHE_FILE)
+IN_FILES = ["context", "yank", "user", "tree"]
+CFG_PATH = os.path.join(VIMLM_DIR, CFG_FILE)
+LOG_PATH = os.path.join(VIMLM_DIR, LOG_FILE)
+LTM_PATH = os.path.join(VIMLM_DIR, LTM_FILE)
 OUT_PATH = os.path.join(WATCH_DIR, OUT_FILE) 
 
 if os.path.exists(WATCH_DIR):
     shutil.rmtree(WATCH_DIR)
 os.makedirs(WATCH_DIR)
 
+try:
+    with open(CFG_PATH, "r") as f:
+        cfg = cfg.load(f)
+    DEBUG = config.get("DEBUG", DEBUG)
+    NUM_TOKEN = config.get("NUM_TOKEN", NUM_TOKEN)
+    SEP_CMD = config.get("SEP_CMD", SEP_CMD)
+except:
+    with open(CFG_PATH, 'w') as f:
+        json.dump(dict(DEBUG=DEBUG, NUM_TOKEN=NUM_TOKEN, SEP_CMD=SEP_CMD), f, indent=2)
+
 def toout(s, key='tovim'):
     with open(OUT_PATH, 'w', encoding='utf-8') as f:
         f.write(s)
@@ -102,11 +114,11 @@ def split_str(doc, max_len=2000, get_len=len):
             chunks.append("".join(current_chunk))
     return chunks
 
-def get_context(src_path, max_len=2000, get_len=len):
+def retrieve(src_path, max_len=2000, get_len=len):
     src_path = os.path.expanduser(src_path)
     result = {}
     if not os.path.exists(src_path):
-        tolog(f"The path {src_path} does not exist.", 'get_context')
+        tolog(f"The path {src_path} does not exist.", 'retrieve')
         return result
     if os.path.isfile(src_path):
         try:
@@ -114,7 +126,7 @@ def get_context(src_path, max_len=2000, get_len=len):
                 content = f.read()
             result = {src_path:dict(timestamp=os.path.getmtime(src_path), list_str=split_str(content, max_len=max_len, get_len=get_len))}
         except Exception as e:
-            tolog(f'Skipped {filename} due to {e}', 'get_context')
+            tolog(f'Skipped {filename} due to {e}', 'retrieve')
         return result
     for filename in os.listdir(src_path):
         try:
@@ -126,7 +138,7 @@ def get_context(src_path, max_len=2000, get_len=len):
                     content = f.read()
                 result[file_path] = dict(timestamp=os.path.getmtime(file_path), list_str=split_str(content, max_len=max_len, get_len=get_len))
         except Exception as e:
-            tolog(f'Skipped {filename} due to {e}', 'get_context')
+            tolog(f'Skipped {filename} due to {e}', 'retrieve')
             continue
     return result
 
@@ -134,12 +146,12 @@ def get_ntok(s):
     return len(chat.tokenizer.encode(s)[0])
 
 def ingest(src):
-    def load_cache(cache_path=CACHE_PATH):
+    def load_cache(cache_path=LTM_PATH):
         if os.path.exists(cache_path):
             with open(cache_path, 'r', encoding='utf-8') as f:
                 return json.load(f)
         return {}
-    def dump_cache(new_data, cache_path=CACHE_PATH):
+    def dump_cache(new_data, cache_path=LTM_PATH):
         current_data = load_cache(cache_path)
         for k, v in new_data.items():
             if k not in current_data or v['timestamp'] > current_data[k]['timestamp']:
@@ -149,7 +161,7 @@ def ingest(src):
     toout('Ingesting...')
     format_ingest = '{volat}{incoming}\n\n---\n\nPlease provide a succint bullet point summary for above:'
     format_volat = 'Here is a summary of part 1 of **{k}**:\n\n---\n\n{newsum}\n\n---\n\nHere is the next part:\n\n---\n\n'
-    dict_doc = get_context(src, get_len=get_ntok)
+    dict_doc = retrieve(src, get_len=get_ntok)
     dict_sum = {}
     cache = load_cache()
     for k, v in dict_doc.items():
@@ -197,10 +209,10 @@ async def monitor_directory():
     async for changes in awatch(WATCH_DIR):
         found_files = {os.path.basename(f) for _, f in changes}
         tolog(f'{found_files=}') # DEBUG
-        if FILES[-1] in found_files and set(FILES).issubset(set(os.listdir(WATCH_DIR))):
+        if IN_FILES[-1] in found_files and set(IN_FILES).issubset(set(os.listdir(WATCH_DIR))):
             tolog(f'listdir()={os.listdir(WATCH_DIR)}') # DEBUG
             data = {}
-            for file in FILES:
+            for file in IN_FILES:
                 path = os.path.join(WATCH_DIR, file)
                 with open(path, 'r', encoding='utf-8') as f:
                     data[file] = f.read().strip()
@@ -241,8 +253,9 @@ async def process_files(data):
     prompt = str_template.format(**data)
     tolog(prompt, 'tollm')
     toout('')
-    response = chat(prompt, max_new=NUM_TOKEN - get_ntok(prompt), verbose=False, stream=OUT_PATH)[0][:-10].strip()
-    toout(response)
+    response = chat(prompt, max_new=NUM_TOKEN - get_ntok(prompt), verbose=False, stream=OUT_PATH)
+    toout(response[0][:-10].strip())
+    tolog(response[-1], 'tps')
 
 VIMLMSCRIPT = Template(r"""
 let s:watched_dir = expand('$WATCH_DIR')

vimlm-0.0.3.dist-info/RECORD

@@ -1,7 +0,0 @@
-vimlm.py,sha256=ku7NOIrcCNHzRYP9_8qp1NX8w5uCiZAlf0ijMKJXTlw,12996
-vimlm-0.0.3.dist-info/LICENSE,sha256=f1xgK8fAXg_intwnbc9nLkHf7ODPLtgpHs7DetQHOro,11343
-vimlm-0.0.3.dist-info/METADATA,sha256=i4W8tmpaMH1m86vzEwNpwo9BUfYcLzr3nksEU_iGW6Y,3178
-vimlm-0.0.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-vimlm-0.0.3.dist-info/entry_points.txt,sha256=mU5V4MYsuIzCc6YB-Ro-6USSHWN5vHw8UDnTEoq0isw,36
-vimlm-0.0.3.dist-info/top_level.txt,sha256=I8GjqoiP--scYsO3AfLhha-6Ax9ci3IvbWvVbPv8g94,6
-vimlm-0.0.3.dist-info/RECORD,,