feat:Added pre-commit hooks

Author: ayusrjn

.gitignore

@@ -182,9 +182,9 @@ cython_debug/
 .abstra/
 
 # Visual Studio Code
-#  Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore 
+#  Visual Studio Code specific template is maintained in a separate VisualStudioCode.gitignore
 #  that can be found at https://github.com/github/gitignore/blob/main/Global/VisualStudioCode.gitignore
-#  and can be added to the global gitignore or merged into this file. However, if you prefer, 
+#  and can be added to the global gitignore or merged into this file. However, if you prefer,
 #  you could uncomment the following to ignore the entire vscode folder
 # .vscode/
 

.pre-commit-config.yaml

@@ -0,0 +1,14 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.5.0
+    hooks:
+      - id: trailing-whitespace
+      - id: end-of-file-fixer
+      - id: check-yaml
+
+  - repo: https://github.com/astral-sh/ruff-pre-commit
+    rev: v0.3.0
+    hooks:
+      - id: ruff
+        args: [ --fix ]
+      - id: ruff-format

README.md

@@ -71,4 +71,4 @@ Distributed under the Apache 2.0 License. See `LICENSE` for more information.
 
 ## Attribution
 
-- Lambda icon by [shohanur.rahman13](https://www.flaticon.com/authors/shohanur-rahman13) from [Flaticon](https://www.flaticon.com/free-icons/lambda)
+- Lambda icon by [shohanur.rahman13](https://www.flaticon.com/authors/shohanur-rahman13) from [Flaticon](https://www.flaticon.com/free-icons/lambda)

lambda/agent.py

@@ -1,18 +1,20 @@
-import json
 from .config import API_KEY, MODEL_NAME
 from .tools import TOOL_EXECUTORS, TOOL_FUNCTIONS
 
 try:
     import google.generativeai as genai
 except ImportError:
-    print("Warning: google-generativeai package is not installed. Please `pip install google-generativeai`.")
+    print(
+        "Warning: google-generativeai package is not installed. Please `pip install google-generativeai`."
+    )
+
 
 class Agent:
     def __init__(self):
         # Configure Gemini API
         genai.configure(api_key=API_KEY)
         self.model_name = MODEL_NAME
-        
+
         system_instruction = (
             "You are Lambda, a minimal and highly efficient AI coding agent. "
             "Your primary goal is to help the user by writing code, executing commands, "
@@ -21,14 +23,14 @@ def __init__(self):
             "something that requires these tools, you should use them autonomously. "
             "Be concise and professional."
         )
-        
+
         # Initialize the generative model with the built tools and system instructions
         self.model = genai.GenerativeModel(
             model_name=self.model_name,
             system_instruction=system_instruction,
-            tools=TOOL_FUNCTIONS
+            tools=TOOL_FUNCTIONS,
         )
-        
+
         # Gemini manages history cleanly through the ChatSession
         self.chat_session = self.model.start_chat()
 
@@ -44,44 +46,48 @@ def chat(self, user_input: str) -> str:
             try:
                 # 1. Check if the model returned a function_call in any part
                 tool_calls = [
-                    part.function_call for part in response.parts if getattr(part, 'function_call', None)
+                    part.function_call
+                    for part in response.parts
+                    if getattr(part, "function_call", None)
                 ]
-                
+
                 # 2. If it did, act on each function call
                 if tool_calls:
                     tool_responses = []
-                    
+
                     for function_call in tool_calls:
                         function_name = function_call.name
-                        
+
                         # Convert protobuf args to dict
-                        arguments = {key: value for key, value in function_call.args.items()}
+                        arguments = {
+                            key: value for key, value in function_call.args.items()
+                        }
                         print(f"\\n[Lambda is executing: {function_name}({arguments})]")
-                        
+
                         # 3. Execute the tool locally
                         if function_name in TOOL_EXECUTORS:
                             function_to_call = TOOL_EXECUTORS[function_name]
                             # Call the function dynamically
                             tool_result = function_to_call(**arguments)
                         else:
                             tool_result = f"Error: Tool {function_name} not found."
-                            
+
                         # Format the result back into Gemini's expected Response format
-                        tool_responses.append({
-                            "function_response": {
-                                "name": function_name,
-                                "response": {"result": str(tool_result)}
+                        tool_responses.append(
+                            {
+                                "function_response": {
+                                    "name": function_name,
+                                    "response": {"result": str(tool_result)},
+                                }
                             }
-                        })
+                        )
 
-                    
-                    # 4. Send ALL the tool responses back to the model 
+                    # 4. Send ALL the tool responses back to the model
                     # so it can continue reasoning based on the new information
                     response = self.chat_session.send_message(tool_responses)
-                    continue # Start the loop over to see if it calls more tools
+                    continue  # Start the loop over to see if it calls more tools
                 else:
                     # No more tool calls; the LLM has generated a final text response.
                     return response.text
             except Exception as e:
                 return f"An error occurred in the agent loop: {str(e)}"
-

lambda/config.py

@@ -1,16 +1,17 @@
 import os
-import sys
 
 try:
     from dotenv import load_dotenv
+
     load_dotenv()
 except ImportError:
-    print("Warning: python-dotenv not installed. If you are using a .env file, it will not be loaded.")
+    print(
+        "Warning: python-dotenv not installed. If you are using a .env file, it will not be loaded."
+    )
 
 # Default to using Gemini
-API_KEY = os.getenv("API_KEY") # You can name this GEMINI_API_KEY in your .env
+API_KEY = os.getenv("API_KEY")  # You can name this GEMINI_API_KEY in your .env
 MODEL_NAME = os.getenv("MODEL_NAME", "gemini-3-flash-preview")
 
 if not API_KEY:
     print("WARNING: API_KEY is not set in the environment or .env file.")
-

lambda/main.py

@@ -1,10 +1,10 @@
-import sys
 from .agent import Agent
 
+
 def main():
     print(r"""
 ========================================================
-██╗      █████╗ ███╗   ███╗██████╗ ██████╗  █████╗ 
+██╗      █████╗ ███╗   ███╗██████╗ ██████╗  █████╗
 ██║     ██╔══██╗████╗ ████║██╔══██╗██╔══██╗██╔══██╗
 ██║     ███████║██╔████╔██║██████╔╝██║  ██║███████║
 ██║     ██╔══██║██║╚██╔╝██║██╔══██╗██║  ██║██╔══██║
@@ -18,25 +18,26 @@ def main():
         agent = Agent()
         print("Lambda is ready! Type 'exit' or 'quit' to stop.")
         print("-" * 40)
-        
+
         while True:
             try:
-                user_input = input("\\nYou: ")
-                if user_input.lower() in ['exit', 'quit']:
+                user_input = input("\nYou: ")
+                if user_input.lower() in ["exit", "quit"]:
                     print("Goodbye!")
                     break
-                
+
                 if not user_input.strip():
                     continue
-                    
+
                 response = agent.chat(user_input)
-                print(f"\\nLambda: {response}")
-                
+                print(f"\nLambda: {response}")
+
             except KeyboardInterrupt:
-                print("\\nGoodbye!")
+                print("\nGoodbye!")
                 break
     except Exception as e:
         print(f"Failed to initialize Lambda: {str(e)}")
 
+
 if __name__ == "__main__":
     main()

lambda/tools.py

@@ -1,45 +1,43 @@
-import os
 import subprocess
 
+
 def read_file(path: str) -> str:
     """Reads the contents of a file.
-    
+
     Args:
         path: The absolute or relative path to the file.
     """
     try:
-        with open(path, 'r', encoding='utf-8') as f:
+        with open(path, "r", encoding="utf-8") as f:
             return f.read()
     except Exception as e:
         return f"Error reading file {path}: {str(e)}"
 
+
 def write_file(path: str, content: str) -> str:
     """Writes content to a specific file path.
-    
+
     Args:
         path: The path to the file to write.
         content: The text content to write to the file.
     """
     try:
-        with open(path, 'w', encoding='utf-8') as f:
+        with open(path, "w", encoding="utf-8") as f:
             f.write(content)
         return f"Successfully wrote to {path}"
     except Exception as e:
         return f"Error writing to file {path}: {str(e)}"
 
+
 def run_command(command: str) -> str:
     """Executes a shell command on the host system.
-    
+
     Args:
         command: The shell command to execute.
     """
     try:
         result = subprocess.run(
-            command,
-            shell=True,
-            capture_output=True,
-            text=True,
-            timeout=30
+            command, shell=True, capture_output=True, text=True, timeout=30
         )
         output = result.stdout + result.stderr
         return output if output else "Command executed successfully with no output."
@@ -48,6 +46,7 @@ def run_command(command: str) -> str:
     except Exception as e:
         return f"Error executing command: {str(e)}"
 
+
 # A dictionary mapping tool names to Python functions for dynamic execution
 TOOL_EXECUTORS = {
     "read_file": read_file,
@@ -56,4 +55,4 @@ def run_command(command: str) -> str:
 }
 
 # The list of raw Python functions for the Gemini SDK to auto-generate schemas
-TOOL_FUNCTIONS = [read_file, write_file, run_command]
+TOOL_FUNCTIONS = [read_file, write_file, run_command]

requirements.txt

@@ -1,8 +1,11 @@
 annotated-types==0.7.0
 certifi==2026.2.25
 cffi==2.0.0
+cfgv==3.5.0
 charset-normalizer==3.4.7
 cryptography==46.0.7
+distlib==0.4.0
+filelock==3.25.2
 google-ai-generativelanguage==0.6.15
 google-api-core==2.30.2
 google-api-python-client==2.193.0
@@ -13,7 +16,11 @@ googleapis-common-protos==1.74.0
 grpcio==1.80.0
 grpcio-status==1.71.2
 httplib2==0.31.2
+identify==2.6.18
 idna==3.11
+nodeenv==1.10.0
+platformdirs==4.9.6
+pre_commit==4.5.1
 proto-plus==1.27.2
 protobuf==5.29.6
 pyasn1==0.6.3
@@ -22,10 +29,13 @@ pycparser==3.0
 pydantic==2.12.5
 pydantic_core==2.41.5
 pyparsing==3.3.2
+python-discovery==1.2.2
 python-dotenv==1.2.2
+PyYAML==6.0.3
 requests==2.33.1
 tqdm==4.67.3
 typing-inspection==0.4.2
 typing_extensions==4.15.0
 uritemplate==4.2.0
 urllib3==2.6.3
+virtualenv==21.2.0