diff --git a/.gitattributes b/.gitattributes new file mode 100644 index 0000000000000000000000000000000000000000..e69de29bb2d1d6434b8b29ae775ad8c2e48c5391 diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000000000000000000000000000000000000..04ad18820c66d89e64a8750edbbe95785419c607 --- /dev/null +++ b/.gitignore @@ -0,0 +1,13 @@ +node_modules +dist +.DS_Store +server/public +vite.config.ts.* +*.tar.gz +.agent_state* +.vscode +.latest* +rapid* +ac4a* +*.bin +*.gguf diff --git a/.replit b/.replit new file mode 100644 index 0000000000000000000000000000000000000000..adcfadc85500604d8f275e7e2219e3fbe623e9cc --- /dev/null +++ b/.replit @@ -0,0 +1,39 @@ +modules = ["nodejs-20", "web", "postgresql-16"] +run = "npm run dev" +hidden = [".config", ".git", "generated-icon.png", "node_modules", "dist"] + +[nix] +channel = "stable-24_05" + +[deployment] +deploymentTarget = "autoscale" +build = ["npm", "run", "build"] +run = ["npm", "run", "start"] + +[[ports]] +localPort = 5000 +externalPort = 80 + +[env] +PORT = "5000" + +[workflows] +runButton = "Project" + +[[workflows.workflow]] +name = "Project" +mode = "parallel" +author = "agent" + +[[workflows.workflow.tasks]] +task = "workflow.run" +args = "Start application" + +[[workflows.workflow]] +name = "Start application" +author = "agent" + +[[workflows.workflow.tasks]] +task = "shell.exec" +args = "npm run dev" +waitForPort = 5000 diff --git a/LICENSE.md b/LICENSE.md new file mode 100644 index 0000000000000000000000000000000000000000..07340b73d2d326c522f70c7547af4c879824ad2f --- /dev/null +++ b/LICENSE.md @@ -0,0 +1,201 @@ + Apache License + Version 2.0, January 2004 + http://www.apache.org/licenses/ + +TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION + +1. Definitions. + + "License" shall mean the terms and conditions for use, reproduction, + and distribution as defined by Sections 1 through 9 of this document. + + "Licensor" shall mean the copyright owner or entity authorized by + the copyright owner that is granting the License. + + "Legal Entity" shall mean the union of the acting entity and all + other entities that control, are controlled by, or are under common + control with that entity. For the purposes of this definition, + "control" means (i) the power, direct or indirect, to cause the + direction or management of such entity, whether by contract or + otherwise, or (ii) ownership of fifty percent (50%) or more of the + outstanding shares, or (iii) beneficial ownership of such entity. + + "You" (or "Your") shall mean an individual or Legal Entity + exercising permissions granted by this License. + + "Source" form shall mean the preferred form for making modifications, + including but not limited to software source code, documentation + source, and configuration files. + + "Object" form shall mean any form resulting from mechanical + transformation or translation of a Source form, including but + not limited to compiled object code, generated documentation, + and conversions to other media types. + + "Work" shall mean the work of authorship, whether in Source or + Object form, made available under the License, as indicated by a + copyright notice that is included in or attached to the work + (an example is provided in the Appendix below). + + "Derivative Works" shall mean any work, whether in Source or Object + form, that is based on (or derived from) the Work and for which the + editorial revisions, annotations, elaborations, or other modifications + represent, as a whole, an original work of authorship. For the purposes + of this License, Derivative Works shall not include works that remain + separable from, or merely link (or bind by name) to the interfaces of, + the Work and Derivative Works thereof. + + "Contribution" shall mean any work of authorship, including + the original version of the Work and any modifications or additions + to that Work or Derivative Works thereof, that is intentionally + submitted to Licensor for inclusion in the Work by the copyright owner + or by an individual or Legal Entity authorized to submit on behalf of + the copyright owner. For the purposes of this definition, "submitted" + means any form of electronic, verbal, or written communication sent + to the Licensor or its representatives, including but not limited to + communication on electronic mailing lists, source code control systems, + and issue tracking systems that are managed by, or on behalf of, the + Licensor for the purpose of discussing and improving the Work, but + excluding communication that is conspicuously marked or otherwise + designated in writing by the copyright owner as "Not a Contribution." + + "Contributor" shall mean Licensor and any individual or Legal Entity + on behalf of whom a Contribution has been received by Licensor and + subsequently incorporated within the Work. + +2. Grant of Copyright License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + copyright license to reproduce, prepare Derivative Works of, + publicly display, publicly perform, sublicense, and distribute the + Work and such Derivative Works in Source or Object form. + +3. Grant of Patent License. Subject to the terms and conditions of + this License, each Contributor hereby grants to You a perpetual, + worldwide, non-exclusive, no-charge, royalty-free, irrevocable + (except as stated in this section) patent license to make, have made, + use, offer to sell, sell, import, and otherwise transfer the Work, + where such license applies only to those patent claims licensable + by such Contributor that are necessarily infringed by their + Contribution(s) alone or by combination of their Contribution(s) + with the Work to which such Contribution(s) was submitted. If You + institute patent litigation against any entity (including a + cross-claim or counterclaim in a lawsuit) alleging that the Work + or a Contribution incorporated within the Work constitutes direct + or contributory patent infringement, then any patent licenses + granted to You under this License for that Work shall terminate + as of the date such litigation is filed. + +4. Redistribution. You may reproduce and distribute copies of the + Work or Derivative Works thereof in any medium, with or without + modifications, and in Source or Object form, provided that You + meet the following conditions: + + (a) You must give any other recipients of the Work or + Derivative Works a copy of this License; and + + (b) You must cause any modified files to carry prominent notices + stating that You changed the files; and + + (c) You must retain, in the Source form of any Derivative Works + that You distribute, all copyright, patent, trademark, and + attribution notices from the Source form of the Work, + excluding those notices that do not pertain to any part of + the Derivative Works; and + + (d) If the Work includes a "NOTICE" text file as part of its + distribution, then any Derivative Works that You distribute must + include a readable copy of the attribution notices contained + within such NOTICE file, excluding those notices that do not + pertain to any part of the Derivative Works, in at least one + of the following places: within a NOTICE text file distributed + as part of the Derivative Works; within the Source form or + documentation, if provided along with the Derivative Works; or, + within a display generated by the Derivative Works, if and + wherever such third-party notices normally appear. The contents + of the NOTICE file are for informational purposes only and + do not modify the License. You may add Your own attribution + notices within Derivative Works that You distribute, alongside + or as an addendum to the NOTICE text from the Work, provided + that such additional attribution notices cannot be construed + as modifying the License. + + You may add Your own copyright statement to Your modifications and + may provide additional or different license terms and conditions + for use, reproduction, or distribution of Your modifications, or + for any such Derivative Works as a whole, provided Your use, + reproduction, and distribution of the Work otherwise complies with + the conditions stated in this License. + +5. Submission of Contributions. Unless You explicitly state otherwise, + any Contribution intentionally submitted for inclusion in the Work + by You to the Licensor shall be under the terms and conditions of + this License, without any additional terms or conditions. + Notwithstanding the above, nothing herein shall supersede or modify + the terms of any separate license agreement you may have executed + with Licensor regarding such Contributions. + +6. Trademarks. This License does not grant permission to use the trade + names, trademarks, service marks, or product names of the Licensor, + except as required for reasonable and customary use in describing the + origin of the Work and reproducing the content of the NOTICE file. + +7. Disclaimer of Warranty. Unless required by applicable law or + agreed to in writing, Licensor provides the Work (and each + Contributor provides its Contributions) on an "AS IS" BASIS, + WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or + implied, including, without limitation, any warranties or conditions + of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A + PARTICULAR PURPOSE. You are solely responsible for determining the + appropriateness of using or redistributing the Work and assume any + risks associated with Your exercise of permissions under this License. + +8. Limitation of Liability. In no event and under no legal theory, + whether in tort (including negligence), contract, or otherwise, + unless required by applicable law (such as deliberate and grossly + negligent acts) or agreed to in writing, shall any Contributor be + liable to You for damages, including any direct, indirect, special, + incidental, or consequential damages of any character arising as a + result of this License or out of the use or inability to use the + Work (including but not limited to damages for loss of goodwill, + work stoppage, computer failure or malfunction, or any and all + other commercial damages or losses), even if such Contributor + has been advised of the possibility of such damages. + +9. Accepting Warranty or Additional Liability. While redistributing + the Work or Derivative Works thereof, You may choose to offer, + and charge a fee for, acceptance of support, warranty, indemnity, + or other liability obligations and/or rights consistent with this + License. However, in accepting such obligations, You may act only + on Your own behalf and on Your sole responsibility, not on behalf + of any other Contributor, and only if You agree to indemnify, + defend, and hold each Contributor harmless for any liability + incurred by, or claims asserted against, such Contributor by reason + of your accepting any such warranty or additional liability. + +END OF TERMS AND CONDITIONS + +APPENDIX: How to apply the Apache License to your work. + + To apply the Apache License to your work, attach the following + boilerplate notice, with the fields enclosed by brackets "[]" + replaced with your own identifying information. (Don't include + the brackets!) The text should be enclosed in the appropriate + comment syntax for the file format. We also recommend that a + file or class name and description of purpose be included on the + same "printed page" as the copyright notice for easier + identification within third-party archives. + +Copyright [yyyy] [name of copyright owner] + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. diff --git a/attached_assets/Pasted-Here-s-a-high-to-medium-level-outline-formatted-roadmap-for-turning-your-existing-embeddinggemma-30-1757473542795_1757473542795.txt b/attached_assets/Pasted-Here-s-a-high-to-medium-level-outline-formatted-roadmap-for-turning-your-existing-embeddinggemma-30-1757473542795_1757473542795.txt new file mode 100644 index 0000000000000000000000000000000000000000..0cbb6cf670b4752b511d29c29fc4a88212e16bff --- /dev/null +++ b/attached_assets/Pasted-Here-s-a-high-to-medium-level-outline-formatted-roadmap-for-turning-your-existing-embeddinggemma-30-1757473542795_1757473542795.txt @@ -0,0 +1,82 @@ +Here’s a high-to-medium-level, outline-formatted roadmap for turning your existing embeddinggemma-300m setup into a unified CLI tool powered by Phi-2. No code snippets—just clear steps and gap-checks. + +1. Define Your Modular Surface +Embedder + +Uses google/embeddinggemma-300m to turn text into vectors + +Retriever + +Indexes and searches embeddings (e.g., FAISS or Chroma) + +Generator + +Runs Phi-2 (2.7 B) for response generation based on retrieved context + +CLI + +Ties everything together with a command-line interface (e.g., Typer) + +2. Prepare Your Conda Environment +Confirm sentence-transformers and transformers are installed + +Add your vector store library (FAISS/Chroma) + +Install any quantization/runtime helpers for Phi-2 (GGUF, bitsandbytes, etc.) + +Allocate model cache paths under your project (e.g., ./models/) + +3. Acquire and Optimize Phi-2 +Download the Phi-2 weights from Hugging Face or Azure AI Studio + +Quantize to Q4/Q5 to fit within ~6–8 GB of RAM on your i5/16 GB rig + +Validate loading speed and memory footprint in isolation before integration + +4. Sketch Your Data Flow +User Input: typed command + query text + +Embedding: call Gemma → vector + +Retrieval: query vector → top-k document vectors + +Context Assembly: stitch retrieved snippets + +Generation: feed context + user prompt into Phi-2 → text answer + +Output: print or save response + +5. Fill Logic Holes +You need a vector store backend; plain in-memory lists won’t scale + +Chunk larger documents before embedding (e.g., 500 token windows) + +Watch Phi-2’s context window (typically ~2 K tokens) and trim if needed + +Plan caching: don’t re-embed the same files on every run + +Design a config file for model paths, thresholds, and CLI defaults + +6. Organize Your Directory +Code +gemma-phi2-cli/ +├── models/ +│ ├── embeddinggemma-300m/ +│ └── phi2-quantized/ +├── data/ +│ └── index/ # FAISS/Chroma files +├── src/ +│ ├── embedder.py # wraps Gemma +│ ├── retriever.py # builds & queries index +│ ├── generator.py # wraps Phi-2 +│ └── cli.py # Typer entry point +├── config.yaml # model locations, hyperparams +└── README.md +7. Plan Your CLI Commands +init → create or update index from a target directory + +search → semantic lookup, returns file names + scores + +ask → retrieval + generation in one step + +serve → optional local API for integrations \ No newline at end of file diff --git a/cli.py b/cli.py new file mode 100644 index 0000000000000000000000000000000000000000..4adf806cb8c69c6bcf04658047f5f9b3c00b8634 --- /dev/null +++ b/cli.py @@ -0,0 +1,100 @@ +import typer + +from src.embedder import embed_documents +from src.retriever import search_documents +from src.generator import generate_commit_message, fallback_commit_message +from src.diff_analyzer import get_staged_diff_chunks +from src.config_loader import load_config + + +# Initialize Typer app +app = typer.Typer() +config = load_config("config.yaml") +print("Config loaded:", config) + + +# ===== CLI Commands ===== +# `init` — Index Documents +@app.command() +def init(path: str = typer.Argument(...)): + embeddings = embed_documents(path, config) + if embeddings: + typer.echo(f"Documents indexed successfully: {len(embeddings)} documents") + else: + typer.echo("No documents were indexed.") + + +# `search` — Semantic Lookup +@app.command() +def search(query: str): + """ + Perform a semantic search on indexed documents. + + Args: + query (str): The search query string. + """ + results = search_documents(query, config) + for i, result in enumerate(results): + typer.echo(f"[{i+1}] {result}") + + +# `ask` — Retrieval-Augmented Generation (RAG) +@app.command() +def ask(query: str): + """ + Perform a retrieval-augmented generation (RAG) process to answer a user query. + + Args: + query (str): The user query string. + """ + context = search_documents(query, config) + prompt = "\n".join(context) + f"\n\nUser question: {query}" + try: + response = generate_commit_message(prompt, config) + except ( + RuntimeError + ): # Replace with the specific exception type raised by generate_commit_message + response = fallback_commit_message([]) + typer.echo(response) + + +# `commit` — Git Diff + Message Generation +@app.command() +def commit(preview: bool = True, apply: bool = False, dry_run: bool = False): + """ + Generate a commit message based on staged Git diff chunks. + + Args: + preview (bool): If True, display the suggested commit message. + apply (bool): If True, apply the commit with the generated message. + dry_run (bool): If True, display staged files and chunks without generating a message. + """ + file_list, diff_chunks = get_staged_diff_chunks() + + if dry_run: + typer.echo(f"Files staged: {len(file_list)}") + typer.echo(f"Chunks: {len(diff_chunks)}") + typer.echo( + f"Estimated tokens: {sum(len(chunk.split()) for chunk in diff_chunks)}" + ) + return + + prompt = "\n".join(diff_chunks) + "\n\nGenerate a commit message:" + try: + message = generate_commit_message(prompt, config) + except Exception: + message = fallback_commit_message(file_list) + + if preview: + typer.echo(f"Suggested commit message:\n{message}") + if apply: + import subprocess + + subprocess.run(["git", "commit", "-m", message]) + + +from sentence_transformers import SentenceTransformer + +model = SentenceTransformer("./models/embeddinggemma-300m") +emb = model.encode("Test string") +print("Embedding shape:", emb.shape) # Should be (768,) diff --git a/client/index.html b/client/index.html new file mode 100644 index 0000000000000000000000000000000000000000..0d05af327082f583ac298ab6103296eee7a0b3fd --- /dev/null +++ b/client/index.html @@ -0,0 +1,16 @@ + + + + + + + + + + +
+ + + + + \ No newline at end of file diff --git a/client/src/App.tsx b/client/src/App.tsx new file mode 100644 index 0000000000000000000000000000000000000000..77bf6836b1417ccae1192042eb5c545a33cfeef6 --- /dev/null +++ b/client/src/App.tsx @@ -0,0 +1,31 @@ +import { Switch, Route } from "wouter"; +import { queryClient } from "./lib/queryClient"; +import { QueryClientProvider } from "@tanstack/react-query"; +import { Toaster } from "@/components/ui/toaster"; +import { TooltipProvider } from "@/components/ui/tooltip"; +import Dashboard from "./pages/dashboard"; +import NotFound from "@/pages/not-found"; + +function Router() { + return ( + + + + + ); +} + +function App() { + return ( + + +
+ + +
+ + + ); +} + +export default App; diff --git a/client/src/components/sections/commands.tsx b/client/src/components/sections/commands.tsx new file mode 100644 index 0000000000000000000000000000000000000000..d0ec8e141721a1de916a463f04f1d2e42b53d947 --- /dev/null +++ b/client/src/components/sections/commands.tsx @@ -0,0 +1,117 @@ +import { Card } from "@/components/ui/card"; +import { TerminalComponent } from "@/components/ui/terminal"; +import { Terminal, Play, Settings } from "lucide-react"; + +export function CommandsSection() { + const coreCommands = [ + { + command: "gemma-phi2-cli init [path]", + description: "Initialize or update index from target directory", + color: "text-accent" + }, + { + command: "gemma-phi2-cli search [query]", + description: "Semantic lookup returning file names and scores", + color: "text-accent" + }, + { + command: "gemma-phi2-cli ask [question]", + description: "Retrieval + generation in one step", + color: "text-accent" + }, + { + command: "gemma-phi2-cli serve", + description: "Launch local API server for integrations", + color: "text-accent" + }, + { + command: "gemma-phi2-cli commit [--preview|--apply]", + description: "Generate commit messages from staged changes", + color: "text-accent" + }, + { + command: "gemma-phi2-cli commit --dry-run", + description: "Show file count, chunks, and token estimates", + color: "text-accent" + } + ]; + + const utilityCommands = [ + { + command: "gemma-phi2-cli status", + description: "Show current configuration and model status", + color: "text-accent" + }, + { + command: "gemma-phi2-cli config [key] [value]", + description: "Get or set configuration parameters", + color: "text-accent" + }, + { + command: "gemma-phi2-cli clear", + description: "Clear vector store and cached embeddings", + color: "text-accent" + }, + { + command: "gemma-phi2-cli --help", + description: "Display detailed help and usage examples", + color: "text-accent" + }, + { + command: "gemma-phi2-cli commit --preview", + description: "Preview generated commit message without applying", + color: "text-accent" + }, + { + command: "gemma-phi2-cli config --list-presets", + description: "Show available commit style presets", + color: "text-accent" + } + ]; + + return ( +
+

CLI Commands

+ +
+ + +
+ +

+ + Core Commands +

+
+ {coreCommands.map((cmd, index) => ( + +
+ {cmd.command} +
+
{cmd.description}
+ + ))} +
+ + + +

+ + Utility Commands +

+
+ {utilityCommands.map((cmd, index) => ( + +
+ {cmd.command} +
+
{cmd.description}
+ + ))} +
+ +
+
+
+ ); +} diff --git a/client/src/components/sections/configuration.tsx b/client/src/components/sections/configuration.tsx new file mode 100644 index 0000000000000000000000000000000000000000..ce9166bc213ac62598e8cf38092ad33a76978d85 --- /dev/null +++ b/client/src/components/sections/configuration.tsx @@ -0,0 +1,105 @@ +import { Card } from "@/components/ui/card"; +import { Button } from "@/components/ui/button"; +import { FileCode, Edit, Save } from "lucide-react"; + +export function ConfigurationSection() { + const configContent = `# Gemma-Phi2-CLI Configuration +models: + embedder: + path: "./models/embeddinggemma-300m" + batch_size: 32 + device: "cpu" + generator: + path: "./models/phi2-quantized.gguf" + quantization: "Q4_0" + max_tokens: 2048 + temperature: 0.7 + format: "gguf" + +embedding: + dim: 768 + truncate_to: 128 + sentence_transformers: true + +vector_store: + type: "faiss" + index_path: "./data/index" + persistence: true + +retrieval: + top_k: 5 + similarity_threshold: 0.75 + rerank: false + +performance: + cache_embeddings: true + chunk_size: 500 + chunk_overlap: 50 + +commit: + tone: "imperative" + style: "conventional" # Options: conventional | descriptive | terse + max_length: 72 + preview_by_default: true + fallback_enabled: true + dry_run_default: false + +logging: + level: "INFO" + file: "./logs/gemma-phi2.log" + verbose: true + telemetry: false + rotate_logs: true + +environment: + conda_env: "gemma-phi2-cli" + python_version: "3.10" + platform: "windows"`; + + return ( +
+

Configuration Management

+ + +
+

+ + config.yaml +

+
+ + +
+
+
+
+
+ {configContent.split('\n').map((line, index) => { + let className = ''; + if (line.trim().startsWith('#')) { + className = 'comment'; + } else if (line.includes(':') && !line.trim().startsWith(' ')) { + className = 'keyword'; + } else if (line.includes('"')) { + className = 'string'; + } + + return ( +
+ {line || '\u00A0'} +
+ ); + })} +
+
+
+ +
+ ); +} \ No newline at end of file diff --git a/client/src/components/sections/deployment.tsx b/client/src/components/sections/deployment.tsx new file mode 100644 index 0000000000000000000000000000000000000000..e0a48fd178739f761924c03e5f5b4590add18350 --- /dev/null +++ b/client/src/components/sections/deployment.tsx @@ -0,0 +1,124 @@ +import { Card } from "@/components/ui/card"; +import { Button } from "@/components/ui/button"; +import { Progress } from "@/components/ui/progress"; +import { Server, Download, Rocket } from "lucide-react"; + +export function DeploymentSection() { + const systemRequirements = [ + { label: "CPU", value: "Intel i5+ / AMD Ryzen 5+" }, + { label: "RAM", value: "16 GB minimum" }, + { label: "Storage", value: "10 GB+ SSD" }, + { label: "Platform", value: "Windows 11 (primary)" }, + { label: "Python", value: "3.10 (Conda managed)" }, + { label: "Models", value: "CPU-only execution" } + ]; + + const installationSteps = [ + { name: "Conda Environment", progress: 100, status: "✓ Complete", color: "bg-green-400" }, + { name: "sentence-transformers", progress: 100, status: "✓ Complete", color: "bg-green-400" }, + { name: "FAISS + Typer + PyYAML", progress: 100, status: "✓ Complete", color: "bg-green-400" }, + { name: "EmbeddingGemma-300m", progress: 75, status: "⟳ Downloading", color: "bg-yellow-400" }, + { name: "Phi-2 GGUF (Q4_0)", progress: 0, status: "⏳ Pending", color: "bg-muted" } + ]; + + const quickStartSteps = [ + { + number: 1, + title: "Conda Setup", + commands: ["conda env create -f environment.yml", "conda activate gemma-phi2-cli", "pip install -e ."], + color: "bg-primary" + }, + { + number: 2, + title: "Model Download", + commands: ["python download_models.py", "# Downloads EmbeddingGemma + Phi-2 GGUF"], + color: "bg-accent" + }, + { + number: 3, + title: "CLI Usage", + commands: ["gemma-phi2-cli init ./my-project", "git add .", "gemma-phi2-cli commit --preview"], + color: "bg-green-400" + } + ]; + + return ( +
+

Deployment & Setup

+ +
+ {/* System Requirements */} + +

+ + System Requirements +

+
+ {systemRequirements.map((req, index) => ( +
+ {req.label} + + {req.value} + +
+ ))} +
+ + + {/* Installation Progress */} + +

+ + Installation Progress +

+
+ {installationSteps.map((step, index) => ( +
+
+ {step.name} + + {step.status} + +
+ +
+ ))} +
+ +
+ + {/* Quick Start Guide */} + +

+ + Quick Start Guide +

+
+ {quickStartSteps.map((step, index) => ( +
+
+ {step.number} +
+

{step.title}

+ + {step.commands.map((command, cmdIndex) => ( +
+ {command} +
+ ))} + +
+ ))} +
+ +
+ ); +} diff --git a/client/src/components/sections/modules.tsx b/client/src/components/sections/modules.tsx new file mode 100644 index 0000000000000000000000000000000000000000..3b951e8383f1d3718470a52bd21b57a5525eee9d --- /dev/null +++ b/client/src/components/sections/modules.tsx @@ -0,0 +1,395 @@ +import { Card } from "@/components/ui/card"; +import { Input } from "@/components/ui/input"; +import { Label } from "@/components/ui/label"; +import { Select, SelectContent, SelectItem, SelectTrigger, SelectValue } from "@/components/ui/select"; +import { Badge } from "@/components/ui/badge"; +import { Brain, Search, Wand2, GitBranch, ChevronDown, CheckCircle, Clock } from "lucide-react"; +import { useState } from "react"; + +export function ModulesSection() { + const [expandedModules, setExpandedModules] = useState>(new Set(["embedder"])); + + const toggleModule = (moduleId: string) => { + const newExpanded = new Set(expandedModules); + if (newExpanded.has(moduleId)) { + newExpanded.delete(moduleId); + } else { + newExpanded.add(moduleId); + } + setExpandedModules(newExpanded); + }; + + return ( +
+

Module Configuration

+ + {/* Embedder Module */} + +
+
+
+ +

Embedder Module

+ + Active + +
+ +
+
+ {expandedModules.has("embedder") && ( +
+
+
+ + +
+
+ + +
+
+ +

+ + Implementation Status +

+
+
+ + Model loading interface +
+
+ + Text preprocessing pipeline +
+
+ + Batch processing optimization +
+
+ +
+ )} + + + {/* Retriever Module */} + +
+
+
+ +

Retriever Module

+ + Active + +
+ +
+
+ {expandedModules.has("retriever") && ( +
+
+
+ + +
+
+ + +
+
+ + +
+
+ +

Index Statistics

+
+
+
1,247
+
Documents
+
+
+
89,432
+
Embeddings
+
+
+
2.3GB
+
Index Size
+
+
+ +
+ )} + + + {/* DiffAnalyzer Module */} + +
+
+
+ +

DiffAnalyzer Module

+ + Active + +
+ +
+
+ {expandedModules.has("diffanalyzer") && ( +
+
+
+ + +
+
+ + +
+
+ +

+ + Git Integration +

+
+
+ + Git diff parsing +
+
+ + Staged changes detection +
+
+ + Logical section chunking +
+
+ + +

+ + Sample Diff Chunk +

+
+
diff --git a/src/embedder.py b/src/embedder.py
+
index 1a2b3c4..5d6e7f8 100644
+
--- a/src/embedder.py
+
+++ b/src/embedder.py
+
@@ -45,6 +45,12 @@ class Embedder:
+
- return self.model.encode(texts)
+
+ embeddings = self.model.encode(texts)
+
+ # Cache embeddings for faster retrieval
+
+ self._cache_embeddings(texts, embeddings)
+
+ return embeddings
+
+ +
+ )} + + + {/* Generator Module */} + +
+
+
+ +

Generator Module

+ + Loading + +
+ +
+
+ {expandedModules.has("generator") && ( +
+
+
+ + +
+
+ + +
+
+ + +
+
+
+
+ + +
+
+ + +
+
+ + +
+
+ +

+ + Commit Message Generation +

+
+
+ + Conventional commit format +
+
+ + Imperative tone enforcement +
+
+ + Context-aware suggestions +
+
+ + Fallback logic for model failures +
+
+ + Style presets (conventional/descriptive/terse) +
+
+ +
+ )} + +
+ ); +} diff --git a/client/src/components/sections/overview.tsx b/client/src/components/sections/overview.tsx new file mode 100644 index 0000000000000000000000000000000000000000..403cac91efeb69f04c6e9733f9bc8130cb6b996b --- /dev/null +++ b/client/src/components/sections/overview.tsx @@ -0,0 +1,154 @@ +import { Card } from "@/components/ui/card"; +import { Badge } from "@/components/ui/badge"; +import { Brain, Search, Wand2, FolderTree, GitCommit } from "lucide-react"; + +export function OverviewSection() { + const features = [ + { + icon: Brain, + title: "EmbeddingGemma-300m", + description: "High-precision text embedding for semantic search", + color: "text-primary" + }, + { + icon: Search, + title: "FAISS Vector Store", + description: "Efficient similarity search and clustering", + color: "text-accent" + }, + { + icon: Wand2, + title: "Phi-2 Generator", + description: "2.7B parameter model for contextual responses", + color: "text-purple-400" + }, + { + icon: FolderTree, + title: "Commit Assistant", + description: "Git diff analysis and intelligent commit message generation", + color: "text-orange-400" + } + ]; + + const projectStructure = [ + { type: "folder", name: "gemma-phi2-cli/", level: 0 }, + { type: "folder", name: "models/", level: 1, badge: "AI Models" }, + { type: "folder", name: "embeddinggemma-300m/", level: 2 }, + { type: "folder", name: "phi2-quantized/", level: 2 }, + { type: "folder", name: "data/", level: 1 }, + { type: "folder", name: "index/", level: 2, badge: "Vector Store" }, + { type: "folder", name: "src/", level: 1 }, + { type: "file", name: "embedder.py", level: 2 }, + { type: "file", name: "retriever.py", level: 2 }, + { type: "file", name: "generator.py", level: 2 }, + { type: "file", name: "diff_analyzer.py", level: 2 }, + { type: "file", name: "cli.py", level: 2 }, + { type: "folder", name: "tests/", level: 1, badge: "Unit Tests" }, + { type: "file", name: "test_embedder.py", level: 2 }, + { type: "file", name: "test_retriever.py", level: 2 }, + { type: "file", name: "test_generator.py", level: 2 }, + { type: "folder", name: "examples/", level: 1, badge: "Sample Repo" }, + { type: "file", name: "sample_repo/", level: 2 }, + { type: "file", name: "config.yaml", level: 1 }, + { type: "file", name: "environment.yml", level: 1 }, + { type: "file", name: "README.md", level: 1 }, + ]; + + return ( +
+
+
+
+
+

Project Overview

+

+ A seamless CLI assistant that combines EmbeddingGemma's precision with Phi-2's generative capabilities. +

+
+ + v1.0.0 + +
+ +
+ {features.map((feature, index) => ( + +
+ +

{feature.title}

+
+

{feature.description}

+ + ))} +
+ + +

+ + Data Flow Pipeline +

+
+
+
+
1
+ Git Diff Analysis +
+ diff_analyzer.py +
+
+
↓
+
+
+
+
2
+ Text Embedding +
+ embedder.py +
+
+
↓
+
+
+
+
3
+ Context Retrieval +
+ retriever.py +
+
+
↓
+
+
+
+
4
+ Message Generation +
+ generator.py +
+
+ + + +

+ + Project Structure +

+
+ {projectStructure.map((item, index) => ( +
+ + {item.name} + {item.badge && ( + + {item.badge} + + )} +
+ ))} +
+ +
+
+
+ ); +} diff --git a/client/src/components/ui/accordion.tsx b/client/src/components/ui/accordion.tsx new file mode 100644 index 0000000000000000000000000000000000000000..e6a723d06574ee5cec8b00759b98f3fbe1ac7cc9 --- /dev/null +++ b/client/src/components/ui/accordion.tsx @@ -0,0 +1,56 @@ +import * as React from "react" +import * as AccordionPrimitive from "@radix-ui/react-accordion" +import { ChevronDown } from "lucide-react" + +import { cn } from "@/lib/utils" + +const Accordion = AccordionPrimitive.Root + +const AccordionItem = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)) +AccordionItem.displayName = "AccordionItem" + +const AccordionTrigger = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, children, ...props }, ref) => ( + + svg]:rotate-180", + className + )} + {...props} + > + {children} + + + +)) +AccordionTrigger.displayName = AccordionPrimitive.Trigger.displayName + +const AccordionContent = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, children, ...props }, ref) => ( + +
{children}
+ +)) + +AccordionContent.displayName = AccordionPrimitive.Content.displayName + +export { Accordion, AccordionItem, AccordionTrigger, AccordionContent } diff --git a/client/src/components/ui/alert-dialog.tsx b/client/src/components/ui/alert-dialog.tsx new file mode 100644 index 0000000000000000000000000000000000000000..8722561cf6bda62d62f9a0c67730aefda971873a --- /dev/null +++ b/client/src/components/ui/alert-dialog.tsx @@ -0,0 +1,139 @@ +import * as React from "react" +import * as AlertDialogPrimitive from "@radix-ui/react-alert-dialog" + +import { cn } from "@/lib/utils" +import { buttonVariants } from "@/components/ui/button" + +const AlertDialog = AlertDialogPrimitive.Root + +const AlertDialogTrigger = AlertDialogPrimitive.Trigger + +const AlertDialogPortal = AlertDialogPrimitive.Portal + +const AlertDialogOverlay = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)) +AlertDialogOverlay.displayName = AlertDialogPrimitive.Overlay.displayName + +const AlertDialogContent = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + + + + +)) +AlertDialogContent.displayName = AlertDialogPrimitive.Content.displayName + +const AlertDialogHeader = ({ + className, + ...props +}: React.HTMLAttributes) => ( +
+) +AlertDialogHeader.displayName = "AlertDialogHeader" + +const AlertDialogFooter = ({ + className, + ...props +}: React.HTMLAttributes) => ( +
+) +AlertDialogFooter.displayName = "AlertDialogFooter" + +const AlertDialogTitle = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)) +AlertDialogTitle.displayName = AlertDialogPrimitive.Title.displayName + +const AlertDialogDescription = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)) +AlertDialogDescription.displayName = + AlertDialogPrimitive.Description.displayName + +const AlertDialogAction = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)) +AlertDialogAction.displayName = AlertDialogPrimitive.Action.displayName + +const AlertDialogCancel = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)) +AlertDialogCancel.displayName = AlertDialogPrimitive.Cancel.displayName + +export { + AlertDialog, + AlertDialogPortal, + AlertDialogOverlay, + AlertDialogTrigger, + AlertDialogContent, + AlertDialogHeader, + AlertDialogFooter, + AlertDialogTitle, + AlertDialogDescription, + AlertDialogAction, + AlertDialogCancel, +} diff --git a/client/src/components/ui/alert.tsx b/client/src/components/ui/alert.tsx new file mode 100644 index 0000000000000000000000000000000000000000..41fa7e0561a3fdb5f986c1213a35e563de740e96 --- /dev/null +++ b/client/src/components/ui/alert.tsx @@ -0,0 +1,59 @@ +import * as React from "react" +import { cva, type VariantProps } from "class-variance-authority" + +import { cn } from "@/lib/utils" + +const alertVariants = cva( + "relative w-full rounded-lg border p-4 [&>svg~*]:pl-7 [&>svg+div]:translate-y-[-3px] [&>svg]:absolute [&>svg]:left-4 [&>svg]:top-4 [&>svg]:text-foreground", + { + variants: { + variant: { + default: "bg-background text-foreground", + destructive: + "border-destructive/50 text-destructive dark:border-destructive [&>svg]:text-destructive", + }, + }, + defaultVariants: { + variant: "default", + }, + } +) + +const Alert = React.forwardRef< + HTMLDivElement, + React.HTMLAttributes & VariantProps +>(({ className, variant, ...props }, ref) => ( +
+)) +Alert.displayName = "Alert" + +const AlertTitle = React.forwardRef< + HTMLParagraphElement, + React.HTMLAttributes +>(({ className, ...props }, ref) => ( +
+)) +AlertTitle.displayName = "AlertTitle" + +const AlertDescription = React.forwardRef< + HTMLParagraphElement, + React.HTMLAttributes +>(({ className, ...props }, ref) => ( +
+)) +AlertDescription.displayName = "AlertDescription" + +export { Alert, AlertTitle, AlertDescription } diff --git a/client/src/components/ui/aspect-ratio.tsx b/client/src/components/ui/aspect-ratio.tsx new file mode 100644 index 0000000000000000000000000000000000000000..c4abbf37f217c715a0eaade7f45ac78600df419f --- /dev/null +++ b/client/src/components/ui/aspect-ratio.tsx @@ -0,0 +1,5 @@ +import * as AspectRatioPrimitive from "@radix-ui/react-aspect-ratio" + +const AspectRatio = AspectRatioPrimitive.Root + +export { AspectRatio } diff --git a/client/src/components/ui/avatar.tsx b/client/src/components/ui/avatar.tsx new file mode 100644 index 0000000000000000000000000000000000000000..51e507ba9d08bcdbb1fb630498f1cbdf2bf50093 --- /dev/null +++ b/client/src/components/ui/avatar.tsx @@ -0,0 +1,50 @@ +"use client" + +import * as React from "react" +import * as AvatarPrimitive from "@radix-ui/react-avatar" + +import { cn } from "@/lib/utils" + +const Avatar = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)) +Avatar.displayName = AvatarPrimitive.Root.displayName + +const AvatarImage = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)) +AvatarImage.displayName = AvatarPrimitive.Image.displayName + +const AvatarFallback = React.forwardRef< + React.ElementRef, + React.ComponentPropsWithoutRef +>(({ className, ...props }, ref) => ( + +)) +AvatarFallback.displayName = AvatarPrimitive.Fallback.displayName + +export { Avatar, AvatarImage, AvatarFallback } diff --git a/client/src/components/ui/badge.tsx b/client/src/components/ui/badge.tsx new file mode 100644 index 0000000000000000000000000000000000000000..f000e3ef5176395b067dfc3f3e1256a80c450015 --- /dev/null +++ b/client/src/components/ui/badge.tsx @@ -0,0 +1,36 @@ +import * as React from "react" +import { cva, type VariantProps } from "class-variance-authority" + +import { cn } from "@/lib/utils" + +const badgeVariants = cva( + "inline-flex items-center rounded-full border px-2.5 py-0.5 text-xs font-semibold transition-colors focus:outline-none focus:ring-2 focus:ring-ring focus:ring-offset-2", + { + variants: { + variant: { + default: + "border-transparent bg-primary text-primary-foreground hover:bg-primary/80", + secondary: + "border-transparent bg-secondary text-secondary-foreground hover:bg-secondary/80", + destructive: + "border-transparent bg-destructive text-destructive-foreground hover:bg-destructive/80", + outline: "text-foreground", + }, + }, + defaultVariants: { + variant: "default", + }, + } +) + +export interface BadgeProps + extends React.HTMLAttributes, + VariantProps {} + +function Badge({ className, variant, ...props }: BadgeProps) { + return ( +
+ ) +} + +export { Badge, badgeVariants } diff --git a/client/src/components/ui/breadcrumb.tsx b/client/src/components/ui/breadcrumb.tsx new file mode 100644 index 0000000000000000000000000000000000000000..60e6c96f72f0350d08b47e4730cab8f3975dc853 --- /dev/null +++ b/client/src/components/ui/breadcrumb.tsx @@ -0,0 +1,115 @@ +import * as React from "react" +import { Slot } from "@radix-ui/react-slot" +import { ChevronRight, MoreHorizontal } from "lucide-react" + +import { cn } from "@/lib/utils" + +const Breadcrumb = React.forwardRef< + HTMLElement, + React.ComponentPropsWithoutRef<"nav"> & { + separator?: React.ReactNode + } +>(({ ...props }, ref) =>