synapse-pkg 2.0.0a3__tar.gz → 2.0.0a5__tar.gz

{synapse_pkg-2.0.0a3 → synapse_pkg-2.0.0a5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-pkg
-Version: 2.0.0a3
+Version: 2.0.0a5
 Summary: Knowledge graph extraction library powered by LLMs
 Author: Piyush Garg
 Requires-Python: >=3.10

{synapse_pkg-2.0.0a3 → synapse_pkg-2.0.0a5}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "synapse-pkg"
-version = "2.0.0a3"
+version = "2.0.0a5"
 description = "Knowledge graph extraction library powered by LLMs"
 readme = "README.md"
 requires-python = ">=3.10"

{synapse_pkg-2.0.0a3 → synapse_pkg-2.0.0a5}/synapse/cli.py

@@ -1,5 +1,4 @@
 import sys
-
 from . import commands
 
 
@@ -11,6 +10,7 @@ def main():
         print("Available commands:")
         print("  init")
         print("  knowledge")
+        print("  setup")
         return
 
     command = sys.argv[1].lower()
@@ -20,6 +20,9 @@ def main():
 
     elif command == "knowledge":
         commands.knowledge()
+    
+    elif command == "setup":
+        commands.setup()
 
     else:
         print(f"Unknown command: {command}")

synapse_pkg-2.0.0a5/synapse/commands.py

@@ -0,0 +1,39 @@
+from .persistence import initialize_synapse, save_config
+from .processor import build_repository_context
+from .extractor import build_compressed_context
+from .provider import get_ollama_model
+
+
+def setup():
+    models = get_ollama_model()
+
+    for i,model in enumerate(models,start=1):
+        print(f"{i}, {model}")
+    choice = int(input("select a model"))
+    selected_model = models[choice-1]
+    save_config(
+        {
+            "provider":"ollama",
+            "model": selected_model
+        }
+    )
+    print("configuration saved")
+
+
+def init():
+    initialize_synapse()
+
+
+def knowledge():
+    print("Initializing repository context...")
+
+    build_repository_context()
+
+    print(
+        "Repository context initialized.\n"
+        "Building compressed repository context..."
+    )
+
+    build_compressed_context()
+
+    print("Successfully initialized compressed repository context.")

{synapse_pkg-2.0.0a3 → synapse_pkg-2.0.0a5}/synapse/extractor.py

@@ -1,5 +1,5 @@
 import git
-from .persistence import load_compressed_context, save_compressed_context, load_metadata, save_metadata
+from .persistence import load_compressed_context, save_compressed_context, load_metadata, save_metadata, load_config
 from .git_manager import (
     get_repository_changes,
     split_commit_by_file
@@ -9,6 +9,8 @@ import json
 
 repo = git.Repo(search_parent_directories=True)
 
+config = load_config()
+
 
 MAX_PROMPT_SIZE = 30000
 MAX_FILES_PER_COMMIT = 3
@@ -625,9 +627,16 @@ def clean_json_response(raw: str) -> dict:
 
 def get_response(prompt, num_predict=4096):
 
+    if config["model"]:
+        model = config["model"]
+    else:
+        print("no model configured.")
+        print("Run 'synapse setup' first.")
+        return 
+
     try:
         response = chat(
-            model="qwen3:14b-q4_K_M",
+            model=model,
             messages=[
                 {
                     "role": "system",

synapse_pkg-2.0.0a5/synapse/provider.py

@@ -0,0 +1,8 @@
+import ollama
+def get_ollama_model():
+    response = ollama.list()
+    model_names=[]
+    for model in response.models:
+        model_names.append(model.model)
+    return model_names
+print(get_ollama_model())

{synapse_pkg-2.0.0a3 → synapse_pkg-2.0.0a5}/synapse_pkg.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: synapse-pkg
-Version: 2.0.0a3
+Version: 2.0.0a5
 Summary: Knowledge graph extraction library powered by LLMs
 Author: Piyush Garg
 Requires-Python: >=3.10

{synapse_pkg-2.0.0a3 → synapse_pkg-2.0.0a5}/synapse_pkg.egg-info/SOURCES.txt

@@ -8,6 +8,7 @@ synapse/extractor.py
 synapse/git_manager.py
 synapse/persistence.py
 synapse/processor.py
+synapse/provider.py
 synapse_pkg.egg-info/PKG-INFO
 synapse_pkg.egg-info/SOURCES.txt
 synapse_pkg.egg-info/dependency_links.txt

synapse_pkg-2.0.0a3/synapse/commands.py

@@ -1,22 +0,0 @@
-from .persistence import initialize_synapse
-from .processor import build_repository_context
-from .extractor import build_compressed_context
-
-
-def init():
-    initialize_synapse()
-
-
-def knowledge():
-    print("Initializing repository context...")
-
-    build_repository_context()
-
-    print(
-        "Repository context initialized.\n"
-        "Building compressed repository context..."
-    )
-
-    build_compressed_context()
-
-    print("Successfully initialized compressed repository context.")